mz/video-gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复图片生成积分后台计算逻辑不对没有加上每张积分

Browse Source
This commit is contained in:
root
2026-07-28 11:43:15 +08:00
parent 9b3541264f
commit 570292266e
2 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
video-gen-api/app/services/generation/ai/task_create_service.py
+6
View File
@@ -281,6 +281,7 @@ async def create_generation_task_group(
gen_type=GenerationType.IMAGE.value,
image_size=size,
engine_id=engine.id,
input_image_count=reference_image_count if reference_image_count > 0 else None,
project_name="AI生成任务",
description_prefix="AI创作-",
owner_type=OWNER_CHAT_GENERATION_TASK,
@@ -340,6 +341,9 @@ async def create_generation_task_group(
raise HTTPException(status_code=400, detail=f"视频时长不能超过 {engine.max_duration} 秒")
input_video_duration = _validate_video_references(refs, max_audio_count=engine.max_audio_count)
reference_image_count = sum(
1 for ref in refs if (ref.get("type") or "").lower() == GenerationType.IMAGE.value
)
provider_generation_resolution, upscale_enabled_snapshot, upscale_snapshot_json = await build_video_upscale_snapshot(
db,
target_resolution=resolution,
@@ -363,6 +367,7 @@ async def create_generation_task_group(
resolution=resolution,
engine_id=engine.id,
input_video_duration=input_video_duration if input_video_duration > 0 else None,
input_image_count=reference_image_count if reference_image_count > 0 else None,
project_name="AI生成任务",
description_prefix="AI创作-",
owner_type=OWNER_CHAT_GENERATION_TASK,
@@ -437,6 +442,7 @@ async def create_generation_task_group(
resolution=resolution,
engine_id=engine.id,
input_video_duration=input_video_duration if input_video_duration > 0 else None,
input_image_count=reference_image_count if reference_image_count > 0 else None,
project_name="AI生成任务",
description_prefix=f"AI创作-第{generation_index}份-",
owner_type=OWNER_CHAT_GENERATION_TASK,