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
+6 -2
View File
@@ -140,7 +140,9 @@ 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
usage = data.get("usage")
if not usage:
info = data.get("info", {}) info = data.get("info", {})
usage = info.get("usage") if info else None usage = info.get("usage") if info else None
if usage: if usage:
@@ -151,7 +153,9 @@ 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
usage = data.get("usage")
if not usage:
info = data.get("info", {}) info = data.get("info", {})
usage = info.get("usage") if info else None usage = info.get("usage") if info else None
message = ChatMessage( message = ChatMessage(
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 = []