Update app.py

app.py

@@ -6,7 +6,8 @@ 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
 
 # Define the system prompt
 SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.
@@ -32,8 +33,10 @@ class MyAgent:
         self.model = PatchedOpenAIServerModel(model_id="gpt-4-turbo")
         self.agent = CodeAgent(tools=[
             DuckDuckGoSearchTool(),
-            WikipediaSearchTool(),
-            WikipediaTool()
+            WikipediaToolWrapper(),
+            WikipediaSearchToolWrapper()
+            #WikipediaSearchTool(),
+            #WikipediaTool()
         ], model=self.model)
 
     def __call__(self, task: dict) -> str: