[DOC] New functionality: Adding VoyageAI text embedding integration

chromadb/test/ef/test_ef.py

@@ -15,6 +15,7 @@ def test_get_builtins_holds() -> None:
     expected_builtins = {
         "AmazonBedrockEmbeddingFunction",
         "CohereEmbeddingFunction",
+        "VoyageAIEmbeddingFunction",
         "GoogleGenerativeAiEmbeddingFunction",
         "GooglePalmEmbeddingFunction",
         "GoogleVertexEmbeddingFunction",

chromadb/utils/embedding_functions/voyageai_embedding_function.py

@@ -0,0 +1,27 @@
+import logging
+
+from chromadb.api.types import Documents, EmbeddingFunction, Embeddings
+
+logger = logging.getLogger(__name__)
+
+
+class VoyageAIEmbeddingFunction(EmbeddingFunction[Documents]):
+    def __init__(self, api_key: str, model_name: str):
+        try:
+            import voyageai
+        except ImportError:
+            raise ValueError(
+                "The voyageai python package is not installed. Please install it with `pip install voyageai`"
+            )
+
+        self._client = voyageai.Client(api_key=api_key)
+        self._model_name = model_name
+
+    def __call__(self, input: Documents) -> Embeddings:
+        # Call Cohere Embedding API for each document.
+        return [
+            embeddings
+            for embeddings in self._client.embed(
+                texts=input, model=self._model_name
+            )
+        ]

clients/js/package.json

@@ -9,9 +9,9 @@
     "@jest/globals": "^29.7.0",
     "@jest/types": "^29.6.3",
     "@openapi-generator-plus/typescript-fetch-client-generator": "^1.5.0",
+    "@types/bcrypt": "^5.0.2",
     "@types/jest": "^29.5.0",
     "@types/node": "^20.8.10",
-    "@types/bcrypt": "^5.0.2",
     "bcrypt": "^5.1.1",
     "jest": "^29.5.0",
     "npm-run-all": "^4.1.5",
@@ -69,7 +69,8 @@
   "peerDependencies": {
     "@google/generative-ai": "^0.1.1",
     "cohere-ai": "^5.0.0 || ^6.0.0 || ^7.0.0",
-    "openai": "^3.0.0 || ^4.0.0"
+    "openai": "^3.0.0 || ^4.0.0",
+    "voyageai": "^0.0.3-1"
   },
   "peerDependenciesMeta": {
     "@google/generative-ai": {
@@ -80,6 +81,9 @@
     },
     "openai": {
       "optional": true
+    },
+    "voyageai": {
+      "optional": true
     }
   }
 }