diff --git a/microservices-modules/microservices-modules-zone/src/main/java/com/microservices/zone/resource/enums/AiResourceCategory.java b/microservices-modules/microservices-modules-zone/src/main/java/com/microservices/zone/resource/enums/AiResourceCategory.java index f8fa256b2..38696b4fa 100644 --- a/microservices-modules/microservices-modules-zone/src/main/java/com/microservices/zone/resource/enums/AiResourceCategory.java +++ b/microservices-modules/microservices-modules-zone/src/main/java/com/microservices/zone/resource/enums/AiResourceCategory.java @@ -24,4 +24,8 @@ public enum AiResourceCategory { public String getDesc() { return desc; } + + public static boolean isAiResourceCategory(String code) { + return SKILL.getCode().equals(code) || MODEL.getCode().equals(code) || DATASET.getCode().equals(code); + } } diff --git a/microservices-modules/microservices-modules-zone/src/main/java/com/microservices/zone/resource/service/impl/ZoneResourceServiceImpl.java b/microservices-modules/microservices-modules-zone/src/main/java/com/microservices/zone/resource/service/impl/ZoneResourceServiceImpl.java index 36bad06f5..801dc5439 100644 --- a/microservices-modules/microservices-modules-zone/src/main/java/com/microservices/zone/resource/service/impl/ZoneResourceServiceImpl.java +++ b/microservices-modules/microservices-modules-zone/src/main/java/com/microservices/zone/resource/service/impl/ZoneResourceServiceImpl.java @@ -607,7 +607,7 @@ public class ZoneResourceServiceImpl implements IZoneResourceService { if (resource == null) { throw new ServiceException("资源不存在(资源id[" + id + "])"); } - if (!AiResourceCategory.SKILL.getCode().equals(resource.getResourceCategory())) { + if (!AiResourceCategory.isAiResourceCategory(resource.getResourceCategory())) { throw new ServiceException("该资源不是AI资源"); } return resource;