1

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

@@ -342,6 +342,41 @@ async def create_async_generation_task(db: AsyncSession, current_user: User, req
     return task
 
 
+def _resolve_error_message(error_message: str | None) -> str | None:
+    """匹配 ARK_ERRORS 字典，将原始错误码转换为友好提示。
+    与 app/api/v1/generation.py 的 _record_to_out 保持一致。
+
+    注意：celery 任务中已调用 extract_error_message 将错误码转为中文提示后存入数据库，
+    所以到达此函数的 message 可能是：
+    1. 已翻译的中文提示（ARK_ERRORS 的 value）→ 直接返回
+    2. 原始错误字符串（含 code='...' 或 JSON 格式）→ 匹配 ARK_ERRORS
+    3. 未知内容 → 返回 "生成失败"
+    """
+    if not error_message:
+        return error_message
+
+    from app.services.error_codes import ARK_ERRORS
+
+    # 如果已经是 ARK_ERRORS 中已翻译的中文值，直接返回
+    if error_message in ARK_ERRORS.values():
+        return error_message
+
+    import re
+    # 匹配 code='InputVideoSensitiveContentDetected.PrivacyInformation' 格式
+    match = re.search(r"code='([^']+)'", error_message)
+    if match:
+        code = match.group(1)
+        if code in ARK_ERRORS:
+            return ARK_ERRORS[code]
+    else:
+        parts = error_message.split(":")
+        if len(parts) >= 2 and parts[1].strip() in ARK_ERRORS:
+            return ARK_ERRORS[parts[1].strip()]
+
+    # 没有匹配到已知错误码时，直接返回"生成失败"
+    return "生成失败"
+
+
 def record_to_out(
     task: ChatGenerationTask,
     is_admin: bool = False,
@@ -410,7 +445,7 @@ def record_to_out(
         video_tokens_used=task.video_tokens_used or 0,
         retry_count=task.retry_count or 0,
         poll_count=task.poll_count or 0,
-        error_message=task.error_message,
+        error_message=_resolve_error_message(task.error_message),
         created_at=task.created_at,
         generated_at=task.generated_at,
     )
@@ -611,7 +646,7 @@ def generation_record_to_history_out(
         video_tokens_used=record.video_tokens_used or 0,
         retry_count=0,
         poll_count=0,
-        error_message=record.error_message,
+        error_message=_resolve_error_message(record.error_message),
         created_at=record.created_at,
         generated_at=record.generated_at,
     )