diff --git a/src/lambdas/agent-runner/handler.py b/src/lambdas/agent-runner/handler.py index ab74dd4..5773a7f 100644 --- a/src/lambdas/agent-runner/handler.py +++ b/src/lambdas/agent-runner/handler.py @@ -224,6 +224,8 @@ def handler(event, context): event = json.loads(data) except (json.JSONDecodeError, ValueError): continue + if not isinstance(event, dict): + continue # Extract text delta from contentBlockDelta delta = event.get('event', {}).get('contentBlockDelta', {}).get('delta', {}) token = delta.get('text', '') or event.get('data', '') diff --git a/test-bot.py b/test-bot.py index 37f75ad..d17166f 100644 --- a/test-bot.py +++ b/test-bot.py @@ -71,6 +71,8 @@ def main(): event = json.loads(data) except (json.JSONDecodeError, ValueError): continue + if not isinstance(event, dict): + continue # Text delta lives in event.contentBlockDelta.delta.text delta = event.get('event', {}).get('contentBlockDelta', {}).get('delta', {}) token = delta.get('text', '') or event.get('data', '')