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

Administrator

23 hours ago 0325d413502474062e1d400df319bfd390e94956

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

@@ -80,7 +80,7 @@
        query.last("limit 1");
        AiMemberTalk aiMemberTalk = this.getByQuery(query);
        if (ObjectUtil.isNull(aiMemberTalk)){
            aiMemberTalk = this.add(memberUuid,companyId,productId,nowTime);
            aiMemberTalk = this.add(memberUuid,companyId,productId,nowTime,5);
        }

        List<AiMessage> aiMessageDtoList = aiMemberTalkItemService.getQuestionUpDownContext(aiMemberTalk.getId(),AiTypeEnum.QUESTION.getCode());
@@ -112,7 +112,7 @@
        queryWrapper.eq(AiMemberTalkItem::getType,AiTypeEnum.QUESTION_ANSWER.getCode());
        queryWrapper.orderByDesc(AiMemberTalkItem::getCreatedTime);
        queryWrapper.last("limit 1");
        AiMemberTalkItem byQuery = aiMemberTalkItemService.getByQuery(queryWrapper);
        AiMemberTalkItem byQuery = aiMemberTalkItemService.getOneByQuery(queryWrapper);
        if (ObjectUtil.isNotNull(byQuery)){
            apiMemberTalkReloadVo.setContext(byQuery.getContext());
            apiMemberTalkReloadVo.setMemberTalkId(memberTalkId);
@@ -155,7 +155,7 @@
                query.last("limit 1");
                AiMemberTalk aiMemberTalk = this.getByQuery(query);
                if (ObjectUtil.isNull(aiMemberTalk)) {
                    aiMemberTalk = this.add(memberUuid,companyId, productId, nowTime);
                    aiMemberTalk = this.add(memberUuid,companyId, productId, nowTime,5);
                }
                try {
                    if (aiResponse.getCode().equals("200")) {
@@ -236,7 +236,7 @@
        memberTalkItemQuery.eq(AiMemberTalkItem::getType,1);
        memberTalkItemQuery.orderByDesc(AiMemberTalkItem::getCreatedTime);
        memberTalkItemQuery.last("limit 1");
        AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.getByQuery(memberTalkItemQuery);
        AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.getOneByQuery(memberTalkItemQuery);
        aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),companyId,2,reqContext,new Date());

        String format = StrUtil.format(ANSWER_FORMAT, aiMemberTalkItem.getContext(), reqContext);
@@ -268,13 +268,16 @@
        return new FebsResponse().success().data(apiMemberTalkVo);
    }
    @Override
    public AiMemberTalk add(String memberUuid,String companyId, 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;
    }
@@ -293,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())
        );
    }

}