From 266e0a86a79c5a7f2f190d00bbd829d9667cda21 Mon Sep 17 00:00:00 2001
From: otto <731554297@qq.com>
Date: Thu, 21 May 2026 15:31:37 +0800
Subject: [PATCH] =?UTF-8?q?feat(AI=E8=B5=84=E6=BA=90=E5=B9=BF=E5=9C=BA):?=
 =?UTF-8?q?=20=E6=A8=A1=E5=9E=8B=E5=92=8C=E6=95=B0=E6=8D=AE=E9=9B=86?=
 =?UTF-8?q?=E8=B5=84=E6=BA=90=E7=9B=B8=E5=85=B3=E5=8A=9F=E8=83=BD=E5=BC=80?=
 =?UTF-8?q?=E5=8F=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

AI资源判断增加模型和数据集
---
 .../microservices/zone/resource/enums/AiResourceCategory.java | 4 ++++
 .../zone/resource/service/impl/ZoneResourceServiceImpl.java   | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

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;