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

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-02-02 09:02:34 -06:00
parent 767343dc5b
commit b0c5f7b668
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/open_webui/models/chat_messages.py
View File

@@ -317,11 +317,12 @@ class ChatMessageTable:
func.json_extract(ChatMessage.usage, "$.output_tokens"), Integer
)
elif dialect == "postgresql":
# Use json_extract_path_text for PostgreSQL JSON columns
input_tokens = cast(
ChatMessage.usage["input_tokens"].astext, Integer
func.json_extract_path_text(ChatMessage.usage, "input_tokens"), Integer
)
output_tokens = cast(
ChatMessage.usage["output_tokens"].astext, Integer
func.json_extract_path_text(ChatMessage.usage, "output_tokens"), Integer
)
else:
raise NotImplementedError(f"Unsupported dialect: {dialect}")
@@ -374,11 +375,12 @@ class ChatMessageTable:
func.json_extract(ChatMessage.usage, "$.output_tokens"), Integer
)
elif dialect == "postgresql":
# Use json_extract_path_text for PostgreSQL JSON columns
input_tokens = cast(
ChatMessage.usage["input_tokens"].astext, Integer
func.json_extract_path_text(ChatMessage.usage, "input_tokens"), Integer
)
output_tokens = cast(
ChatMessage.usage["output_tokens"].astext, Integer
func.json_extract_path_text(ChatMessage.usage, "output_tokens"), Integer
)
else:
raise NotImplementedError(f"Unsupported dialect: {dialect}")