Skip to content

Commit 7b9c973

Browse files
author
Bob Strahan
committed
Fix S3 Vector API parameter names to match service specification
1 parent a3b51e9 commit 7b9c973

File tree

1 file changed

+14
-18
lines changed
  • options/bedrockkb/src/s3_vectors_manager

1 file changed

+14
-18
lines changed

options/bedrockkb/src/s3_vectors_manager/handler.py

Lines changed: 14 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -280,18 +280,14 @@ def create_s3_vector_resources(s3vectors_client, bucket_name, index_name, embedd
280280
logger.info(f"Creating vector bucket: {bucket_name}")
281281

282282
create_bucket_params = {
283-
'Bucket': bucket_name
283+
'vectorBucketName': bucket_name
284284
}
285285

286286
# Add KMS encryption if provided
287287
if kms_key_arn:
288-
create_bucket_params['CreateBucketConfiguration'] = {
289-
'BucketConfiguration': {
290-
'BucketEncryption': {
291-
'SSEAlgorithm': 'aws:kms',
292-
'KMSMasterKeyID': kms_key_arn
293-
}
294-
}
288+
create_bucket_params['encryptionConfiguration'] = {
289+
'SSEAlgorithm': 'aws:kms',
290+
'KMSMasterKeyID': kms_key_arn
295291
}
296292
logger.info(f"Using KMS encryption for bucket with key: {kms_key_arn}")
297293

@@ -302,10 +298,10 @@ def create_s3_vector_resources(s3vectors_client, bucket_name, index_name, embedd
302298
# Step 2: Create vector index in the bucket
303299
logger.info(f"Creating vector index: {index_name} in bucket: {bucket_name}")
304300
index_response = s3vectors_client.create_vector_index(
305-
Bucket=bucket_name,
306-
IndexName=index_name,
307-
EmbeddingConfig={
308-
'EmbeddingModelArn': f"arn:aws:bedrock:*::foundation-model/{embedding_model}"
301+
vectorBucketName=bucket_name,
302+
indexName=index_name,
303+
embeddingConfiguration={
304+
'embeddingModelArn': f"arn:aws:bedrock:*::foundation-model/{embedding_model}"
309305
}
310306
)
311307
index_arn = index_response['IndexArn']
@@ -337,8 +333,8 @@ def delete_s3_vector_resources(s3vectors_client, bucket_name, index_name):
337333
try:
338334
logger.info(f"Deleting vector index: {index_name} from bucket: {bucket_name}")
339335
s3vectors_client.delete_vector_index(
340-
Bucket=bucket_name,
341-
IndexName=index_name
336+
vectorBucketName=bucket_name,
337+
indexName=index_name
342338
)
343339
logger.info(f"Deleted vector index: {index_name}")
344340
except ClientError as e:
@@ -350,7 +346,7 @@ def delete_s3_vector_resources(s3vectors_client, bucket_name, index_name):
350346
try:
351347
logger.info(f"Deleting vector bucket: {bucket_name}")
352348
s3vectors_client.delete_vector_bucket(
353-
Bucket=bucket_name
349+
vectorBucketName=bucket_name
354350
)
355351
logger.info(f"Deleted vector bucket: {bucket_name}")
356352
except ClientError as e:
@@ -365,13 +361,13 @@ def get_s3_vector_info(s3vectors_client, bucket_name, index_name):
365361
"""Get information about existing S3 Vector resources."""
366362
try:
367363
# Get bucket info
368-
bucket_response = s3vectors_client.get_vector_bucket(Bucket=bucket_name)
364+
bucket_response = s3vectors_client.get_vector_bucket(vectorBucketName=bucket_name)
369365
bucket_arn = bucket_response['BucketArn']
370366

371367
# Get index info
372368
index_response = s3vectors_client.describe_vector_index(
373-
Bucket=bucket_name,
374-
IndexName=index_name
369+
vectorBucketName=bucket_name,
370+
indexName=index_name
375371
)
376372
index_arn = index_response['IndexArn']
377373

0 commit comments

Comments
 (0)