diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java index 1146ada56..007aedaa7 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java @@ -22,6 +22,8 @@ public class AsyncModelServiceService { public void startModelService(ModelServiceDO updateObj) { try { Thread.sleep(30000); + updateObj.setStatus(2); + modelServiceMapper.updateById(updateObj); }catch(Exception e){ updateObj.setStatus(5); modelServiceMapper.updateById(updateObj); @@ -33,6 +35,8 @@ public class AsyncModelServiceService { public void stopModelService(ModelServiceDO updateObj) { try { Thread.sleep(30000); + updateObj.setStatus(3); + modelServiceMapper.updateById(updateObj); }catch(Exception e){ updateObj.setStatus(5); modelServiceMapper.updateById(updateObj); diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java index d81676bab..093ce6e22 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java @@ -95,9 +95,6 @@ public class ModelServiceServiceImpl implements ModelServiceService { //todo 开启模型服务 asyncModelServiceService.startModelService(updateObj); - - updateObj.setStatus(2); - modelServiceMapper.updateById(updateObj); } @Override @@ -109,8 +106,6 @@ public class ModelServiceServiceImpl implements ModelServiceService { //todo 关闭模型服务 asyncModelServiceService.stopModelService(updateObj); -/* updateObj.setStatus(3); - modelServiceMapper.updateById(updateObj);*/ }