From 02b6da900c129fc1272c633de7ef4c477bc22206 Mon Sep 17 00:00:00 2001
From: sunxiqing <2240398334@qq.com>
Date: Fri, 14 Mar 2025 19:54:26 +0800
Subject: [PATCH] =?UTF-8?q?refactor(llm):=20=E4=BF=AE=E6=94=B9=E5=88=A0?=
 =?UTF-8?q?=E9=99=A4=E5=9F=BA=E5=BA=A7=E6=A8=A1=E5=9E=8B=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 将删除模型接口的参数从 BaseModelPageReqVO 改为 Long 类型的 id
- 更新了 BaseModelService接口和 BaseModelServiceImpl 实现类中的方法签名
- 优化了删除模型的逻辑，直接使用 id 进行查询和删除操作
---
 .../llm/controller/admin/basemodel/BaseModelController.java  | 4 ++--
 .../yudao/module/llm/service/basemodel/BaseModelService.java | 2 +-
 .../module/llm/service/basemodel/BaseModelServiceImpl.java   | 5 ++---
 3 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java
index 35f225123..09fd06928 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java
@@ -84,8 +84,8 @@ public class BaseModelController {
 
     @DeleteMapping("/deleteTheModel")
     @Operation(summary = "删除基座模型")
-    public CommonResult<Boolean> deleteTheModel(@Valid BaseModelPageReqVO pageReqVO) {
-        baseModelService.deleteTheModel(pageReqVO);
+    public CommonResult<Boolean> deleteTheModel(@RequestParam("id") Long id) {
+        baseModelService.deleteTheModel(id);
         baseModelTaskService.updateTheBaseModelState();
         return success(true);
     }
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java
index dadf9cfda..9dee47ad2 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java
@@ -85,5 +85,5 @@ public interface BaseModelService {
 
     BaseModelDO getById(Long fineTuningTask);
 
-    void deleteTheModel(BaseModelPageReqVO pageReqVO);
+    void deleteTheModel(Long id);
 }
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java
index 025ad1c67..e953b93d0 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java
@@ -229,10 +229,9 @@ public class BaseModelServiceImpl implements BaseModelService {
     }
 
     @Override
-    public void deleteTheModel(BaseModelPageReqVO pageReqVO) {
+    public void deleteTheModel(Long id) {
         try {
-            Map<String,String> map = new HashMap<>();
-            map.put("model_name",pageReqVO.getModelName());
+            BaseModelDO pageReqVO = baseModelMapper.selectById(id);
 //            String resStr = HttpUtils.del(llmBackendProperties.getDeleteTheModel()+"/"+pageReqVO.getModelName(), null);