修复异常处理机制

This commit is contained in:
2026-06-03 18:00:32 +08:00
parent b4f8cdb99d
commit 290508a178
+6 -4
View File
@@ -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,