From a1484171b57bbcba140ddc949ac1b4c42250b199 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 28 Dec 2023 17:15:16 +0530 Subject: [PATCH] (docs) voyage ai embeddings --- .../docs/embedding/supported_embedding.md | 23 +++++++++++++++++++ docs/my-website/docs/providers/voyage.md | 4 +++- 2 files changed, 26 insertions(+), 1 deletion(-) diff --git a/docs/my-website/docs/embedding/supported_embedding.md b/docs/my-website/docs/embedding/supported_embedding.md index 7697e4323..462cc1e70 100644 --- a/docs/my-website/docs/embedding/supported_embedding.md +++ b/docs/my-website/docs/embedding/supported_embedding.md @@ -224,3 +224,26 @@ print(response) | mistral-embed | `embedding(model="mistral/mistral-embed", input)` | +## Voyage AI Embedding Models + +### Usage - Embedding +```python +from litellm import embedding +import os + +os.environ['VOYAGE_API_KEY'] = "" +response = embedding( + model="voyage/voyage-01", + input=["good morning from litellm"], +) +print(response) +``` + +## Supported Models +All models listed here https://docs.voyageai.com/embeddings/#models-and-specifics are supported + +| Model Name | Function Call | +|--------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------| +| voyage-01 | `embedding(model="voyage/voyage-01", input)` | +| voyage-lite-01 | `embedding(model="voyage/voyage-lite-01", input)` | +| voyage-lite-01-instruct | `embedding(model="voyage/voyage-lite-01-instruct", input)` | diff --git a/docs/my-website/docs/providers/voyage.md b/docs/my-website/docs/providers/voyage.md index 352ff14fd..8dd9e6b6f 100644 --- a/docs/my-website/docs/providers/voyage.md +++ b/docs/my-website/docs/providers/voyage.md @@ -25,6 +25,8 @@ All models listed here https://docs.voyageai.com/embeddings/#models-and-specific | Model Name | Function Call | |--------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------| -| mistral-embed | `embedding(model="voyage/voyage-01", input)` | +| voyage-01 | `embedding(model="voyage/voyage-01", input)` | +| voyage-lite-01 | `embedding(model="voyage/voyage-lite-01", input)` | +| voyage-lite-01-instruct | `embedding(model="voyage/voyage-lite-01-instruct", input)` |