update

app.py

@@ -156,11 +156,14 @@ class BasicAgent:
             max_steps=10,
         ) 
         
-        self.code_agent = CodeAgent(
+        self.manager_agent = CodeAgent(
             tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_image, get_text_file],
             model=InferenceClientModel(
                 model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
             ),
+            managed_agents=[
+                self.multimodal_agent
+            ],
             additional_authorized_imports=[
                 "requests",
                 "bs4",
@@ -185,52 +188,22 @@ class BasicAgent:
                     - Browse the web to find information.
                     - Solving chess problems.
                 
-                Please follow hints below:
-                1. `pandas` Python package is provided. Please try to use it FIRST when there is need to extract structured data (such as tables) from HTML content.
-                2. `wikipedia` Python package is provided to interact with Wikipedia. Try to work with raw wikipedia HTML content and use `pandas` to parse first.
-                3. `chess` Python package is provided. Please use it when there is need to solve chess problems.
-                4. Please take the question literally! Do not add any additional information or assumptions.
+               
             """,
             verbosity_level=0,
             max_steps=10,
         )
 
-        self.manager_agent = CodeAgent(
-            model=InferenceClientModel(
-                "Qwen/Qwen2.5-32B-Instruct"
-            ),
-            tools=[get_image, get_text_file, VisitWebpageTool(), DuckDuckGoSearchTool()],
-            managed_agents=[
-                self.multimodal_agent, 
-                self.code_agent],
-            additional_authorized_imports=[
-                "requests",
-                "bs4",
-                "markdownify",
-                "wikipedia",
-                "pandas",
-                "io",
-                "PIL",
-                "chess",
-                "img2text",
-                "chess.pgn",
-                "PIL.Image",
-                "bytes",
-                "cv2",
-                "numpy",
-                "chess.engine",
-            ],
-            planning_interval=5,
-            max_steps=15,
-        )
-
     def __call__(self, question: str, question_id: str, has_file: bool) -> str:
         print(f"Agent received question: {question}")
         prompt = f"""
             Answer the following question: 
                "{question} {"The file name for this question is: " if has_file else ""} {question_id if has_file else ""}"
-            Please follow hints below:
-                1. Try to use managed agents to answer the question first.
+             Please follow hints below:
+                1. `pandas` Python package is provided. Please try to use it FIRST when there is need to extract structured data (such as tables) from HTML content.
+                2. `wikipedia` Python package is provided to interact with Wikipedia. Try to work with raw wikipedia HTML content and use `pandas` to parse first.
+                3. `chess` Python package is provided. Please use it when there is need to solve chess problems.
+                4. Please take the question literally! Do not add any additional information or assumptions.
         """
         result = self.manager_agent.run(prompt)
         print(f"Agent responded with: {result}")