sion/sionrui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加导入

Browse Source
This commit is contained in:
shenaowei
2026-02-22 23:19:48 +08:00
parent 8ace847b9a
commit 547953cd00
10 changed files with 403 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/AiAgentController.java
View File

@@ -2,6 +2,8 @@ package cn.iocoder.yudao.module.tik.muye.aiagent;
import cn.iocoder.yudao.module.tik.muye.aiagent.dal.AiAgentDO;
import cn.iocoder.yudao.module.tik.muye.aiagent.service.AiAgentService;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentImportExcelVO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentImportRespVO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentPageReqVO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentRespVO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentSaveReqVO;
@@ -18,6 +20,8 @@ import jakarta.servlet.http.*;
import java.util.*;
import java.io.IOException;
import org.springframework.web.multipart.MultipartFile;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
@@ -101,4 +105,32 @@ public class AiAgentController {
BeanUtils.toBean(list, AiAgentRespVO.class));
}
@GetMapping("/get-import-template")
@Operation(summary = "获得AI智能体导入模板")
public void importTemplate(HttpServletResponse response) throws IOException {
// 构造示例数据
List<AiAgentImportExcelVO> list = Arrays.asList(
AiAgentImportExcelVO.builder()
.agentId("12345")
.agentName("示例智能体")
.categoryName("文案创作")
.icon("https://example.com/icon.png")
.status(1)
.description("这是一个示例智能体的描述")
.systemPrompt("你是一个有帮助的AI助手")
.remark("备注信息")
.build()
);
// 导出 Excel
ExcelUtils.write(response, "AI智能体导入模板.xls", "智能体列表", AiAgentImportExcelVO.class, list);
}
@PostMapping("/import")
@Operation(summary = "导入AI智能体")
@PreAuthorize("@ss.hasPermission('muye:ai-agent:create')")
public CommonResult<AiAgentImportRespVO> importExcel(@RequestParam("file") MultipartFile file) throws Exception {
List<AiAgentImportExcelVO> list = ExcelUtils.read(file, AiAgentImportExcelVO.class);
return success(aiAgentService.importAiAgentList(list));
}
}

10
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/service/AiAgentService.java
View File

@@ -3,6 +3,8 @@ package cn.iocoder.yudao.module.tik.muye.aiagent.service;
import java.util.*;
import cn.iocoder.yudao.module.tik.muye.aiagent.dal.AiAgentDO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentImportExcelVO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentImportRespVO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentPageReqVO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentSaveReqVO;
import jakarta.validation.*;
@@ -67,4 +69,12 @@ public interface AiAgentService {
*/
List<AiAgentDO> getEnabledAgentList();
/**
* 批量导入AI智能体
*
* @param importAgents 导入数据列表
* @return 导入结果
*/
AiAgentImportRespVO importAiAgentList(List<AiAgentImportExcelVO> importAgents);
}

72
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/service/AiAgentServiceImpl.java
View File

