Update tools.py

updated search tool to use googleSearch.

tools.py

@@ -1,5 +1,4 @@
 from smolagents import Tool, tool
-from duckduckgo_search import DDGS
 import requests
 from bs4 import BeautifulSoup
 import os
@@ -34,31 +33,41 @@ class TranscriptTool(Tool):
 transcription_tool = TranscriptTool()
         
 
-class SimpleSearchTool(Tool):
+class GoogleSearchTool(Tool):
     name = "web_search"
-    description = "Performs a google-like search for a query and returns the top results with snippets."
-    inputs = {'query': {'type': 'string', 'description': 'The search query to perform.'}}
+    description = "Searches the web using Google. Essential for finding specific articles and papers."
+    inputs = {'query': {'type': 'string', 'description': 'The search query.'}}
     output_type = "string"
 
     def forward(self, query: str) -> str:
+        api_key = os.getenv("SERPER_API_KEY")
+        if not api_key:
+            return "Error: SERPER_API_KEY not found in environment variables."
+        
+        url = "https://google.serper.dev/search"
+        payload = {"q": query}
+        headers = {
+            'X-API-KEY': api_key,
+            'Content-Type': 'application/json'
+        }
+
         try:
-            results = []
-            # Try the standard backend first
-            with DDGS() as ddgs:
-                for r in ddgs.text(query, max_results=5):
-                    if r.get('body'):
-                        results.append(f"Title: {r['title']}\nLink: {r['href']}\nSnippet: {r['body']}")
-            
-            if not results:
-                 return "No results found. Try a broader query."
+            response = requests.post(url, headers=headers, json=payload)
+            response.raise_for_status()
+            results = response.json()
             
-            return "\n\n".join(results)
+            if 'organic' not in results:
+                return "No results found."
 
+            output = []
+            for item in results['organic'][:5]: # Take top 5 results
+                output.append(f"Title: {item['title']}\nLink: {item['link']}\nSnippet: {item['snippet']}\n")
+            
+            return "\n---\n".join(output)
         except Exception as e:
-            return f"Search failed: {str(e)}"
+            return f"Google Search failed: {str(e)}"
 
-# Instantiate web search tool
-search_tool = SimpleSearchTool()
+search_tool = GoogleSearchTool()
 
 class VisitWebpageTool(Tool):
     name = "visit_webpage"