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 
- 引入 BaseModelTaskService 服务
- 将模型服务初始状态从 3改为 1
- 优化了与基础模型关联的逻辑
This commit is contained in:
sunxiqing 2025-03-14 21:13:49 +08:00
parent 2038f9a40b
commit 972ad6acd9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java
View File

@ -19,6 +19,7 @@ import cn.iocoder.yudao.module.llm.framework.backend.config.LLMBackendProperties
import cn.iocoder.yudao.module.llm.service.application.ApplicationService;
import cn.iocoder.yudao.module.llm.service.async.AsyncModelServiceService;
import cn.iocoder.yudao.module.llm.service.basemodel.BaseModelService;
import cn.iocoder.yudao.module.llm.service.basemodel.BaseModelTaskService;
import cn.iocoder.yudao.module.llm.service.http.ModelService;
import cn.iocoder.yudao.module.llm.service.http.TrainHttpService;
import cn.iocoder.yudao.module.llm.service.modelassesstaskauto.ModelAssessTaskAutoService;
@ -89,6 +90,8 @@ public class ModelServiceServiceImpl implements ModelServiceService {
@Resource
@Lazy
private BaseModelService baseModelService;
@Resource
private BaseModelTaskService baseModelTaskService;
@Override
public Long createModelService (ModelServiceSaveReqVO createReqVO) {
@ -118,7 +121,7 @@ public class ModelServiceServiceImpl implements ModelServiceService {
String replace = llmBackendProperties.getModelServiceApi().replace("apiKey", apikey);
modelService.setApiUrl(replace);
modelService.setStatus(3);
modelService.setStatus(1);
BaseModelDO byAigcId = baseModelService.getById(modelService.getFineTuningTask());
if (byAigcId != null){
modelService.setBaseModelName(byAigcId.getModelName());