Update tools/WebSearchTool.py

tools/WebSearchTool.py

@@ -1,18 +1,15 @@
 import os
+import requests
 from smolagents import Tool
-from googleapiclient.discovery import build
 
-class GoogleSearchTool(Tool):
+class SerpApiSearchTool(Tool):
     """
-    A tool to perform web search Google searches using the Custom Search Engine (CSE) API.
-    Use this tool first to get the necessary information for calculation or further tools.
+    A tool to perform web search using the SerpApi engine.
+    This tool should be used to find current information and general knowledge.
     """
-    # Use a descriptive name to guide the agent
-    name = "google_search"
-    # Update the description to reflect the new functionality
-    description = "Use Google to find current information and general knowledge. Returns a snippet and URL."
-
-    # Define the required input structure for the agent framework
+    name = "serpapi_search"
+    description = "Use SerpApi to find current information and general knowledge. Returns a snippet and URL."
+    
     inputs = {
         "query": {"type": "string", "description": "The search term to look up."}
     }
@@ -20,62 +17,58 @@ class GoogleSearchTool(Tool):
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
+        # Retrieve API key from environment variables
+        self.api_key = os.getenv("SERPAPI_KEY")
 
-        # Retrieve credentials from environment variables
-        self.api_key = os.getenv("GOOGLE_API_KEY")
-        self.cse_id = os.getenv("GOOGLE_CSE_ID")
-
-        # Check for mandatory credentials
-        if not self.api_key or not self.cse_id:
-            raise ValueError("GOOGLE_API_KEY or GOOGLE_CSE_ID secret not found. Check environment variables.")
-
-        # Initialize the Google Custom Search service
-        # 'customsearch' is the API name, 'v1' is the version
-        self.service = build(
-            "customsearch", "v1", developerKey=self.api_key
-        )
+        if not self.api_key:
+            raise ValueError("SERPAPI_KEY secret not found. Check environment variables.")
 
     def forward(self, query: str) -> str:
         """
-        Executes a Google search query and formats the top results (up to 3).
-
-        The output is formatted to prioritize the Title and Snippet (informational content)
-        over the URL to address the confusion that it was only returning links.
-
+        Executes a SerpApi search query and formats the top results (up to 3).
+        
         Args:
             query: The search term provided by the agent.
-
+            
         Returns:
             A formatted string of search results, or an error message.
         """
-        print(f"Executing Google search for: '{query}'")
+        print(f"Executing SerpApi search for: '{query}'")
+        
+        search_url = "https://serpapi.com/search"
+        params = {
+            "q": query,
+            "engine": "google",  # Use Google search engine via SerpApi
+            "api_key": self.api_key,
+            "num": 3,
+        }
+
         try:
-            # Execute the search request for up to 3 results
-            res = self.service.cse().list(
-                q=query,
-                cx=self.cse_id,
-                num=3
-            ).execute()
+            # Make the API request
+            response = requests.get(search_url, params=params)
+            response.raise_for_status() # Raise an exception for bad status codes (4xx or 5xx)
+            data = response.json()
 
-            items = res.get('items', [])
+            # Extract the organic results
+            items = data.get('organic_results', [])
 
             if not items:
                 # Return the specific failure message expected by the agent
                 return "XX record info: No results found."
 
             search_results = []
+            # Process the top 3 results (since 'num=3' was requested)
             for i, item in enumerate(items):
-                # We've adjusted the format here to put the snippet first,
-                # which is usually the most important information for the agent.
                 search_results.append(
-                    f"RESULT {i+1}: '{item.get('title')}'\n"
-                    f"CONTENT: {item.get('snippet')}\n"
-                    f"SOURCE: {item.get('link')}" # The URL is now clearly labeled as the SOURCE
+                    f"RESULT {i+1}: '{item.get('title', 'N/A')}'\n"
+                    f"CONTENT: {item.get('snippet', 'No snippet available.')}\n"
+                    f"SOURCE: {item.get('link', 'N/A')}"
                 )
 
             # Join the results with a clear separator
             return "\n\n---SEPARATOR---\n\n".join(search_results)
 
+        except requests.exceptions.RequestException as e:
+            return f"Error during SerpApi Request: {e}"
         except Exception as e:
-            # Provide an informative error message upon API failure
-            return f"Error during Google Search API call: {e}"
+            return f"Error processing SerpApi results: {e}"