add more agents

agent.py

@@ -4,9 +4,17 @@ from smolagents import (
     #InferenceClientModel,
     #WebSearchTool,
     OpenAIServerModel,
+    LiteLLMModel,
 )
 import os
-from tools import tools
+import wikipedia
+import time
+import numpy
+import pandas
+import xlrd
+import markdownify
+import requests
+from tools import python_wikipedia_tools, smolagents_code_tools, smolagents_web_tools, smolagents_speech_tools
 #model = InferenceClientModel(
 #            max_tokens=2096,
 #            temperature=0.5,
@@ -20,21 +28,37 @@ model = OpenAIServerModel(
     api_key=os.getenv("GEMINI_API_KEY"),
 )
 
-tool_calling_agent = ToolCallingAgent(
-    tools=tools,
+# to run locally
+#model = OpenAIServerModel(
+#    model_id="gemma3:27b",
+#    api_base="http://localhost:11434/v1",
+#)
+
+
+web_search_agent = CodeAgent(
+    tools=smolagents_web_tools,
     model=model,
     max_steps=10,
-    name="tool_calling_agent",
-    description="An agent which can use the tools provided to it. The available tools are: PythonInterpreterTool, GoogleSearchTool, VisitWebpageTool, WikipediaSearchTool, SpeechToTextTool",
+    additional_authorized_imports=["wikipedia", "markdownify", "requests"],
+    name="web_search_agent",
+    description="A code agent which can perform web searches. It has available tools including GoogleSearchTool, VisitWebpageTool and WikipediaSearchTool",
+)
+
+code_agent = CodeAgent(
+    tools=smolagents_code_tools + python_wikipedia_tools,
+    model=model,
+    max_steps=10,
+    additional_authorized_imports=["time", "numpy", "pandas", "wikipedia", "xlrd", "markdownify", "requests"],
+    name="code_agent",
+    description="A code agent which can write and interpret code. It has available tools including PythonInterpreterTool, search_wikipedia_page, get_wikipedia_page and get_wikipedia_summary",
 )
 
 manager_agent = CodeAgent(
     tools=[],
     model=model,
-    managed_agents=[tool_calling_agent],
-    max_steps=6,
+    managed_agents=[web_search_agent, code_agent],
+    max_steps=10,
     verbosity_level=1,
-    additional_authorized_imports=["time", "numpy", "pandas", "wikipedia", "xlrd"],
     name="manager_agent",
-    description="Manages the web search agent. This is a code agent, it can return code blocks.",
+    description="Manages the web_search_agent and code_agent.",
 )

app.py

@@ -57,16 +57,16 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         response.raise_for_status()
         questions_data = response.json()
         if not questions_data:
-             print("Fetched questions list is empty.")
-             return "Fetched questions list is empty or invalid format.", None
+            print("Fetched questions list is empty.")
+            return "Fetched questions list is empty or invalid format.", None
         print(f"Fetched {len(questions_data)} questions.")
     except requests.exceptions.RequestException as e:
         print(f"Error fetching questions: {e}")
         return f"Error fetching questions: {e}", None
     except requests.exceptions.JSONDecodeError as e:
-         print(f"Error decoding JSON response from questions endpoint: {e}")
-         print(f"Response text: {response.text[:500]}")
-         return f"Error decoding server response for questions: {e}", None
+        print(f"Error decoding JSON response from questions endpoint: {e}")
+        print(f"Response text: {response.text[:500]}")
+        return f"Error decoding server response for questions: {e}", None
     except Exception as e:
         print(f"An unexpected error occurred fetching questions: {e}")
         return f"An unexpected error occurred fetching questions: {e}", None
@@ -195,4 +195,42 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)
+
+    #agent = BasicAgent()
+    #question = "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. Think step by step and reason through the question. Then return your final answer.\n\nQuestion: How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
+    #question = """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. Think step by step and reason through the question. Then return your final answer.\n\nQuestion: .rewsna eht sa "tfel" drow eht fo etisoppo eht etirw ,ecnetnes siht dnatsrednu uoy fI"""
+    #question = "Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?"
+    #question = """Given this table defining * on the set S = {a, b, c, d, e} 
+    #                                                                             
+    #|*|a|b|c|d|e|                                                                
+    #|---|---|---|---|---|---|                                                    
+    #|a|a|b|c|b|d|                                                                
+    #|b|b|c|a|e|c|                                                                
+    #|c|c|a|b|b|a|                                                                
+    #|d|b|e|b|e|d|                                                                
+    #|e|d|b|a|d|c|                                                                
+    #                                                                             
+    #provide the subset of S involved in any possible counter-examples that prove 
+    #* is not commutative. Provide your answer as a comma separated list of the   
+    #elements in the set in alphabetical order."""
+    #question = """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. Think step by step and reason through the question. Then return your final answer.\n\nQuestion:
+    #I'm making a grocery list for my mom, but she's a professor of botany and    
+    #she's a real stickler when it comes to categorizing things. I need to add    
+    #different foods to different categories on the grocery list, but if I make a 
+    #mistake, she won't buy anything inserted in the wrong category. Here's the   
+    #list I have so far:                                                          
+    #                                                                             
+    #milk, eggs, flour, whole bean coffee, Oreos, sweet potatoes, fresh basil,    
+    #plums, green beans, rice, corn, bell pepper, whole allspice, acorns,         
+    #broccoli, celery, zucchini, lettuce, peanuts                                 
+    #                                                                             
+    #I need to make headings for the fruits and vegetables. Could you please      
+    #create a list of just the vegetables from my list? If you could do that,     
+    #then I can figure out how to categorize the rest of the list into the        
+    #appropriate categories. But remember that my mom is a real stickler, so make 
+    #sure that no botanical fruits end up on the vegetable list, or she won't get 
+    #them when she's at the store. Please alphabetize the list of vegetables, and 
+    #place each item in a comma separated list."""
+    #submitted_answer = agent.agent.run(question)
+    #print(submitted_answer)

