add two agents

agent.py

@@ -0,0 +1,36 @@
+from smolagents import (
+    CodeAgent,
+    ToolCallingAgent,
+    HfApiModel,
+    WebSearchTool,
+)
+import yaml
+
+#model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
+model = HfApiModel(
+            max_tokens=2096,
+            temperature=0.5,
+            model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+            custom_role_conversions=None,
+        )
+
+web_agent = ToolCallingAgent(
+    tools=[WebSearchTool()],
+    model=model,
+    max_steps=10,
+    name="web_search_agent",
+    description="Runs web searches.",
+)
+
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
+
+manager_agent = CodeAgent(
+    tools=[],
+    model=model,
+    managed_agents=[web_agent],
+    max_steps=6,
+    verbosity_level=1,
+    additional_authorized_imports=["time", "numpy", "pandas"],
+    prompt_templates=prompt_templates
+)

app.py

@@ -5,16 +5,16 @@ import inspect
 import pandas as pd
 from smolagents import CodeAgent, ToolCallingAgent, InferenceClientModel, WebSearchTool, LiteLLMModel, HfApiModel
 import yaml
+from agent import manager_agent
 
-# (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
+        self.agent = manager_agent
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         fixed_answer = "This is a default answer."
@@ -42,22 +42,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        #agent = BasicAgent()
-        model = HfApiModel(
-            max_tokens=2096,
-            temperature=0.5,
-            model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-            custom_role_conversions=None,
-        )
-        with open("prompts.yaml", 'r') as stream:
-            prompt_templates = yaml.safe_load(stream)
-        agent = CodeAgent(
-            model=model,
-            tools=[WebSearchTool()],
-            max_steps=10,
-            verbosity_level=1,
-            prompt_templates=prompt_templates
-        )
+        agent = BasicAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None

prompts.yaml

@@ -15,17 +15,3 @@
   - You can use imports in your code, but only from the following list of modules: {{authorized_imports}}
   - The state persists between code executions: so if in one step you've created variables or imported modules, these will all persist.
   - Don't give up! You're in charge of solving the task, not providing directions to solve it.
-
-managed_agent: |-
-  You are a general AI assistant. You will be asked a question. 
-  Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. 
-  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. 
-  If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 
-  If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. 
-  If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
-
-planning: |-
-  Think step by step. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. 
-
-final_answer: |-
-  FINAL ANSWER: {{final_answer}}