From aa2a971ffa999f6cc0d38c0948efa6a7762ed89e Mon Sep 17 00:00:00 2001
From: wwwwwwwww <526125649@qq.com>
Date: Mon, 15 Jun 2026 16:19:07 +0800
Subject: [PATCH] 1

---
 video-gen-api/app/api/v1/admin.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/video-gen-api/app/api/v1/admin.py b/video-gen-api/app/api/v1/admin.py
index 48a90145..0ccf96cd 100644
--- a/video-gen-api/app/api/v1/admin.py
+++ b/video-gen-api/app/api/v1/admin.py
@@ -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