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/dal/mysql/model/AiModelMapper.java | 25 ++++++++++++++++++++++++-
1 files changed, 24 insertions(+), 1 deletions(-)
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiModelMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiModelMapper.java
index d6d6b4c..ffefbe6 100644
--- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiModelMapper.java
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiModelMapper.java
@@ -19,12 +19,27 @@
.last("LIMIT 1"));
}
+ /**
+ * 鏌ヨ鎸囧畾銆岀被鍨� + 鑳藉姏銆嶄笅宸插惎鐢ㄧ殑妯″瀷
+ */
+ default AiModelDO selectFirstByTypeAndCapabilityAndStatus(Integer type, String capability, Integer status) {
+ return selectOne(new LambdaQueryWrapperX<AiModelDO>()
+ .eq(AiModelDO::getType, type)
+ .eq(AiModelDO::getCapability, capability)
+ .eq(AiModelDO::getStatus, status)
+ .orderByAsc(AiModelDO::getSort)
+ .last("LIMIT 1"));
+ }
+
default PageResult<AiModelDO> selectPage(AiModelPageReqVO pageReqVO) {
return selectPage(pageReqVO, new LambdaQueryWrapperX<AiModelDO>()
.eqIfPresent(AiModelDO::getType, pageReqVO.getType())
+ .eqIfPresent(AiModelDO::getCapability, pageReqVO.getCapability())
.eqIfPresent(AiModelDO::getStatus, pageReqVO.getStatus())
.eqIfPresent(AiModelDO::getPlatform, pageReqVO.getPlatform())
.likeIfPresent(AiModelDO::getName, pageReqVO.getName())
+ .likeIfPresent(AiModelDO::getModel, pageReqVO.getModel())
+ .orderByAsc(AiModelDO::getType)
.orderByAsc(AiModelDO::getSort));
}
@@ -32,7 +47,15 @@
return selectList(new LambdaQueryWrapperX<AiModelDO>()
.eqIfPresent(AiModelDO::getStatus, status)
.eqIfPresent(AiModelDO::getType, type)
- .eqIfPresent(AiModelDO::getPlatform, platform));
+ .eqIfPresent(AiModelDO::getPlatform, platform)
+ .orderByAsc(AiModelDO::getSort));
+ }
+
+ default List<AiModelDO> selectListByTypeAndCapabilityAndStatus(Integer type, String capability, Integer status) {
+ return selectList(new LambdaQueryWrapperX<AiModelDO>()
+ .eq(AiModelDO::getType, type)
+ .eq(AiModelDO::getCapability, capability)
+ .eq(AiModelDO::getStatus, status));
}
}
--
Gitblit v1.9.3