From 1f20747f9ced1e3c70349ec77dcea8bc5cce933b Mon Sep 17 00:00:00 2001
From: Liuyang <2746366019@qq.com>
Date: Fri, 14 Feb 2025 10:03:20 +0800
Subject: [PATCH] =?UTF-8?q?refactor(llm):=20=E9=87=8D=E6=9E=84=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E5=BE=AE=E8=B0=83=E7=9B=B8=E5=85=B3=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 将 AigcFineTuningCreateReqVO 和 FineTuningTaskDO 中的 learningRate 字段类型从 double/Integer 改为 BigDecimal，以提高精度
- 优化 ModelService 中的 createFineTuningTask 方法，异常处理改为抛出 RuntimeException
---
 .../llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java   | 4 +++-
 .../iocoder/yudao/module/llm/service/http/ModelService.java   | 2 +-
 .../module/llm/service/http/vo/AigcFineTuningCreateReqVO.java | 4 +++-
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java
index 348562494..647d9f67b 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java
@@ -1,6 +1,8 @@
 package cn.iocoder.yudao.module.llm.dal.dataobject.finetuningtask;
 
 import lombok.*;
+
+import java.math.BigDecimal;
 import java.util.*;
 import java.time.LocalDateTime;
 import java.time.LocalDateTime;
@@ -71,7 +73,7 @@ public class FineTuningTaskDO extends BaseDO {
     /**
      * 学习率
      */
-    private Integer learningRate;
+    private BigDecimal learningRate;
     /**
      * 训练轮数
      */
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/ModelService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/ModelService.java
index a9ad51fe6..7117339f0 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/ModelService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/ModelService.java
@@ -90,7 +90,7 @@ public class ModelService {
             }
             return null;
         } catch (Exception e) {
-            return null;
+          throw new RuntimeException(e);
         }
     }
 
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/AigcFineTuningCreateReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/AigcFineTuningCreateReqVO.java
index 7ba30c8a1..83741b186 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/AigcFineTuningCreateReqVO.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/AigcFineTuningCreateReqVO.java
@@ -4,6 +4,8 @@ import lombok.AllArgsConstructor;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 
+import java.math.BigDecimal;
+
 @Data
 @AllArgsConstructor
 @NoArgsConstructor
@@ -20,7 +22,7 @@ public class AigcFineTuningCreateReqVO {
         private int evalBatchSize;
         private int accumulationSteps;
         private int procPerNode;
-        private double learningRate;
+        private BigDecimal learningRate;
         private int modelMaxLength;
         private boolean lora;
 }