enum 增加 valueOfType

cherishsince
1 parent e4dc7154
Showing 2 changed files with 21 additions and 0 deletions
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/ChatRoleSourceEnum.java
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/ChatRoleVisibilityEnum.java
@@ -23,4 +23,14 @@ public enum ChatRoleSourceEnum {
     private String type;
     private String name;
+
+
+    public static ChatRoleSourceEnum valueOfType(String type) {
+        for (ChatRoleSourceEnum itemEnum : ChatRoleSourceEnum.values()) {
+            if (itemEnum.getType().equals(type)) {
+                return itemEnum;
+            }
+        }
+        throw new IllegalArgumentException("Invalid MessageType value: " + type);
+    }
 }
@@ -22,4 +22,15 @@ public enum ChatRoleVisibilityEnum {
     private String type;
     private String name;
+
+
+    public static ChatRoleVisibilityEnum valueOfType(String type) {
+        for (ChatRoleVisibilityEnum itemEnum : ChatRoleVisibilityEnum.values()) {
+            if (itemEnum.getType().equals(type)) {
+                return itemEnum;
+            }
+        }
+        throw new IllegalArgumentException("Invalid MessageType value: " + type);
+    }
+
 }