Administrator
2 days ago fa14d18506141aab48ed716767b298de495f36b6
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java
@@ -74,7 +74,7 @@
        }
        AiProduct aiProduct = aiProductService.getById(productId);
        Integer questionCount = aiProduct.getQuestionCount();
        Integer talkCnt = aiProduct.getTalkCnt();
        String companyId = aiProductRoleLink.getCompanyId();
@@ -85,19 +85,32 @@
        query.eq(AiMemberTalk::getState,AiTypeEnum.AI_MEMBER_TALK_STATE_ING.getCode());
        query.orderByDesc(AiMemberTalk::getCreatedTime);
        query.last("limit 1");
        AiMemberTalk aiMemberTalk = aiMemberTalkService.getByQuery(query);
        if (ObjectUtil.isNull(aiMemberTalk)){
            aiMemberTalk = aiMemberTalkService.add(memberUuid,companyId,productId,nowTime,questionCount);
        }
        ApiMemberTalkStreamVo apiMemberTalkVo = new ApiMemberTalkStreamVo();
        AiMemberTalk aiMemberTalk = aiMemberTalkService.getByQuery(query);
        String title = "加载时间过程,请重试!";
        AiProductQuestion question = aiProductQuestionService.createQuestion(productId);
        if (ObjectUtil.isNotNull(question)){
            title = question.getTitle();
            AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 1, title, nowTime);
            apiMemberTalkVo.setMemberTalkItemId(aiMemberTalkItem.getId());
            aiMemberTalkService.updateTimeUpdate(nowTime,aiMemberTalk.getId());
        if (ObjectUtil.isNull(aiMemberTalk)){
            aiMemberTalk = aiMemberTalkService.add(memberUuid,companyId,productId,nowTime,talkCnt);
            AiProductQuestion question = aiProductQuestionService.createQuestion(productId);
            if (ObjectUtil.isNotNull(question)){
                title = question.getTitle();
                AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 1, title, nowTime);
                apiMemberTalkVo.setMemberTalkItemId(aiMemberTalkItem.getId());
                aiMemberTalkService.updateTimeUpdate(nowTime,aiMemberTalk.getId());
            }
        }else{
            AiMemberTalkItem oneByQuery = aiMemberTalkItemService.getOneByQuery(
                    Wrappers.lambdaQuery(AiMemberTalkItem.class)
                            .eq(AiMemberTalkItem::getMemberTalkId, aiMemberTalk.getId())
                            .eq(AiMemberTalkItem::getType, 1)
                            .orderByDesc(AiMemberTalkItem::getCreatedTime)
                            .last("limit 1")
            );
            if (ObjectUtil.isNotNull(oneByQuery)){
                title = oneByQuery.getContext();
                apiMemberTalkVo.setMemberTalkItemId(oneByQuery.getId());
            }
        }
        apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId());
        apiMemberTalkVo.setType(1);
@@ -347,7 +360,7 @@
        AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.getOneByQuery(memberTalkItemQuery);
        String question = aiMemberTalkItem.getContext();
        String promptHead = aiProductRole.getPromptHead();
        String promptHead = aiProductRole.getModelId();
        String answer = dto.getReqContext();
        List<LlmStrategyDto> llmStrategyDtoList = new ArrayList<>();