add bocha search tool

src/agents/agent.py

@@ -17,7 +17,7 @@
 
 from src.bert.sentence import SentenceBert
 from src.memory import KnowledgeBase
-from src.browser.search import BingSearch, GoogleSearch, DuckDuckGoSearch
+from src.browser.search import BingSearch, GoogleSearch, DuckDuckGoSearch,BoChaSearch
 from src.browser import Browser
 from src.browser import start_interaction
 from src.filesystem import ReadCode
@@ -85,6 +85,8 @@ def search_queries(self, queries: list, project_name: str) -> dict:
             web_search = BingSearch()
         elif self.engine == "google":
             web_search = GoogleSearch()
+        elif self.engine == "bocha":
+            web_search = BoChaSearch()
         else:
             web_search = DuckDuckGoSearch()
 

src/browser/search.py

@@ -6,6 +6,31 @@
 from html import unescape
 import orjson
 
+class BoChaSearch:
+    def __init__(self):
+        self.config = Config()
+        self.bocha_api_key = self.config.get_bocha_api_key()
+        self.bocha_api_endpoint = self.config.get_bocha_api_endpoint()
+        self.query_result = None
+
+    def search(self, query):
+        headers = {
+        'Authorization': f'Bearer {self.bocha_api_key}',  # 请替换为你的API密钥
+        'Content-Type': 'application/json'
+    }
+
+        params = {"query": query}
+
+        try:
+            response = requests.post(self.bocha_api_endpoint, headers=headers, json=params)
+            response.raise_for_status()
+            self.query_result = response.json()
+            return self.query_result
+        except Exception as error:
+            return error
+
+    def get_first_link(self):
+        return self.query_result["data"]["webPages"]["value"][0]["url"]
 
 class BingSearch:
     def __init__(self):

src/config.py

@@ -48,6 +48,12 @@ def get_bing_api_endpoint(self):
     def get_bing_api_key(self):
         return self.config["API_KEYS"]["BING"]
 
+    def get_bocha_api_endpoint(self):
+        return self.config["API_ENDPOINTS"]["BoCha"]
+
+    def get_bocha_api_key(self):
+        return self.config["API_KEYS"]["BoCha"]
+
     def get_google_search_api_key(self):
         return self.config["API_KEYS"]["GOOGLE_SEARCH"]
 
@@ -119,6 +125,14 @@ def set_bing_api_endpoint(self, endpoint):
         self.config["API_ENDPOINTS"]["BING"] = endpoint
         self.save_config()
 
+    def set_bocha_api_key(self, key):
+        self.config["API_KEYS"]["BoCha"] = key
+        self.save_config()
+
+    def set_bocha_api_endpoint(self, endpoint):
+        self.config["API_ENDPOINTS"]["BoCha"] = endpoint
+        self.save_config()
+
     def set_google_search_api_key(self, key):
         self.config["API_KEYS"]["GOOGLE_SEARCH"] = key
         self.save_config()