云
12 小时以前 95ae0c83d2aa58d86b4039e3f7f16ee93afb8f00
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));
    }
}