sem história de mensagens

agent.py

@@ -37,46 +37,12 @@ class Agent:
                 text_inverter_tool, check_table_commutativity_tool],
             prompt= SUPERVISOR_PROMPT,
             add_handoff_back_messages=True,
-            output_mode="full_history",
+            output_mode="final_response",
         ).compile()
         
         print("Agent initialized.")     
         
     
-    
-    def _call_antiga(self, question: str, task_id: str, task_file_name: str) -> str:
-        print(f"Agent received question({task_id}) (first 50 chars): {question[:50]}...")
-        
-        file_prefix = ""
-        if task_file_name:
-            print(f"Task com arquivo {task_file_name}")
-            File_Util.baixa_arquivo_task(task_file_name)
-            file_prefix = f"File: {task_file_name}  . "
-        
-        for chunk in self.supervisor.stream(
-            {
-                "messages": [
-                    {
-                        "role": "user",
-                        "content": f"{file_prefix}{question}",
-                    }
-                ]
-            },
-        ):
-            Agent_Util.pretty_print_messages(chunk, last_message=True)
-            final_chunk = chunk
-
-        print("Extraindo a resposta do agente")
-        agent_answer = final_chunk["supervisor"]["messages"]
-        print(f"resposta: {agent_answer}")
-        
-                
-        final_answer = re.sub(r"^FINAL ANSWER:\s*", "", agent_answer, flags=re.IGNORECASE)
-        print(f"Agent returning answer for task {task_id}: {final_answer}")
-        
-        return final_answer
-    
-    
     def __call__(self, question: str, task_id: str, task_file_name: str) -> str:
         print(f"Agent (nova forma de invocar) received question({task_id}) (first 50 chars): {question[:50]}...")
     
@@ -98,16 +64,10 @@ class Agent:
             }
         )
 
-        print(f"resposta LLM: {response}")
-        messages = response.get("supervisor", {}).get("messages", [])
-        if not messages:
-            print(f"Nenhuma mensagem retornada para task {task_id}.")
-            return "Desculpe, não houve resposta."
-
-        print("Extraindo last_message")
-        last_msg = messages[-1]["content"]
-        print(f"Last Message: {last_msg}")
-        final_answer = re.sub(r"^FINAL ANSWER:\s*", "", last_msg.strip(), flags=re.IGNORECASE)
+        print(f"Resposta LLM: {response}")
+        answer_llm = response.get("output", "").strip()
+        print(f"Resposta pré-processada: {answer_llm}")
+        final_answer = re.sub(r"^FINAL ANSWER:\s*", "", answer_llm, flags=re.IGNORECASE)
         print(f"Agent returning answer for task {task_id}: {final_answer}")
         
         return final_answer