Modified files

agent.py

@@ -1,14 +1,12 @@
-
-
 import os
 import re
 import requests
+import serpapi
 from smolagents import CodeAgent, ToolCallingAgent, WebSearchTool, tool
 from smolagents import InferenceClientModel
 from dotenv import load_dotenv
 from markdownify import markdownify
 from requests.exceptions import RequestException
-from serpapi import GoogleSearch
 
 def initialize_agent():
     # Load environment variables from .env file
@@ -63,12 +61,8 @@ def initialize_agent():
             The search results, or an error message if the search fails.
         """
         try:
-            params = {
-                "q": query,
-                "api_key": os.environ.get("SERPAPI_API_KEY"),
-            }
-            search = GoogleSearch(params)
-            results = search.get_dict()
+            client = serpapi.Client(api_key=os.environ.get("SERPAPI_API_KEY"))
+            results = client.search(q=query, engine="google")
             if "ai_overview" in results:
                 ai_overview = results["ai_overview"]
                 output = ""
@@ -108,10 +102,8 @@ def initialize_agent():
             model=model,
             managed_agents=[web_agent],
             additional_authorized_imports=["time", "numpy", "pandas"],
-            instructions="""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 $ or 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 of whether the element to be put in the list is a number or a string."""
+            instructions='''You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the a new line and 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 $ or 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 of whether the element to be put in the list is a number or a string.'''
         )
         return manager_agent
     else:
-        return None
-
-
+        return None