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
2025-12-27 01:41:06 +04:00
parent ac0ae2ae20
commit a9609585af
1 changed files with 15 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/routers/chats.py
+15 -17
View File
@@ -222,6 +222,18 @@ def calculate_chat_stats(user_id, skip=0, limit=10, filter=None):
chat_stats_export_list = [] chat_stats_export_list = []
def get_message_content_length(message):
content = message.get("content", "")
if isinstance(content, str):
return len(content)
elif isinstance(content, list):
return sum(
len(item.get("text", ""))
for item in content
if item.get("type") == "text"
)
return 0
for chat in result.items: for chat in result.items:
try: try:
messages_map = chat.chat.get("history", {}).get("messages", {}) messages_map = chat.chat.get("history", {}).get("messages", {})
@@ -235,13 +247,7 @@ def calculate_chat_stats(user_id, skip=0, limit=10, filter=None):
export_messages = {} export_messages = {}
for key, message in messages_map.items(): for key, message in messages_map.items():
try: try:
content = message.get("content", "") content_length = get_message_content_length(message)
if isinstance(content, str):
content_length = len(content)
else:
content_length = (
0 # Handle cases where content might be None or not string
)
# Extract rating safely # Extract rating safely
rating = message.get("annotation", {}).get("rating") rating = message.get("annotation", {}).get("rating")
@@ -277,22 +283,14 @@ def calculate_chat_stats(user_id, skip=0, limit=10, filter=None):
# Calculate Averages # Calculate Averages
average_user_message_content_length = ( average_user_message_content_length = (
sum( sum(get_message_content_length(m) for m in history_user_messages)
len(m.get("content", ""))
for m in history_user_messages
if isinstance(m.get("content"), str)
)
/ len(history_user_messages) / len(history_user_messages)
if history_user_messages if history_user_messages
else 0 else 0
) )
average_assistant_message_content_length = ( average_assistant_message_content_length = (
sum( sum(get_message_content_length(m) for m in history_assistant_messages)
len(m.get("content", ""))
for m in history_assistant_messages
if isinstance(m.get("content"), str)
)
/ len(history_assistant_messages) / len(history_assistant_messages)
if history_assistant_messages if history_assistant_messages
else 0 else 0