hit google limits.

basic_agent.py

@@ -23,9 +23,6 @@ class BasicAgent:
     # Logs appear to be swallowed.
     LOG.warning("logging BasicAgent initialized.")
 
-    # Force google for now.
-    model_id = "google"
-
     # Assume we will use the default model creation
     self.model = None
 
@@ -63,35 +60,55 @@ class BasicAgent:
       )
     print(f"NEW2: BasicAgent {self.model_id=} {self.model=}")
 
-    self.tools = [
+    web_search_tools = [
         smolagents.DuckDuckGoSearchTool(),
         smolagents.VisitWebpageTool(),
         smolagents.FinalAnswerTool()
     ]
 
     print("BasicAgent making search tool.")
-    self.search_agent = smolagents.CodeAgent(
-        name="search_agent",
+    self.web_search_agent = smolagents.CodeAgent(
+        name="web_search_agent",
         description="Search the web",
         model=self.model,
-        tools=self.tools,
+        tools=web_search_tools,
         max_steps=6,
         verbosity_level=2,
         planning_interval=None,
         additional_authorized_imports=["duckduckgo_search"],
     )
 
+    wiki_search_tools = [
+        smolagents.WikipediaSearchTool(),
+        smolagents.VisitWebpageTool(),
+        smolagents.FinalAnswerTool()
+    ]
+
+    self.wiki_search_agent = smolagents.CodeAgent(
+        name="wikipedia_search_agent",
+        description="Search wikipedia",
+        model=self.model,
+        tools=wiki_search_tools,
+        max_steps=6,
+        verbosity_level=2,
+        planning_interval=None,
+        additional_authorized_imports=["wikipedia-api"],
+    )
+
+    print("BasicAgent making wikipedia search tool.")
+
+
     print("BasicAgent making manager.")
     self.manager_agent = smolagents.CodeAgent(
         name="manager_agent",
         description="Manger of other agents",
         tools=[smolagents.FinalAnswerTool()],
         model=self.model,
-        max_steps=6,
+        max_steps=10,
         verbosity_level=2,
         planning_interval=None,
-        additional_authorized_imports=["duckduckgo_search"],
-        managed_agents=[self.search_agent])
+        additional_authorized_imports=["duckduckgo_search", "wikipedia-api"],
+        managed_agents=[self.web_search_agent, self.wiki_search_agent])
 
   def __call__(self, question: str) -> str:
     # Avoid rate limiting issues

basic_agent_test.py

@@ -8,8 +8,12 @@ LOG = logging.getLogger(__name__)
 
 ba = basic_agent.BasicAgent()
 
+"""
 answer = ba(
     "Who is the 47th president of the united states? If necessary, use a web search to get the most up to date information."
 )
+"""
+answer = ba("What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?"
+)
 
 LOG.warning(f"{answer=}")

requirements.txt

@@ -3,6 +3,7 @@ requests
 huggingface_hub[hf_xet]
 smolagents
 smolagents[openai]
+wikipedia-api
 
 # Only needed for running locally
 #compressed-tensors