diff --git a/docs/my-website/docs/embedding/supported_embedding.md b/docs/my-website/docs/embedding/supported_embedding.md index 87dd805cfd..945897eaa5 100644 --- a/docs/my-website/docs/embedding/supported_embedding.md +++ b/docs/my-website/docs/embedding/supported_embedding.md @@ -68,7 +68,7 @@ os.environ["AWS_REGION_NAME"] = "" # us-east-1, us-east-2, us-west-1, us-west-2 ```python from litellm import embedding response = embedding( - model="bedrock/amazon.titan-embed-text-v1", + model="amazon.titan-embed-text-v1", input=["good morning from litellm"], ) print(response) @@ -76,7 +76,7 @@ print(response) | Model Name | Function Call | |----------------------|---------------------------------------------| -| Titan Embeddings - G1 | `embedding(model="bedrock/amazon.titan-embed-text-v1", input=input)` | +| Titan Embeddings - G1 | `embedding(model="amazon.titan-embed-text-v1", input=input)` | ## Cohere Embedding Models diff --git a/docs/my-website/docs/providers/bedrock.md b/docs/my-website/docs/providers/bedrock.md index 87967273f5..4f28b9b80f 100644 --- a/docs/my-website/docs/providers/bedrock.md +++ b/docs/my-website/docs/providers/bedrock.md @@ -161,7 +161,7 @@ os.environ["AWS_REGION_NAME"] = "" # us-east-1, us-east-2, us-west-1, ```python from litellm import embedding response = embedding( - model="bedrock/amazon.titan-embed-text-v1", + model="amazon.titan-embed-text-v1", input=["good morning from litellm"], ) print(response) @@ -171,4 +171,4 @@ print(response) | Model Name | Function Call | |----------------------|---------------------------------------------| -| Titan Embeddings - G1 | `embedding(model="bedrock/amazon.titan-embed-text-v1", input=input)` | +| Titan Embeddings - G1 | `embedding(model="amazon.titan-embed-text-v1", input=input)` | diff --git a/litellm/__init__.py b/litellm/__init__.py index 5eba89b12b..7bc023bc23 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -317,6 +317,7 @@ longer_context_model_fallback_dict: dict = { ####### EMBEDDING MODELS ################### open_ai_embedding_models: List = ["text-embedding-ada-002"] cohere_embedding_models: List = ["embed-english-v2.0", "embed-english-light-v2.0", "embed-multilingual-v2.0"] +bedrock_embedding_models: List = ["amazon.titan-embed-text-v1"] from .timeout import timeout from .testing import * diff --git a/litellm/utils.py b/litellm/utils.py index 3798cbe79b..4ef1b38d6e 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1700,6 +1700,8 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None, api_ # cohere embeddings elif model in litellm.cohere_embedding_models: custom_llm_provider = "cohere" + elif model in litellm.bedrock_embedding_models: + custom_llm_provider = "bedrock" if custom_llm_provider is None or custom_llm_provider=="": print()