Update tools/WebSearchTool.py

tools/WebSearchTool.py

@@ -1,30 +1,35 @@
 from smolagents import Tool
-from duckduckgo_search import DDGS
+from serpapi import GoogleSearch
+import os
 
-class DuckDuckGoSearchTool(Tool):
+class SerpApiSearchTool(Tool):
     name = "web_search"
-    description = "Use DuckDuckGo to search the web. Returns short snippets and URLs."
-    
+    description = "Use SerpAPI (Google search) to get reliable snippets."
+
     inputs = {
-        "query": {"type": "string", "description": "Search term"}
+        "query": {"type": "string"}
     }
     output_type = "string"
 
     def forward(self, query: str) -> str:
-        try:
-            results = []
-            with DDGS() as ddgs:
-                for r in ddgs.text(query, max_results=3):
-                    results.append(
-                        f"Title: {r.get('title')}\n"
-                        f"Snippet: {r.get('body')}\n"
-                        f"URL: {r.get('href')}"
-                    )
-            
-            if not results:
-                return "No search results found."
-            
-            return "\n---\n".join(results)
+        params = {
+            "q": query,
+            "api_key": os.getenv("SERPAPI_KEY"),
+            "num": 3
+        }
+
+        search = GoogleSearch(params)
+        results = search.get_dict()
+
+        if "organic_results" not in results:
+            return "XX record info: No results."
+
+        formatted = []
+        for item in results.get("organic_results", []):
+            formatted.append(
+                f"Title: {item.get('title')}\n"
+                f"Snippet: {item.get('snippet')}\n"
+                f"URL: {item.get('link')}"
+            )
 
-        except Exception as e:
-            return f"DuckDuckGo search error: {e}"
+        return "\n---\n".join(formatted) if formatted else "No record info: No results."