diff --git a/backend/app/routers/stats.py b/backend/app/routers/stats.py index 5c9d7e5..7d50965 100644 --- a/backend/app/routers/stats.py +++ b/backend/app/routers/stats.py @@ -68,7 +68,7 @@ async def get_stats_summary( # 今日调用次数和 token 消耗 today_stats = db.query( func.count(AICallEvent.id).label('calls'), - func.coalesce(func.sum(AICallEvent.total_tokens), 0).label('tokens') + func.coalesce(func.sum(AICallEvent.input_tokens + AICallEvent.output_tokens), 0).label('tokens') ).filter( func.date(AICallEvent.created_at) == today ).first() @@ -77,7 +77,7 @@ async def get_stats_summary( week_start = today - timedelta(days=today.weekday()) week_stats = db.query( func.count(AICallEvent.id).label('calls'), - func.coalesce(func.sum(AICallEvent.total_tokens), 0).label('tokens') + func.coalesce(func.sum(AICallEvent.input_tokens + AICallEvent.output_tokens), 0).label('tokens') ).filter( func.date(AICallEvent.created_at) >= week_start ).first() @@ -104,7 +104,7 @@ async def get_stats_trend( query = db.query( func.date(AICallEvent.created_at).label('date'), func.count(AICallEvent.id).label('calls'), - func.coalesce(func.sum(AICallEvent.total_tokens), 0).label('tokens') + func.coalesce(func.sum(AICallEvent.input_tokens + AICallEvent.output_tokens), 0).label('tokens') ).filter( func.date(AICallEvent.created_at) >= start_date, func.date(AICallEvent.created_at) <= end_date