From b0c5f7b6685c38a5db887a4552e73ea7505fc342 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Mon, 2 Feb 2026 09:02:34 -0600 Subject: [PATCH] refac --- backend/open_webui/models/chat_messages.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/backend/open_webui/models/chat_messages.py b/backend/open_webui/models/chat_messages.py index 8c9d326a3..bc7508d45 100644 --- a/backend/open_webui/models/chat_messages.py +++ b/backend/open_webui/models/chat_messages.py @@ -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}")