diff --git a/src/api/models/bedrock.py b/src/api/models/bedrock.py index 4b589510..18ded8c0 100644 --- a/src/api/models/bedrock.py +++ b/src/api/models/bedrock.py @@ -69,7 +69,7 @@ def get_inference_region_prefix(): "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", # "amazon.titan-embed-image-v1": "Titan Multimodal Embeddings G1" } @@ -852,6 +852,8 @@ def get_embeddings_model(model_id: str) -> BedrockEmbeddingsModel: match model_name: case "Cohere Embed Multilingual" | "Cohere Embed English": return CohereEmbeddingsModel() + case "Titan Embeddings G2 - Text": + return TitanEmbeddingsModel() case _: logger.error("Unsupported model id " + model_id) raise HTTPException(