From 2ef897d378675be4816266e5e7d466564880920b Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Wed, 27 Aug 2025 15:32:52 +0800 Subject: [PATCH] feat(ai): 添加再练一次(重新回答)功能 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java | 33 +++++++++++++++++++++++++++++++++ 1 files changed, 33 insertions(+), 0 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java index 539dbac..2cabe56 100644 --- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java @@ -1,22 +1,29 @@ package cc.mrbird.febs.ai.service.impl; +import cc.mrbird.febs.ai.entity.AiProductQuestion; import cc.mrbird.febs.ai.entity.AiTalk; 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; +import cc.mrbird.febs.ai.service.AiProductQuestionService; 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.utils.UUID; import cc.mrbird.febs.common.entity.FebsResponse; import cc.mrbird.febs.common.utils.LoginUserUtil; +import cn.hutool.core.collection.CollUtil; import cn.hutool.core.date.DateTime; import cn.hutool.core.date.DateUtil; import cn.hutool.core.util.StrUtil; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import lombok.RequiredArgsConstructor; @@ -24,7 +31,9 @@ import org.springframework.stereotype.Service; import reactor.core.publisher.Flux; +import java.util.ArrayList; import java.util.Date; +import java.util.List; @Slf4j @Service @@ -32,7 +41,26 @@ public class AiTalkServiceImpl extends ServiceImpl<AiTalkMapper, AiTalk> implements AiTalkService { private final AiTalkItemService aiTalkItemService; + private final AiProductQuestionService aiProductQuestionService; private final AiService aiService; + + @Override + public FebsResponse questionList() { + List<ApiTalkQuestionVo> list = new ArrayList<>(); + LambdaQueryWrapper<AiProductQuestion> queryWrapper = Wrappers.lambdaQuery(AiProductQuestion.class); + queryWrapper.orderByDesc(AiProductQuestion::getCreatedTime); + queryWrapper.last("limit 30"); + List<AiProductQuestion> listByQuery = aiProductQuestionService.getListByQuery(queryWrapper); + if (CollUtil.isNotEmpty(listByQuery)){ + for (AiProductQuestion aiProductQuestion : listByQuery){ + ApiTalkQuestionVo apiTalkQuestionVo = new ApiTalkQuestionVo(); + apiTalkQuestionVo.setContext(aiProductQuestion.getTitle()); + list.add(apiTalkQuestionVo); + } + } + return new FebsResponse().success().data(list); + } + @Override public FebsResponse talkOpen(ApiTalkDto dto) { String talkId = dto.getTalkId(); @@ -88,4 +116,9 @@ return aiTalkItemService.historyPage(dto); } + @Override + public Flux<FebsResponse> answerStreamV2(AiTalkAnswerStream dto) { + return aiService.answerStreamV2(dto); + } + } -- Gitblit v1.9.1