Listen for all object events to keep knowledge base in sync

Author: kris-szlapa

packages/cdk/constructs/S3LambdaNotification.ts

@@ -12,10 +12,11 @@ export class S3LambdaNotification extends Construct {
   constructor(scope: Construct, id: string, props: S3LambdaNotificationProps) {
     super(scope, id)
 
-    // Use CDK's built-in S3 notification
-    props.bucket.addEventNotification(
-      EventType.OBJECT_CREATED,
-      new LambdaDestination(props.lambdaFunction)
-    )
+    const lambdaDestination = new LambdaDestination(props.lambdaFunction)
+
+    // Listen for all object events to keep knowledge base in sync
+    props.bucket.addEventNotification(EventType.OBJECT_CREATED, lambdaDestination)
+    props.bucket.addEventNotification(EventType.OBJECT_REMOVED, lambdaDestination)
+    props.bucket.addEventNotification(EventType.OBJECT_RESTORE_COMPLETED, lambdaDestination)
   }
 }

packages/syncKnowledgeBaseFunction/app.py

@@ -17,8 +17,8 @@ def handler(event, context):
     """
     Lambda handler that processes S3 events and triggers Bedrock Knowledge Base ingestion.
 
-    This function is triggered when documents are uploaded to the S3 bucket and automatically
-    starts an ingestion job to update the knowledge base with new content.
+    This function is triggered when documents are created, updated, deleted, or restored in the S3 bucket
+    and automatically starts an ingestion job to sync the knowledge base with current content.
     """
     # Record start time for performance tracking
     start_time = time.time()
@@ -88,12 +88,12 @@ def handler(event, context):
                     },
                 )
 
-                # Start ingestion job for the knowledge base
+                # Start ingestion job for the knowledge base (handles all event types)
                 ingestion_start_time = time.time()
                 response = bedrock_agent.start_ingestion_job(
                     knowledgeBaseId=knowledge_base_id,
                     dataSourceId=data_source_id,
-                    description=f"Auto-sync triggered by S3 event: {event_name} on {key}",
+                    description=f"Auto-sync triggered by S3 {event_name} on {key}",
                 )
                 ingestion_request_time = time.time() - ingestion_start_time