added support for google search api, corrected set_gemini_api_key

config.toml

@@ -8,10 +8,13 @@ REPOS_DIR = "repos"
 
 [API_KEYS]
 BING = "<YOUR_BING_API_KEY>"
+GOOGLE_SEARCH = "<YOUR_GOOGLE_SEARCH_API_KEY>"
+GOOGLE_SEARCH_ENGINE_ID = "<YOUR_GOOGLE_SEARCH_ENGINE_ID>"
 CLAUDE = "<YOUR_CLAUDE_API_KEY>"
 NETLIFY = "<YOUR_NETLIFY_API_KEY>"
 OPENAI = "<YOUR_OPENAI_API_KEY>"
 GEMINI = "<YOUR_GEMINI_API_KEY>"
 
 [API_ENDPOINTS]
 BING = "https://api.bing.microsoft.com/v7.0/search"
+GOOGLE_SEARCH = "https://www.googleapis.com/customsearch/v1"

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,DuckDuckGoSearch
+from src.browser.search import BingSearch,DuckDuckGoSearch,GoogleSearch
 from src.browser import Browser
 from src.browser import start_interaction
 from src.filesystem import ReadCode
@@ -64,7 +64,8 @@ def search_queries(self, queries: list, project_name: str) -> dict:
 
         knowledge_base = KnowledgeBase()
         # web_search = BingSearch()
-        web_search = DuckDuckGoSearch()
+        # web_search = DuckDuckGoSearch()
+        web_search = GoogleSearch()
         browser = Browser()
 
         for query in queries:

src/browser/search.py

@@ -1,7 +1,6 @@
 import requests
 from src.config import Config
 from duckduckgo_search import DDGS
-
 class BingSearch:
     def __init__(self):
         self.config = Config()
@@ -33,4 +32,30 @@ def search(self, query):
             return err
 
     def get_first_link(self):
-        return self.query_result[0]["href"]
+        return self.query_result[0]["href"]
+
+class GoogleSearch:
+     def __init__(self):
+        self.config = Config()
+        self.google_search_api_key = self.config.get_google_search_api_key()
+        self.google_search_engine_ID = self.config.get_google_search_engine_id()
+        self.google_search_api_endpoint = self.config.get_google_search_api_endpoint()
+        self.query_result = None
+
+     def search(self, query):
+        try:
+            params = {
+                'q': query,
+                'key': self.google_search_api_key,
+                'cx': self.google_search_engine_ID
+            }
+            response = requests.get(self.google_search_api_endpoint, params=params)
+            self.query_result = response.json()
+        except Exception as err:
+            return err
+
+     def get_first_link(self):
+        item = ""
+        if 'items' in self.query_result:
+            item = self.query_result['items'][0]['link']
+        return item

src/config.py

@@ -13,6 +13,15 @@ def get_bing_api_key(self):
     def get_bing_api_endpoint(self):
         return self.config["API_ENDPOINTS"]["BING"]
 
+    def get_google_search_api_key(self):
+        return self.config["API_KEYS"]["GOOGLE_SEARCH"]
+
+    def get_google_search_engine_id(self):
+        return self.config["API_KEYS"]["GOOGLE_SEARCH_ENGINE_ID"]
+
+    def get_google_search_api_endpoint(self):
+        return self.config["API_ENDPOINTS"]["GOOGLE_SEARCH"]
+
     def get_claude_api_key(self):
         return self.config["API_KEYS"]["CLAUDE"]
 
@@ -50,6 +59,18 @@ def set_bing_api_key(self, key):
     def set_bing_api_endpoint(self, endpoint):
         self.config["API_ENDPOINTS"]["BING"] = endpoint
         self.save_config()
+
+    def set_google_search_api_key(self, key):
+        self.config["API_KEYS"]["GOOGLE_SEARCH"] = key
+        self.save_config()
+
+    def set_google_search_engine_id(self, key):
+        self.config["API_KEYS"]["GOOGLE_SEARCH_ENGINE_ID"] = key
+        self.save_config()
+
+    def set_google_search_api_endpoint(self, endpoint):
+        self.config["API_ENDPOINTS"]["GOOGLE_SEARCH"] = endpoint
+        self.save_config()
 
     def set_claude_api_key(self, key):
         self.config["API_KEYS"]["CLAUDE"] = key
@@ -59,7 +80,7 @@ def set_openai_api_key(self, key):
         self.config["API_KEYS"]["OPENAI"] = key
         self.save_config()
 
-    def set_openai_api_key(self, key):
+    def set_gemini_api_key(self, key):
         self.config["API_KEYS"]["GEMINI"] = key
         self.save_config()