code agent test

agent.py

@@ -3,6 +3,7 @@ from smolagents import (
     HfApiModel,
     CodeAgent,
     load_tool,
+    Tool,
     InferenceClientModel,
     ToolCallingAgent,
     FinalAnswerTool,
@@ -22,11 +23,15 @@ load_dotenv()
 login(os.environ["HF_API_KEY"])
 
 # Tools
+
+# wikipedia = Tool.from_langchain(load_tool("wikipedia", trust_remote_code=True))
+
 tools = [
     DuckDuckGoSearchTool(),
     VisitWebpageTool(),
     PythonInterpreterTool(),
     FinalAnswerTool(),
+    # wikipedia
 ]
 
 question = QUESTIONS[0]
@@ -51,10 +56,10 @@ codeAgent = CodeAgent(
 
 codeAgent.logger.console.width = 66
 
-# answer = codeAgent.run(question)
 # answer = codeAgent.run(
 #     "Who are the pitchers with the number before and after Taishō Tamai's number as of July 2023? Give them to me in the form Pitcher Before, Pitcher After, use their last names only, in Roman characters."
 # )
+# answer = codeAgent.run("search wikipedia for: what the name of the main character of the fast and furious movie")
 # print(answer)
 
 

app.py

@@ -15,8 +15,8 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
-        # return codeAgent.run(question)
-        return toolCallingAgent.run(question)
+        return codeAgent.run(question)
+        # return toolCallingAgent.run(question)
         # print(f"Agent received question (first 50 chars): {question[:50]}...")
         # fixed_answer = "This is a default answer."
         # print(f"Agent returning fixed answer: {fixed_answer}")

requirements.txt

@@ -5,4 +5,5 @@ dotenv
 duckduckgo-search
 markdownify
 pandas
-numpy
+numpy
+langchain

tools/wikipedia_summarize.py

@@ -0,0 +1,21 @@
+from smolagents import tool
+import requests
+
+@tool
+def wiki_summarize(topic: str) -> str:
+    """Get the first paragraph summary for a Wikipedia topic.
+    rgs:
+        topic: The Wikipedia page title to summarize.
+    Returns:
+        A short summary of the page.
+
+    """
+    url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{topic}"
+    resp = requests.get(url)
+    if resp.status_code == 200:
+        data = resp.json()
+        return data.get("extract", "No summary available.")
+    else:
+        return f"Error fetching Wikipedia summary (status {resp.status_code})."
+
+# wiki_summarize_tool = wiki_summarize.push_to_hub()