fix

app.py

@@ -34,36 +34,37 @@ def serper_search(query: str) -> str:
         return "Serper key missing, please set SERPER_API_KEY."
 
 # --- Other Tools (unchanged) ---
+from smolagents.tools import tool
+import os, requests
+
 @tool
 def serper_search(query: str) -> str:
     """
-    Search the web using the Serper API to find current factual information.
+    Performs a Google search using the Serper API.
 
     Args:
-        query (str): The search query string.
+        query (str): The search query to look up.
 
     Returns:
-        str: A formatted string of top search results, or an error message.
+        str: A formatted string with search results or an error message.
     """
     api_key = os.getenv("SERPER_API_KEY")
-    if api_key:
-        try:
-            url = "https://google.serper.dev/search"
-            payload = {"q": query, "num": 10}
-            headers = {'X-API-KEY': api_key}
-            r = requests.post(url, headers=headers, json=payload, timeout=15)
-            r.raise_for_status()
-            data = r.json()
-            snippets = []
-            if kg := data.get("knowledgeGraph"):
-                snippets.append(f"{kg.get('title')}: {kg.get('description')}")
-            for item in data.get("organic", [])[:5]:
-                snippets.append(f"{item.get('title')}\n{item.get('snippet')}\n{item.get('link')}")
-            return "\n\n".join(snippets) if snippets else "No results."
-        except Exception as e:
-            return f"Serper error: {e}"
-    else:
-        return "Serper key missing, please set SERPER_API_KEY."
+    if not api_key:
+        return "Serper API key is missing."
+
+    try:
+        url = "https://google.serper.dev/search"
+        headers = {'X-API-KEY': api_key}
+        payload = {"q": query, "num": 5}
+        response = requests.post(url, headers=headers, json=payload, timeout=10)
+        data = response.json()
+        results = []
+        for item in data.get("organic", []):
+            results.append(f"{item.get('title')}\n{item.get('snippet')}\n{item.get('link')}")
+        return "\n\n".join(results) if results else "No results found."
+    except Exception as e:
+        return f"Error during search: {e}"
+
 
 
 @tool