Update app.py

app.py

@@ -60,10 +60,8 @@ def get_stock_price(stock: str) -> str:
         return f"Error fetching price for stock '{stock}': {str(e)}"
 
 
-from typing import Any
-
 @tool
-def web_search(query: str) -> Any:
+def web_search(query: str) -> str:
     """
     Fetches the result from a web search using DuckDuckGo.
 
@@ -71,24 +69,19 @@ def web_search(query: str) -> Any:
         query (str): A search query or question.
 
     Returns:
-        dict: A dictionary containing the search query and the top result.
+        str: The top search result as a string.
     """
     try:
-        # Initialize the search tool
         search_tool = DuckDuckGoSearchTool()
-        
-        # Perform the search query
         search_results = search_tool.run(query)
-        
-        # If search_results is a list, extract the first relevant result
+
         if isinstance(search_results, list) and len(search_results) > 0:
-            return {"query": query, "result": search_results[0]}
-        
-        return {"query": query, "error": "No results found."}
-    
-    except Exception as e:
-        return {"query": query, "error": f"Error during search: {str(e)}"}
+            return f"Top result for '{query}': {search_results[0]}"
+
+        return f"No results found for '{query}'."
 
+    except Exception as e:
+        return f"Error during search: {str(e)}"
 
     
 final_answer = FinalAnswerTool()