From 4926ca75447ea3b3d326f3861b43dd49e4d071df Mon Sep 17 00:00:00 2001 From: limin Date: Sun, 5 Jan 2025 14:16:40 +0800 Subject: [PATCH] =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E6=9C=8D=E5=8A=A1=E7=8A=B6?= =?UTF-8?q?=E6=80=81=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../module/llm/service/async/AsyncModelServiceService.java | 4 ++++ .../llm/service/modelservice/ModelServiceServiceImpl.java | 5 ----- 2 files changed, 4 insertions(+), 5 deletions(-) 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);*/ }