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:
ire 2025-01-02 14:08:59 +08:00
parent 6d4246b1c4
commit 49aea9b285
4 changed files with 30 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
View File

@ -32,7 +32,7 @@ public class AsyncFineTuningTaskService {
@Async
public void startFineTuningTask() {
try {
Thread.sleep(30000);
}catch(Exception e){
};
@ -42,7 +42,7 @@ public class AsyncFineTuningTaskService {
@Async
public void stopFineTuningTask() {
try {
Thread.sleep(30000);
}catch(Exception e){
};

19
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java
View File

@ -1,6 +1,8 @@
package cn.iocoder.yudao.module.llm.service.async;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelservice.ModelServiceDO;
import cn.iocoder.yudao.module.llm.dal.mysql.finetuningtask.FineTuningTaskMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.modelservice.ModelServiceMapper;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
@ -12,23 +14,28 @@ public class AsyncModelServiceService {
@Resource
private FineTuningTaskMapper fineTuningTaskMapper;
@Resource
private ModelServiceMapper modelServiceMapper;
//模型服务开启
@Async
public void startModelService() {
public void startModelService(ModelServiceDO updateObj) {
try {
Thread.sleep(30000);
}catch(Exception e){
updateObj.setStatus(5);
modelServiceMapper.updateById(updateObj);
};
}
//调型服务关闭
@Async
public void stopModelService() {
public void stopModelService(ModelServiceDO updateObj) {
try {
Thread.sleep(30000);
}catch(Exception e){
updateObj.setStatus(5);
modelServiceMapper.updateById(updateObj);
};
}
}

8
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java
View File

@ -136,13 +136,19 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
fineTuningTaskMapper.stopStartTask(id,1);
//todo 调用模型服务，开启调优任务
asyncFineTuningTaskService.startFineTuningTask();
fineTuningTaskMapper.stopStartTask(id,2);
}
@Override
public void stopFineTuningTask(Long id) {
fineTuningTaskMapper.stopStartTask(id,0);
fineTuningTaskMapper.stopStartTask(id,1);
//todo 调用模型服务，停止调优任务
asyncFineTuningTaskService.stopFineTuningTask();
fineTuningTaskMapper.stopStartTask(id,0);
}

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

@ -87,7 +87,10 @@ public class ModelServiceServiceImpl implements ModelServiceService {
modelServiceMapper.updateById(updateObj);
//todo 开启模型服务
asyncModelServiceService.startModelService();
asyncModelServiceService.startModelService(updateObj);
updateObj.setStatus(2);
modelServiceMapper.updateById(updateObj);
}
@Override
@ -97,7 +100,10 @@ public class ModelServiceServiceImpl implements ModelServiceService {
modelServiceMapper.updateById(updateObj);
//todo 关闭模型服务
asyncModelServiceService.stopModelService();
asyncModelServiceService.stopModelService(updateObj);
updateObj.setStatus(3);
modelServiceMapper.updateById(updateObj);
}