feat(ai): 新增 AI 陪练相关功能

Administrator

23 hours ago 0325d413502474062e1d400df319bfd390e94956

 src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java

@@ -7,11 +7,9 @@
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.memberTalk.ApiMemberTalkAnswerDto;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkDto;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkItemPageDto;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkStreamDto;
import cc.mrbird.febs.ai.req.memberTalk.*;
import cc.mrbird.febs.ai.res.ai.AiResponse;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkReloadVo;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkVo;
import cc.mrbird.febs.ai.service.AiMemberTalkItemService;
import cc.mrbird.febs.ai.service.AiMemberTalkService;
@@ -73,6 +71,7 @@
        if(ObjectUtil.isNull(aiProductRoleLink)){
            throw new FebsException("产品AI陪练不存在");
        }
        String companyId = aiProductRoleLink.getCompanyId();

        Date nowTime = new Date();
        LambdaQueryWrapper<AiMemberTalk> query = Wrappers.lambdaQuery(AiMemberTalk.class);
@@ -81,23 +80,45 @@
        query.last("limit 1");
        AiMemberTalk aiMemberTalk = this.getByQuery(query);
        if (ObjectUtil.isNull(aiMemberTalk)){
            aiMemberTalk = this.add(memberUuid,productId,nowTime);
            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);
        if(aiResponse.getCode().equals("200")){
            aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),1,aiResponse.getResContext(),nowTime);
            this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
        }else{
            throw new FebsException(aiResponse.getDescription());
        }

        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 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
@@ -117,6 +138,7 @@
                callback.accept(aiResponse);
                return;
            }
            String companyId = aiProductRoleLink.getCompanyId();

            // 构造AI请求
            AiRequest aiRequest = new AiRequest();
@@ -133,14 +155,14 @@
                query.last("limit 1");
                AiMemberTalk aiMemberTalk = this.getByQuery(query);
                if (ObjectUtil.isNull(aiMemberTalk)) {
                    aiMemberTalk = this.add(memberUuid, productId, nowTime);
                    aiMemberTalk = this.add(memberUuid,companyId, productId, nowTime,5);
                }
                try {
                    if (aiResponse.getCode().equals("200")) {
                        // 如果是最终结果（包含报告）
                        if (aiResponse.getReport() != null) {
                            // 保存完整响应到数据库
                            aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(), 1, aiResponse.getResContext(), nowTime);
                            aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 1, aiResponse.getResContext(), nowTime);
                            this.updateTimeUpdate(nowTime, aiMemberTalk.getId());

                            callback.accept(aiResponse);
@@ -196,6 +218,8 @@
            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");
@@ -212,14 +236,17 @@
        memberTalkItemQuery.eq(AiMemberTalkItem::getType,1);
        memberTalkItemQuery.orderByDesc(AiMemberTalkItem::getCreatedTime);
        memberTalkItemQuery.last("limit 1");
        AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.getByQuery(memberTalkItemQuery);
        aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),2,reqContext,new Date());
        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;
@@ -229,34 +256,37 @@
            if ("null".equals( context)){
                context = aiResponse.getResContext();
            }
            aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),3, context,nowTime);
            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());
        }
        ApiMemberTalkVo apiMemberTalkVo = new ApiMemberTalkVo();
        apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId());
        apiMemberTalkVo.setType(3);
        apiMemberTalkVo.setContext(context);
        return new FebsResponse().success().data(apiMemberTalkVo);
    }
    @Override
    public AiMemberTalk add(String memberUuid, String productId, Date nowTime) {
    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
    public FebsResponse historyPage(ApiMemberTalkItemPageDto dto) {

        String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
        dto.setMemberUuid(memberUuid);
        return aiMemberTalkItemService.historyPage(dto);
        return new FebsResponse().success().data(aiMemberTalkItemService.historyPage(dto));
    }

    @Override
@@ -266,4 +296,23 @@
        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())
        );
    }

}