Update crewai/tools/clarifai_tools.py

crewai/tools/clarifai_tools.py

@@ -13,7 +13,13 @@ import json
 import os
 from google.api_core import exceptions
 
+#Constants
+
 PAT = os.getenv('CLARIFAI_PAT')
+USER_ID = 'openai'
+APP_ID = 'dall-e'
+MODEL_ID = 'dall-e-3'
+MODEL_VERSION_ID = 'dc9dcb6ee67543cebc0b9a025861b868'
 
 # Ensure the API key is available
 if not PAT:
@@ -26,11 +32,15 @@ from langchain.tools import tool
 class ClarifaiTools():
   @tool("Clarifai Image Tool")
   def clarifai_image(prompt):
-    # Constants
-    USER_ID = 'openai'
-    APP_ID = 'dall-e'
-    MODEL_ID = 'dall-e-3'
-    MODEL_VERSION_ID = 'dc9dcb6ee67543cebc0b9a025861b868'
+
+    """
+    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.
+    """  
 
     # Clarifai gRPC setup
     channel = ClarifaiChannel.get_grpc_channel()