Update agent.py

agent.py

@@ -81,6 +81,7 @@ class TestAgent:
                    wikipedia_tool,                  # tool from langchain with extra parmaeters
                    #youtube_tools,                   # tool from MCP server
                    FinalAnswerTool()],
+            additional_authorized_imports=["pandas"],
             model=model,
             max_steps=3,
             verbosity_level=2
@@ -89,6 +90,6 @@ class TestAgent:
     def __call__(self, question: str) -> str:
 
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        answer = self.agent.invoke(question)
+        answer = self.agent.run(question)
         print(f"Agent returning his answer: {answer}")
         return answer