From 7f89aeef84b418bf7343f2adc1054f412514c893 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Fri, 28 Feb 2025 17:16:35 +0000
Subject: [PATCH] Fix Google AI Studio embedder initialization error handling

Co-Authored-By: Joe Moura <joao@crewai.com>
---
 src/crewai/crew.py                      | 13 +++++++
 tests/knowledge/test_google_embedder.py | 51 +++++++++++++++++++++++++
 2 files changed, 64 insertions(+)
 create mode 100644 tests/knowledge/test_google_embedder.py

diff --git a/src/crewai/crew.py b/src/crewai/crew.py
index 9cecfed3a..2b5484b1c 100644
--- a/src/crewai/crew.py
+++ b/src/crewai/crew.py
@@ -314,6 +314,19 @@ class Crew(BaseModel):
                         collection_name="crew",
                     )
 
+            except ValueError as e:
+                # Check if the error is related to missing google-generativeai package
+                if "The Google Generative AI python package is not installed" in str(e):
+                    self._logger.log(
+                        "error",
+                        "Google AI Studio embedder requires the google-generativeai package. "
+                        "Please install it with `pip install google-generativeai`",
+                        color="red",
+                    )
+                else:
+                    self._logger.log(
+                        "warning", f"Failed to init knowledge: {e}", color="yellow"
+                    )
             except Exception as e:
                 self._logger.log(
                     "warning", f"Failed to init knowledge: {e}", color="yellow"
diff --git a/tests/knowledge/test_google_embedder.py b/tests/knowledge/test_google_embedder.py
new file mode 100644
index 000000000..7a7fbf333
--- /dev/null
+++ b/tests/knowledge/test_google_embedder.py
@@ -0,0 +1,51 @@
+import pytest
+from unittest.mock import patch, MagicMock
+from crewai import Crew, Agent, Task
+from crewai.knowledge.source.json_knowledge_source import JSONKnowledgeSource
+
+def test_google_embedder_missing_package():
+    """Test that a helpful error message is displayed when google-generativeai is not installed."""
+    # Create a simple agent and task
+    agent = Agent(
+        role="Test Agent",
+        goal="Test the knowledge component",
+        backstory="I am a test agent",
+    )
+
+    task = Task(
+        description="Test task",
+        expected_output="Test output",
+        agent=agent,
+    )
+
+    # Create a mock JSONKnowledgeSource
+    json_knowledge_source = MagicMock(spec=JSONKnowledgeSource)
+
+    # Mock the GoogleGenerativeAiEmbeddingFunction to raise the import error
+    with patch("chromadb.utils.embedding_functions.google_embedding_function.GoogleGenerativeAiEmbeddingFunction.__init__") as mock_init:
+        mock_init.side_effect = ValueError("The Google Generative AI python package is not installed. Please install it with `pip install google-generativeai`")
+        
+        # Mock the logger to capture the error message
+        with patch("crewai.utilities.logger.Logger.log") as mock_log:
+            # Create a crew with Google embedder
+            crew = Crew(
+                agents=[agent],
+                tasks=[task],
+                verbose=True,
+                knowledge_sources=[json_knowledge_source],
+                embedder={
+                    "provider": "google",
+                    "config": {
+                        "api_key": "fake-api-key",
+                        "model": 'models/embedding-001'
+                    }
+                }
+            )
+            
+            # Verify that the error message was logged correctly
+            mock_log.assert_any_call(
+                "error",
+                "Google AI Studio embedder requires the google-generativeai package. "
+                "Please install it with `pip install google-generativeai`",
+                color="red"
+            )