ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-01-02 22:55:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-12-26 16:16:10 +04:00
parent 51dd488417
commit 39a2b9789c
2 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
backend/open_webui/models/chats.py
View file

@ -218,6 +218,7 @@ class MessageStats(BaseModel):
token_count: Optional[int] = None token_count: Optional[int] = None
timestamp: Optional[int] = None timestamp: Optional[int] = None
rating: Optional[int] = None # Derived from message.annotation.rating rating: Optional[int] = None # Derived from message.annotation.rating
tags: Optional[list[str]] = None # Derived from message.annotation.tags
class ChatHistoryStats(BaseModel): class ChatHistoryStats(BaseModel):

3
backend/open_webui/routers/chats.py
View file

@ -274,6 +274,8 @@ async def export_chat_stats(
# Extract rating safely # Extract rating safely
rating = message.get("annotation", {}).get("rating") rating = message.get("annotation", {}).get("rating")
tags = message.get("annotation", {}).get("tags")
message_stat = MessageStats( message_stat = MessageStats(
id=message.get("id"), id=message.get("id"),
role=message.get("role"), role=message.get("role"),
@ -282,6 +284,7 @@ async def export_chat_stats(
content_length=content_length, content_length=content_length,
token_count=None, # Populate if available, e.g. message.get("info", {}).get("token_count") token_count=None, # Populate if available, e.g. message.get("info", {}).get("token_count")
rating=rating, rating=rating,
tags=tags,
) )
export_messages[key] = message_stat export_messages[key] = message_stat