From c7cee264bccb5026fd42a9f0dc83d274aaee40cf Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Thu, 30 Oct 2025 10:05:28 +0800
Subject: [PATCH] feat(ai): 新增AI陪练流式对话V2接口
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java | 346 ++++++++++++++++++++++++++++++++++++++++++++++----------
1 files changed, 281 insertions(+), 65 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 8aaa3f9..c30e579 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
@@ -1,16 +1,46 @@
package cc.mrbird.febs.ai.service.impl;
+import cc.mrbird.febs.ai.enumerates.AiTypeEnum;
import cc.mrbird.febs.ai.entity.AiMemberTalk;
+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.AiMessage;
+import cc.mrbird.febs.ai.req.ai.AiRequest;
+import cc.mrbird.febs.ai.req.member.ApiMemberTalkPageDto;
+import cc.mrbird.febs.ai.req.memberTalk.*;
+import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkListDto;
+import cc.mrbird.febs.ai.res.ai.AiResponse;
+import cc.mrbird.febs.ai.res.member.ApiMemberTalkPageVo;
+import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkListVo;
+import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkReloadVo;
+import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkVo;
+import cc.mrbird.febs.ai.res.memberTalk.ApiTalkMemberListVo;
+import cc.mrbird.febs.ai.service.AiMemberTalkItemService;
import cc.mrbird.febs.ai.service.AiMemberTalkService;
+import cc.mrbird.febs.ai.service.AiProductRoleLinkService;
+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;
+import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
+import reactor.core.publisher.Flux;
+import java.util.Date;
import java.util.List;
+import java.util.function.Consumer;
/**
* AI用户对话训练记录 Service实现类
@@ -24,98 +54,284 @@
public class AiMemberTalkServiceImpl extends ServiceImpl<AiMemberTalkMapper, AiMemberTalk> implements AiMemberTalkService {
private final AiMemberTalkMapper aiMemberTalkMapper;
+ private final AiProductRoleLinkService aiProductRoleLinkService;
+ private final AiMemberTalkItemService aiMemberTalkItemService;
+ private final AiService aiService;
+
@Override
public AiMemberTalk getById(String id) {
- return this.getById(id);
+ return aiMemberTalkMapper.selectById( id);
}
@Override
- public List<AiMemberTalk> getByMemberId(String memberId) {
- LambdaQueryWrapper<AiMemberTalk> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiMemberTalk::getMemberId, memberId);
- queryWrapper.orderByDesc(AiMemberTalk::getCreatedTime);
- return this.list(queryWrapper);
+ @Transactional
+ public FebsResponse start(ApiMemberTalkDto dto) {
+
+ 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)){
+ throw new FebsException("产品AI陪练不存在");
+ }
+ String companyId = aiProductRoleLink.getCompanyId();
+
+ 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,companyId,productId,nowTime,5);
+ }
+
+ List<AiMessage> aiMessageDtoList = aiMemberTalkItemService.getQuestionUpDownContext(aiMemberTalk.getId(),AiTypeEnum.QUESTION.getCode());
+
+ AiResponse aiResponse = aiService.start(aiMessageDtoList,AiTypeEnum.QUESTION.getCode(),aiProductRoleLink.getProductRoleId(),AiTypeEnum.QUESTION.getName(), null);
+
+ ApiMemberTalkVo apiMemberTalkVo = new ApiMemberTalkVo();
+ apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId());
+ apiMemberTalkVo.setType(1);
+ apiMemberTalkVo.setContext(aiResponse.getResContext());
+ if(aiResponse.getCode().equals("200")){
+ AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 1, aiResponse.getResContext(), nowTime);
+ apiMemberTalkVo.setMemberTalkItemId(aiMemberTalkItem.getId());
+ this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
+ }else{
+ throw new FebsException(aiResponse.getDescription());
+ }
+ return new FebsResponse().success().data(apiMemberTalkVo);
}
@Override
- public List<AiMemberTalk> getByProductId(String productId) {
- LambdaQueryWrapper<AiMemberTalk> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiMemberTalk::getProductId, productId);
- queryWrapper.orderByDesc(AiMemberTalk::getCreatedTime);
- return this.list(queryWrapper);
+ public FebsResponse reload(ApiMemberTalkReloadDto dto) {
+
+ String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ ApiMemberTalkReloadVo apiMemberTalkReloadVo = new ApiMemberTalkReloadVo();
+ String memberTalkId = dto.getMemberTalkId();
+ LambdaQueryWrapper<AiMemberTalkItem> queryWrapper = Wrappers.lambdaQuery(AiMemberTalkItem.class);
+ queryWrapper.eq(AiMemberTalkItem::getMemberTalkId,memberTalkId);
+ queryWrapper.eq(AiMemberTalkItem::getType,AiTypeEnum.QUESTION_ANSWER.getCode());
+ queryWrapper.orderByDesc(AiMemberTalkItem::getCreatedTime);
+ queryWrapper.last("limit 1");
+ AiMemberTalkItem byQuery = aiMemberTalkItemService.getOneByQuery(queryWrapper);
+ if (ObjectUtil.isNotNull(byQuery)){
+ apiMemberTalkReloadVo.setContext(byQuery.getContext());
+ apiMemberTalkReloadVo.setMemberTalkId(memberTalkId);
+ apiMemberTalkReloadVo.setType(AiTypeEnum.QUESTION_ANSWER.getCode());
+ }
+ return new FebsResponse().success().data(apiMemberTalkReloadVo);
}
@Override
- public List<AiMemberTalk> getByProductRoleId(String productRoleId) {
- LambdaQueryWrapper<AiMemberTalk> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiMemberTalk::getProductRoleId, productRoleId);
- queryWrapper.orderByDesc(AiMemberTalk::getCreatedTime);
- return this.list(queryWrapper);
- }
-
- @Override
- public List<AiMemberTalk> getByCompanyId(String companyId) {
- LambdaQueryWrapper<AiMemberTalk> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiMemberTalk::getCompanyId, companyId);
- queryWrapper.orderByDesc(AiMemberTalk::getCreatedTime);
- return this.list(queryWrapper);
- }
-
- @Override
- @Transactional(rollbackFor = Exception.class)
- public boolean saveTalk(AiMemberTalk aiMemberTalk) {
+ public void startStream(ApiMemberTalkDto dto, Consumer<AiResponse> callback) {
try {
- return this.save(aiMemberTalk);
+ 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;
+ }
+ String companyId = aiProductRoleLink.getCompanyId();
+
+ // 构造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,companyId, productId, nowTime,5);
+ }
+ try {
+ if (aiResponse.getCode().equals("200")) {
+ // 如果是最终结果（包含报告）
+ if (aiResponse.getReport() != null) {
+ // 保存完整响应到数据库
+ aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 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("保存对话记录失败: ", e);
- return false;
+ log.error("流式调用start方法异常", e);
+
+ AiResponse aiResponse = new AiResponse();
+ aiResponse.setCode("500");
+ aiResponse.setDescription("流式调用start方法异常");
+ callback.accept(aiResponse);
}
}
@Override
- @Transactional(rollbackFor = Exception.class)
- public boolean saveBatchTalks(List<AiMemberTalk> talks) {
- try {
- return this.saveBatch(talks);
- } catch (Exception e) {
- log.error("批量保存对话记录失败: ", e);
- return false;
- }
+ public AiMemberTalk getByQuery(LambdaQueryWrapper<AiMemberTalk> query) {
+ return aiMemberTalkMapper.selectOne( query);
}
@Override
- @Transactional(rollbackFor = Exception.class)
- public boolean updateTalk(AiMemberTalk aiMemberTalk) {
- try {
- return this.updateById(aiMemberTalk);
- } catch (Exception e) {
- log.error("更新对话记录失败: ", e);
- return false;
+ public void updateTimeUpdate(Date nowTime, String id) {
+ aiMemberTalkMapper.update(
+ null,
+ Wrappers.lambdaUpdate(AiMemberTalk.class)
+ .set(AiMemberTalk::getUpdatedTime,nowTime)
+ .eq(AiMemberTalk::getId,id));
+ }
+
+
+ public static final String ANSWER_FORMAT = "###题目：{}###用户回答：{}";
+ @Override
+ @Transactional
+ 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 FebsException("产品AI陪练对话不存在");
}
+
+ String companyId = aiMemberTalk.getCompanyId();
+
+ LambdaQueryWrapper<AiProductRoleLink> productLinkQuery = Wrappers.lambdaQuery(AiProductRoleLink.class);
+ productLinkQuery.eq(AiProductRoleLink::getProductId,aiMemberTalk.getProductId());
+ productLinkQuery.last("limit 1");
+ AiProductRoleLink aiProductRoleLink = aiProductRoleLinkService.getByQuery(productLinkQuery);
+ if(ObjectUtil.isNull(aiProductRoleLink)){
+ throw new FebsException("产品AI陪练不存在");
+ }
+
+ String reqContext = dto.getReqContext();
+
+ LambdaQueryWrapper<AiMemberTalkItem> memberTalkItemQuery = Wrappers.lambdaQuery(AiMemberTalkItem.class);
+ memberTalkItemQuery.eq(AiMemberTalkItem::getMemberId,memberUuid);
+ memberTalkItemQuery.eq(AiMemberTalkItem::getMemberTalkId,memberTalkId);
+ memberTalkItemQuery.eq(AiMemberTalkItem::getType,1);
+ memberTalkItemQuery.orderByDesc(AiMemberTalkItem::getCreatedTime);
+ memberTalkItemQuery.last("limit 1");
+ AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.getOneByQuery(memberTalkItemQuery);
+ aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),companyId,2,reqContext,new Date());
+
+ String format = StrUtil.format(ANSWER_FORMAT, aiMemberTalkItem.getContext(), reqContext);
+ log.info("format:{}",format);
+// AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(), format);
+
+
+ ApiMemberTalkVo apiMemberTalkVo = new ApiMemberTalkVo();
+ apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId());
+ apiMemberTalkVo.setType(3);
+ List<AiMessage> aiMessageDtoList = aiMemberTalkItemService.getQuestionUpDownContext(aiMemberTalk.getId(),AiTypeEnum.ANSWER.getCode());
+ AiResponse aiResponse = aiService.start(aiMessageDtoList,AiTypeEnum.ANSWER.getCode(),aiProductRoleLink.getProductRoleId(), reqContext,aiMemberTalkItem.getContext());
+ String context = null;
+ if(aiResponse.getCode().equals("200")){
+ Date nowTime = new Date();
+ context = String.valueOf(JSONUtil.parse(aiResponse.getReport()));
+ if ("null".equals( context)){
+ context = aiResponse.getResContext();
+ }
+ AiMemberTalkItem talkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 3, context, nowTime);
+ this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
+
+
+ apiMemberTalkVo.setContext(context);
+ apiMemberTalkVo.setMemberTalkItemId(talkItem.getId());
+ }else{
+ throw new FebsException(aiResponse.getDescription());
+ }
+ return new FebsResponse().success().data(apiMemberTalkVo);
+ }
+ @Override
+ public AiMemberTalk add(String memberUuid,String companyId, String productId, Date nowTime,Integer questionCount) {
+ AiMemberTalk aiMemberTalk = new AiMemberTalk();
+ aiMemberTalk.setId(UUID.getSimpleUUIDString());
+ aiMemberTalk.setCompanyId(companyId);
+ aiMemberTalk.setCreatedTime(nowTime);
+ aiMemberTalk.setMemberId(memberUuid);
+ aiMemberTalk.setProductId(productId);
+ aiMemberTalk.setQuestionCnt(questionCount);
+ aiMemberTalk.setDoneCnt(0);
+ aiMemberTalk.setState(AiTypeEnum.AI_MEMBER_TALK_STATE_ING.getCode());
+ aiMemberTalkMapper.insert(aiMemberTalk);
+ return aiMemberTalk;
}
@Override
- @Transactional(rollbackFor = Exception.class)
- public boolean deleteById(String id) {
- try {
- return this.removeById(id);
- } catch (Exception e) {
- log.error("删除对话记录失败: ", e);
- return false;
- }
+ public FebsResponse historyPage(ApiMemberTalkItemPageDto dto) {
+ String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ dto.setMemberUuid(memberUuid);
+ return new FebsResponse().success().data(aiMemberTalkItemService.historyPage(dto));
}
@Override
- @Transactional(rollbackFor = Exception.class)
- public boolean deleteByMemberId(String memberId) {
- try {
- LambdaQueryWrapper<AiMemberTalk> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiMemberTalk::getMemberId, memberId);
- return this.remove(queryWrapper);
- } catch (Exception e) {
- log.error("根据用户ID删除对话记录失败: ", e);
- return false;
- }
+ public Flux<FebsResponse> answerStream(String question) {
+
+// String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ return aiService.answerStream(question);
}
+
+ @Override
+ public List<AiMemberTalk> getListByCompanyId(String companyId) {
+ return aiMemberTalkMapper.selectList(
+ Wrappers.lambdaQuery(AiMemberTalk.class)
+ .eq(AiMemberTalk::getCompanyId, companyId)
+ .eq(AiMemberTalk::getState, AiTypeEnum.AI_MEMBER_TALK_STATE_DONE.getCode())
+ );
+ }
+
+ @Override
+ public List<AiMemberTalk> getListByCompanyIdAndMemberUuid(String companyId, String memberUuid) {
+ return aiMemberTalkMapper.selectList(
+ Wrappers.lambdaQuery(AiMemberTalk.class)
+ .eq(AiMemberTalk::getCompanyId, companyId)
+ .eq(AiMemberTalk::getMemberId, memberUuid)
+ .eq(AiMemberTalk::getState, AiTypeEnum.AI_MEMBER_TALK_STATE_DONE.getCode())
+ );
+ }
+
+ @Override
+ public IPage<ApiMemberTalkListVo> getPageByDto(ApiMemberTalkListDto dto) {
+ // 创建分页对象，传入当前页和每页大小
+ Page<ApiMemberTalkListVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
+ return aiMemberTalkMapper.getPageByDto(page, dto);
+ }
+
+ @Override
+ public Page<ApiMemberTalkPageVo> getTalkPage(Page<ApiMemberTalkPageVo> page, ApiMemberTalkPageDto dto) {
+ return aiMemberTalkMapper.getTalkPage(page, dto);
+ }
+
}
--
Gitblit v1.9.1