【调整】调整AI角色模块

This commit is contained in:
cherishsince
2024-05-07 11:34:20 +08:00
parent c124681c3a
commit f0a1666e84
10 changed files with 142 additions and 128 deletions

View File

@ -12,7 +12,7 @@ import lombok.Getter;
*/
@AllArgsConstructor
@Getter
public enum AiChatRoleClassifyEnum {
public enum AiChatRoleCategoryEnum {
WRITING("writing", "写作"),
@ -21,17 +21,17 @@ public enum AiChatRoleClassifyEnum {
;
private String classify;
private String category;
private String name;
public static AiChatRoleClassifyEnum valueOfClassify(String classify) {
for (AiChatRoleClassifyEnum itemEnum : AiChatRoleClassifyEnum.values()) {
if (itemEnum.getClassify().equals(classify)) {
public static AiChatRoleCategoryEnum valueOfCategory(String category) {
for (AiChatRoleCategoryEnum itemEnum : AiChatRoleCategoryEnum.values()) {
if (itemEnum.getCategory().equals(category)) {
return itemEnum;
}
}
throw new IllegalArgumentException("Invalid MessageType value: " + classify);
throw new IllegalArgumentException("Invalid MessageType value: " + category);
}
}

View File

@ -1,36 +0,0 @@
package cn.iocoder.yudao.module.ai.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
/**
* chat角色 可见范围
*
* @author fansili
* @time 2024/4/24 16:44
* @since 1.0
*/
@AllArgsConstructor
@Getter
public enum AiChatRoleEnableEnum {
OPEN("open", "公开"),
CLOSE("close", "关闭"),
;
private String type;
private String name;
public static AiChatRoleEnableEnum valueOfType(String type) {
for (AiChatRoleEnableEnum itemEnum : AiChatRoleEnableEnum.values()) {
if (itemEnum.getType().equals(type)) {
return itemEnum;
}
}
throw new IllegalArgumentException("Invalid MessageType value: " + type);
}
}