Add Titan Embeddings G2 (#94)

Author: UniMa007

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"
 }
 
@@ -860,6 +860,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(