Update app.py

app.py

@@ -1102,44 +1102,48 @@ def search_knowledge_base(query):
 
 
 # Function to perform an advanced Google search using SERP API
-def google_search(query, site=None, exclude_terms=None, freshness="last_week", gl=None, hl=None):
-    try:
-        # Configure SERP API client
-        search_client = Client(api_key=serper_api_key)
-        
-        # Build advanced search parameters
-        search_params = {
-            "engine": "google",
-            "q": query,
-            "num": 10,  # Number of results to fetch
-        }
-        if site:
-            search_params["q"] += f" site:{site}"  # Restrict results to a specific domain
-        if exclude_terms:
-            search_params["exclude_terms"] = exclude_terms
-        if freshness:
-            search_params["freshness"] = freshness  # Options: last_day, last_week, last_month
-        if gl:
-            search_params["gl"] = gl  # Country code (e.g., "us" for United States)
-        if hl:
-            search_params["hl"] = hl  # Language code (e.g., "en" for English)
-
-        # Perform the search
-        results = search_client.search(search_params)
-        
-        # Extract snippets and URLs
-        organic_results = results.get("organic_results", [])
-        valid_results = [
-            {"snippet": result["snippet"], "link": result["link"]}
-            for result in organic_results
-            if "snippet" in result and "link" in result
-        ]
-        
-        return valid_results
+def brave_search(query, api_key, 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.
     
-    except Exception as e:
-        logger.error(f"Error in Google search: {e}")
-        return [{"snippet": "Error occurred during Google search", "link": ""}]
+    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.brave.com/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()
+
+
 
 def rag_response(query, selected_doc_ids=None):
     """
@@ -1230,22 +1234,35 @@ 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",
+#     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
+# )#
+    api_key = "BSA8wB9NbA2fM76yt0lnU7OEpTQhAm6"  # Replace with your Brave API key
+
+results = brave_search(
+            api_key=api_key,
+            max_results=12,
+            search_depth="advanced",
+            days=1,
+            include_answer=True,
+            include_raw_content=True,
+            exclude_domains=["example.com"]
+)
+        
+
 #Compile all tool functions into a list
 tools = [
     knowledge_base_tool,  # Tool for querying the knowledge base and retrieving responses
-    tavily_tool,
+    results,
     #google_search_tool,  # Tool for performing a Google search and retrieving search result snippets
 ]