Commit c5dd1d50e7521914a92e5d61dd0ca52990300adf
1 parent
d8c2d0d6
增加chat role 分类
Showing
1 changed file
with
37 additions
and
0 deletions
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/ChatRoleClassifyEnum.java
0 → 100644
| 1 | +package cn.iocoder.yudao.module.ai.enums; | ||
| 2 | + | ||
| 3 | +import lombok.AllArgsConstructor; | ||
| 4 | +import lombok.Getter; | ||
| 5 | + | ||
| 6 | +/** | ||
| 7 | + * 聊天role 分类 | ||
| 8 | + * | ||
| 9 | + * @author fansili | ||
| 10 | + * @time 2024/4/24 16:41 | ||
| 11 | + * @since 1.0 | ||
| 12 | + */ | ||
| 13 | +@AllArgsConstructor | ||
| 14 | +@Getter | ||
| 15 | +public enum ChatRoleClassifyEnum { | ||
| 16 | + | ||
| 17 | + WRITING("writing", "写作"), | ||
| 18 | + | ||
| 19 | + ENTERTAINMENT("entertainment", "娱乐"), | ||
| 20 | + | ||
| 21 | + ; | ||
| 22 | + | ||
| 23 | + | ||
| 24 | + private String classify; | ||
| 25 | + | ||
| 26 | + private String name; | ||
| 27 | + | ||
| 28 | + | ||
| 29 | + public static ChatRoleClassifyEnum valueOfClassify(String classify) { | ||
| 30 | + for (ChatRoleClassifyEnum itemEnum : ChatRoleClassifyEnum.values()) { | ||
| 31 | + if (itemEnum.getClassify().equals(classify)) { | ||
| 32 | + return itemEnum; | ||
| 33 | + } | ||
| 34 | + } | ||
| 35 | + throw new IllegalArgumentException("Invalid MessageType value: " + classify); | ||
| 36 | + } | ||
| 37 | +} |