1

2026-06-15 16:19:07 +08:00
parent 5ee682d06e
commit aa2a971ffa
1 changed files with 4 additions and 3 deletions
@@ -9,6 +9,7 @@ from app.dependencies import get_db, get_admin_user
 from app.models.user import User
 from app.models.project import Project
 from app.models.generation_record import GenerationRecord
+from app.models.chat_generation_task import ChatGenerationTask
 from app.models.credit_record import CreditRecord
 from app.models.model_config import ModelConfig
 from app.models.system_config import SystemConfig
@@ -1175,9 +1176,9 @@ async def get_stats(
    )).scalar() or 0
    
    total_generations = (await db.execute(
-        select(func.count(GenerationAITask.id)).where(
-            GenerationAITask.created_at >= date_start,
-            GenerationAITask.created_at <= date_end,
+        select(func.count(ChatGenerationTask.id)).where(
+            ChatGenerationTask.created_at >= date_start,
+            ChatGenerationTask.created_at <= date_end,
        )
    )).scalar() or 0