Skip to content

Commit 9356a35

Browse files
committed
Updated the collection ARN to be different from embedding model ARN
1 parent 0dec32e commit 9356a35

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/example_code/bedrock-agent/knowledge_bases/knowledge_base.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def create_knowledge_base(bedrock_agent_client, name, role_arn, description=None
5454
"storageConfiguration": {
5555
"type": "OPENSEARCH_SERVERLESS",
5656
"opensearchServerlessConfiguration": {
57-
"collectionArn": "arn:aws:bedrock:us-east-1::foundation-model/amazon.titan-embed-text-v1",
57+
"collectionArn": "arn:aws:aoss:us-east-1::123456789012:collection/abcdefgh12345678defgh",
5858
"fieldMapping": {
5959
"metadataField": "metadata",
6060
"textField": "text",
@@ -316,7 +316,7 @@ def run_knowledge_base_scenario():
316316
print("\nScenario completed successfully!")
317317

318318
except ClientError as error:
319-
print("Operation failed: " + error)
319+
print("Operation failed: " + str(error))
320320
# Clean up resources on error
321321
if knowledge_base_id:
322322
try:
@@ -329,7 +329,7 @@ def run_knowledge_base_scenario():
329329
print("Attempting to delete IAM role: " + role_name + " ...")
330330
delete_knowledge_base_role(iam_client, role_name)
331331
except Exception as e:
332-
print("Failed to delete IAM role: " + e)
332+
print("Failed to delete IAM role: " + str(e))
333333

334334
print("-" * 88)
335335

0 commit comments

Comments
 (0)