Skip to content

Commit e4439b5

Browse files
author
Bob Strahan
committed
Update bedrock knowledge base embedding configuration and remove deprecated model option
1 parent 76c1ee4 commit e4439b5

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

options/bedrockkb/src/s3_vectors_manager/handler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,7 @@ def create_knowledge_base_s3_vectors(bedrock_agent_client, name, description, ro
264264
'vectorKnowledgeBaseConfiguration': {
265265
'embeddingModelConfiguration': {
266266
'bedrockEmbeddingModelConfiguration': {
267-
'dimensions': 1024, # Console uses 1024
267+
'dimensions': 1024, # All embedding models in picklist output 1024
268268
'embeddingDataType': 'FLOAT32'
269269
}
270270
},
@@ -356,7 +356,7 @@ def create_s3_vector_resources(s3vectors_client, bucket_name, index_name, embedd
356356
vectorBucketName=bucket_name,
357357
indexName=index_name,
358358
dataType="float32",
359-
dimension=1024, # Console uses 1024 for Titan v2
359+
dimension=1024, # All embedding models in picklist output 1024
360360
distanceMetric="cosine",
361361
metadataConfiguration={
362362
"nonFilterableMetadataKeys": [

options/bedrockkb/template.yaml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,6 @@ Parameters:
9090
Type: String
9191
AllowedValues:
9292
- amazon.titan-embed-text-v2:0
93-
- amazon.titan-embed-text-v1
9493
- cohere.embed-english-v3
9594
- cohere.embed-multilingual-v3
9695
Default: amazon.titan-embed-text-v2:0

0 commit comments

Comments
 (0)