Added fetch_wikipedia_page_content

app.py

@@ -13,6 +13,8 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 
+import wikipedia
+
 huggingface_hub.login(os.getenv('HF_TOKEN'))
 
 # (Keep Constants as is)
@@ -63,6 +65,15 @@ class VisitWebpageTool(Tool):
 
 search_tool = DuckDuckGoSearchTool()
 
+@tool
+def fetch_wikipedia_page_content(topic:str)-> 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 fetchs the contents of a page topic on Wikipedia
+    Args:
+        topic: the topic
+    """
+    return wikipedia.page(topic).content
+
 @tool
 def say_hello() -> str:
     """A tool that says hello
@@ -123,7 +134,7 @@ with open("prompts.yaml", 'r') as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, say_hello, image_generation_tool, search_tool, VisitWebpageTool()], ## add your tools here (don't remove final answer)
+    tools=[fetch_wikipedia_page_content, final_answer, say_hello, image_generation_tool, search_tool, VisitWebpageTool()], ## add your tools here (don't remove final answer)
     max_steps=10,
     verbosity_level=1,
     grammar=None,

requirements.txt

@@ -4,3 +4,4 @@ requests
 markdownify
 duckduckgo_search
 pandas
+wikipedia