BUGFIX:获取当前生成来自哪个模型

2025-06-16 11:03:40 +08:00
parent 78239d3ead
commit 86a991db81
1 changed files with 2 additions and 1 deletions
--- 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") )){