feat(ai): 优化 AI 陪练问答流程

Administrator

2025-08-18 a46c4d2db30c2f534400a6179cd82f7beb07a29d

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

@@ -1,15 +1,27 @@
package cc.mrbird.febs.ai.service.impl;

import cc.mrbird.febs.ai.controller.enumerates.AiTypeEnum;
import cc.mrbird.febs.ai.entity.AiMemberTalk;
import cc.mrbird.febs.ai.entity.AiMemberTalkItem;
import cc.mrbird.febs.ai.mapper.AiMemberTalkItemMapper;
import cc.mrbird.febs.ai.req.ai.AiMessage;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkItemPageDto;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkItemVo;
import cc.mrbird.febs.ai.service.AiMemberTalkItemService;
import cc.mrbird.febs.ai.utils.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
import cn.hutool.core.collection.CollUtil;
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 com.volcengine.ark.runtime.model.completion.chat.ChatMessageRole;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
@@ -26,88 +38,62 @@
    private final AiMemberTalkItemMapper aiMemberTalkItemMapper;

    @Override
    public AiMemberTalkItem getById(String id) {
        return this.getById(id);
    public void add(String memberUuid, String id, int type, String resContext,Date createdTime) {
        AiMemberTalkItem aiMemberTalkItem = new AiMemberTalkItem();
        aiMemberTalkItem.setId(UUID.getSimpleUUIDString());
        aiMemberTalkItem.setCreatedTime(createdTime);
        aiMemberTalkItem.setMemberId(memberUuid);
        aiMemberTalkItem.setMemberTalkId(id);
        aiMemberTalkItem.setType(type);
        aiMemberTalkItem.setContext(resContext);
        aiMemberTalkItemMapper.insert(aiMemberTalkItem);
    }

    @Override
    public List<AiMemberTalkItem> getByMemberTalkId(String memberTalkId) {
        LambdaQueryWrapper<AiMemberTalkItem> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(AiMemberTalkItem::getMemberTalkId, memberTalkId);
        queryWrapper.orderByAsc(AiMemberTalkItem::getCreatedTime);
        return this.list(queryWrapper);
    public AiMemberTalkItem getByQuery(LambdaQueryWrapper<AiMemberTalkItem> memberTalkItemQuery) {
        return aiMemberTalkItemMapper.selectOne(memberTalkItemQuery);
    }

    @Override
    public List<AiMemberTalkItem> getByMemberId(String memberId) {
        LambdaQueryWrapper<AiMemberTalkItem> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(AiMemberTalkItem::getMemberId, memberId);
        queryWrapper.orderByAsc(AiMemberTalkItem::getCreatedTime);
        return this.list(queryWrapper);
    public FebsResponse historyPage(ApiMemberTalkItemPageDto dto) {
        // 创建分页对象，传入当前页和每页大小
        Page<ApiMemberTalkItemVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
        Page<ApiMemberTalkItemVo> pageListByQuery = aiMemberTalkItemMapper.getPageListByQuery(page, dto);
        return new FebsResponse().success().data(pageListByQuery);
    }

    @Override
    public List<AiMemberTalkItem> getByType(Integer type) {
        LambdaQueryWrapper<AiMemberTalkItem> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(AiMemberTalkItem::getType, type);
        queryWrapper.orderByAsc(AiMemberTalkItem::getCreatedTime);
        return this.list(queryWrapper);
    public List<AiMessage> getQuestionUpDownContext(String memberTalkId, int code) {
        List<AiMessage> aiMessages = new ArrayList<>();

        LambdaQueryWrapper<AiMemberTalkItem> query = Wrappers.lambdaQuery(AiMemberTalkItem.class);
            if (AiTypeEnum.QUESTION.getCode() == code){
                query.eq(AiMemberTalkItem::getType,AiTypeEnum.QUESTION_ANSWER.getCode());
                query.eq(AiMemberTalkItem::getMemberTalkId,memberTalkId);
                query.orderByDesc(AiMemberTalkItem::getCreatedTime);
                query.last("limit 1");
                List<AiMemberTalkItem> aiMemberTalkItems = aiMemberTalkItemMapper.selectList(query);
                if (CollUtil.isNotEmpty(aiMemberTalkItems)){
                    AiMessage assistantMessage = new AiMessage();
                    assistantMessage.setRole(ChatMessageRole.ASSISTANT);
                    assistantMessage.setContent(aiMemberTalkItems.get(0).getContext());
                    aiMessages.add(assistantMessage);
                }
            }
            if (AiTypeEnum.ANSWER.getCode() == code){
                query.eq(AiMemberTalkItem::getType,AiTypeEnum.ANSWER_ANALYSIS.getCode());
                query.eq(AiMemberTalkItem::getMemberTalkId,memberTalkId);
                query.orderByDesc(AiMemberTalkItem::getCreatedTime);
                query.last("limit 1");
                List<AiMemberTalkItem> aiMemberTalkItems = aiMemberTalkItemMapper.selectList(query);
                if (CollUtil.isNotEmpty(aiMemberTalkItems)){
                    AiMessage assistantMessage = new AiMessage();
                    assistantMessage.setRole(ChatMessageRole.ASSISTANT);
                    assistantMessage.setContent(aiMemberTalkItems.get(0).getContext());
                    aiMessages.add(assistantMessage);
                }
            }
        return aiMessages;
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean saveTalkItem(AiMemberTalkItem aiMemberTalkItem) {
        try {
            return this.save(aiMemberTalkItem);
        } catch (Exception e) {
            log.error("保存对话项失败: ", e);
            return false;
        }
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean saveBatchTalkItems(List<AiMemberTalkItem> talkItems) {
        try {
            return this.saveBatch(talkItems);
        } catch (Exception e) {
            log.error("批量保存对话项失败: ", e);
            return false;
        }
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean updateTalkItem(AiMemberTalkItem aiMemberTalkItem) {
        try {
            return this.updateById(aiMemberTalkItem);
        } catch (Exception e) {
            log.error("更新对话项失败: ", e);
            return false;
        }
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean deleteById(String id) {
        try {
            return this.removeById(id);
        } catch (Exception e) {
            log.error("删除对话项失败: ", e);
            return false;
        }
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean deleteByMemberTalkId(String memberTalkId) {
        try {
            LambdaQueryWrapper<AiMemberTalkItem> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.eq(AiMemberTalkItem::getMemberTalkId, memberTalkId);
            return this.remove(queryWrapper);
        } catch (Exception e) {
            log.error("根据对话ID删除对话项失败: ", e);
            return false;
        }
    }
}