diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index f1bcad7ee..1b66715a1 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -1487,6 +1487,16 @@ async def process_chat_response( else: choices = data.get("choices", []) if not choices: + usage = data.get("usage", {}) + if usage: + await event_emitter( + { + "type": "chat:completion", + "data": { + "usage": usage, + }, + } + ) continue delta = choices[0].get("delta", {})