From 9b3541264fbd196ccd1a018bb28ec8381ea1719a Mon Sep 17 00:00:00 2001
From: wwwwwwwww <526125649@qq.com>
Date: Tue, 28 Jul 2026 10:34:48 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E8=A1=8C=E4=B8=9A=E6=8F=90?=
 =?UTF-8?q?=E7=A4=BA=E8=AF=8D=E4=BC=98=E5=8C=96=E6=9C=80=E7=BB=88=E4=B8=8D?=
 =?UTF-8?q?=E4=BD=BF=E7=94=A8=E9=99=84=E4=BB=B6=E8=BF=98=E6=98=AF=E4=BC=A0?=
 =?UTF-8?q?=E5=88=B0=E6=A8=A1=E5=9E=8B=E7=9A=84=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 video-gen-api/app/services/generation/provider_service.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/video-gen-api/app/services/generation/provider_service.py b/video-gen-api/app/services/generation/provider_service.py
index f2a65218..8463a26b 100644
--- a/video-gen-api/app/services/generation/provider_service.py
+++ b/video-gen-api/app/services/generation/provider_service.py
@@ -51,7 +51,9 @@ def _snapshot_owner(task: GenerationOwner) -> SimpleNamespace:
         for key, value in vars(task).items()
         if key != "_sa_instance_state"
     }
-    values.setdefault("generation_mode", getattr(task, "generation_mode", None) or "generation_record")
+    # 只有原对象确实有 generation_mode 时才保留，避免 SimpleNamespace 快照被误判为 ChatGenerationTask
+    if getattr(task, "generation_mode", None):
+        values.setdefault("generation_mode", task.generation_mode)
     return SimpleNamespace(**values)