Update app.py

app.py

@@ -1168,7 +1168,7 @@ def knowledge_base_tool(query: str):
 @tool
 def google_search_tool(query: str):
     """Perform a Google search using the SERPER API."""
-    return brave_search(query)
+    return tavily_tool_with_validation(query)
 
 
 # Initialize Redis connection
@@ -1205,12 +1205,12 @@ tavily_tool = TavilySearchResults(
     exclude_domains=['example.com'],
 )
 
-def validate_tavily_results(tavily_tool):
+def validate_tavily_results(query):
     """
     Fetch and validate results from TavilySearchResults.
     Ensures that results do not lead to 'Page Not Found' pages.
     """
-    results = tavily_tool.search()
+    results = tavily_tool.search(query=query)
     valid_results = []
 
     for result in results:
@@ -1241,12 +1241,17 @@ def validate_tavily_results(tavily_tool):
 
     return valid_results
 
+# Define a custom tool for Tavily
+def tavily_tool_with_validation(query):
+    valid_results = validate_tavily_results(query)
+    return "\n".join([f"{res.get('title')}: {res.get('url')}" for res in valid_results])
+
 
 
 #Compile all tool functions into a list
 tools = [
     knowledge_base_tool,  # Tool for querying the knowledge base and retrieving responses
-    {"tool": tavily_tool, "validate": validate_tavily_results(tavily_tool)},
+    google_search_tool,
     #google_search_tool,  # Tool for performing a Google search and retrieving search result snippets
 ]