tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-11 21:46:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Cohere Embed v4 support for AWS Bedrock

Browse Source 
Add Cohere Embed v4 support for AWS Bedrock
This commit is contained in:
Krish Dholakia
2025-10-07 19:10:22 -07:00
committed by GitHub
parent 5d6cd3ffed d71d801e4d
commit 3f4d2c6ada
4 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/my-website/docs/providers/bedrock_embedding.md
+1
View File
@@ -265,6 +265,7 @@ print(response)
| TwelveLabs Marengo Embed 2.7 | `embedding(model="bedrock/us.twelvelabs.marengo-embed-2-7-v1:0", input=input)` | Supports multimodal input (text, video, audio, image) |
| Cohere Embeddings - English | `embedding(model="bedrock/cohere.embed-english-v3", input=input)` | [here](https://github.com/BerriAI/litellm/blob/f5905e100068e7a4d61441d7453d7cf5609c2121/litellm/llms/bedrock/embed/cohere_transformation.py#L18)
| Cohere Embeddings - Multilingual | `embedding(model="bedrock/cohere.embed-multilingual-v3", input=input)` | [here](https://github.com/BerriAI/litellm/blob/f5905e100068e7a4d61441d7453d7cf5609c2121/litellm/llms/bedrock/embed/cohere_transformation.py#L18)
| Cohere Embed v4 | `embedding(model="bedrock/cohere.embed-v4:0", input=input)` | Supports text and image input, configurable dimensions (256, 512, 1024, 1536), 128k context length |
### Advanced - [Drop Unsupported Params](https://docs.litellm.ai/docs/completion/drop_params#openai-proxy-usage)
litellm/constants.py
+1
View File
@@ -871,6 +871,7 @@ bedrock_embedding_models: set = set(
"amazon.titan-embed-text-v1",
"cohere.embed-english-v3",
"cohere.embed-multilingual-v3",
"cohere.embed-v4:0",
"twelvelabs.marengo-embed-2-7-v1:0",
]
)
litellm/model_prices_and_context_window_backup.json
+10
View File
@@ -5294,6 +5294,16 @@
"output_cost_per_token": 0.0,
"supports_embedding_image_input": true
},
"cohere.embed-v4:0": {
"input_cost_per_token": 1.2e-07,
"litellm_provider": "bedrock",
"max_input_tokens": 128000,
"max_tokens": 128000,
"mode": "embedding",
"output_cost_per_token": 0.0,
"output_vector_size": 1536,
"supports_embedding_image_input": true
},
"cohere.rerank-v3-5:0": {
"input_cost_per_query": 0.002,
"input_cost_per_token": 0.0,
model_prices_and_context_window.json
+10
View File
@@ -5294,6 +5294,16 @@
"output_cost_per_token": 0.0,
"supports_embedding_image_input": true
},
"cohere.embed-v4:0": {
"input_cost_per_token": 1.2e-07,
"litellm_provider": "bedrock",
"max_input_tokens": 128000,
"max_tokens": 128000,
"mode": "embedding",
"output_cost_per_token": 0.0,
"output_vector_size": 1536,
"supports_embedding_image_input": true
},
"cohere.rerank-v3-5:0": {
"input_cost_per_query": 0.002,
"input_cost_per_token": 0.0,