From 989d5fd4e2ce285edf4475a1e13f0981a78d3821 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Tue, 5 May 2026 04:05:15 +0900 Subject: [PATCH] refac --- backend/open_webui/utils/middleware.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index 5b1da36d37..f3e829f22c 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -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)