diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/datarefluxdata/DataRefluxDataController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/datarefluxdata/DataRefluxDataController.java index 6ca925490..2316a7d16 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/datarefluxdata/DataRefluxDataController.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/datarefluxdata/DataRefluxDataController.java @@ -67,6 +67,7 @@ public class DataRefluxDataController { @Parameter(name = "id", description = "编号", required = true, example = "1024") @PreAuthorize("@ss.hasPermission('llm:data-reflux-data:query')") public CommonResult getDataRefluxData(@RequestParam("id") Long id) { + //TODO 查看按钮后面没具体内容 DataRefluxDataDO dataRefluxData = dataRefluxDataService.getDataRefluxData(id); return success(BeanUtils.toBean(dataRefluxData, DataRefluxDataRespVO.class)); } @@ -79,6 +80,16 @@ public class DataRefluxDataController { return success(BeanUtils.toBean(pageResult, DataRefluxDataRespVO.class)); } + @GetMapping("/review-the-data") + @Operation(summary = "数据回流配置保存数据") + public void saveTheData(@RequestParam("id") DataRefluxDataReqVO pageReqVO){ + if (pageReqVO.getId() != null){ + pageReqVO.setIds(Collections.singletonList(pageReqVO.getId())); + } + //TODO 数据回流配置保存数据 + } + + @GetMapping("/export-excel") @Operation(summary = "导出数据回流 —— 数据 Excel") @PreAuthorize("@ss.hasPermission('llm:data-reflux-data:export')") @@ -92,4 +103,4 @@ public class DataRefluxDataController { BeanUtils.toBean(list, DataRefluxDataRespVO.class)); } -} \ No newline at end of file +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/datarefluxdata/vo/DataRefluxDataReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/datarefluxdata/vo/DataRefluxDataReqVO.java new file mode 100644 index 000000000..56af95bee --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/datarefluxdata/vo/DataRefluxDataReqVO.java @@ -0,0 +1,16 @@ +package cn.iocoder.yudao.module.llm.controller.admin.datarefluxdata.vo; + +import io.swagger.v3.oas.annotations.media.Schema; +import lombok.Data; + +import java.util.List; + +@Schema(description = "管理后台 - 数据回流 —— 保存数据") +@Data +public class DataRefluxDataReqVO { + @Schema(description = "配置ID",example = "16823") + private Long id; + + @Schema(description = "多个配置ID",example = "16823") + private List ids; +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java index 9eaea3892..9c2ae0497 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java @@ -2,6 +2,8 @@ package cn.iocoder.yudao.module.llm.service.datarefluxdata; import cn.hutool.core.util.ObjectUtil; import cn.iocoder.yudao.module.llm.controller.admin.datarefluxconfig.vo.DataRefluxConfigPageReqVO; +import cn.iocoder.yudao.module.llm.dal.dataobject.datarefluxconfig.DataRefluxConfigDO; +import cn.iocoder.yudao.module.llm.dal.mysql.datarefluxconfig.DataRefluxConfigMapper; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import org.springframework.stereotype.Service; import javax.annotation.Resource; @@ -31,11 +33,19 @@ public class DataRefluxDataServiceImpl implements DataRefluxDataService { @Resource private DataRefluxDataMapper dataRefluxDataMapper; + @Resource + private DataRefluxConfigMapper dataRefluxConfigMapper; @Override public Long createDataRefluxData(DataRefluxDataSaveReqVO createReqVO) { // 插入 DataRefluxDataDO dataRefluxData = BeanUtils.toBean(createReqVO, DataRefluxDataDO.class); + LambdaQueryWrapper eq = new LambdaQueryWrapper() + .eq(DataRefluxConfigDO::getModelServiceId, dataRefluxData.getModelServiceId()) + .eq(DataRefluxConfigDO::getModelType, dataRefluxData.getModelType()) + .like(DataRefluxConfigDO::getCreator, dataRefluxData.getCreator()); + List dataRefluxConfigDOS = dataRefluxConfigMapper.selectList(eq); + dataRefluxData.setConfigId(dataRefluxConfigDOS.get(0).getId()); dataRefluxDataMapper.insert(dataRefluxData); // 返回 return dataRefluxData.getId();