diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/framework/backend/config/LLMBackendProperties.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/framework/backend/config/LLMBackendProperties.java index 7faf9ce02..49c146b63 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/framework/backend/config/LLMBackendProperties.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/framework/backend/config/LLMBackendProperties.java @@ -116,6 +116,8 @@ public class LLMBackendProperties { private String deployModel; + private String deleteModel; + /** * 知识库向量嵌入 */ diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java index 9a5dd479e..33ed2dc64 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java @@ -166,9 +166,17 @@ public class BaseModelServiceImpl implements BaseModelService { public void unActive (BaseModelSaveReqVO updateReqVO) { // 校验使用 validateModelUse(updateReqVO.getId()); + try { + BaseModelDO baseModelDO = baseModelMapper.selectById(updateReqVO.getId()); + Long deploy_id = baseModelDO.getModelId(); + String resStr = HttpUtils.post(llmBackendProperties.getDeleteModel(), null, JSON.toJSONString(deploy_id)); + // 禁用模型 + updateReqVO.setIsActive(0); + updateBaseModel(updateReqVO); + }catch (Exception e){ + log.error("禁用基础模型状态时发生异常: {}", e.getMessage(), e); + } - updateReqVO.setIsActive(0); - updateBaseModel(updateReqVO); } @Override diff --git a/yudao-server/src/main/resources/application-local.yaml b/yudao-server/src/main/resources/application-local.yaml index 87acd576c..e9b7563d9 100644 --- a/yudao-server/src/main/resources/application-local.yaml +++ b/yudao-server/src/main/resources/application-local.yaml @@ -332,6 +332,8 @@ llm: base_model_status: http://36.103.199.248:5123/llm/deploy/list # 模型部署 POST deploy_model: http://36.103.199.248:5123/llm/deploy + # 模型删除 + delete_model: http://36.103.199.248:5123/llm/deploy/stop #################### 30000: 大模型对话相关API。 ################### #### 大模型对话