Update app.py

app.py

@@ -6,8 +6,8 @@ import pandas as pd
 from smolagents import CodeAgent, DuckDuckGoSearchTool
 from smolagents.models import OpenAIServerModel
 
-#from tools import WikipediaTool, WikipediaSearchTool
-from tools import WikipediaToolWrapper, WikipediaSearchToolWrapper
+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.
@@ -33,10 +33,10 @@ class MyAgent:
         self.model = PatchedOpenAIServerModel(model_id="gpt-4-turbo")
         self.agent = CodeAgent(tools=[
             DuckDuckGoSearchTool(),
-            WikipediaToolWrapper(),
-            WikipediaSearchToolWrapper()
-            #WikipediaSearchTool(),
-            #WikipediaTool()
+            #WikipediaToolWrapper(),
+            #WikipediaSearchToolWrapper()
+            WikipediaSearchTool(),
+            WikipediaTool()
         ], model=self.model)
 
     def __call__(self, task: dict) -> str: