Update crewai/tools/gemini_tools.py

crewai/tools/gemini_tools.py

@@ -23,6 +23,16 @@ model = genai.GenerativeModel('gemini-pro')
 class GeminiSearchTools():
   @tool("Gemini search the internet")
   def gemini_search(query):
+    """
+    Searches for content based on the provided query using the Gemini model.
+    Handles DeadlineExceeded exceptions from the Google API.
+
+    Args:
+        query (str): The search query.
+
+    Returns:
+        str: The response text from the Gemini model or an error message.
+    """
     try:
         response = model.generate_content(query)
         return response.text
@@ -32,9 +42,20 @@ class GeminiSearchTools():
         # 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):
+    """
+    Searches for content based on the provided query using the Gemini model.
+    Handles DeadlineExceeded exceptions from the Google API.
+
+    Args:
+        query (str): The search query.
+
+    Returns:
+        str: The response text from the Gemini model or an error message.
+    """
     try:
         response = model.generate_content(query)
         return response.text
@@ -42,4 +63,4 @@ class GeminiSearchTools():
         # 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."
+        return "Error: The request timed out. Please try again later."