From bc7eee681db17069783fa8c46d41a6506c092c2a Mon Sep 17 00:00:00 2001
From: wwwwwwwww <526125649@qq.com>
Date: Fri, 24 Jul 2026 15:56:15 +0800
Subject: [PATCH] =?UTF-8?q?=E7=88=86=E6=AC=BE=E5=92=8C=E6=8B=86=E9=95=9Cer?=
 =?UTF-8?q?ror=E6=B6=88=E6=81=AF=E6=AD=A3=E5=88=99=E4=B8=AD=E6=96=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../app/services/hot_opening_replicate_service.py     | 11 ++++++++++-
 .../app/services/shot_replicate_flow_service.py       | 11 ++++++++++-
 2 files changed, 20 insertions(+), 2 deletions(-)

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,