add titan G1 embeddings (#152)

2025-06-16 23:09:22 -04:00
parent aed57307bc
commit 844efec086
1 changed files with 2 additions and 1 deletions
--- a/src/api/models/bedrock.py
+++ b/src/api/models/bedrock.py
@@ -68,8 +68,9 @@ cr_inference_prefix = get_inference_region_prefix()
 SUPPORTED_BEDROCK_EMBEDDING_MODELS = {
    "cohere.embed-multilingual-v3": "Cohere Embed Multilingual",
    "cohere.embed-english-v3": "Cohere Embed English",
-    # Disable Titan embedding.
+    "amazon.titan-embed-text-v1": "Titan Embeddings G1 - Text",
    "amazon.titan-embed-text-v2:0": "Titan Embeddings G2 - Text",
+    # Disable Titan embedding.
    # "amazon.titan-embed-image-v1": "Titan Multimodal Embeddings G1"
 }