From 11932d0576a073d83f38a418cbdf6b2d8d4ff46f Mon Sep 17 00:00:00 2001 From: Igor Ribeiro Lima Date: Fri, 27 Dec 2024 20:27:26 -0300 Subject: [PATCH] Add Gemini embedding doc (#7436) --- .../docs/embedding/supported_embedding.md | 30 +++++++++++++++++-- 1 file changed, 28 insertions(+), 2 deletions(-) diff --git a/docs/my-website/docs/embedding/supported_embedding.md b/docs/my-website/docs/embedding/supported_embedding.md index 603e04dd92..1f877ecc37 100644 --- a/docs/my-website/docs/embedding/supported_embedding.md +++ b/docs/my-website/docs/embedding/supported_embedding.md @@ -394,6 +394,32 @@ print(response) |--------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------| | mistral-embed | `embedding(model="mistral/mistral-embed", input)` | +## Gemini AI Embedding Models + +### API keys + +This can be set as env variables or passed as **params to litellm.embedding()** +```python +import os +os.environ["GEMINI_API_KEY"] = "" +``` + +### Usage - Embedding +```python +from litellm import embedding +response = embedding( + model="gemini/text-embedding-004", + input=["good morning from litellm"], +) +print(response) +``` + +All models listed [here](https://ai.google.dev/gemini-api/docs/models/gemini) are supported: + +| Model Name | Function Call | +| :--- | :--- | +| text-embedding-004 | `embedding(model="gemini/text-embedding-004", input)` | + ## Vertex AI Embedding Models @@ -411,7 +437,7 @@ response = embedding( print(response) ``` -## Supported Models +### Supported Models All models listed [here](https://github.com/BerriAI/litellm/blob/57f37f743886a0249f630a6792d49dffc2c5d9b7/model_prices_and_context_window.json#L835) are supported | Model Name | Function Call | @@ -509,4 +535,4 @@ curl -X POST 'http://0.0.0.0:4000/v1/embeddings' \ }' ``` - \ No newline at end of file +