[BUG]: Fix Voyage AI EF to match new API (#4392)

Author: tazarov

chromadb/test/ef/test_voyageai_ef.py

@@ -0,0 +1,19 @@
+import os
+import pytest
+from chromadb.utils.embedding_functions.voyageai_embedding_function import (
+    VoyageAIEmbeddingFunction,
+)
+
+voyageai = pytest.importorskip("voyageai", reason="voyageai not installed")
+
+
+def test_with_embedding_dimensions() -> None:
+    if os.environ.get("CHROMA_VOYAGE_API_KEY") is None:
+        pytest.skip("CHROMA_VOYAGE_API_KEY not set")
+    ef = VoyageAIEmbeddingFunction(
+        api_key=os.environ["CHROMA_VOYAGE_API_KEY"]
+    )
+    embeddings = ef(["hello world"])
+    assert embeddings is not None
+    assert len(embeddings) == 1
+    assert len(embeddings[0]) == 1536

chromadb/utils/embedding_functions/voyageai_embedding_function.py

@@ -55,7 +55,9 @@ def __call__(self, input: Documents) -> Embeddings:
         embeddings = self._client.embed(texts=input, model=self.model_name)
 
         # Convert to numpy arrays
-        return [np.array(embedding, dtype=np.float32) for embedding in embeddings]
+        return [
+            np.array(embedding, dtype=np.float32) for embedding in embeddings.embeddings
+        ]
 
     @staticmethod
     def name() -> str: