Guard isinstance(event, dict) in SSE parser
This commit is contained in:
@@ -224,6 +224,8 @@ def handler(event, context):
|
|||||||
event = json.loads(data)
|
event = json.loads(data)
|
||||||
except (json.JSONDecodeError, ValueError):
|
except (json.JSONDecodeError, ValueError):
|
||||||
continue
|
continue
|
||||||
|
if not isinstance(event, dict):
|
||||||
|
continue
|
||||||
# Extract text delta from contentBlockDelta
|
# Extract text delta from contentBlockDelta
|
||||||
delta = event.get('event', {}).get('contentBlockDelta', {}).get('delta', {})
|
delta = event.get('event', {}).get('contentBlockDelta', {}).get('delta', {})
|
||||||
token = delta.get('text', '') or event.get('data', '')
|
token = delta.get('text', '') or event.get('data', '')
|
||||||
|
|||||||
@@ -71,6 +71,8 @@ def main():
|
|||||||
event = json.loads(data)
|
event = json.loads(data)
|
||||||
except (json.JSONDecodeError, ValueError):
|
except (json.JSONDecodeError, ValueError):
|
||||||
continue
|
continue
|
||||||
|
if not isinstance(event, dict):
|
||||||
|
continue
|
||||||
# Text delta lives in event.contentBlockDelta.delta.text
|
# Text delta lives in event.contentBlockDelta.delta.text
|
||||||
delta = event.get('event', {}).get('contentBlockDelta', {}).get('delta', {})
|
delta = event.get('event', {}).get('contentBlockDelta', {}).get('delta', {})
|
||||||
token = delta.get('text', '') or event.get('data', '')
|
token = delta.get('text', '') or event.get('data', '')
|
||||||
|
|||||||
Reference in New Issue
Block a user