replace web search tool, add system prompt

agent.py

@@ -1,13 +1,30 @@
-from smolagents import CodeAgent, InferenceClientModel, WikipediaSearchTool
+from smolagents import CodeAgent, InferenceClientModel, WikipediaSearchTool, GoogleSearchTool, VisitWebpageTool, \
+    PythonInterpreterTool
+
+
+def get_prompt():
+    with open("prompt.txt", "r") as f:
+        return f.read()
 
 
 class GAIAAgent:
     def __init__(self):
         self.agent = CodeAgent(
-            tools=[WikipediaSearchTool()],
+            tools=[
+                GoogleSearchTool(provider="serper"),
+                VisitWebpageTool(),
+                WikipediaSearchTool(),
+                PythonInterpreterTool(),
+            ],
             model=InferenceClientModel(),
-            add_base_tools=True
+            add_base_tools=False
         )
+        self.prompt = get_prompt()
 
     def __call__(self, question: str) -> str:
-        return self.agent.run(question)
+        return self.agent.run(self.prompt, additional_args={"question": question})
+
+
+if __name__ == '__main__':
+    agent = GAIAAgent()
+    agent("What is the meaning of life?")

prompt.txt

@@ -0,0 +1 @@
+You are a general AI assistant. I will ask you a question. Report your thoughts and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma-separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ nor percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g., for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma-separated list, apply the above rules depending on whether the element to be put in the list is a number or a string.