Update tools/WebSearchTool.py

tools/WebSearchTool.py

@@ -1,17 +1,19 @@
-import requests
-from smolagents import Tool
 import os
-import json
+from smolagents import Tool
+# We use googleapiclient for the official Google Custom Search API
+from googleapiclient.discovery import build
 
-class SerpApiSearchTool(Tool):
+class GoogleSearchTool(Tool):
     """
-    A reliable web search tool using the SerpApi service, which wraps major 
-    search engines (like Google) to provide structured results. This is ideal 
-    for high reliability when keyless options are unstable.
+    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.
     """
-    name = "serpapi_search"
-    description = "Use the reliable SerpApi to fetch current information from Google Search results. Requires an API key."
-    
+    # Use a descriptive name to guide the agent
+    name = "google_search"
+    # Reverting to a proper, concise tool description.
+    description = "Use Google to find current information and general knowledge. Returns a snippet and URL."
+
+    # Define the required input structure for the agent framework
     inputs = {
         "query": {"type": "string", "description": "The search term to look up."}
     }
@@ -19,63 +21,59 @@ class SerpApiSearchTool(Tool):
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
-        # Retrieve API key from environment variables
-        self.api_key = os.getenv("SERPAPI_KEY")
-        self.endpoint = "https://serpapi.com/search"
 
-        if not self.api_key:
-            raise ValueError("SERPAPI_API_KEY secret not found. You need a key from the SerpApi free tier.")
+        # Retrieve credentials from environment variables
+        # NOTE: This requires GOOGLE_API_KEY (the developer key) and GOOGLE_CSE_ID (the ID of your search engine)
+        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
+        )
 
     def forward(self, query: str) -> str:
         """
-        Executes a SerpApi search query and formats the top results (up to 3).
-        
+        Executes a Google 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 SerpApi Search for: '{query}'")
-        
-        params = {
-            "api_key": self.api_key,
-            "q": query,
-            "num": 3,  # Request up to 3 results
-            "engine": "google", # Use Google search engine
-        }
-
+        print(f"Executing Google search for: '{query}'")
         try:
-            # Make the API request
-            response = requests.get(self.endpoint, params=params, timeout=10)
-            response.raise_for_status() 
-            data = response.json()
+            # Execute the search request for up to 3 results
+            res = self.service.cse().list(
+                q=query,
+                cx=self.cse_id,
+                num=3
+            ).execute()
 
-            # Extract the organic results
-            organic_results = data.get('organic_results', [])
+            items = res.get('items', [])
 
-            if not organic_results:
-                # Check for a knowledge panel or featured snippet as a fallback
-                featured_snippet = data.get('knowledge_graph', {}).get('description')
-                if featured_snippet:
-                    return f"RESULT 1 (Featured Snippet): '{query}'\nCONTENT: {featured_snippet}\nSOURCE: Knowledge Panel"
-                
-                # If nothing useful is found:
+            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 organic results
-            for i, item in enumerate(organic_results[:3]):
+            for i, item in enumerate(items):
+                # Formatting the output for clarity (Title, Content, Source URL)
                 search_results.append(
-                    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')}"
+                    f"RESULT {i+1}: '{item.get('title')}'\n"
+                    f"CONTENT: {item.get('snippet')}\n"
+                    f"SOURCE: {item.get('link')}"
                 )
 
             # 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 Search API Request: {e}"
         except Exception as e:
-            return f"Error processing SerpApi results: {e}"
+            # Provide an informative error message upon API failure
+            return f"Error during Google Search API call: {e}"