Update agents.py

agents.py

@@ -14,7 +14,7 @@ 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
+from langchain_core.messages import AIMessage
 
 class WrappedChatGroq(ChatGroq):
     def invoke(self, *args, **kwargs):
@@ -23,6 +23,10 @@ class WrappedChatGroq(ChatGroq):
             return AIMessage(content=result["content"])
         return result
 
+    def __call__(self, *args, **kwargs):
+        return self.invoke(*args, **kwargs)
+
+
 # === TOOLS ===
 class GetTaskFileTool(Tool):
     name = "get_task_file_tool"