Update app.py

app.py

@@ -1102,46 +1102,7 @@ def search_knowledge_base(query):
 
 
 # Function to perform an advanced Google search using SERP API
-def brave_search(query, api_key="BSA8wB9NbA2fM76yt0lnU7OEpTQhAm6", max_results=12, search_depth="advanced", days=3, include_answer=True, include_raw_content=True, exclude_domains=None):
-    """
-    Perform a search using the Brave Search API.
-    
-    Args:
-        query (str): The search query string.
-        api_key (str): Your Brave Search API key.
-        max_results (int): Maximum number of results to retrieve.
-        search_depth (str): Search depth ("basic" or "advanced").
-        days (int): Limit results to the past specified days.
-        include_answer (bool): Include direct answers in the response.
-        include_raw_content (bool): Include raw content in the response.
-        exclude_domains (list): List of domains to exclude from the results.
-    
-    Returns:
-        dict: Parsed search results from the Brave Search API.
-    """
-    url = "https://api.search.brave.com/res/v1/web/search?q=brave+search"
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json",
-    }
 
-    # Payload to customize the search
-    payload = {
-        "q": query,
-        "count": max_results,
-        "search_depth": search_depth,
-        "days": days,
-        "include_answer": include_answer,
-        "include_raw_content": include_raw_content,
-    }
-
-    if exclude_domains:
-        payload["exclude_domains"] = exclude_domains
-
-    # Make the API request
-    response = requests.post(url, json=payload, headers=headers)
-    response.raise_for_status()  # Raise an exception for HTTP errors
-    return response.json()
 
 
 
@@ -1234,24 +1195,25 @@ def cache_response(query, response, ttl=3600):
 
 
 
-# tavily_tool = TavilySearchResults(
-#     max_results=12,
-#     search_depth="advanced",
-#     days=3,
-#     include_answer=True,
-#     include_raw_content=True,
-#     # include_domains=[...],
-#     exclude_domains=['example.com'],
-#     # name="...",            # overwrite default tool name
-#     # description="...",     # overwrite default tool description
-#     # args_schema=...,       # overwrite default args_schema: BaseModel
-# )#
 
+tavily_tool = TavilySearchResults(
+    max_results=12,
+    search_depth="advanced",
+    topic="news",
+    days=1,
+    include_answer=True,
+    include_raw_content=True,
+    # include_domains=[...],
+    exclude_domains=['example.com'],
+    # name="...",            # overwrite default tool name
+    # description="...",     # overwrite default tool description
+    # args_schema=...,       # overwrite default args_schema: BaseModel
+)
 
 #Compile all tool functions into a list
 tools = [
     knowledge_base_tool,  # Tool for querying the knowledge base and retrieving responses
-    google_search_tool,
+    tavily_tool,
     #google_search_tool,  # Tool for performing a Google search and retrieving search result snippets
 ]