This commit is contained in:
Timothy Jaeryang Baek
2026-05-05 04:05:15 +09:00
parent 4e6a7baab7
commit 989d5fd4e2
+6
View File
@@ -3945,6 +3945,12 @@ async def streaming_chat_response_handler(response, ctx):
response_id = response_metadata.pop('response_id', None)
if response_id:
last_response_id = response_id
# Normalize and capture usage for DB persistence
if response_metadata.get('usage'):
response_metadata['usage'] = normalize_usage(response_metadata['usage'])
usage = response_metadata['usage']
processed_data.update(response_metadata)
processed_data.pop('done', None)