blame | history | patch | commit | commitdiff | ignore whitespace
Administrator
2025-09-02 3d762c09e6b25a87c8f4c5e026ed6de255b43377 
 src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java


@@ -2,12 +2,12 @@




import cc.mrbird.febs.ai.entity.AiProductQuestion;


import cc.mrbird.febs.ai.entity.AiTalk;


import cc.mrbird.febs.ai.enumerates.AiPromptEnum;


import cc.mrbird.febs.ai.mapper.AiTalkMapper;


import cc.mrbird.febs.ai.req.talk.AiTalkAnswerStream;


import cc.mrbird.febs.ai.req.talk.ApiTalkDto;


import cc.mrbird.febs.ai.req.talk.ApiTalkItemPageDto;


import cc.mrbird.febs.ai.req.talk.ApiTalkPageDto;


import cc.mrbird.febs.ai.res.memberAnswer.ApiMemberProductWorkVo;


import cc.mrbird.febs.ai.res.talk.ApiTalkPageVo;


import cc.mrbird.febs.ai.res.talk.ApiTalkQuestionVo;


import cc.mrbird.febs.ai.res.talk.ApiTalkVo;


@@ -15,6 +15,9 @@


import cc.mrbird.febs.ai.service.AiService;


import cc.mrbird.febs.ai.service.AiTalkItemService;


import cc.mrbird.febs.ai.service.AiTalkService;


import cc.mrbird.febs.ai.strategy.LlmStrategyFactory;


import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyEnum;


import cc.mrbird.febs.ai.strategy.param.LlmStrategyDto;


import cc.mrbird.febs.ai.utils.UUID;


import cc.mrbird.febs.common.entity.FebsResponse;


import cc.mrbird.febs.common.utils.LoginUserUtil;


@@ -22,6 +25,7 @@


import cn.hutool.core.date.DateTime;


import cn.hutool.core.date.DateUtil;


import cn.hutool.core.util.StrUtil;


import com.alibaba.dashscope.common.Role;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;


import com.baomidou.mybatisplus.core.toolkit.Wrappers;


import com.baomidou.mybatisplus.extension.plugins.pagination.Page;


@@ -43,6 +47,7 @@


    private final AiTalkItemService aiTalkItemService;


    private final AiProductQuestionService aiProductQuestionService;


    private final AiService aiService;


    private final LlmStrategyFactory llmStrategyFactory;




    @Override


    public FebsResponse questionList() {


@@ -121,4 +126,23 @@


        return aiService.answerStreamV2(dto);


    }




    @Override


    public Flux<FebsResponse> answerStreamV3(AiTalkAnswerStream dto) {


        ArrayList<LlmStrategyDto> llmStrategyDtoList = new ArrayList<>();


        if (dto.getPrompt() != null){


            LlmStrategyDto llmStrategyDto = new LlmStrategyDto();


            llmStrategyDto.setRole(Role.SYSTEM.getValue());


            llmStrategyDto.setContent(AiPromptEnum.STREAM_NORMAL.getPrompt());


            llmStrategyDtoList.add(llmStrategyDto);


        }


        if (dto.getQuestion() != null){


            LlmStrategyDto llmStrategyDto = new LlmStrategyDto();


            llmStrategyDto.setRole(Role.USER.getValue());


            llmStrategyDto.setContent(dto.getQuestion());


            llmStrategyDtoList.add(llmStrategyDto);


        }


        String modelName = LlmStrategyEnum.getName(aiService.getSystemSetAiType());


        return llmStrategyFactory.getCalculationStrategyMap().get(modelName).llmInvokeStreamingWithThink(llmStrategyDtoList);


    }




}