From 59b3a64e64bef83ed3b72e052c06920e48e21207 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 1 Nov 2023 14:12:47 -0700 Subject: [PATCH] (docs) add bedrock embeddings() --- .../docs/embedding/supported_embedding.md | 27 +++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/docs/my-website/docs/embedding/supported_embedding.md b/docs/my-website/docs/embedding/supported_embedding.md index 2ee71fbd2ec..87dd805cfd0 100644 --- a/docs/my-website/docs/embedding/supported_embedding.md +++ b/docs/my-website/docs/embedding/supported_embedding.md @@ -52,6 +52,33 @@ print(response) h/t to [Mikko](https://www.linkedin.com/in/mikkolehtimaki/) for this integration + +## Bedrock Embedding + +### API keys +This can be set as env variables or passed as **params to litellm.embedding()** +```python +import os +os.environ["AWS_ACCESS_KEY_ID"] = "" # Access key +os.environ["AWS_SECRET_ACCESS_KEY"] = "" # Secret access key +os.environ["AWS_REGION_NAME"] = "" # us-east-1, us-east-2, us-west-1, us-west-2 +``` + +### Usage +```python +from litellm import embedding +response = embedding( + model="bedrock/amazon.titan-embed-text-v1", + input=["good morning from litellm"], +) +print(response) +``` + +| Model Name | Function Call | +|----------------------|---------------------------------------------| +| Titan Embeddings - G1 | `embedding(model="bedrock/amazon.titan-embed-text-v1", input=input)` | + + ## Cohere Embedding Models https://docs.cohere.com/reference/embed