Skip to content

Commit 7580980

Browse files
Backport PR #778: Add Titan embedding model v2 (#779)
Co-authored-by: Sanjiv Das <[email protected]>
1 parent 90649f1 commit 7580980

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)