Update tools/WebSearchTool.py

tools/WebSearchTool.py

@@ -3,14 +3,14 @@ from smolagents import Tool
 import os
 import json
 
-class ScaleSerpSearchTool(Tool):
+class SerpApiSearchTool(Tool):
     """
-    A reliable web search tool using the Scale SERP API. This tool is chosen 
-    for its high reliability and accessible free tier when other search providers 
-    fail due to quota or instability.
+    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.
     """
-    name = "scaleserp_search"
-    description = "Use the reliable Scale SERP API to fetch current and general knowledge from Google Search results. Requires an API key."
+    name = "serpapi_search"
+    description = "Use the reliable SerpApi to fetch current information from Google Search results. Requires an API key."
     
     inputs = {
         "query": {"type": "string", "description": "The search term to look up."}
@@ -20,15 +20,15 @@ class ScaleSerpSearchTool(Tool):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         # Retrieve API key from environment variables
-        self.api_key = os.getenv("SCALESERP_API_KEY")
-        self.endpoint = "https://api.scaleserp.com/search"
+        self.api_key = os.getenv("SERPAPI_API_KEY")
+        self.endpoint = "https://serpapi.com/search"
 
         if not self.api_key:
-            raise ValueError("SCALESERP_API_KEY secret not found. You need a key from the Scale SERP free tier.")
+            raise ValueError("SERPAPI_API_KEY secret not found. You need a key from the SerpApi free tier.")
 
     def forward(self, query: str) -> str:
         """
-        Executes a Scale SERP search query and formats the top results (up to 3).
+        Executes a SerpApi search query and formats the top results (up to 3).
         
         Args:
             query: The search term provided by the agent.
@@ -36,13 +36,13 @@ class ScaleSerpSearchTool(Tool):
         Returns:
             A formatted string of search results, or an error message.
         """
-        print(f"Executing Scale SERP Search for: '{query}'")
+        print(f"Executing SerpApi Search for: '{query}'")
         
         params = {
             "api_key": self.api_key,
             "q": query,
             "num": 3,  # Request up to 3 results
-            "gl": "us" # Limit results to US region for consistency
+            "engine": "google", # Use Google search engine
         }
 
         try:
@@ -69,13 +69,13 @@ class ScaleSerpSearchTool(Tool):
                 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('url', 'N/A')}"
+                    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 Scale SERP API Request: {e}"
+            return f"Error during SerpApi Search API Request: {e}"
         except Exception as e:
-            return f"Error processing Scale SERP results: {e}"
+            return f"Error processing SerpApi results: {e}"