Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Tim Baek
2026-02-01 10:11:13 +04:00
parent 75e5a485d2
commit 3da4323ef3
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/models/chat_messages.py
+10 -6
View File
@@ -140,9 +140,11 @@ class ChatMessageTable:
) )
if "error" in data: if "error" in data:
existing.error = data.get("error") existing.error = data.get("error")
# Extract usage from info.usage if present # Extract usage - check direct field first, then info.usage
info = data.get("info", {}) usage = data.get("usage")
usage = info.get("usage") if info else None if not usage:
info = data.get("info", {})
usage = info.get("usage") if info else None
if usage: if usage:
existing.usage = usage existing.usage = usage
existing.updated_at = now existing.updated_at = now
@@ -151,9 +153,11 @@ class ChatMessageTable:
return ChatMessageModel.model_validate(existing) return ChatMessageModel.model_validate(existing)
else: else:
# Insert new # Insert new
# Extract usage from info.usage if present # Extract usage - check direct field first, then info.usage
info = data.get("info", {}) usage = data.get("usage")
usage = info.get("usage") if info else None if not usage:
info = data.get("info", {})
usage = info.get("usage") if info else None
message = ChatMessage( message = ChatMessage(
id=message_id, id=message_id,
chat_id=chat_id, chat_id=chat_id,
backend/open_webui/utils/middleware.py
+1
View File
@@ -3042,6 +3042,7 @@ async def process_chat_response(
async def stream_body_handler(response, form_data): async def stream_body_handler(response, form_data):
nonlocal content nonlocal content
nonlocal content_blocks nonlocal content_blocks
nonlocal usage
response_tool_calls = [] response_tool_calls = []