Update app.py

app.py

@@ -60,7 +60,7 @@ def get_stock_price(stock: str) -> str:
         return f"Error fetching price for stock '{stock}': {str(e)}"
 
 @tool
-def web_search(query: str) -> json:
+def web_search(query: str) -> dict:
     """Fetches the result from a web search.
     
     Args:
@@ -78,12 +78,12 @@ def web_search(query: str) -> json:
         
         # If search_results is a list, extract the first relevant result
         if isinstance(search_results, list) and len(search_results) > 0:
-            return {search_results[0]}
+            return {"query": query, "result": search_results[0]}
         
-        return f"Could not find {query}. Please try again."
+        return {"query": query, "error": "No results found."}
     
     except Exception as e:
-        return f"Error finding '{query}': {str(e)}"
+        return {"query": query, "error": f"Error during search: {str(e)}"}