Update crewai/tools/gemini_tools.py

crewai/tools/gemini_tools.py

@@ -12,59 +12,21 @@ GOOGLE_AI_STUDIO = os.environ.get('GOOGLE_AI_STUDIO2')
 if not GOOGLE_AI_STUDIO:
     raise ValueError("API key not found. Please set the GOOGLE_AI_STUDIO2 environment variable.")
 
-
-
 import requests
 from langchain.tools import tool
 
 
 class GeminiSearchTools():
   @tool("Gemini search the internet")
-  def gemini_search_internet(query):
+  def gemini_search(query):
     """Useful to search the internet 
     about a a given topic and return relevant results"""
-    top_result_to_return = 4
-    url = "https://google.serper.dev/search"
-    payload = json.dumps({"q": query})
-    headers = {
-        'X-API-KEY': os.environ['SERPER_API_KEY'],
-        'content-type': 'application/json'
-    }
-    response = requests.request("POST", url, headers=headers, data=payload)
-    results = response.json()['organic']
-    string = []
-    for result in results[:top_result_to_return]:
-      try:
-        string.append('\n'.join([
-            f"Title: {result['title']}", f"Link: {result['link']}",
-            f"Snippet: {result['snippet']}", "\n-----------------"
-        ]))
-      except KeyError:
-        next
-
-    return '\n'.join(string)
+    response = model.generate_content(query)
+    return response.text
 
   @tool("Gemini search news on the internet")
   def gemini_search_news(query):
     """Useful to search news about a company, stock or any other
     topic and return relevant results"""""
-    top_result_to_return = 4
-    url = "https://google.serper.dev/news"
-    payload = json.dumps({"q": query})
-    headers = {
-        'X-API-KEY': os.environ['SERPER_API_KEY'],
-        'content-type': 'application/json'
-    }
-    response = requests.request("POST", url, headers=headers, data=payload)
-    results = response.json()['news']
-    string = []
-    for result in results[:top_result_to_return]:
-      try:
-        string.append('\n'.join([
-            f"Title: {result['title']}", f"Link: {result['link']}",
-            f"Snippet: {result['snippet']}", "\n-----------------"
-        ]))
-      except KeyError:
-        next
-
-    return '\n'.join(string)
+    response = model.generate_content(query)
+    return response.text