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

feat: enhance stats command to aggregate conversation token usage

Browse Source
This commit is contained in:
Soulter
2026-04-27 12:53:46 +08:00
parent 8ca8231176
commit 2ce6b1b885
1 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,4 +1,4 @@
from sqlalchemy import select
from sqlalchemy import case, func, select
from sqlmodel import col
from astrbot.api import sp, star
@@ -267,14 +267,27 @@ class ConversationCommands:
db = self.context.get_db()
async with db.get_db() as session:
result = await session.execute(
select(ProviderStat).where(
select(
func.count(case((col(ProviderStat.id).is_not(None), 1))).label(
"record_count",
),
func.coalesce(func.sum(ProviderStat.token_input_other), 0).label(
"total_input_other",
),
func.coalesce(func.sum(ProviderStat.token_input_cached), 0).label(
"total_input_cached",
),
func.coalesce(func.sum(ProviderStat.token_output), 0).label(
"total_output",
),
).where(
col(ProviderStat.agent_type) == "internal",
col(ProviderStat.conversation_id) == cid,
)
)
records = result.scalars().all()
stats = result.one()
if not records:
if stats.record_count == 0:
message.set_result(
MessageEventResult().message(
"📊 No stats available for this conversation yet."
@@ -282,9 +295,9 @@ class ConversationCommands:
)
return
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_input_other = stats.total_input_other
total_input_cached = stats.total_input_cached
total_output = stats.total_output
total_tokens = total_input_other + total_input_cached + total_output
ret = (