From 290508a17897c2bb52a7d943af13d5805c14835b Mon Sep 17 00:00:00 2001 From: GinHa <15201596918@163.com> Date: Wed, 3 Jun 2026 18:00:32 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=BC=82=E5=B8=B8=E5=A4=84?= =?UTF-8?q?=E7=90=86=E6=9C=BA=E5=88=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- video-gen-api/app/api/v1/generation.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/video-gen-api/app/api/v1/generation.py b/video-gen-api/app/api/v1/generation.py index 75937ee1..83e949f9 100644 --- a/video-gen-api/app/api/v1/generation.py +++ b/video-gen-api/app/api/v1/generation.py @@ -198,6 +198,7 @@ async def optimize( current_user: User = Depends(get_current_user), db: AsyncSession = Depends(get_db), ): + record = None # Validate parameters based on generation type if req.gen_type == GenerationType.video: if req.duration not in DURATIONS: @@ -285,10 +286,11 @@ async def optimize( await db.commit() except Exception as e: from app.services.error_codes import extract_error_message - record.status = "failed" - record.error_message = extract_error_message(e, "提示词") - await db.flush() - await db.commit() + if record: + record.status = "failed" + record.error_message = extract_error_message(e, "提示词") + await db.flush() + await db.commit() error_message = extract_error_message(e, "提示词") raise HTTPException( status_code=502,