Modified files

agent.py

@@ -8,7 +8,7 @@ from smolagents import InferenceClientModel
 from dotenv import load_dotenv
 from markdownify import markdownify
 from requests.exceptions import RequestException
-from langchain_community.utilities import SerpAPIWrapper
+from serpapi import GoogleSearch
 
 def initialize_agent():
     # Load environment variables from .env file
@@ -63,8 +63,33 @@ def initialize_agent():
             The search results, or an error message if the search fails.
         """
         try:
-            serpapi = SerpAPIWrapper()
-            return serpapi.run(query)
+            params = {
+                "q": query,
+                "api_key": os.environ.get("SERPAPI_API_KEY"),
+            }
+            search = GoogleSearch(params)
+            results = search.get_dict()
+            if "ai_overview" in results:
+                ai_overview = results["ai_overview"]
+                output = ""
+                for block in ai_overview.get("text_blocks", []):
+                    if block["type"] == "paragraph":
+                        output += block["snippet"] + "\n\n"
+                    elif block["type"] == "heading":
+                        output += f"### {block['snippet']}\n\n"
+                    elif block["type"] == "list":
+                        for item in block["list"]:
+                            output += f"- **{item['title']}** {item['snippet']}\n"
+                        output += "\n"
+                if "references" in ai_overview:
+                    output += "\n**References:**\n"
+                    for ref in ai_overview["references"]:
+                        output += f"- [{ref['title']}]({ref['link']})\n"
+                return output
+            elif "organic_results" in results:
+                return str(results["organic_results"])
+            else:
+                return "No results found."
         except Exception as e:
             return f"Error performing Google search: {str(e)}"
 
@@ -89,3 +114,4 @@ def initialize_agent():
     else:
         return None
 
+

requirements.txt

@@ -6,4 +6,4 @@ gradio
 markdownify
 duckduckgo-search
 wikipedia
-serpapi-python
+serpapi