adding support of "amazon.titan-embed-text-v2:0"

jdixosnd · May 14, 2024 · 1c5bef2 · 1c5bef2
1 parent 0e02fd6
commit 1c5bef2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/...edrock/src/haystack_integrations/components/embedders/amazon_bedrock/document_embedder.py b/...edrock/src/haystack_integrations/components/embedders/amazon_bedrock/document_embedder.py
@@ -16,7 +16,7 @@
 
 logger = logging.getLogger(__name__)
 
-SUPPORTED_EMBEDDING_MODELS = ["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3"]
+SUPPORTED_EMBEDDING_MODELS = ["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3", "amazon.titan-embed-text-v2:0"]
 
 
 @component
@@ -51,7 +51,7 @@ class AmazonBedrockDocumentEmbedder:
 
  def __init__(
  self,
- model: Literal["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3"],
+ model: Literal["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3", "amazon.titan-embed-text-v2:0"],
  aws_access_key_id: Optional[Secret] = Secret.from_env_var("AWS_ACCESS_KEY_ID", strict=False), # noqa: B008
  aws_secret_access_key: Optional[Secret] = Secret.from_env_var( # noqa: B008
  "AWS_SECRET_ACCESS_KEY", strict=False

diff --git a/...on_bedrock/src/haystack_integrations/components/embedders/amazon_bedrock/text_embedder.py b/...on_bedrock/src/haystack_integrations/components/embedders/amazon_bedrock/text_embedder.py
@@ -14,7 +14,7 @@
 
 logger = logging.getLogger(__name__)
 
-SUPPORTED_EMBEDDING_MODELS = ["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3"]
+SUPPORTED_EMBEDDING_MODELS = ["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3", "amazon.titan-embed-text-v2:0"]
 
 
 @component
@@ -44,7 +44,7 @@ class AmazonBedrockTextEmbedder:
 
  def __init__(
  self,
- model: Literal["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3"],
+ model: Literal["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3", "amazon.titan-embed-text-v2:0"],
  aws_access_key_id: Optional[Secret] = Secret.from_env_var("AWS_ACCESS_KEY_ID", strict=False), # noqa: B008
  aws_secret_access_key: Optional[Secret] = Secret.from_env_var( # noqa: B008
  "AWS_SECRET_ACCESS_KEY", strict=False