refactor(ai): 调整 AiTypeEnum 包名

Administrator

2025-08-19 df6ebc091fe184609c8ed5aad97e9fc82d185f35

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

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

import cc.mrbird.febs.ai.enumerates.AiTypeEnum;
import cc.mrbird.febs.ai.entity.AiMemberTalk;
import cc.mrbird.febs.ai.entity.AiMemberTalkItem;
import cc.mrbird.febs.ai.entity.AiProductRoleLink;
import cc.mrbird.febs.ai.mapper.AiMemberTalkMapper;
import cc.mrbird.febs.ai.req.ai.AiMessage;
import cc.mrbird.febs.ai.req.ai.AiRequest;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkAnswerDto;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkDto;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkItemPageDto;
import cc.mrbird.febs.ai.res.ai.AiResponse;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkVo;
import cc.mrbird.febs.ai.service.AiMemberTalkItemService;
@@ -19,14 +22,17 @@
import cc.mrbird.febs.common.utils.LoginUserUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.Date;
import java.util.List;
import java.util.function.Consumer;

/**
@@ -52,6 +58,7 @@
    }

    @Override
    @Transactional
    public FebsResponse start(ApiMemberTalkDto dto) {

        String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
@@ -75,7 +82,9 @@
            aiMemberTalk = this.add(memberUuid,productId,nowTime);
        }

        AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(),"<strong>\"生成题目\"</strong>");
        List<AiMessage> aiMessageDtoList = aiMemberTalkItemService.getQuestionUpDownContext(aiMemberTalk.getId(),AiTypeEnum.QUESTION.getCode());

        AiResponse aiResponse = aiService.start(aiMessageDtoList,AiTypeEnum.QUESTION.getCode(),aiProductRoleLink.getProductRoleId(),AiTypeEnum.QUESTION.getName(), null);
        if(aiResponse.getCode().equals("200")){
            aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),1,aiResponse.getResContext(),nowTime);
            this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
@@ -86,8 +95,6 @@
        apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId());
        apiMemberTalkVo.setType(1);
        apiMemberTalkVo.setContext(aiResponse.getResContext());
        apiMemberTalkVo.setReport(aiResponse.getReport());

        return new FebsResponse().success().data(apiMemberTalkVo);
    }

@@ -176,8 +183,9 @@
    }


    public static final String ANSWER_FORMAT = "{}/n[回答]{}/n";
    public static final String ANSWER_FORMAT = "###题目：{}###用户回答：{}";
    @Override
    @Transactional
    public FebsResponse answer(ApiMemberTalkAnswerDto dto) {
        String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
        String memberTalkId = dto.getId();
@@ -207,11 +215,19 @@

        String format = StrUtil.format(ANSWER_FORMAT, aiMemberTalkItem.getContext(), reqContext);
        log.info("format:{}",format);
        AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(), format);
//        AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(), reqContext);
//        AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(), format);


        List<AiMessage> aiMessageDtoList = aiMemberTalkItemService.getQuestionUpDownContext(aiMemberTalk.getId(),AiTypeEnum.ANSWER.getCode());
        AiResponse aiResponse = aiService.start(aiMessageDtoList,AiTypeEnum.ANSWER.getCode(),aiProductRoleLink.getProductRoleId(), reqContext,aiMemberTalkItem.getContext());
        String context = null;
        if(aiResponse.getCode().equals("200")){
            Date nowTime = new Date();
            aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),3,aiResponse.getResContext(),nowTime);
            context = String.valueOf(JSONUtil.parse(aiResponse.getReport()));
            if ("null".equals( context)){
                context = aiResponse.getResContext();
            }
            aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),3, context,nowTime);
            this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
        }else{
            throw new FebsException(aiResponse.getDescription());
@@ -219,8 +235,7 @@
        ApiMemberTalkVo apiMemberTalkVo = new ApiMemberTalkVo();
        apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId());
        apiMemberTalkVo.setType(3);
        apiMemberTalkVo.setContext(aiResponse.getResContext());
        apiMemberTalkVo.setReport(aiResponse.getReport());
        apiMemberTalkVo.setContext(context);
        return new FebsResponse().success().data(apiMemberTalkVo);
    }
    @Override
@@ -233,4 +248,12 @@
        aiMemberTalkMapper.insert(aiMemberTalk);
        return aiMemberTalk;
    }

    @Override
    public FebsResponse historyPage(ApiMemberTalkItemPageDto dto) {

        String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
        dto.setMemberUuid(memberUuid);
        return aiMemberTalkItemService.historyPage(dto);
    }
}