Update tools/WebSearchTool.py

tools/WebSearchTool.py

@@ -1,31 +1,24 @@
-import os
 import requests
 from smolagents import Tool
+import json
 
-class SerpApiSearchTool(Tool):
+class DuckDuckGoSearchTool(Tool):
     """
-    A tool to perform web search using the SerpApi engine.
-    This tool should be used to find current information and general knowledge.
+    A tool to perform general web searches using the DuckDuckGo Instant Answer API.
+    This tool is a free and accessible alternative when other search APIs require keys or payment.
+    It's best for obtaining quick, summarized facts and related links.
     """
-    name = "serpapi_search"
-    description = "Use SerpApi to find current information and general knowledge. Returns a snippet and URL."
+    name = "duckduckgo_search"
+    description = "Use DuckDuckGo to find current information and general knowledge, returning summarized answers and relevant search results."
     
     inputs = {
         "query": {"type": "string", "description": "The search term to look up."}
     }
     output_type = "string"
 
-    def __init__(self, **kwargs):
-        super().__init__(**kwargs)
-        # Retrieve API key from environment variables
-        self.api_key = os.getenv("SERPAPI_KEY")
-
-        if not self.api_key:
-            raise ValueError("SERPAPI_KEY secret not found. Check environment variables.")
-
     def forward(self, query: str) -> str:
         """
-        Executes a SerpApi search query and formats the top results (up to 3).
+        Executes a DuckDuckGo search query and formats the top results.
         
         Args:
             query: The search term provided by the agent.
@@ -33,42 +26,63 @@ class SerpApiSearchTool(Tool):
         Returns:
             A formatted string of search results, or an error message.
         """
-        print(f"Executing SerpApi search for: '{query}'")
+        print(f"Executing DuckDuckGo Search for: '{query}'")
         
-        search_url = "https://serpapi.com/search"
+        # DuckDuckGo Zero-click API endpoint
+        search_url = "https://api.duckduckgo.com/"
         params = {
             "q": query,
-            "engine": "google",  # Use Google search engine via SerpApi
-            "api_key": self.api_key,
-            "num": 3,
+            "format": "json",
+            "no_redirect": "1",  # Do not redirect to a search page
+            "no_html": "1",      # Remove HTML from results
+            "t": "smolagents"    # Application name for logging
         }
 
         try:
             # 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)
+            response.raise_for_status()
             data = response.json()
 
-            # Extract the organic results
-            items = data.get('organic_results', [])
+            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):
-                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')}"
+            # 1. Instant Answer (if available)
+            abstract_text = data.get('AbstractText')
+            abstract_url = data.get('AbstractURL')
+            
+            if abstract_text and abstract_url:
+                results.append(
+                    f"RESULT 1 (Instant Answer): '{data.get('Heading', 'Summary')}'\n"
+                    f"CONTENT: {abstract_text}\n"
+                    f"SOURCE: {abstract_url}"
                 )
 
+            # 2. Related Topics (for more search results)
+            # This often contains deep links or snippets for further reading
+            topics = data.get('RelatedTopics', [])
+            
+            # Filter and take up to the next 2 relevant topics
+            for i, topic in enumerate(topics[:2]):
+                # Check for a standard result structure
+                if 'Result' in topic and 'Text' in topic:
+                    # Parse the snippet/link info from the 'Result' field (usually HTML but 'no_html' helps)
+                    snippet = topic['Text']
+                    link = topic.get('FirstURL', 'N/A')
+                    title = snippet.split(' - ')[0] if ' - ' in snippet else 'Related Topic'
+                    
+                    results.append(
+                        f"RESULT {len(results) + 1}: '{title}'\n"
+                        f"CONTENT: {snippet}\n"
+                        f"SOURCE: {link}"
+                    )
+            
+            if not results:
+                return "XX record info: No results found."
+
             # Join the results with a clear separator
-            return "\n\n---SEPARATOR---\n\n".join(search_results)
+            return "\n\n---SEPARATOR---\n\n".join(results)
 
         except requests.exceptions.RequestException as e:
-            return f"Error during SerpApi Request: {e}"
+            return f"Error during DuckDuckGo API Request: {e}"
         except Exception as e:
-            return f"Error processing SerpApi results: {e}"
+            return f"Error processing DuckDuckGo results: {e}"