1

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

@@ -457,8 +457,13 @@ async def charge_generation_media_by_params(
     record_id: str,
     gen_type: str,
     image_size: str | None = None,
+    image_px: str | None = None,
+    aspect_ratio: str | None = None,
     duration: int | None = None,
     resolution: str | None = None,
+    fps: int | float | None = None,
+    generate_audio: bool | None = None,
+    inference_mode: str | None = None,
     engine_id: str | None = None,
     input_video_duration: float | None = None,
     project_name: str | None = None,
@@ -470,6 +475,8 @@ async def charge_generation_media_by_params(
     source_step_id: str | None = None,
     source_step_code: str | None = None,
     billing_scene: str | None = None,
+    media_references: Any = None,
+    requested_output_count: int = 1,
 ) -> BillingSummary:
     """图片/视频媒体生成扣费。
 
@@ -504,6 +511,17 @@ async def charge_generation_media_by_params(
         source_step_id=source_step_id,
         source_step_code=source_step_code,
         billing_scene=billing_scene,
+        media_references=media_references,
+        image_size=image_size,
+        image_px=image_px,
+        aspect_ratio=aspect_ratio,
+        duration=duration,
+        resolution=resolution,
+        fps=fps,
+        generate_audio=generate_audio,
+        inference_mode=inference_mode,
+        input_video_duration=input_video_duration,
+        requested_output_count=requested_output_count,
     )
 
     if gen_type == "image":
@@ -561,6 +579,8 @@ async def charge_generation_media_for_record(
         record_id=record.id,
         gen_type=record.gen_type,
         image_size=record.image_size,
+        image_px=record.image_px,
+        aspect_ratio=record.aspect_ratio or record.image_proportion,
         duration=record.duration,
         resolution=record.resolution,
         project_name=project_name,
@@ -568,4 +588,5 @@ async def charge_generation_media_for_record(
         owner_type=OWNER_GENERATION_RECORD,
         attempt_no=attempt_no,
         source_module=CreditRecordSourceModule.GENERATION_RECORD.value,
+        media_references=record.media_references,
     )