trying single agent again

agents.py

@@ -20,7 +20,7 @@ gemini_api_key = os.environ.get("GEMINI_API_KEY", None)
 open_router_api_key = os.environ.get("OPENROUTER_API_KEY", None)
 gemini_model = LiteLLMModel(
     # model_id="gemini/gemini-2.0-flash",
-    model_id="gemini/gemini-2.5-pro",
+    model_id="gemini/gemini-2.0-flash-001",
     api_key=gemini_api_key,
     tools=[google_search_tool],
 )
@@ -39,14 +39,14 @@ python_tool = PythonInterpreterTool()
 visit_webpage_tool = VisitWebpageTool()
 my_search_tool = MySearchTool()
 
-# with open("prompts.yaml", "r") as stream:
-#     prompt_templates = yaml.safe_load(stream)
+with open("prompts.yaml", "r") as stream:
+    prompt_templates = yaml.safe_load(stream)
 
-# prompt_templates["final_answer"] = {
-#     "pre_messages": "",
-#     "final_answer": "Here is the final answer from your managed agent '{{name}}':\n{{final_answer}}",
-#     "post_messages": "",
-# }
+prompt_templates["final_answer"] = {
+    "pre_messages": "",
+    "final_answer": "Here is the final answer from your managed agent '{{name}}':\n{{final_answer}}",
+    "post_messages": "",
+}
 
 
 class SkynetMultiAgent:
@@ -57,16 +57,9 @@ class SkynetMultiAgent:
             name="web_search_agent",
             description="""
                 This agent has access to Google Search to search for information about a specific topic.
-                
-                IMPORTANT: This agent MUST use the built-in Google Search capability to find accurate and up-to-date information.
-                DO NOT respond that you cannot access external websites or APIs.
-                DO NOT claim you are unable to search the web.
-                This agent is specifically configured with Google Search capability.
-                
-                It provides a comprehensive answer based on the search results.
-                Includes relevant facts, figures, and details.
             """,
             max_steps=2,
+            prompt_templates=prompt_templates,
         )
         self.agent = CodeAgent(
             tools=[],
@@ -74,7 +67,7 @@ class SkynetMultiAgent:
             name="manager_agent",
             description="Manages the web agents and it runs code.",
             managed_agents=[web_agent],
-            # prompt_templates=prompt_templates,
+            prompt_templates=prompt_templates,
             max_steps=2,
         )
 
@@ -96,10 +89,11 @@ class SkynetMultiAgent:
 class SkynetSingleAgent:
     def __init__(self):
         self.agent = ToolCallingAgent(
-            tools=[search_tool, wiki_tool, visit_webpage_tool, python_tool],
+            tools=[],
             model=gemini_model,
             name="skynet_agent",
-            description="Agent to answer questions and run code",
+            description="Agent to answer questions and to search in the web using google search tool",
+            prompt_templates=prompt_templates,
             # additional_authorized_imports=[
             #     "time",
             #     "numpy",

app.py

@@ -31,8 +31,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = SkynetMultiAgent()
-        single_agent = SkynetSingleAgent()
+        # agent = SkynetMultiAgent()
+        agent = SkynetSingleAgent()
         print(f"Agent instantiated successfully: {agent}")
     except Exception as e:
         print(f"Error instantiating agent: {e}")