From 86a991db816f0bacb8171687dc48b1d887b1fb4a Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Mon, 16 Jun 2025 11:03:40 +0800
Subject: [PATCH] =?UTF-8?q?BUGFIX:=E8=8E=B7=E5=8F=96=E5=BD=93=E5=89=8D?=
 =?UTF-8?q?=E7=94=9F=E6=88=90=E6=9D=A5=E8=87=AA=E5=93=AA=E4=B8=AA=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
index 5d4f185a..5c071a83 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -2017,7 +2017,8 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
     private String resolveModelType(String taskId){
         // 判断当前task来自哪个模型
         Generate generate = selectByUniqueId(taskId);
-        if (!StringUtil.isNullOrEmpty(generate.getModelName()) &&
+        if (Objects.nonNull(generate) &&
+                !StringUtil.isNullOrEmpty(generate.getModelName()) &&
                 (generate.getModelName().equals("wx")
                         || generate.getModelName().equals("freepik")
                         || generate.getModelName().equals("flux") )){