Update crewai/tools/gemini_tools.py

crewai/tools/gemini_tools.py

@@ -4,6 +4,7 @@ import json
 import os
 
 import google.generativeai as genai
+from google.api_core import exceptions
 
 # Retrieve API Key from Environment Variable
 GOOGLE_AI_STUDIO = os.environ.get('GOOGLE_AI_STUDIO2')
@@ -22,14 +23,23 @@ model = genai.GenerativeModel('gemini-pro')
 class GeminiSearchTools():
   @tool("Gemini search the internet")
   def gemini_search(query):
-    """Useful to search the internet 
-    about a a given topic and return relevant results"""
-    response = model.generate_content(query)
-    return response.text
+    try:
+        response = model.generate_content(query)
+        return response.text
+    except exceptions.DeadlineExceeded as e:
+        # Handle the DeadlineExceeded exception here
+        print("Error: Deadline Exceeded -", str(e))
+        # You can return a custom message or take other appropriate actions
+        return "Error: The request timed out. Please try again later."
+
 
   @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"""""
-    response = model.generate_content(query)
-    return response.text
+    try:
+        response = model.generate_content(query)
+        return response.text
+    except exceptions.DeadlineExceeded as e:
+        # Handle the DeadlineExceeded exception here
+        print("Error: Deadline Exceeded -", str(e))
+        # You can return a custom message or take other appropriate actions
+        return "Error: The request timed out. Please try again later."