Update app.py

added wiki_tool

app.py

@@ -1,7 +1,8 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, LiteLLMModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, Tool, HfApiModel, LiteLLMModel, load_tool, tool
 import datetime
 import pytz
 import time
+import wikipediaapi
 import yaml
 from tools.final_answer import FinalAnswerTool
 
@@ -28,8 +29,31 @@ model = LiteLLMModel(
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
+class WikipediaSearchTool(Tool):
+    name = "wikipedia_search"
+    description = "Fetches a summary of a Wikipedia page based on a given search query (only one word or group of words)."
+
+    inputs = {
+        "query": {"type": "string", "description": "The search term for the Wikipedia page (only one word or group of words)."}
+    }
+
+    output_type = "string"
+
+    def __init__(self, lang="en"):
+        super().__init__()
+        self.wiki = wikipediaapi.Wikipedia(
+            language=lang, user_agent="MinimalAgent/1.0")
+
+    def forward(self, query: str):
+        page = self.wiki.page(query)
+        if not page.exists():
+            return "No Wikipedia page found."
+        return page.summary[:1000]
+
+wiki_tool = WikipediaSearchTool()
+
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def search_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that provides web search via duckduckgo 
     Args:
@@ -60,7 +84,7 @@ class BasicAgent:
             prompt_templates = yaml.safe_load(stream) 
         self.agent = CodeAgent(
             model=model,
-            tools=[final_answer, my_custom_tool],
+            tools=[final_answer, search_tool, wiki_tool],
             max_steps=6,
             verbosity_level=1,
             grammar=None,