From 01f7e850572af1f75ff79fa69f48e32b3a0989c5 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 28 Dec 2023 16:54:26 +0530 Subject: [PATCH] (docs) add mistral embeddings --- .../docs/embedding/supported_embedding.md | 23 +++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/docs/my-website/docs/embedding/supported_embedding.md b/docs/my-website/docs/embedding/supported_embedding.md index d860e0074..7697e4323 100644 --- a/docs/my-website/docs/embedding/supported_embedding.md +++ b/docs/my-website/docs/embedding/supported_embedding.md @@ -201,3 +201,26 @@ response = embedding( | microsoft/codebert-base | `embedding('huggingface/microsoft/codebert-base', input=input)` | `os.environ['HUGGINGFACE_API_KEY']` | | BAAI/bge-large-zh | `embedding('huggingface/BAAI/bge-large-zh', input=input)` | `os.environ['HUGGINGFACE_API_KEY']` | | any-hf-embedding-model | `embedding('huggingface/hf-embedding-model', input=input)` | `os.environ['HUGGINGFACE_API_KEY']` | + + +## Mistral AI Embedding Models +All models listed here https://docs.mistral.ai/platform/endpoints are supported + +### Usage +```python +from litellm import embedding +import os + +os.environ['MISTRAL_API_KEY'] = "" +response = embedding( + model="mistral/mistral-embed", + input=["good morning from litellm"], +) +print(response) +``` + +| Model Name | Function Call | +|--------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------| +| mistral-embed | `embedding(model="mistral/mistral-embed", input)` | + +