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());