[BUG] Update Cohere embedding function to agree with new API

chroma-core · Jan 8, 2025 · bc75a89 · bc75a89
1 parent fe13636
commit bc75a89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chromadb/utils/embedding_functions/cohere_embedding_function.py b/chromadb/utils/embedding_functions/cohere_embedding_function.py
@@ -23,5 +23,5 @@ def __call__(self, input: Documents) -> Embeddings:
             embeddings
             for embeddings in self._client.embed(
                 texts=input, model=self._model_name, input_type="search_document"
-            )
+            ).embeddings
         ]