aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/dev/dev' into dev/dev

Browse Source
This commit is contained in:
shahaibo
2025-06-16 13:45:25 +08:00
parent 3db0a94981 891b26d493
commit 774a20257d
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
View File

@@ -2024,7 +2024,8 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
private String resolveModelType(String taskId){ private String resolveModelType(String taskId){
// 判断当前task来自哪个模型 // 判断当前task来自哪个模型
Generate generate = selectByUniqueId(taskId); Generate generate = selectByUniqueId(taskId);
if (!StringUtil.isNullOrEmpty(generate.getModelName()) && if (Objects.nonNull(generate) &&
!StringUtil.isNullOrEmpty(generate.getModelName()) &&
(generate.getModelName().equals("wx") (generate.getModelName().equals("wx")
|| generate.getModelName().equals("freepik") || generate.getModelName().equals("freepik")
|| generate.getModelName().equals("flux") )){ || generate.getModelName().equals("flux") )){