@@ -1,10 +1,14 @@
package cn.iocoder.yudao.module.tik.muye.aiagent.service;
import cn.iocoder.yudao.framework.common.exception.ErrorCode;
import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
import cn.iocoder.yudao.module.tik.muye.aiagent.dal.AiAgentDO;
import cn.iocoder.yudao.module.tik.muye.aiagent.mapper.AiAgentMapper;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentImportExcelVO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentImportRespVO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentPageReqVO;
import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AiAgentSaveReqVO;
import cn.hutool.core.collection.CollUtil;
import org.springframework.stereotype.Service;
import jakarta.annotation.Resource;
import org.springframework.validation.annotation.Validated;
@@ -12,6 +16,7 @@ import org.springframework.validation.annotation.Validated;
import java.util.*;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
/**
@@ -80,4 +85,71 @@ public class AiAgentServiceImpl implements AiAgentService {
return aiAgentMapper.selectEnabledList();
}
@Override
public AiAgentImportRespVO importAiAgentList(List<AiAgentImportExcelVO> importAgents) {
// 1. 参数校验
if (CollUtil.isEmpty(importAgents)) {
throw exception(new ErrorCode(1005, "导入数据不能为空"));
}
// 2. 初始化返回结果
AiAgentImportRespVO respVO = AiAgentImportRespVO.builder()
.createAgentNames(new ArrayList<>())
.failureAgentNames(new LinkedHashMap<>())
.build();
// 3. 获取当前登录用户信息
Long operatorId = SecurityFrameworkUtils.getLoginUserId();
String operatorName = String.valueOf(SecurityFrameworkUtils.getLoginUserNickname());
// 4. 遍历逐条处理
importAgents.forEach(importAgent -> {
try {
// 4.1 必填字段校验
if (importAgent.getAgentId() == null || importAgent.getAgentId().isEmpty()) {
throw new RuntimeException("智能体ID不能为空");
}
if (importAgent.getAgentName() == null || importAgent.getAgentName().isEmpty()) {
throw new RuntimeException("智能体名称不能为空");
}
if (importAgent.getCategoryName() == null || importAgent.getCategoryName().isEmpty()) {
throw new RuntimeException("分类不能为空");
}
if (importAgent.getIcon() == null || importAgent.getIcon().isEmpty()) {
throw new RuntimeException("图标URL不能为空");
}
if (importAgent.getStatus() == null) {
throw new RuntimeException("状态不能为空");
}
if (importAgent.getDescription() == null || importAgent.getDescription().isEmpty()) {
throw new RuntimeException("设定描述不能为空");
}
if (importAgent.getSystemPrompt() == null || importAgent.getSystemPrompt().isEmpty()) {
throw new RuntimeException("预置提示词不能为空");
}
// 4.2 检查 agentId 是否已存在
AiAgentDO existAgent = aiAgentMapper.selectOne(
new LambdaQueryWrapperX<AiAgentDO>()
.eq(AiAgentDO::getAgentId, importAgent.getAgentId())
);
if (existAgent != null) {
respVO.getFailureAgentNames().put(importAgent.getAgentName(), "智能体ID已存在");
return;
}
// 4.3 创建智能体
AiAgentDO aiAgent = BeanUtils.toBean(importAgent, AiAgentDO.class);
aiAgent.setOperatorId(operatorId);
aiAgent.setOperatorName(operatorName);
aiAgentMapper.insert(aiAgent);
respVO.getCreateAgentNames().add(importAgent.getAgentName());
} catch (Exception ex) {
respVO.getFailureAgentNames().put(importAgent.getAgentName(), ex.getMessage());
}
});
return respVO;
}
}

46
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/vo/AiAgentImportExcelVO.java Normal file
View File

@@ -0,0 +1,46 @@
package cn.iocoder.yudao.module.tik.muye.aiagent.vo;
import cn.idev.excel.annotation.ExcelProperty;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
/**
* AI智能体导入 Excel VO
*
* @author 芋道源码
*/
@Schema(description = "管理后台 - AI智能体导入 Excel VO")
@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class AiAgentImportExcelVO {
@ExcelProperty("智能体ID")
private String agentId;
@ExcelProperty("智能体名称")
private String agentName;
@ExcelProperty("分类")
private String categoryName;
@ExcelProperty("图标URL")
private String icon;
@ExcelProperty("状态")
private Integer status;
@ExcelProperty("设定描述")
private String description;
@ExcelProperty("预置提示词")
private String systemPrompt;
@ExcelProperty("备注")
private String remark;
}

34
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/vo/AiAgentImportRespVO.java Normal file
View File

@@ -0,0 +1,34 @@
package cn.iocoder.yudao.module.tik.muye.aiagent.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
/**
* AI智能体导入结果 Response VO
*
* @author 芋道源码
*/
@Schema(description = "管理后台 - AI智能体导入结果 Response VO")
@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class AiAgentImportRespVO {
@Schema(description = "创建成功的智能体名称数组")
@Builder.Default
private List<String> createAgentNames = new ArrayList<>();
@Schema(description = "导入失败的智能体集合，key 为智能体名称，value 为失败原因")
@Builder.Default
private Map<String, String> failureAgentNames = new LinkedHashMap<>();
}