DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

refactor(llm): 将参数量字段类型从 Integer 改为 Double

Browse Source 
- 修改了 BaseModelDO、BaseModelPageReqVO、BaseModelRespVO、BaseModelSaveReqVO 中的 parameterCount 字段类型- 更新了 BaseModelTaskService 中的相关代码，以适应新的 Double 类型参数量
- 这个改动可以更精确地表示模型的参数量，特别是对于参数量较大的模型
This commit is contained in:
sunxiqing 2025-08-11 10:34:39 +08:00
parent 56c67cc89f
commit 64f5c3fe93
5 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelPageReqVO.java
View File

@ -19,7 +19,7 @@ public class BaseModelPageReqVO extends PageParam {
private String modelName;
@Schema(description = "参数量", example = "12388")
private Integer parameterCount;
private Double parameterCount;
@Schema(description = "最大上下文长度")
private Integer maxContextLength;

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelRespVO.java
View File

@ -25,7 +25,7 @@ public class BaseModelRespVO {
@Schema(description = "参数量", requiredMode = Schema.RequiredMode.REQUIRED, example = "12388")
@ExcelProperty("参数量")
private Integer parameterCount;
private Double parameterCount;
@Schema(description = "最大上下文长度", requiredMode = Schema.RequiredMode.REQUIRED)
@ExcelProperty("最大上下文长度")

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelSaveReqVO.java
View File

@ -18,7 +18,7 @@ public class BaseModelSaveReqVO {
@Schema(description = "参数量", requiredMode = Schema.RequiredMode.REQUIRED, example = "12388")
@NotNull(message = "参数量不能为空")
private Integer parameterCount;
private Double parameterCount;
@Schema(description = "最大上下文长度", requiredMode = Schema.RequiredMode.REQUIRED)
@NotNull(message = "最大上下文长度不能为空")

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/basemodel/BaseModelDO.java
View File

@ -34,7 +34,7 @@ public class BaseModelDO extends BaseDO {
/**
* 参数量
*/
private Integer parameterCount;
private Double parameterCount;
/**
* 最大上下文长度
*/

6
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java
View File

@ -179,12 +179,12 @@ public class BaseModelTaskService {
baseModelSaveReqVO.setIsActive(1);
baseModelSaveReqVO.setIsFinetuned(modelListRe.getIsFineTuning());
baseModelSaveReqVO.setNotes(modelListRe.getRemark());
baseModelSaveReqVO.setParameterCount(1);
baseModelSaveReqVO.setParameterCount(1.0);
baseModelService.createBaseModel(baseModelSaveReqVO);
} else {
BaseModelSaveReqVO baseModelSaveReqVO = new BaseModelSaveReqVO();
baseModelSaveReqVO.setId(byAigcId.getId());
baseModelSaveReqVO.setParameterCount(1);
baseModelSaveReqVO.setParameterCount(1.0);
baseModelSaveReqVO.setModelName(modelListRe.getModelName());
baseModelSaveReqVO.setModelType(modelListRe.getModelType());
baseModelSaveReqVO.setMaxContextLength(modelListRe.getMaxTokens());
@ -311,7 +311,7 @@ public class BaseModelTaskService {
PedestalModelVO pedestalModelVo = collect.get(0);
// 查找本地模型
List<ModelServiceDO> localModels = modelServiceDOS.stream()
.filter(baseModel -> pedestalModelVo.getDeploymentName().equals(baseModel.getBaseModelName()))
.filter(baseModel -> pedestalModelVo.getDeploymentName().equals(baseModel.getBaseModelName()) && baseModel.getStatus() != 3)
.collect(Collectors.toList());
if (localModels != null && localModels.size() > 0) {