DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

refactor(llm): 更新模型状态逻辑

Browse Source 
- 在更新模型状态时，增加了设置 modelId 的操作- 优化了模型状态为 running 时的处理逻辑
This commit is contained in:
sunxiqing 2025-03-05 18:37:44 +08:00
parent 1135d3b1b9
commit b5329b5323
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java
View File

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