From a01ef141b18dd249df9adc93612501d782c466a7 Mon Sep 17 00:00:00 2001
From: houzhongjian <houzhongyi@126.com>
Date: 星期四, 12 六月 2025 11:29:15 +0800
Subject: [PATCH] ai服务代码更新
---
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