From fc7ebc6bf97d7c4a4a8911c7a5d7cadcbda24719 Mon Sep 17 00:00:00 2001 From: houzhongjian <houzhongyi@126.com> Date: 星期四, 12 六月 2025 17:41:47 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- iailab-module-ai/iailab-module-ai-biz/src/main/java/com/iailab/module/ai/service/questiontemplate/QuestionTemplateServiceImpl.java | 30 ++++++++++++++++++++++++++++++ 1 files changed, 30 insertions(+), 0 deletions(-) diff --git a/iailab-module-ai/iailab-module-ai-biz/src/main/java/com/iailab/module/ai/service/questiontemplate/QuestionTemplateServiceImpl.java b/iailab-module-ai/iailab-module-ai-biz/src/main/java/com/iailab/module/ai/service/questiontemplate/QuestionTemplateServiceImpl.java index 9c70d05..1c1edcc 100644 --- a/iailab-module-ai/iailab-module-ai-biz/src/main/java/com/iailab/module/ai/service/questiontemplate/QuestionTemplateServiceImpl.java +++ b/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); } -- Gitblit v1.9.3