新增数据集类型 官方数据集

yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/vo/DatasetPageReqVO.java

@@ -40,4 +40,7 @@ public class DatasetPageReqVO extends PageParam {
     @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND)
     private LocalDateTime[] createTime;
 
-}
+    @Schema(description = "数据集类型 0 普通 1 官方", example = "1")
+    private Integer type;
+
+}

yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/vo/DatasetRespVO.java

@@ -48,4 +48,6 @@ public class DatasetRespVO {
     @ExcelProperty("创建时间")
     private LocalDateTime createTime;
 
-}
+    @Schema(description = "数据集类型 0 普通 1 官方", example = "1")
+    private Integer type;
+}

yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/vo/DatasetSaveReqVO.java

@@ -35,4 +35,7 @@ public class DatasetSaveReqVO {
     @Schema(description = "文件URL地址", example = "https://www.iocoder.cn")
     private String datasetFileUrl;
 
-}
+    @Schema(description = "数据集类型 0 普通 1 官方", example = "1")
+    private Integer type;
+
+}

yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/dataset/DatasetDO.java

@@ -55,5 +55,8 @@ public class DatasetDO extends BaseDO {
      * 文件URL地址
      */
     private String datasetFileUrl;
-
-}
+    /**
+     * 数据集类型 0 普通 1 官方
+     */
+    private Integer type;
+}

yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetMapper.java

@@ -21,6 +21,7 @@ public interface DatasetMapper extends BaseMapperX<DatasetDO> {
         return selectPage(reqVO, new LambdaQueryWrapperX<DatasetDO>()
                 .likeIfPresent(DatasetDO::getDatasetName, reqVO.getDatasetName())
                 .eqIfPresent(DatasetDO::getDatasetCategory, reqVO.getDatasetCategory())
+                .eqIfPresent(DatasetDO::getType, reqVO.getType())
                 .eqIfPresent(DatasetDO::getStatus, reqVO.getStatus())
                 .eqIfPresent(DatasetDO::getDatasetIntro, reqVO.getDatasetIntro())
                 .eqIfPresent(DatasetDO::getDatasetFile, reqVO.getDatasetFile())
@@ -30,4 +31,4 @@ public interface DatasetMapper extends BaseMapperX<DatasetDO> {
                 .orderByDesc(DatasetDO::getId));
     }
 
-}
+}

yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java

@@ -28,6 +28,9 @@ public class DatasetServiceImpl implements DatasetService {
 
     @Override
     public Long createDataset(DatasetSaveReqVO createReqVO) {
+        if(createReqVO.getType() == null){
+            createReqVO.setType(0);
+        }
         // 插入
         DatasetDO dataset = BeanUtils.toBean(createReqVO, DatasetDO.class);
         datasetMapper.insert(dataset);