VoyageAI embedding function

docs/embeddings.md

@@ -10,16 +10,17 @@ Chroma provides lightweight wrappers around popular embedding providers, making
 
 <div class="special_table"></div>
 
-|              | Python | JS |
-|--------------|-----------|---------------|
-| [OpenAI](/embeddings/openai) | ✅  | ✅ |
-| [Google Generative AI](/embeddings/google-gemini) | ✅  | ✅ |
-| [Cohere](/embeddings/cohere) | ✅  | ✅ |
-| [Google PaLM](/embeddings/google-palm) | ✅  | ➖ |
-| [Hugging Face](/embeddings/hugging-face) | ✅  | ➖ |
-| [Instructor](/embeddings/instructor) | ✅  | ➖ |
+|                                                                            | Python | JS |
+|----------------------------------------------------------------------------|-----------|---------------|
+| [OpenAI](/embeddings/openai)                                               | ✅  | ✅ |
+| [Google Generative AI](/embeddings/google-gemini)                          | ✅  | ✅ |
+| [Cohere](/embeddings/cohere)                                               | ✅  | ✅ |
+| [Google PaLM](/embeddings/google-palm)                                     | ✅  | ➖ |
+| [Hugging Face](/embeddings/hugging-face)                                   | ✅  | ➖ |
+| [Instructor](/embeddings/instructor)                                       | ✅  | ➖ |
 | [Hugging Face Embedding Server](/embeddings/hugging-face-embedding-server) | ✅  | ✅ |
-| [Jina AI](/embeddings/jinaai) | ✅  | ✅ |
+| [Jina AI](/embeddings/jinaai)                                              | ✅  | ✅ |
+| [VoyageAI](/embeddings/voyageai)                                           | ✅  | ✅ |
 | [Roboflow](/embeddings/roboflow-api) | ✅  | ➖ |
 | [Ollama](/embeddings/ollama)                                               | ✅  | ✅ |    
 

docs/embeddings/voyageai.md

@@ -0,0 +1,53 @@
+---
+---
+
+# VoyageAI
+
+import Tabs from '@theme/Tabs';
+import TabItem from '@theme/TabItem';
+
+<div class="select-language">Select a language</div>
+
+<Tabs queryString groupId="lang">
+<TabItem value="py" label="Python"></TabItem>
+<TabItem value="js" label="JavaScript"></TabItem>
+</Tabs>
+
+Chroma also provides a convenient wrapper around VoyageAI's embedding API. This embedding function runs remotely on VoyageAI’s servers, and requires an API key. You can get an API key by signing up for an account at [VoyageAI](https://dash.voyageai.com/api-keys).
+
+<Tabs queryString groupId="lang" className="hideTabSwitcher">
+<TabItem value="py" label="Python">
+
+This embedding function relies on the `voyageai` python package, which you can install with `pip install voyageai`.
+
+```python
+from chromadb.utils.embedding_functions import VoyageAIEmbeddingFunction
+voyageai_ef  = VoyageAIEmbeddingFunction(api_key="YOUR_API_KEY",  model_name="voyage-law-2", input_type=VoyageAIEmbeddingFunction.InputType.DOCUMENT)
+result = voyageai_ef(input=["document1","document2"])
+```
+
+</TabItem>
+<TabItem value="js" label="JavaScript">
+
+```javascript
+const {VoyageAIEmbeddingFunction, InputType} = require('chromadb');
+// const {VoyageAIEmbeddingFunction, InputType}  from "chromadb"; // ESM import
+const embedder = new VoyageAIEmbeddingFunction("apiKey", "voyage-law-2", InputType.DOCUMENT)
+
+// use directly 
+const embeddings = embedder.generate(["document1","document2"])
+
+// pass documents to query for .add and .query
+const collection = await client.createCollection({name: "name", embeddingFunction: embedder})
+const collectionGet = await client.getCollection({name:"name", embeddingFunction: embedder})
+```
+
+</TabItem>
+
+</Tabs>
+
+
+
+You should pass in the `model_name` argument, which lets you choose which VoyageAI embeddings model to use. You can see the available models [here](https://docs.voyageai.com/docs/embeddings).
+
+

sidebars.js

@@ -95,6 +95,7 @@ const sidebars = {
         'embeddings/roboflow-api',
         'embeddings/hugging-face-embedding-server',
         'embeddings/jinaai',
+        'embeddings/voyageai',
         'embeddings/ollama',
       ],
     },