fz-mirrors/AstrBot
1
0
Fork 0
mirror of https://github.com/AstrBotDevs/AstrBot synced 2026-07-16 17:47:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: reorder conversation stats output for better readability

Browse Source 
Co-authored-by: Copilot <copilot@github.com>
This commit is contained in:
Soulter
2026-04-27 12:00:12 +08:00
parent f02444146d
commit 6ba01a4775
1 changed files with 8 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
astrbot/builtin_stars/builtin_commands/commands/conversation.py
View File

@@ -1,4 +1,5 @@
from sqlalchemy import select
from sqlmodel import col
from astrbot.api import sp, star
from astrbot.api.event import AstrMessageEvent, MessageEventResult
@@ -267,8 +268,8 @@ class ConversationCommands:
async with db.get_db() as session:
result = await session.execute(
select(ProviderStat).where(
ProviderStat.agent_type == "internal",
ProviderStat.conversation_id == cid,
col(ProviderStat.agent_type) == "internal",
col(ProviderStat.conversation_id) == cid,
)
)
records = result.scalars().all()
@@ -281,23 +282,17 @@ class ConversationCommands:
)
return
total_calls = len(records)
total_input_other = sum(r.token_input_other for r in records)
total_input_cached = sum(r.token_input_cached for r in records)
total_output = sum(r.token_output for r in records)
total_tokens = total_input_other + total_input_cached + total_output
success_count = sum(1 for r in records if r.status != "error")
ret = (
f"📊 Conversation Stats (ID: {cid[:8]}...)\n"
f"───\n"
f"Total Calls: {total_calls}\n"
f"Successful: {success_count}\n"
f"───\n"
f"Input Tokens (other): {total_input_other:,}\n"
f"Input Tokens (cached): {total_input_cached:,}\n"
f"Output Tokens: {total_output:,}\n"
f"Total Tokens: {total_tokens:,}"
f"📊 Token usage (ID: {cid[:8]}...)\n"
f"Total: {total_tokens:,}\n"
f"Input (other): {total_input_other:,}\n"
f"Input (cached): {total_input_cached:,}\n"
f"Output: {total_output:,}\n"
)
message.set_result(MessageEventResult().message(ret))