Quellcode durchsuchen

enum 增加 valueOfType

cherishsince vor 1 Jahr
Ursprung
Commit
c75e415b61

+ 10 - 0
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/ChatRoleSourceEnum.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);
+    }
 }

+ 11 - 0
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/ChatRoleVisibilityEnum.java

@@ -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);
+    }
+
 }