Skip to content

Commit ee2d54b

Browse files
jdixosndmasci
andauthored
adding support of "amazon.titan-embed-text-v2:0" (#735)
* adding support of "amazon.titan-embed-text-v2:0" * rectifying the format --------- Co-authored-by: Massimiliano Pippi <[email protected]>
1 parent 428c2a8 commit ee2d54b

File tree

2 files changed

+24
-4
lines changed

2 files changed

+24
-4
lines changed

integrations/amazon_bedrock/src/haystack_integrations/components/embedders/amazon_bedrock/document_embedder.py

100644100755
+12-2
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,12 @@
1616

1717
logger = logging.getLogger(__name__)
1818

19-
SUPPORTED_EMBEDDING_MODELS = ["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3"]
19+
SUPPORTED_EMBEDDING_MODELS = [
20+
"amazon.titan-embed-text-v1",
21+
"cohere.embed-english-v3",
22+
"cohere.embed-multilingual-v3",
23+
"amazon.titan-embed-text-v2:0",
24+
]
2025

2126

2227
@component
@@ -51,7 +56,12 @@ class AmazonBedrockDocumentEmbedder:
5156

5257
def __init__(
5358
self,
54-
model: Literal["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3"],
59+
model: Literal[
60+
"amazon.titan-embed-text-v1",
61+
"cohere.embed-english-v3",
62+
"cohere.embed-multilingual-v3",
63+
"amazon.titan-embed-text-v2:0",
64+
],
5565
aws_access_key_id: Optional[Secret] = Secret.from_env_var("AWS_ACCESS_KEY_ID", strict=False), # noqa: B008
5666
aws_secret_access_key: Optional[Secret] = Secret.from_env_var( # noqa: B008
5767
"AWS_SECRET_ACCESS_KEY", strict=False

integrations/amazon_bedrock/src/haystack_integrations/components/embedders/amazon_bedrock/text_embedder.py

100644100755
+12-2
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,12 @@
1414

1515
logger = logging.getLogger(__name__)
1616

17-
SUPPORTED_EMBEDDING_MODELS = ["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3"]
17+
SUPPORTED_EMBEDDING_MODELS = [
18+
"amazon.titan-embed-text-v1",
19+
"cohere.embed-english-v3",
20+
"cohere.embed-multilingual-v3",
21+
"amazon.titan-embed-text-v2:0",
22+
]
1823

1924

2025
@component
@@ -44,7 +49,12 @@ class AmazonBedrockTextEmbedder:
4449

4550
def __init__(
4651
self,
47-
model: Literal["amazon.titan-embed-text-v1", "cohere.embed-english-v3", "cohere.embed-multilingual-v3"],
52+
model: Literal[
53+
"amazon.titan-embed-text-v1",
54+
"cohere.embed-english-v3",
55+
"cohere.embed-multilingual-v3",
56+
"amazon.titan-embed-text-v2:0",
57+
],
4858
aws_access_key_id: Optional[Secret] = Secret.from_env_var("AWS_ACCESS_KEY_ID", strict=False), # noqa: B008
4959
aws_secret_access_key: Optional[Secret] = Secret.from_env_var( # noqa: B008
5060
"AWS_SECRET_ACCESS_KEY", strict=False

0 commit comments

Comments
 (0)