From 627e031fcba8ec8c6eca9367061a16d5c34b295e Mon Sep 17 00:00:00 2001
From: sunxiqing <2240398334@qq.com>
Date: Fri, 8 Aug 2025 16:49:27 +0800
Subject: [PATCH] =?UTF-8?q?refactor(llm):=20=E4=BC=98=E5=8C=96=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E6=9C=8D=E5=8A=A1=E7=9B=B8=E5=85=B3=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 移除了不必要的日志输出
- 修改了模型状态更新逻辑
- 优化了模型 URL 的构建方式
- 调整了模型状态码的处理方式
---
 .../llm/service/basemodel/BaseModelTaskService.java    | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java
index a3cdf6cdc..695bd6d24 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java
@@ -220,7 +220,7 @@ public class BaseModelTaskService {
             // 调用接口获取远程模型状态
             log.info("开始请求：{}", llmBackendProperties.getAListOfAvailableModels());
             String resStr = HttpUtils.get(llmBackendProperties.getAListOfAvailableModels(), null);
-            log.info("获取aicg模型列表返回数据内容: {}", resStr);
+//            log.info("获取aicg模型列表返回数据内容: {}", resStr);
 
             List<AListOfPedestalModelsVO> aListOfPedestalModelsVOS = JSONArray.parseArray(resStr, AListOfPedestalModelsVO.class);
             //获取所有名称
@@ -279,7 +279,7 @@ public class BaseModelTaskService {
 
             // 调用接口获取远程模型状态
             String resStr = HttpUtils.post(llmBackendProperties.getBaseModelStatus(), null, "");
-            log.info("获取aicg模型列表返回数据内容: {}", resStr);
+//            log.info("获取aicg模型列表返回数据内容: {}", resStr);
 
             JSONArray objects = JSONArray.parseArray(resStr);
             String s = JSON.toJSONString(objects);
@@ -326,7 +326,7 @@ public class BaseModelTaskService {
                             String string1 = pedestalModelVo.getHost() + "/v1/chat/completions";
                             localModel.setStatus(2);
                             localModel.setModelUrl(string1);
-                            localModel.setNumber(2);
+//                            localModel.setNumber(localModel.getNumber()+1);
                             localModel.setJobId((long) pedestalModelVo.getId());
                             modelServiceMapper.updateById(localModel);
                             BaseModelDO baseModelDO = baseModelMapper.selectById(localModel.getFineTuningTask());
@@ -365,10 +365,10 @@ public class BaseModelTaskService {
                         if (localModels != null && localModels.size() > 0) {
                             ModelServiceDO localModel = localModels.get(0);
                             // 如果状态不是 "running"，更新本地状态
-                            if (localModel.getNumber()!=1) {
+                            if (localModel.getNumber()==20) {
                                 localModel.setStatus(3);
                             }else {
-                                localModel.setNumber(2);
+                                localModel.setNumber(localModel.getNumber()+1);
                             }
                             modelServiceMapper.updateById(localModel);
                         }