fix: openai usage display issue

2025-03-04 11:29:59 +00:00 · 2025-02-25 01:35:59 -08:00 · 2025-02-25 01:35:59 -08:00 · 7e2b2ec761
commit 7e2b2ec761
parent b964b175d4
1 changed files with 10 additions and 0 deletions
--- 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", {})