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/controller/admin/questiontemplate/QuestionTemplateController.java | 25 +++++++++++++++++++++++++
1 files changed, 25 insertions(+), 0 deletions(-)
diff --git a/iailab-module-ai/iailab-module-ai-biz/src/main/java/com/iailab/module/ai/controller/admin/questiontemplate/QuestionTemplateController.java b/iailab-module-ai/iailab-module-ai-biz/src/main/java/com/iailab/module/ai/controller/admin/questiontemplate/QuestionTemplateController.java
index d673385..4ecda38 100644
--- a/iailab-module-ai/iailab-module-ai-biz/src/main/java/com/iailab/module/ai/controller/admin/questiontemplate/QuestionTemplateController.java
+++ b/iailab-module-ai/iailab-module-ai-biz/src/main/java/com/iailab/module/ai/controller/admin/questiontemplate/QuestionTemplateController.java
@@ -1,5 +1,8 @@
package com.iailab.module.ai.controller.admin.questiontemplate;
+import com.iailab.module.ai.api.questionTemplate.dto.AiModelDTO;
+import com.iailab.module.ai.api.questionTemplate.dto.QuestionTemplateDTO;
+import com.iailab.module.ai.service.model.AiModelService;
import jakarta.annotation.Resource;
import org.springframework.web.bind.annotation.*;
import org.springframework.validation.annotation.Validated;
@@ -13,12 +16,15 @@
import com.iailab.framework.common.pojo.PageResult;
import com.iailab.framework.common.pojo.CommonResult;
import com.iailab.framework.common.util.object.BeanUtils;
+
import static com.iailab.framework.common.pojo.CommonResult.success;
import com.iailab.module.ai.controller.admin.questiontemplate.vo.*;
import com.iailab.module.ai.dal.dataobject.questiontemplate.QuestionTemplateDO;
import com.iailab.module.ai.service.questiontemplate.QuestionTemplateService;
+
+import java.util.List;
@Tag(name = "管理后台 - 大模型问题模板")
@RestController
@@ -28,6 +34,9 @@
@Resource
private QuestionTemplateService questionTemplateService;
+
+ @Resource
+ private AiModelService aiModelService;
@PostMapping("/create")
@Operation(summary = "创建大模型问题模板")
@@ -59,6 +68,7 @@
@PreAuthorize("@ss.hasPermission('ai:question-template:query')")
public CommonResult<QuestionTemplateRespVO> getQuestionTemplate(@RequestParam("id") String id) {
QuestionTemplateDO questionTemplate = questionTemplateService.getQuestionTemplate(id);
+ questionTemplate.setModelName( aiModelService.getModel(Long.valueOf((questionTemplate.getModelId()))).getName());
return success(BeanUtils.toBean(questionTemplate, QuestionTemplateRespVO.class));
}
@@ -67,7 +77,22 @@
@PreAuthorize("@ss.hasPermission('ai:question-template:query')")
public CommonResult<PageResult<QuestionTemplateRespVO>> getQuestionTemplatePage(@Valid QuestionTemplatePageReqVO pageReqVO) {
PageResult<QuestionTemplateDO> pageResult = questionTemplateService.getQuestionTemplatePage(pageReqVO);
+ pageResult.getList().forEach(questionTemplateDO ->
+ questionTemplateDO.setModelName(
+ aiModelService.getModel(Long.valueOf((questionTemplateDO.getModelId()))).getName()
+ )
+ );
return success(BeanUtils.toBean(pageResult, QuestionTemplateRespVO.class));
}
+ @GetMapping( "/modelList")
+ @Operation(summary = "Ai大模型级联问题模版")
+ List<AiModelDTO> queryModelList(){
+ List<AiModelDTO> list = BeanUtils.toBean(aiModelService.getModelListByStatusAndType(0, 1, null), AiModelDTO.class);
+ list.forEach(item->{
+ item.setChildren(
+ BeanUtils.toBean(questionTemplateService.getQuestionTemplateList(item.getId()), QuestionTemplateDTO.class));
+ });
+ return list;
+ }
}
\ No newline at end of file
--
Gitblit v1.9.3