From 95ae0c83d2aa58d86b4039e3f7f16ee93afb8f00 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期五, 23 十月 2026 17:01:00 +0800
Subject: [PATCH] feat(ai): 完善模型管理和智能体导入功能
---
yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/model/AiModelServiceImpl.java | 79 ++++++++++++++++++++++++++++++++++++---
1 files changed, 73 insertions(+), 6 deletions(-)
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/model/AiModelServiceImpl.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/model/AiModelServiceImpl.java
index 5668da0..4b34498 100644
--- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/model/AiModelServiceImpl.java
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/model/AiModelServiceImpl.java
@@ -5,10 +5,14 @@
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.ai.controller.admin.model.vo.model.AiModelPageReqVO;
import cn.iocoder.yudao.module.ai.controller.admin.model.vo.model.AiModelSaveReqVO;
+import cn.iocoder.yudao.module.ai.controller.admin.model.vo.model.AiModelUpdateStatusReqVO;
import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiApiKeyDO;
import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiModelDO;
import cn.iocoder.yudao.module.ai.dal.mysql.model.AiModelMapper;
+import cn.iocoder.yudao.module.ai.enums.model.AiModelCapabilityEnum;
+import cn.iocoder.yudao.module.ai.enums.model.AiModelTypeEnum;
import cn.iocoder.yudao.module.ai.enums.model.AiPlatformEnum;
+import cn.iocoder.yudao.module.ai.framework.ai.config.YudaoAiProperties;
import cn.iocoder.yudao.module.ai.framework.ai.core.model.AiModelFactory;
import jakarta.annotation.Resource;
import org.springframework.ai.chat.model.ChatModel;
@@ -16,6 +20,7 @@
import org.springframework.ai.vectorstore.VectorStore;
import org.springframework.ai.vectorstore.milvus.MilvusVectorStore;
import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.annotation.Validated;
import java.util.List;
import java.util.Map;
@@ -32,22 +37,38 @@
private AiModelMapper modelMapper;
@Resource
private AiModelFactory modelFactory;
+ @Resource
+ private YudaoAiProperties aiProperties;
@Override
+ @Transactional(rollbackFor = Exception.class)
public Long createModel(AiModelSaveReqVO createReqVO) {
AiPlatformEnum.validatePlatform(createReqVO.getPlatform());
apiKeyService.validateApiKey(createReqVO.getKeyId());
AiModelDO model = BeanUtils.toBean(createReqVO, AiModelDO.class);
+ model.setCapability(resolveCapability(createReqVO.getCapability()));
+ if (CommonStatusEnum.isEnable(model.getStatus())) {
+ disableOtherEnabledModels(model);
+ }
modelMapper.insert(model);
return model.getId();
}
@Override
+ @Transactional(rollbackFor = Exception.class)
public void updateModel(AiModelSaveReqVO updateReqVO) {
- validateModelExists(updateReqVO.getId());
+ AiModelDO existing = validateModelExists(updateReqVO.getId());
AiPlatformEnum.validatePlatform(updateReqVO.getPlatform());
apiKeyService.validateApiKey(updateReqVO.getKeyId());
- modelMapper.updateById(BeanUtils.toBean(updateReqVO, AiModelDO.class));
+ AiModelDO model = BeanUtils.toBean(updateReqVO, AiModelDO.class);
+ // 鏈紶鑳藉姏鏃舵部鐢ㄥ師鍊硷紝閬垮厤鑰佸墠绔彁浜ゆ妸瑙嗚妯″瀷闄嶇骇鎴愭枃鏈兘鍔�
+ model.setCapability(resolveCapability(
+ updateReqVO.getCapability() == null || updateReqVO.getCapability().isBlank()
+ ? existing.getCapability() : updateReqVO.getCapability()));
+ if (CommonStatusEnum.isEnable(model.getStatus())) {
+ disableOtherEnabledModels(model);
+ }
+ modelMapper.updateById(model);
}
@Override
@@ -63,8 +84,17 @@
@Override
public AiModelDO getRequiredDefaultModel(Integer type) {
- AiModelDO model = modelMapper.selectFirstByStatus(type, CommonStatusEnum.ENABLE.getStatus());
- if (model == null) throw exception(MODEL_DISABLE);
+ return getRequiredDefaultModel(type, AiModelCapabilityEnum.defaultCapability());
+ }
+
+ @Override
+ public AiModelDO getRequiredDefaultModel(Integer type, String capability) {
+ AiModelDO model = modelMapper.selectFirstByTypeAndCapabilityAndStatus(
+ type, resolveCapability(capability), CommonStatusEnum.ENABLE.getStatus());
+ if (model == null) {
+ throw exception(MODEL_NOT_ENABLED,
+ AiModelTypeEnum.getName(type), AiModelCapabilityEnum.getName(resolveCapability(capability)));
+ }
return model;
}
@@ -74,15 +104,49 @@
}
@Override
+ @Transactional(rollbackFor = Exception.class)
+ public void updateModelStatus(AiModelUpdateStatusReqVO updateReqVO) {
+ AiModelDO model = validateModelExists(updateReqVO.getId());
+ model.setStatus(updateReqVO.getStatus());
+ if (CommonStatusEnum.isEnable(updateReqVO.getStatus())) {
+ // 鍚敤锛氳嚜鍔ㄥ叧闂悓銆岀被鍨� + 鑳藉姏銆嶄笅鐨勫叾浠栨ā鍨嬶紝淇濊瘉鍚岀粍鍙湁涓�涓惎鐢�
+ disableOtherEnabledModels(model);
+ }
+ modelMapper.updateById(model);
+ }
+
+ @Override
public AiModelDO validateModel(Long id) {
AiModelDO model = validateModelExists(id);
- if (CommonStatusEnum.isDisable(model.getStatus())) throw exception(MODEL_DISABLE);
+ if (CommonStatusEnum.isDisable(model.getStatus())) throw exception(MODEL_DISABLE, model.getName());
return model;
}
@Override
public List<AiModelDO> getModelListByStatusAndType(Integer status, Integer type, String platform) {
return modelMapper.selectListByStatusAndType(status, type, platform);
+ }
+
+ /**
+ * 鍏抽棴鍚屻�岀被鍨� + 鑳藉姏銆嶄笅宸插惎鐢ㄧ殑鍏朵粬妯″瀷锛堜笉鍚嚜韬級
+ */
+ private void disableOtherEnabledModels(AiModelDO model) {
+ List<AiModelDO> sameGroup = modelMapper.selectListByTypeAndCapabilityAndStatus(
+ model.getType(), resolveCapability(model.getCapability()), CommonStatusEnum.ENABLE.getStatus());
+ for (AiModelDO other : sameGroup) {
+ if (other.getId().equals(model.getId())) {
+ continue;
+ }
+ AiModelDO update = new AiModelDO();
+ update.setId(other.getId());
+ update.setStatus(CommonStatusEnum.DISABLE.getStatus());
+ modelMapper.updateById(update);
+ }
+ }
+
+ private static String resolveCapability(String capability) {
+ return capability == null || capability.isBlank()
+ ? AiModelCapabilityEnum.defaultCapability() : capability;
}
private AiModelDO validateModelExists(Long id) {
@@ -98,7 +162,10 @@
AiModelDO model = validateModel(id);
AiApiKeyDO apiKey = apiKeyService.validateApiKey(model.getKeyId());
AiPlatformEnum platform = AiPlatformEnum.validatePlatform(apiKey.getPlatform());
- return modelFactory.getOrCreateChatModel(platform, apiKey.getApiKey(), apiKey.getUrl(), model.getModel());
+ // temperature / maxTokens 鏄� AI 妯″瀷琛ㄤ笂涓�鐩村瓨鍦ㄣ�佸嵈浠庢病浜鸿杩囩殑涓ゅ垪锛�
+ // 涓嶈 temperature锛屽簱閲岄厤鐨勫�煎璋冪敤鏂规槸鍋囩殑锛涗笉璇� maxTokens锛岄暱杈撳嚭浼氳闈欓粯鎴柇
+ return modelFactory.getOrCreateChatModel(platform, apiKey.getApiKey(), apiKey.getUrl(), model.getModel(),
+ model.getTemperature(), model.getMaxTokens(), aiProperties.getTimeout());
}
@Override
--
Gitblit v1.9.3