diff --git a/video-gen-api/app/services/hot_opening_replicate_service.py b/video-gen-api/app/services/hot_opening_replicate_service.py index b8476710..e3d88e9d 100644 --- a/video-gen-api/app/services/hot_opening_replicate_service.py +++ b/video-gen-api/app/services/hot_opening_replicate_service.py @@ -370,6 +370,15 @@ async def _soft_delete_steps_from_index( def _step_to_out(step: ModuleGenerationStep) -> HotOpeningStepOut: + error_message = step.error_message + if error_message: + from app.services.error_codes import ARK_ERRORS + import re + match = re.search(r"code='([^']+)'", error_message) + if match: + code = match.group(1) + if code in ARK_ERRORS: + error_message = ARK_ERRORS[code] return HotOpeningStepOut( id=step.id, project_id=step.project_id, @@ -384,7 +393,7 @@ def _step_to_out(step: ModuleGenerationStep) -> HotOpeningStepOut: chat_task_id=step.chat_task_id, input=_parse_json(step.input_json, {}), output=_parse_json(step.output_json, {}), - error_message=step.error_message, + error_message=error_message, created_at=step.created_at, updated_at=step.updated_at, completed_at=step.completed_at, diff --git a/video-gen-api/app/services/shot_replicate_flow_service.py b/video-gen-api/app/services/shot_replicate_flow_service.py index 8ac13c50..1ecb7ad2 100644 --- a/video-gen-api/app/services/shot_replicate_flow_service.py +++ b/video-gen-api/app/services/shot_replicate_flow_service.py @@ -381,6 +381,15 @@ async def _soft_delete_steps_from_index( def _step_to_out(step: ModuleGenerationStep) -> ShotReplicateStepOut: + error_message = step.error_message + if error_message: + from app.services.error_codes import ARK_ERRORS + import re + match = re.search(r"code='([^']+)'", error_message) + if match: + code = match.group(1) + if code in ARK_ERRORS: + error_message = ARK_ERRORS[code] return ShotReplicateStepOut( id=step.id, project_id=step.project_id, @@ -395,7 +404,7 @@ def _step_to_out(step: ModuleGenerationStep) -> ShotReplicateStepOut: chat_task_id=step.chat_task_id, input=_parse_json(step.input_json, {}), output=_parse_json(step.output_json, {}), - error_message=step.error_message, + error_message=error_message, created_at=step.created_at, updated_at=step.updated_at, completed_at=step.completed_at,