From 972ad6acd989a59d02169b13d712312650d5198c Mon Sep 17 00:00:00 2001
From: sunxiqing <2240398334@qq.com>
Date: Fri, 14 Mar 2025 21:13:49 +0800
Subject: [PATCH] =?UTF-8?q?refactor(llm):=20=E6=9B=B4=E6=96=B0=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E6=9C=8D=E5=8A=A1=E5=88=9B=E5=BB=BA=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 引入 BaseModelTaskService 服务
- 将模型服务初始状态从 3改为 1
- 优化了与基础模型关联的逻辑
---
 .../llm/service/modelservice/ModelServiceServiceImpl.java    | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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 7e0ee8649..6912880ff 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
@@ -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());