Skip to content

Commit 368bcf1

Browse files
Add Titan embedding model v2 (#778)
* Add Titan embeddning model v2 Included the new embedding model, released recently on Amazon Bedrock. * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 759c60c commit 368bcf1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,10 +98,11 @@ class HfHubEmbeddingsProvider(BaseEmbeddingsProvider, HuggingFaceHubEmbeddings):
9898
registry = True
9999

100100

101+
# See model ID list here: https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html
101102
class BedrockEmbeddingsProvider(BaseEmbeddingsProvider, BedrockEmbeddings):
102103
id = "bedrock"
103104
name = "Bedrock"
104-
models = ["amazon.titan-embed-text-v1"]
105+
models = ["amazon.titan-embed-text-v1", "amazon.titan-embed-text-v2:0"]
105106
model_id_key = "model_id"
106107
pypi_package_deps = ["boto3"]
107108
auth_strategy = AwsAuthStrategy()

0 commit comments

Comments
 (0)