From 23f39aba86f01ffb3cab4d68d6971d0b5adf0e52 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 11 Aug 2025 14:28:22 +0800
Subject: [PATCH] feat(ai): 新增雷达图数据项并更新报告模型
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java | 94 ++++++++++++++++++++++++++++++++++++++++++-----
1 files changed, 84 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..db3d4ba 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
@@ -4,6 +4,7 @@
import cc.mrbird.febs.ai.entity.AiMemberTalkItem;
import cc.mrbird.febs.ai.entity.AiProductRoleLink;
import cc.mrbird.febs.ai.mapper.AiMemberTalkMapper;
+import cc.mrbird.febs.ai.req.ai.AiRequest;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkAnswerDto;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkDto;
import cc.mrbird.febs.ai.res.ai.AiResponse;
@@ -14,6 +15,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;
@@ -25,6 +27,7 @@
import org.springframework.stereotype.Service;
import java.util.Date;
+import java.util.function.Consumer;
/**
* AI用户对话训练记录 Service实现类
@@ -59,7 +62,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,19 +75,90 @@
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);
+ }
+
+ @Override
+ public void startStream(ApiMemberTalkDto dto, Consumer<AiResponse> callback) {
+ try {
+ String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ String productId = dto.getId();
+
+ LambdaQueryWrapper<AiProductRoleLink> productLinkQuery = Wrappers.lambdaQuery(AiProductRoleLink.class);
+ productLinkQuery.eq(AiProductRoleLink::getProductId, productId);
+ productLinkQuery.last("limit 1");
+ AiProductRoleLink aiProductRoleLink = aiProductRoleLinkService.getByQuery(productLinkQuery);
+ if (ObjectUtil.isNull(aiProductRoleLink)) {
+ AiResponse aiResponse = new AiResponse();
+ aiResponse.setCode("500");
+ aiResponse.setDescription("产品AI陪练不存在");
+ callback.accept(aiResponse);
+ return;
+ }
+
+ // 构造AI请求
+ AiRequest aiRequest = new AiRequest();
+ aiRequest.setLinkId(aiProductRoleLink.getProductRoleId());
+ aiRequest.setPromptTemplate(aiProductRoleLink.getProductRoleId());
+ aiRequest.setContent("<strong>\"生成题目\"</strong>");
+
+ // 定义AI服务回调处理
+ Consumer<AiResponse> aiCallback = aiResponse -> {
+ Date nowTime = new Date();
+ LambdaQueryWrapper<AiMemberTalk> query = Wrappers.lambdaQuery(AiMemberTalk.class);
+ query.eq(AiMemberTalk::getMemberId, memberUuid);
+ query.eq(AiMemberTalk::getProductId, productId);
+ query.last("limit 1");
+ AiMemberTalk aiMemberTalk = this.getByQuery(query);
+ if (ObjectUtil.isNull(aiMemberTalk)) {
+ aiMemberTalk = this.add(memberUuid, productId, nowTime);
+ }
+ try {
+ if (aiResponse.getCode().equals("200")) {
+ // 如果是最终结果（包含报告）
+ if (aiResponse.getReport() != null) {
+ // 保存完整响应到数据库
+ aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(), 1, aiResponse.getResContext(), nowTime);
+ this.updateTimeUpdate(nowTime, aiMemberTalk.getId());
+
+ callback.accept(aiResponse);
+ } else {
+ // 流式响应片段
+ callback.accept(aiResponse);
+ }
+ } else {
+ callback.accept(aiResponse);
+ }
+ } catch (Exception e) {
+ log.error("处理AI响应异常", e);
+ callback.accept(aiResponse);
+ }
+ };
+
+ // 调用AI服务的流式接口
+ aiService.streamQuestion(aiRequest, aiCallback);
+
+ } catch (Exception e) {
+ log.error("流式调用start方法异常", e);
+
+ AiResponse aiResponse = new AiResponse();
+ aiResponse.setCode("500");
+ aiResponse.setDescription("流式调用start方法异常");
+ callback.accept(aiResponse);
+ }
}
@Override
@@ -102,14 +176,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 +191,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 +208,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