Update crewai/tools/gemini_tools.py

crewai/tools/gemini_tools.py

@@ -1 +1,70 @@
-# tools created using gemini
+# tools created using gemini
+
+import json
+import os
+
+import google.generativeai as genai
+
+# Retrieve API Key from Environment Variable
+GOOGLE_AI_STUDIO = os.environ.get('GOOGLE_AI_STUDIO2')
+
+# Ensure the API key is available
+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):
+    """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)
+
+  @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)