From 5a624d468c8f4eddd89c8cf9b99eb6466fa21481 Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Wed, 06 Aug 2025 17:55:18 +0800 Subject: [PATCH] feat(ai): 增加 AI 陪练报告数据解析功能 - 新增 Report、RadarData 和 Evaluation 类用于解析报告数据 - 在 AiService 接口中添加 extractReportData 方法 - 在 AiServiceImpl 中实现报告数据的提取和解析 - 更新 ApiMemberTalkVo,增加 report 字段用于存储解析后的报告数据 - 修改前端相关的回答格式和类型 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java | 22 ++++++++++++---------- 1 files changed, 12 insertions(+), 10 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java index ddec2bb..6c88c98 100644 --- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java @@ -14,6 +14,7 @@ import cc.mrbird.febs.ai.service.AiService; import cc.mrbird.febs.ai.utils.UUID; import cc.mrbird.febs.common.entity.FebsResponse; +import cc.mrbird.febs.common.exception.FebsException; import cc.mrbird.febs.common.utils.LoginUserUtil; import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.StrUtil; @@ -59,7 +60,7 @@ productLinkQuery.last("limit 1"); AiProductRoleLink aiProductRoleLink = aiProductRoleLinkService.getByQuery(productLinkQuery); if(ObjectUtil.isNull(aiProductRoleLink)){ - throw new RuntimeException("产品AI陪练不存在"); + throw new FebsException("产品AI陪练不存在"); } Date nowTime = new Date(); @@ -72,17 +73,18 @@ aiMemberTalk = this.add(memberUuid,productId,nowTime); } - AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(),"开始出题"); + AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(),"<strong>\"生成题目\"</strong>"); if(aiResponse.getCode().equals("200")){ aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),1,aiResponse.getResContext(),nowTime); this.updateTimeUpdate(nowTime,aiMemberTalk.getId()); }else{ - throw new RuntimeException(aiResponse.getDescription()); + throw new FebsException(aiResponse.getDescription()); } ApiMemberTalkVo apiMemberTalkVo = new ApiMemberTalkVo(); apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId()); apiMemberTalkVo.setType(1); apiMemberTalkVo.setContext(aiResponse.getResContext()); + apiMemberTalkVo.setReport(aiResponse.getReport()); return new FebsResponse().success().data(apiMemberTalkVo); } @@ -102,14 +104,14 @@ } - public static final String ANSWER_FORMAT = "{}/n回答:{}/n请根据回答给出以下四个方面的总结,这个四个方面分别是亮点、建议、参考答案和核心知识点回顾。重点:四个方面的总结都是必须要有内容。"; + public static final String ANSWER_FORMAT = "{}/n[回答]{}/n"; @Override public FebsResponse answer(ApiMemberTalkAnswerDto dto) { String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid(); String memberTalkId = dto.getId(); AiMemberTalk aiMemberTalk = this.getById(memberTalkId); if (ObjectUtil.isNull(aiMemberTalk)){ - throw new RuntimeException("产品AI陪练对话不存在"); + throw new FebsException("产品AI陪练对话不存在"); } LambdaQueryWrapper<AiProductRoleLink> productLinkQuery = Wrappers.lambdaQuery(AiProductRoleLink.class); @@ -117,7 +119,7 @@ productLinkQuery.last("limit 1"); AiProductRoleLink aiProductRoleLink = aiProductRoleLinkService.getByQuery(productLinkQuery); if(ObjectUtil.isNull(aiProductRoleLink)){ - throw new RuntimeException("产品AI陪练不存在"); + throw new FebsException("产品AI陪练不存在"); } String reqContext = dto.getReqContext(); @@ -134,21 +136,21 @@ String format = StrUtil.format(ANSWER_FORMAT, aiMemberTalkItem.getContext(), reqContext); log.info("format:{}",format); AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(), format); +// AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(), reqContext); if(aiResponse.getCode().equals("200")){ Date nowTime = new Date(); aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),3,aiResponse.getResContext(),nowTime); this.updateTimeUpdate(nowTime,aiMemberTalk.getId()); }else{ - throw new RuntimeException(aiResponse.getDescription()); + throw new FebsException(aiResponse.getDescription()); } ApiMemberTalkVo apiMemberTalkVo = new ApiMemberTalkVo(); apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId()); - apiMemberTalkVo.setType(1); + apiMemberTalkVo.setType(3); apiMemberTalkVo.setContext(aiResponse.getResContext()); - + apiMemberTalkVo.setReport(aiResponse.getReport()); return new FebsResponse().success().data(apiMemberTalkVo); } - @Override public AiMemberTalk add(String memberUuid, String productId, Date nowTime) { AiMemberTalk aiMemberTalk = new AiMemberTalk(); -- Gitblit v1.9.1