BJ_ZTEITS-Developers / parking-cloud-yudao

Commit ababc914bd5f6ba65b1348d60b2b78d24ac9f37c

Authored by xiaoxin
1 parent 6e71b721

【新增】AI 脑图

Inline Side-by-side
Showing 6 changed files with 285 additions and 0 deletions
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/mindmap/AiMindMapController.java 0 → 100644
  View file @ababc91
  1 +package cn.iocoder.yudao.module.ai.controller.admin.mindmap;
  2 +
  3 +import cn.iocoder.yudao.framework.common.pojo.CommonResult;
  4 +import cn.iocoder.yudao.module.ai.controller.admin.mindmap.vo.AiMindMapGenerateReqVO;
  5 +import cn.iocoder.yudao.module.ai.service.mindmap.AiMindMapService;
  6 +import io.swagger.v3.oas.annotations.Operation;
  7 +import io.swagger.v3.oas.annotations.tags.Tag;
  8 +import jakarta.annotation.Resource;
  9 +import jakarta.annotation.security.PermitAll;
  10 +import jakarta.validation.Valid;
  11 +import org.springframework.http.MediaType;
  12 +import org.springframework.web.bind.annotation.PostMapping;
  13 +import org.springframework.web.bind.annotation.RequestBody;
  14 +import org.springframework.web.bind.annotation.RequestMapping;
  15 +import org.springframework.web.bind.annotation.RestController;
  16 +import reactor.core.publisher.Flux;
  17 +
  18 +import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;
  19 +
  20 +@Tag(name = "管理后台 - AI 思维导图")
  21 +@RestController
  22 +@RequestMapping("/ai/mind-map")
  23 +public class AiMindMapController {
  24 +
  25 + @Resource
  26 + private AiMindMapService mindMapService;
  27 +
  28 + @PostMapping(value = "/generate-stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
  29 + @Operation(summary = "发送消息（流式）", description = "流式返回，响应较快")
  30 + @PermitAll // 解决 SSE 最终响应的时候，会被 Access Denied 拦截的问题
  31 + public Flux<CommonResult<String>> generateMindMap(@RequestBody @Valid AiMindMapGenerateReqVO generateReqVO) {
  32 + return mindMapService.generateMindMap(generateReqVO, getLoginUserId());
  33 + }
  34 +
  35 +}
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/mindmap/vo/AiMindMapGenerateReqVO.java 0 → 100644
  View file @ababc91
  1 +package cn.iocoder.yudao.module.ai.controller.admin.mindmap.vo;
  2 +
  3 +import io.swagger.v3.oas.annotations.media.Schema;
  4 +import jakarta.validation.constraints.NotBlank;
  5 +import lombok.Data;
  6 +
  7 +@Schema(description = "管理后台 - AI 思维导图生成 Request VO")
  8 +@Data
  9 +public class AiMindMapGenerateReqVO {
  10 + @Schema(description = "思维导图内容提示", example = "Java 学习路线")
  11 + @NotBlank(message = "思维导图内容提示不能为空")
  12 + private String prompt;
  13 +}
0 \ No newline at end of file 14 \ No newline at end of file
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/mindmap/AiMindMapDO.java 0 → 100644
  View file @ababc91
  1 +package cn.iocoder.yudao.module.ai.dal.dataobject.mindmap;
  2 +
  3 +import cn.iocoder.yudao.framework.ai.core.enums.AiPlatformEnum;
  4 +import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
  5 +import com.baomidou.mybatisplus.annotation.IdType;
  6 +import com.baomidou.mybatisplus.annotation.TableId;
  7 +import com.baomidou.mybatisplus.annotation.TableName;
  8 +import lombok.Data;
  9 +
  10 +/**
  11 + * AI 思维导图 DO
  12 + *
  13 + * @author xiaoxin
  14 + */
  15 +@TableName(value = "ai_mind_map", autoResultMap = true)
  16 +@Data
  17 +public class AiMindMapDO extends BaseDO {
  18 +
  19 + /**
  20 + * 编号
  21 + */
  22 + @TableId(type = IdType.AUTO)
  23 + private Long id;
  24 +
  25 + /**
  26 + * 用户编号
  27 + */
  28 + private Long userId;
  29 +
  30 + /**
  31 + * 模型
  32 + */
  33 + private String model;
  34 +
  35 + /**
  36 + * 平台
  37 + * <p>
  38 + * 枚举 {@link AiPlatformEnum}
  39 + */
  40 + private String platform;
  41 +
  42 + /**
  43 + * 生成内容提示
  44 + */
  45 + private String prompt;
  46 +
  47 + /**
  48 + * 生成的内容
  49 + */
  50 + private String generatedContent;
  51 +
  52 + /**
  53 + * 错误信息
  54 + */
  55 + private String errorMessage;
  56 +
  57 +}
0 \ No newline at end of file 58 \ No newline at end of file
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/mindmap/AiMindMapMapper.java 0 → 100644
  View file @ababc91
  1 +package cn.iocoder.yudao.module.ai.dal.mysql.mindmap;
  2 +
  3 +import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
  4 +import cn.iocoder.yudao.module.ai.dal.dataobject.mindmap.AiMindMapDO;
  5 +import org.apache.ibatis.annotations.Mapper;
  6 +
  7 +/**
  8 + * AI 音乐 Mapper
  9 + *
  10 + * @author xiaoxin
  11 + */
  12 +@Mapper
  13 +public interface AiMindMapMapper extends BaseMapperX<AiMindMapDO> {
  14 +}
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapService.java 0 → 100644
  View file @ababc91
  1 +package cn.iocoder.yudao.module.ai.service.mindmap;
  2 +
  3 +import cn.iocoder.yudao.framework.common.pojo.CommonResult;
  4 +import cn.iocoder.yudao.module.ai.controller.admin.mindmap.vo.AiMindMapGenerateReqVO;
  5 +import reactor.core.publisher.Flux;
  6 +
  7 +/**
  8 + * AI 思维导图 Service 接口
  9 + *
  10 + * @author xiaoxin
  11 + */
  12 +public interface AiMindMapService {
  13 +
  14 + /**
  15 + * 生成思维导图内容
  16 + *
  17 + * @param generateReqVO 请求参数
  18 + * @param userId 用户编号
  19 + * @return 生成结果
  20 + */
  21 + Flux<CommonResult<String>> generateMindMap(AiMindMapGenerateReqVO generateReqVO, Long userId);
  22 +
  23 +}
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapServiceImpl.java 0 → 100644
  View file @ababc91
  1 +package cn.iocoder.yudao.module.ai.service.mindmap;
  2 +
  3 +import cn.hutool.core.collection.CollUtil;
  4 +import cn.hutool.core.util.StrUtil;
  5 +import cn.iocoder.yudao.framework.ai.core.enums.AiPlatformEnum;
  6 +import cn.iocoder.yudao.framework.ai.core.util.AiUtils;
  7 +import cn.iocoder.yudao.framework.common.pojo.CommonResult;
  8 +import cn.iocoder.yudao.framework.common.pojo.PageResult;
  9 +import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
  10 +import cn.iocoder.yudao.framework.tenant.core.util.TenantUtils;
  11 +import cn.iocoder.yudao.module.ai.controller.admin.mindmap.vo.AiMindMapGenerateReqVO;
  12 +import cn.iocoder.yudao.module.ai.controller.admin.model.vo.chatRole.AiChatRolePageReqVO;
  13 +import cn.iocoder.yudao.module.ai.dal.dataobject.mindmap.AiMindMapDO;
  14 +import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiChatModelDO;
  15 +import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiChatRoleDO;
  16 +import cn.iocoder.yudao.module.ai.dal.mysql.mindmap.AiMindMapMapper;
  17 +import cn.iocoder.yudao.module.ai.enums.ErrorCodeConstants;
  18 +import cn.iocoder.yudao.module.ai.service.model.AiApiKeyService;
  19 +import cn.iocoder.yudao.module.ai.service.model.AiChatModelService;
  20 +import cn.iocoder.yudao.module.ai.service.model.AiChatRoleService;
  21 +import jakarta.annotation.Resource;
  22 +import lombok.extern.slf4j.Slf4j;
  23 +import org.springframework.ai.chat.messages.Message;
  24 +import org.springframework.ai.chat.messages.SystemMessage;
  25 +import org.springframework.ai.chat.messages.UserMessage;
  26 +import org.springframework.ai.chat.model.ChatModel;
  27 +import org.springframework.ai.chat.model.ChatResponse;
  28 +import org.springframework.ai.chat.prompt.ChatOptions;
  29 +import org.springframework.ai.chat.prompt.Prompt;
  30 +import org.springframework.stereotype.Service;
  31 +import reactor.core.publisher.Flux;
  32 +
  33 +import java.util.ArrayList;
  34 +import java.util.List;
  35 +import java.util.Objects;
  36 +
  37 +import static cn.iocoder.yudao.framework.common.pojo.CommonResult.error;
  38 +import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
  39 +
  40 +/**
  41 + * AI 写作 Service 实现类
  42 + *
  43 + * @author xiaoxin
  44 + */
  45 +@Service
  46 +@Slf4j
  47 +public class AiMindMapServiceImpl implements AiMindMapService {
  48 +
  49 + @Resource
  50 + private AiApiKeyService apiKeyService;
  51 + @Resource
  52 + private AiChatModelService chatModalService;
  53 + @Resource
  54 + private AiChatRoleService chatRoleService;
  55 +
  56 + @Resource
  57 + private AiMindMapMapper mindMapMapper;
  58 +
  59 + private static final String DEFAULT_SYSTEM_MESSAGE = """
  60 + 你是一位非常优秀的思维导图助手，你会把用户的所有提问都总结成思维导图，然后以 Markdown 格式输出。markdown 只需要输出一级标题，二级标题，三级标题，四级标题，最多输出四级，除此之外不要输出任何其他 markdown 标记。下面是一个合格的例子：
  61 + # Geek-AI 助手
  62 +
  63 + ## 完整的开源系统
  64 + ### 前端开源
  65 + ### 后端开源
  66 +
  67 + ## 支持各种大模型
  68 + ### OpenAI
  69 + ### Azure
  70 + ### 文心一言
  71 + ### 通义千问
  72 +
  73 + ## 集成多种收费方式
  74 + ### 支付宝
  75 + ### 微信
  76 +
  77 + 另外，除此之外不要任何解释性语句。
  78 + """;
  79 +
  80 + @Override
  81 + public Flux<CommonResult<String>> generateMindMap(AiMindMapGenerateReqVO generateReqVO, Long userId) {
  82 + // 1.1 获取脑图模型 尝试获取思维导图助手角色，如果没有则使用默认模型
  83 + AiChatRoleDO mindMapRole = selectOneMindMapRole();
  84 + AiChatModelDO model;
  85 + String systemMessage;
  86 + if (Objects.nonNull(mindMapRole)) {
  87 + model = chatModalService.getChatModel(mindMapRole.getModelId());
  88 + systemMessage = mindMapRole.getSystemMessage();
  89 + } else {
  90 + model = chatModalService.getRequiredDefaultChatModel();
  91 + systemMessage = DEFAULT_SYSTEM_MESSAGE;
  92 + }
  93 +
  94 + AiPlatformEnum platform = AiPlatformEnum.validatePlatform(model.getPlatform());
  95 + ChatModel chatModel = apiKeyService.getChatModel(model.getKeyId());
  96 +
  97 + // 1.2 插入思维导图信息
  98 + AiMindMapDO mindMapDO = BeanUtils.toBean(generateReqVO, AiMindMapDO.class, e -> e.setUserId(userId).setModel(model.getModel()).setPlatform(platform.getPlatform()));
  99 + mindMapMapper.insert(mindMapDO);
  100 +
  101 + ChatOptions chatOptions = AiUtils.buildChatOptions(platform, model.getModel(), model.getTemperature(), model.getMaxTokens());
  102 + // 2.1 角色设定
  103 + List<Message> chatMessages = new ArrayList<>();
  104 + if (StrUtil.isNotBlank(systemMessage)) {
  105 + chatMessages.add(new SystemMessage(systemMessage));
  106 + }
  107 + // 2.2 用户输入
  108 + chatMessages.add(new UserMessage(generateReqVO.getPrompt()));
  109 + // 2.3 构建提示词
  110 + Prompt prompt = new Prompt(chatMessages, chatOptions);
  111 +
  112 + Flux<ChatResponse> streamResponse = chatModel.stream(prompt);
  113 + // 2.4 流式返回
  114 + StringBuffer contentBuffer = new StringBuffer();
  115 + return streamResponse.map(chunk -> {
  116 + String newContent = chunk.getResult() != null ? chunk.getResult().getOutput().getContent() : null;
  117 + newContent = StrUtil.nullToDefault(newContent, ""); // 避免 null 的 情况
  118 + contentBuffer.append(newContent);
  119 + // 响应结果
  120 + return success(newContent);
  121 + }).doOnComplete(() -> {
  122 + // 忽略租户，因为 Flux 异步无法透传租户
  123 + TenantUtils.executeIgnore(() ->
  124 + mindMapMapper.updateById(new AiMindMapDO().setId(mindMapDO.getId()).setGeneratedContent(contentBuffer.toString())));
  125 + }).doOnError(throwable -> {
  126 + log.error("[generateWriteContent][generateReqVO({}) 发生异常]", generateReqVO, throwable);
  127 + // 忽略租户，因为 Flux 异步无法透传租户
  128 + TenantUtils.executeIgnore(() ->
  129 + mindMapMapper.updateById(new AiMindMapDO().setId(mindMapDO.getId()).setErrorMessage(throwable.getMessage())));
  130 + }).onErrorResume(error -> Flux.just(error(ErrorCodeConstants.WRITE_STREAM_ERROR)));
  131 +
  132 + }
  133 +
  134 + private AiChatRoleDO selectOneMindMapRole() {
  135 + AiChatRoleDO chatRoleDO = null;
  136 + PageResult<AiChatRoleDO> mindMapRolePage = chatRoleService.getChatRolePage(new AiChatRolePageReqVO().setName("思维导图助手"));
  137 + List<AiChatRoleDO> list = mindMapRolePage.getList();
  138 + if (CollUtil.isNotEmpty(list)) {
  139 + chatRoleDO = list.get(0);
  140 + }
  141 + return chatRoleDO;
  142 + }
  143 +}