ajustando retorno mensagens

agent.py

@@ -65,9 +65,19 @@ class Agent:
         )
 
         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}")
-        
+        messages = response.get("messages", [])
+
+        # Extrair o conteúdo das mensagens do tipo AIMessage
+        final_content = ""
+        for message in messages:
+            if isinstance(message, dict) and message.get("content"):
+                content = message["content"]
+                if "FINAL ANSWER:" in content.upper():
+                    final_content = content
+                    break
+
+        # Extrair o valor final
+        final_answer = re.sub(r"(?i)^.*FINAL ANSWER:\s*", "", final_content).strip()
+        print("Final Answer:", final_answer)
+            
         return final_answer