From 3866266b93629cdf3d42a33d51b93de587c7e731 Mon Sep 17 00:00:00 2001
From: limin <limin@qq.com>
Date: Sat, 25 Jan 2025 11:20:04 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=A8=A1=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../module/llm/service/async/AsyncModelServiceService.java    | 4 ++--
 .../llm/service/modelservice/ModelServiceTaskSyncService.java | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java
index 4db799f3e..634652e95 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java
@@ -130,11 +130,11 @@ public class AsyncModelServiceService {
             if (aigcRespVO.isSuccess()){
                 updateObj.setStatus(0);
             }else {
-                updateObj.setStatus(3);
+//                updateObj.setStatus(3);
             }
             modelServiceMapper.updateById(updateObj);
         }catch(Exception e){
-            updateObj.setStatus(3);
+//            updateObj.setStatus(3);
             modelServiceMapper.updateById(updateObj);
         };
     }
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceTaskSyncService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceTaskSyncService.java
index 429da3054..a0842e722 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceTaskSyncService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceTaskSyncService.java
@@ -62,7 +62,7 @@ public class ModelServiceTaskSyncService {
 
                 // 使用 TypeReference 解析 JSON 字符串为 List<String>
                 try {
-                    String query = "?filter={\"model_id\":"+modelServiceDO.getJobId()+"}";
+                    String query = "?filter={\"deploy_name\":"+modelServiceDO.getBaseModelName()+"}";
                     String res = trainHttpService.modelTableQuery(new HashMap<>(), "model_deploy",query);
                     log.info("获取 aigc model_deploy 表数据 info {}",res);
                     ObjectMapper mapper = new ObjectMapper();