diff --git a/video-gen-api/app/api/v1/generation.py b/video-gen-api/app/api/v1/generation.py index 9f82010e..d2bf5fa4 100644 --- a/video-gen-api/app/api/v1/generation.py +++ b/video-gen-api/app/api/v1/generation.py @@ -271,6 +271,9 @@ async def optimize( db: AsyncSession = Depends(get_db), ): record = None + # 积分不足直接返回 + if (current_user.credits or 0) < 5: + raise HTTPException(status_code=402, detail="积分不足,请充值") # Validate parameters based on generation type if req.gen_type == GenerationType.video: if req.duration not in DURATIONS: