Update app.py

app.py

@@ -6,9 +6,21 @@ import pandas as pd
 from smolagents import CodeAgent, DuckDuckGoSearchTool
 from smolagents.models import OpenAIServerModel
 
-from tools import WikipediaTool, WikipediaSearchTool
+#from tools import WikipediaTool, WikipediaSearchTool
 #from tools import WikipediaToolWrapper, WikipediaSearchToolWrapper
 
+from smolagents import Tool
+from wikipedia_searcher import WikipediaSearcher
+
+wikipedia_search_tool = Tool(
+    name="wikipedia_search",
+    func=WikipediaSearcher().search,
+    description="Search Wikipedia for a given query.",
+    inputs={"query": "string"},
+    output_type="string"
+)
+
+
 # Define the system prompt
 SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.
 Report your thoughts, and finish your answer with the following template:
@@ -35,8 +47,9 @@ class MyAgent:
             DuckDuckGoSearchTool(),
             #WikipediaToolWrapper(),
             #WikipediaSearchToolWrapper()
-            WikipediaSearchTool(),
-            WikipediaTool()
+            #WikipediaSearchTool(),
+            #WikipediaTool()
+            wikipedia_search_tool
         ], model=self.model)
 
     def __call__(self, task: dict) -> str: