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:
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
backend/open_webui/models/chat_messages.py
+6 -4
View File
@@ -317,11 +317,12 @@ class ChatMessageTable:
func.json_extract(ChatMessage.usage, "$.output_tokens"), Integer func.json_extract(ChatMessage.usage, "$.output_tokens"), Integer
) )
elif dialect == "postgresql": elif dialect == "postgresql":
# Use json_extract_path_text for PostgreSQL JSON columns
input_tokens = cast( input_tokens = cast(
ChatMessage.usage["input_tokens"].astext, Integer func.json_extract_path_text(ChatMessage.usage, "input_tokens"), Integer
) )
output_tokens = cast( output_tokens = cast(
ChatMessage.usage["output_tokens"].astext, Integer func.json_extract_path_text(ChatMessage.usage, "output_tokens"), Integer
) )
else: else:
raise NotImplementedError(f"Unsupported dialect: {dialect}") raise NotImplementedError(f"Unsupported dialect: {dialect}")
@@ -374,11 +375,12 @@ class ChatMessageTable:
func.json_extract(ChatMessage.usage, "$.output_tokens"), Integer func.json_extract(ChatMessage.usage, "$.output_tokens"), Integer
) )
elif dialect == "postgresql": elif dialect == "postgresql":
# Use json_extract_path_text for PostgreSQL JSON columns
input_tokens = cast( input_tokens = cast(
ChatMessage.usage["input_tokens"].astext, Integer func.json_extract_path_text(ChatMessage.usage, "input_tokens"), Integer
) )
output_tokens = cast( output_tokens = cast(
ChatMessage.usage["output_tokens"].astext, Integer func.json_extract_path_text(ChatMessage.usage, "output_tokens"), Integer
) )
else: else:
raise NotImplementedError(f"Unsupported dialect: {dialect}") raise NotImplementedError(f"Unsupported dialect: {dialect}")