From 1364cdf71e9c640cac3a7bbffca86ac1f011f223 Mon Sep 17 00:00:00 2001
From: jonathang4 <jonathang4@sina.com>
Date: Fri, 10 Oct 2025 19:53:25 +0800
Subject: [PATCH] =?UTF-8?q?=E9=94=99=E8=AF=AF=E4=BF=A1=E6=81=AF=201?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/lib/services/DatabaseGenerationService.ts | 3 ++-
 src/lib/services/TaskPollingService.ts        | 8 ++++++--
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/lib/services/DatabaseGenerationService.ts b/src/lib/services/DatabaseGenerationService.ts
index cf0a7ed..49ba3cf 100644
--- a/src/lib/services/DatabaseGenerationService.ts
+++ b/src/lib/services/DatabaseGenerationService.ts
@@ -229,7 +229,8 @@ export class DatabaseGenerationService {
                     data: {
                         task_id: taskId,
                         url: result.tos_urls.join(','),
-                        status: result.status === 'success' ? -1 : -2
+                        status: result.status === 'success' ? -1 : -2,
+                        error_message: result.metadata.fail_reason || ""
                     }
                 };
                 
diff --git a/src/lib/services/TaskPollingService.ts b/src/lib/services/TaskPollingService.ts
index 0d361d7..a12a3de 100644
--- a/src/lib/services/TaskPollingService.ts
+++ b/src/lib/services/TaskPollingService.ts
@@ -9,6 +9,7 @@ import TOSService from '@/lib/tos/tos-service.js';
 import { generateImages as originalGenerateImages } from '@/api/controllers/images.js';
 import { generateVideo as originalGenerateVideo } from '@/api/controllers/video.js';
 import { request } from '@/api/controllers/core.js';
+import EX from "@/api/consts/exceptions.ts";
 
 const timeZone = 'Asia/Shanghai';
 const TASK_POLLING_LOG_PATH = path.resolve("./logs/task_polling.log");
@@ -391,8 +392,11 @@ export class TaskPollingService {
                 await this.handleGenerationSuccess(task, item_list, currentTime);
             } else {
                 // 生成失败
-                const errorMessage = failCode === '2038' ? 'Content filtered' : 'Generation failed';
-                await this.handleGenerationFailure(task, errorMessage, failCode, currentTime);
+                if(failCode == '2038' || failCode == '2041'){
+                    await this.handleGenerationFailure(task, EX.API_CONTENT_FILTERED[1] as string, failCode, currentTime);
+                }else{
+                    await this.handleGenerationFailure(task, EX.API_IMAGE_GENERATION_FAILED[1] as string, failCode, currentTime);
+                }
             }
 
         } catch (error) {