Mudança para Invoke

agent.py

@@ -44,7 +44,7 @@ class Agent:
         
     
     
-    def __call__(self, question: str, task_id: str, task_file_name: str) -> str:
+    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 = ""
@@ -74,4 +74,37 @@ class Agent:
         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]}...")
+    
+        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}  . "
+
+        # Chamando sem stream
+        response = self.supervisor.invoke(
+            {
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": f"{file_prefix}{question}",
+                    }
+                ]
+            }
+        )
+
+        messages = response.get("supervisor", {}).get("messages", [])
+        if not messages:
+            print(f"Nenhuma mensagem retornada para task {task_id}.")
+            return "Desculpe, não houve resposta."
+
+        last_msg = messages[-1]["content"]
+        final_answer = re.sub(r"^FINAL ANSWER:\s*", "", last_msg.strip(), flags=re.IGNORECASE)
+        print(f"Agent returning answer for task {task_id}: {final_answer}")
+        
         return final_answer