fix: update embedding functions to inherit from chromadb callable

Author: greysonlalonde

src/crewai/rag/core/base_embeddings_callable.py

@@ -140,3 +140,10 @@ def wrapped_call(self: EmbeddingFunction[D], input: D) -> Embeddings:
             return validate_embeddings(normalized)
 
         cls.__call__ = wrapped_call  # type: ignore[method-assign]
+
+    def embed_query(self, input: D) -> Embeddings:
+        """
+        Get the embeddings for a query input.
+        This method is optional, and if not implemented, the default behavior is to call __call__.
+        """
+        return self.__call__(input=input)

src/crewai/rag/embeddings/providers/ibm/embedding_callable.py

@@ -2,10 +2,9 @@
 
 from typing import cast
 
+from chromadb.api.types import Documents, EmbeddingFunction, Embeddings
 from typing_extensions import Unpack
 
-from crewai.rag.core.base_embeddings_callable import EmbeddingFunction
-from crewai.rag.core.types import Documents, Embeddings
 from crewai.rag.embeddings.providers.ibm.types import WatsonXProviderConfig
 
 
@@ -18,8 +17,14 @@ def __init__(self, **kwargs: Unpack[WatsonXProviderConfig]) -> None:
         Args:
             **kwargs: Configuration parameters for WatsonX Embeddings and Credentials.
         """
+        super().__init__(**kwargs)
         self._config = kwargs
 
+    @staticmethod
+    def name() -> str:
+        """Return the name of the embedding function for ChromaDB compatibility."""
+        return "watsonx"
+
     def __call__(self, input: Documents) -> Embeddings:
         """Generate embeddings for input documents.
 

src/crewai/rag/embeddings/providers/voyageai/embedding_callable.py

@@ -2,10 +2,9 @@
 
 from typing import cast
 
+from chromadb.api.types import Documents, EmbeddingFunction, Embeddings
 from typing_extensions import Unpack
 
-from crewai.rag.core.base_embeddings_callable import EmbeddingFunction
-from crewai.rag.core.types import Documents, Embeddings
 from crewai.rag.embeddings.providers.voyageai.types import VoyageAIProviderConfig
 
 
@@ -33,6 +32,11 @@ def __init__(self, **kwargs: Unpack[VoyageAIProviderConfig]) -> None:
             timeout=kwargs.get("timeout"),
         )
 
+    @staticmethod
+    def name() -> str:
+        """Return the name of the embedding function for ChromaDB compatibility."""
+        return "voyageai"
+
     def __call__(self, input: Documents) -> Embeddings:
         """Generate embeddings for input documents.
 

src/crewai/rag/embeddings/types.py

@@ -11,7 +11,10 @@
     VertexAIProviderSpec,
 )
 from crewai.rag.embeddings.providers.huggingface.types import HuggingFaceProviderSpec
-from crewai.rag.embeddings.providers.ibm.types import WatsonXProviderSpec
+from crewai.rag.embeddings.providers.ibm.types import (
+    WatsonProviderSpec,
+    WatsonXProviderSpec,
+)
 from crewai.rag.embeddings.providers.instructor.types import InstructorProviderSpec
 from crewai.rag.embeddings.providers.jina.types import JinaProviderSpec
 from crewai.rag.embeddings.providers.microsoft.types import AzureProviderSpec
@@ -44,6 +47,7 @@
     | Text2VecProviderSpec
     | VertexAIProviderSpec
     | VoyageAIProviderSpec
+    | WatsonProviderSpec  # Deprecated, use WatsonXProviderSpec
     | WatsonXProviderSpec
 )