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,