From 404beed57cbec5d03b81c3e133c6e63a80ed81a5 Mon Sep 17 00:00:00 2001
From: sunxiqing <2240398334@qq.com>
Date: Fri, 14 Mar 2025 19:19:54 +0800
Subject: [PATCH] =?UTF-8?q?feat(llm):=20=E6=9B=B4=E6=96=B0=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E8=AE=AD=E7=BB=83=E4=BB=BB=E5=8A=A1=E7=8A=B6=E6=80=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在更新 localModel 的同时，增加了对 baseModel 相关信息的更新
- 更新 baseModel 的 jobId、chatUrl 和 apiUrl
- 优化了日志信息，增加了模型状态为 running 时的提示
---
 .../module/llm/service/basemodel/BaseModelTaskService.java   | 5 +++++
 1 file changed, 5 insertions(+)

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 f859c34af..c4c0ce264 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
@@ -313,6 +313,11 @@ public class BaseModelTaskService {
 //                            localModel.setApiUrl(string1);
                             localModel.setJobId((long) pedestalModelVo.getId());
                             modelServiceMapper.updateById(localModel);
+                            BaseModelDO baseModelDO = baseModelMapper.selectById(localModel.getFineTuningTask());
+                            baseModelDO.setModelId((long) pedestalModelVo.getId());
+                            baseModelDO.setChatUrl(string1);
+                            baseModelDO.setApiUrl(string1);
+                            baseModelMapper.updateById(baseModelDO);
                             log.info("模型 {} 状态为 running，无需更新", pedestalModelVo.getDeploymentName());
                         }
                     }