| | |
| | | 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; |
| | |
| | | import org.springframework.stereotype.Service; |
| | | import reactor.core.publisher.Flux; |
| | | |
| | | import java.util.ArrayList; |
| | | import java.util.Date; |
| | | import java.util.List; |
| | | |
| | | @Slf4j |
| | | @Service |
| | |
| | | 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(); |
| | |
| | | return new FebsResponse().success().data(pageListByQuery); |
| | | } |
| | | |
| | | @Override |
| | | public FebsResponse historyPage(ApiTalkItemPageDto dto) { |
| | | String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid(); |
| | | dto.setMemberUuid(memberUuid); |
| | | return aiTalkItemService.historyPage(dto); |
| | | } |
| | | |
| | | @Override |
| | | public Flux<FebsResponse> answerStreamV2(AiTalkAnswerStream dto) { |
| | | return aiService.answerStreamV2(dto); |
| | | } |
| | | |
| | | } |