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

修复模型服务错误

Browse Source
This commit is contained in:
limin 2025-01-14 16:55:44 +08:00
parent 17ac8ae7ad
commit 85d6d8470c
2 changed files with 2 additions and 1 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/async/AsyncModelServiceService.java
View File

@ -99,7 +99,7 @@ public class AsyncModelServiceService {
"float16", 1);
AigcRespVO aigcRespVO = trainHttpService.modelDeploy(new HashMap<>(), aigcModelDeploySaveReq,modelServiceDO.getJobId());
if (aigcRespVO.isSuccess()){
updateObj.setStatus(2);
updateObj.setStatus(1);
}else {
updateObj.setStatus(3);
}

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

@ -65,6 +65,7 @@ public class ModelServiceServiceImpl implements ModelServiceService {
String replace = llmBackendProperties.getModelServiceApi().replace("apiKey", apikey);
modelService.setApiUrl(replace);
modelService.setStatus(0);
modelServiceMapper.insert(modelService);
asyncModelServiceService.createModelService(modelService);
// 返回