update

app.py

@@ -60,10 +60,6 @@ def get_image_content(file_name: str) -> Image.Image | None:
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-        model = InferenceClientModel(
-            model_id="meta-llama/Llama-3.2-11B-Vision-Instruct",
-        )
-
         self.multimodal_agent = CodeAgent(
             tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_image_content],
             model=InferenceClientModel(
@@ -85,12 +81,34 @@ class BasicAgent:
             max_steps=10,
         )
 
+        self.code_agent = CodeAgent(
+            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_image_content],
+            model=InferenceClientModel(
+                model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+            ),
+            additional_authorized_imports=[
+                "requests",
+                "bs4",
+                "markdownify",
+                "wikipedia",
+                "pandas",
+                "io",
+                "PIL",
+                "chess",
+            ],
+            name="code_agent",
+            description="I am strong at writing code.",
+            verbosity_level=0,
+            max_steps=10,
+        )
+
         self.manager_agent = CodeAgent(
             model=InferenceClientModel(
-                "deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096
+                "deepseek-ai/DeepSeek-R1", 
+                provider="together"
             ),
             tools=[get_image_content],
-            managed_agents=[self.multimodal_agent],
+            managed_agents=[self.multimodal_agent, self.code_agent],
             additional_authorized_imports=[
                 "requests",
                 "bs4",
@@ -102,7 +120,6 @@ class BasicAgent:
                 "chess",
             ],
             planning_interval=5,
-            verbosity_level=2,
             max_steps=15,
         )
 
@@ -110,12 +127,13 @@ class BasicAgent:
         print(f"Agent received question: {question}")
         prompt = f"""
             Answer the following question: 
-               "{question} {", with the following file: " if file_name else ""} {file_name}"
+               "{question} {"The file name is: " if file_name else ""} {file_name}"
             
             Please follow hints below:
                 1. `wikipedia` Python package is provided to interact with Wikipedia.
                 2. `pandas` Python package is provided. Please use it when there is need to extract structured data (such as tables) from HTML content.
-                3. Please take the question literally! Do not add any additional information or assumptions.
+                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}")