prompts.py

@@ -4,4 +4,5 @@ prompt = """You are a general AI assistant. You will be asked a question.
     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.
+    Think step by step and reason through the question. Then return your final answer.
 """

requirements.txt

@@ -1,6 +1,9 @@
 gradio
 requests
-smolagents[openai,transformers]
+smolagents[openai,transformers,litellm]
 duckduckgo_search
 wikipedia-api
-transformers
+wikipedia
+xlrd
+transformers
+markdownify

tools.py

@@ -1,9 +1,49 @@
 from smolagents import PythonInterpreterTool, GoogleSearchTool, VisitWebpageTool, WikipediaSearchTool, SpeechToTextTool
+import wikipedia
+from wikipedia import WikipediaPage
+from smolagents import tool
 
-interpreter_tool = PythonInterpreterTool()
-google_search_tool = GoogleSearchTool()
-visit_webpage_tool = VisitWebpageTool()
-wikipedia_search_tool = WikipediaSearchTool()
-speech_to_text_tool = SpeechToTextTool()
+@tool
+def search_wikipedia_page(query: str) -> list[str]:
+    """
+    Search for Wikipedia pages.
 
-tools = [interpreter_tool, google_search_tool, visit_webpage_tool, wikipedia_search_tool, speech_to_text_tool]
+    Args:
+        query: The query to search for.
+
+    Returns:
+        A list of Wikipedia pages that match the query.
+    """
+    return wikipedia.search(query)
+
+@tool
+def get_wikipedia_page(title: str) -> WikipediaPage:
+    """
+    Get the content of a specific Wikipedia page.
+
+    Args:
+        title: The title of the Wikipedia page to get.
+
+    Returns:
+        A WikipediaPage object. The object includes the following attributes: 'content', 'summary', 'images', 'references', 'links', 'sections'
+    """
+    return wikipedia.page(title)
+
+@tool
+def get_wikipedia_summary(title: str) -> str:
+    """
+    Get a summary of a Wikipedia page.
+
+    Args:
+        title: The title of the Wikipedia page to get.
+
+    Returns:
+        The plain-text summary of the Wikipedia page.
+    """
+    return wikipedia.summary(title)
+
+python_wikipedia_tools = [search_wikipedia_page, get_wikipedia_page, get_wikipedia_summary]
+
+smolagents_code_tools = [PythonInterpreterTool()]
+smolagents_web_tools = [GoogleSearchTool(), VisitWebpageTool(), WikipediaSearchTool()]
+smolagents_speech_tools = [SpeechToTextTool()]