BJ_ZTEITS-Developers / parking-cloud-yudao

Commit c5dd1d50e7521914a92e5d61dd0ca52990300adf

Authored by cherishsince
1 parent d8c2d0d6

增加chat role 分类

Inline Side-by-side
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
  View file @c5dd1d5
  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 +}
... ...