Update agent.py

agent.py

@@ -5,12 +5,33 @@ from langchain.agents import initialize_agent, AgentType
 from langchain_community.tools import Tool, WikipediaQueryRun
 from langchain_community.utilities import WikipediaAPIWrapper
 from langchain_experimental.tools.python.tool import PythonREPLTool
-from langchain_community.tools.brave_search import BraveSearch
 from langchain_google_genai import ChatGoogleGenerativeAI
 import pandas as pd
 from pathlib import Path
 from docx import Document
 import fitz  # PyMuPDF
+import requests
+
+class BraveSearchTool:
+    def __init__(self, api_key: str):
+        self.api_key = api_key
+        self.base_url = "https://api.search.brave.com/res/v1/web/search"
+
+    def run(self, query: str) -> str:
+        try:
+            response = requests.get(
+                self.base_url,
+                headers={"Accept": "application/json", "X-Subscription-Token": self.api_key},
+                params={"q": query}
+            )
+            response.raise_for_status()
+            results = response.json().get("web", {}).get("results", [])
+            if results:
+                return results[0].get("title", "") + ": " + results[0].get("url", "")
+            else:
+                return "No results found."
+        except Exception as e:
+            return f"BraveSearchTool ERROR: {str(e)}"
 
 class Agent:
     def __init__(self):
@@ -40,7 +61,7 @@ class Agent:
             ),
             Tool(
                 name="Brave Search",
-                func=BraveSearch(api_key=brave_key).run,
+                func=BraveSearchTool(api_key=brave_key).run,
                 description="Useful for factual and current event queries using Brave search engine."
             )
         ]