feat(ai): 增加 AI 陪练报告数据解析功能 - 新增 Report、RadarData 和 Evaluation 类用于解析报告数据 - 在...

Administrator

2025-08-11 0be80a07e38f08dd40e20175d788a3bfad8e68ef

 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;
@@ -26,6 +27,7 @@
import org.springframework.stereotype.Service;

import java.util.Date;
import java.util.function.Consumer;

/**
 * AI用户对话训练记录 Service实现类
@@ -90,6 +92,76 @@
    }

    @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
    public AiMemberTalk getByQuery(LambdaQueryWrapper<AiMemberTalk> query) {
        return aiMemberTalkMapper.selectOne( query);
    }