修复异常处理机制
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user