fix: openai usage display issue

This commit is contained in:
Timothy Jaeryang Baek 2025-02-25 01:35:59 -08:00
parent b964b175d4
commit 7e2b2ec761

View File

@ -1487,6 +1487,16 @@ async def process_chat_response(
else: else:
choices = data.get("choices", []) choices = data.get("choices", [])
if not choices: if not choices:
usage = data.get("usage", {})
if usage:
await event_emitter(
{
"type": "chat:completion",
"data": {
"usage": usage,
},
}
)
continue continue
delta = choices[0].get("delta", {}) delta = choices[0].get("delta", {})