Update app.py

app.py

@@ -6,6 +6,9 @@ import pandas as pd
 from smolagents import CodeAgent, DuckDuckGoSearchTool
 from smolagents.models import OpenAIServerModel
 
+from tools import WikipediaTool, WikipediaSearchTool
+
+
 # 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:
@@ -28,8 +31,8 @@ class PatchedOpenAIServerModel(OpenAIServerModel):
 
 class MyAgent:
     def __init__(self):
-        self.model = PatchedOpenAIServerModel(model_id="gpt-4")
-        self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=self.model)
+        self.model = PatchedOpenAIServerModel(model_id="gpt-4-turbo") #to be changed gpt-4
+        self.agent = CodeAgent(tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(), WikipediaTool()], model=self.model)  #add a new tool 
 
     def __call__(self, question: str) -> str:
         return self.agent.run(question)