From b5329b53235eff9ef50e7877f38eb14473449ba5 Mon Sep 17 00:00:00 2001
From: sunxiqing <2240398334@qq.com>
Date: Wed, 5 Mar 2025 18:37:44 +0800
Subject: [PATCH] =?UTF-8?q?refactor(llm):=20=E6=9B=B4=E6=96=B0=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E7=8A=B6=E6=80=81=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在更新模型状态时，增加了设置 modelId 的操作- 优化了模型状态为 running 时的处理逻辑
---
 .../module/llm/service/basemodel/BaseModelTaskService.java      | 2 ++
 1 file changed, 2 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 d9a168850..e3ea30a55 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
@@ -238,12 +238,14 @@ public class BaseModelTaskService {
                         if (!"running".equals(pedestalModelVo.getStatus())) {
                             log.info("更新模型状态，模型名称: {}, 状态: {}", pedestalModelVo.getDeploymentName(), pedestalModelVo.getStatus());
                             localModel.setIsActive(2);
+                            localModel.setModelId((long) pedestalModelVo.getId());
                             baseModelService.updetatebyId(localModel);
                         } else {
                             String string1 = pedestalModelVo.getHost() + "/v1/chat/completions";
                             if (!string1.equals(localModel.getChatUrl())) {
                                 localModel.setIsActive(1);
                                 localModel.setChatUrl(string1);
+                                localModel.setModelId((long) pedestalModelVo.getId());
                                 baseModelService.updetatebyId(localModel);
                             }
                             log.debug("模型 {} 状态为 running，无需更新", pedestalModelVo.getDeploymentName());