iailab-module-ai/iailab-module-ai-biz/src/main/java/com/iailab/module/ai/service/questiontemplate/QuestionTemplateServiceImpl.java

@@ -1,9 +1,15 @@
package com.iailab.module.ai.service.questiontemplate;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.iailab.framework.mybatis.core.query.LambdaQueryWrapperX;
import com.iailab.module.ai.controller.admin.questionparamsetting.vo.QuestionParamSettingRespVO;
import com.iailab.module.ai.dal.dataobject.model.AiModelDO;
import com.iailab.module.ai.dal.dataobject.questionparamsetting.QuestionParamSettingDO;
import com.iailab.module.ai.dal.mysql.questionparamsetting.QuestionParamSettingMapper;
import com.iailab.module.ai.service.model.AiModelService;
import jakarta.annotation.Resource;
import org.apache.commons.lang3.ObjectUtils;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;

@@ -34,6 +40,9 @@

    @Resource
    private QuestionParamSettingMapper questionParamSettingMapper;

    @Resource
    private AiModelService aiModelService;

    @Override
    public String createQuestionTemplate(QuestionTemplateSaveReqVO createReqVO) {
@@ -92,6 +101,27 @@
    }

    @Override
    public List<QuestionTemplateDO> getQuestionTemplates(QuestionTemplateReqVO reqVO) {
        String modelName = reqVO.getModelName();
        if(ObjectUtils.isNotEmpty(modelName)) {
            AiModelDO model = aiModelService.getModelByName(modelName);
            List<QuestionTemplateDO> questionTemplateDOS = questionTemplateMapper.selectList(
                    new LambdaQueryWrapperX<QuestionTemplateDO>()
                            .eqIfPresent(QuestionTemplateDO::getModelId, model.getId())
                            .eqIfPresent(QuestionTemplateDO::getQuestionCode, reqVO.getQuestionCode()));
            // 模板数量暂时很少，先循环遍历查询模板设置
            if (questionTemplateDOS != null && questionTemplateDOS.size() > 0) {
                questionTemplateDOS.stream().forEach(questionTemplateDO -> {
                    questionTemplateDO.setSettingList(
                            BeanUtils.toBean(questionParamSettingMapper.selectList("template_id", questionTemplateDO.getId()), QuestionParamSettingRespVO.class));
                });
            }
            return questionTemplateDOS;
        }
        return null;
    }

    @Override
    public PageResult<QuestionTemplateDO> getQuestionTemplatePage(QuestionTemplatePageReqVO pageReqVO) {
        return questionTemplateMapper.selectPage(pageReqVO);
    }