交易流水对账明细导出完成

video-gen-api/app/services/shot_replicate_flow_service.py

@@ -903,6 +903,24 @@ async def generate_image_from_prompt(
         {"type": "image", "url": material.get("material_image_url"), "name": "新产品图片"},
     ]
 
+    step = await _create_step(
+        db,
+        project=project,
+        step_code=ShotReplicateStepCodeEnum.IMAGE_GENERATE.value,
+        status=ModuleStepStatusEnum.PROCESSING.value,
+        parent_step_id=prompt_step.id,
+        source_step_id=prompt_step.id,
+        input_data={
+            "engine_id": req.engine_id,
+            "params": {
+                "image_size": req.image_size,
+                "image_proportion": req.image_proportion,
+                "image_px": req.image_px,
+            },
+            "prompt": optimized_prompt,
+            "media_references": refs,
+        },
+    )
     chat_task = await create_chat_generation_task_for_module(
         db,
         current_user=current_user,
@@ -917,21 +935,30 @@ async def generate_image_from_prompt(
         image_px=req.image_px,
         billing_project_name=project.title or "拆镜复刻",
         billing_description_prefix="拆镜复刻图片生成",
+        billing_source_module=project.module,
+        billing_source_project_id=project.id,
+        billing_source_step_id=step.id,
+        billing_source_step_code=ShotReplicateStepCodeEnum.IMAGE_GENERATE.value,
     )
-    step = await _create_step(
-        db,
-        project=project,
-        step_code=ShotReplicateStepCodeEnum.IMAGE_GENERATE.value,
-        status=ModuleStepStatusEnum.PROCESSING.value,
-        parent_step_id=prompt_step.id,
-        source_step_id=prompt_step.id,
-        chat_task_id=chat_task.id,
-        input_data={
-            "engine_id": chat_task.engine_id,
-            "params": {"image_size": chat_task.image_size, "image_proportion": chat_task.image_proportion, "image_px": chat_task.image_px},
-            "prompt": optimized_prompt,
-            "media_references": refs,
-        },
+    step.chat_task_id = chat_task.id
+    _force_set_json(
+        step,
+        "input_json",
+        _step_input(
+            step_code=ShotReplicateStepCodeEnum.IMAGE_GENERATE.value,
+            source_step_id=prompt_step.id,
+            parent_step_id=prompt_step.id,
+            payload={
+                "engine_id": chat_task.engine_id,
+                "params": {
+                    "image_size": chat_task.image_size,
+                    "image_proportion": chat_task.image_proportion,
+                    "image_px": chat_task.image_px,
+                },
+                "prompt": optimized_prompt,
+                "media_references": refs,
+            },
+        ),
     )
     project.status = ModuleProjectStatusEnum.PROCESSING.value
     project.current_step_code = ShotReplicateStepCodeEnum.IMAGE_GENERATE.value
@@ -1223,6 +1250,25 @@ async def generate_video_from_prompt(
         {"type": "image", "url": _build_file_url_or_data_uri(generated_image_url), "name": "新项目图片"},
     ]
 
+    step = await _create_step(
+        db,
+        project=project,
+        step_code=ShotReplicateStepCodeEnum.VIDEO_GENERATE.value,
+        status=ModuleStepStatusEnum.PROCESSING.value,
+        parent_step_id=prompt_step.id,
+        source_step_id=prompt_step.id,
+        input_data={
+            "engine_id": req.engine_id or prompt_params.get("engine_id"),
+            "params": {
+                "duration": duration,
+                "aspect_ratio": aspect_ratio,
+                "resolution": resolution,
+            },
+            "prompt_schema": prompt_schema,
+            "final_prompt": final_prompt,
+            "media_references": refs,
+        },
+    )
     chat_task = await create_chat_generation_task_for_module(
         db,
         current_user=current_user,
@@ -1237,29 +1283,34 @@ async def generate_video_from_prompt(
         resolution=resolution,
         billing_project_name=project.title or "拆镜复刻",
         billing_description_prefix="拆镜复刻视频生成",
+        billing_source_module=project.module,
+        billing_source_project_id=project.id,
+        billing_source_step_id=step.id,
+        billing_source_step_code=ShotReplicateStepCodeEnum.VIDEO_GENERATE.value,
     )
-    step = await _create_step(
-        db,
-        project=project,
-        step_code=ShotReplicateStepCodeEnum.VIDEO_GENERATE.value,
-        status=ModuleStepStatusEnum.PROCESSING.value,
-        parent_step_id=prompt_step.id,
-        source_step_id=prompt_step.id,
-        chat_task_id=chat_task.id,
-        input_data={
-            "engine_id": chat_task.engine_id,
-            "params": {
-                "duration": chat_task.duration,
-                "aspect_ratio": chat_task.aspect_ratio,
-                "resolution": chat_task.resolution,
-                "image_size": chat_task.image_size,
-                "image_proportion": chat_task.image_proportion,
-                "image_px": chat_task.image_px,
+    step.chat_task_id = chat_task.id
+    _force_set_json(
+        step,
+        "input_json",
+        _step_input(
+            step_code=ShotReplicateStepCodeEnum.VIDEO_GENERATE.value,
+            source_step_id=prompt_step.id,
+            parent_step_id=prompt_step.id,
+            payload={
+                "engine_id": chat_task.engine_id,
+                "params": {
+                    "duration": chat_task.duration,
+                    "aspect_ratio": chat_task.aspect_ratio,
+                    "resolution": chat_task.resolution,
+                    "image_size": chat_task.image_size,
+                    "image_proportion": chat_task.image_proportion,
+                    "image_px": chat_task.image_px,
+                },
+                "prompt_schema": prompt_schema,
+                "final_prompt": final_prompt,
+                "media_references": refs,
             },
-            "prompt_schema": prompt_schema,
-            "final_prompt": final_prompt,
-            "media_references": refs,
-        },
+        ),
     )
     project.status = ModuleProjectStatusEnum.PROCESSING.value
     project.current_step_code = ShotReplicateStepCodeEnum.VIDEO_GENERATE.value