@@ -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