Update agents.py

agents.py

@@ -14,11 +14,13 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, Wikipe
 from langchain_groq import ChatGroq
 
 # === WRAPPER CLASS TO FIX 'content' ERROR ===
+from langchain_core.messages import AIMessage  # <-- import AIMessage
+
 class WrappedChatGroq(ChatGroq):
     def invoke(self, *args, **kwargs):
         result = super().invoke(*args, **kwargs)
         if isinstance(result, dict) and "content" in result:
-            return SimpleNamespace(content=result["content"])
+            return AIMessage(content=result["content"])
         return result
 
 # === TOOLS ===