package cn.iocoder.yudao.module.ai.enums.model;
|
|
import cn.iocoder.yudao.framework.common.core.ArrayValuable;
|
import lombok.AllArgsConstructor;
|
import lombok.Getter;
|
import java.util.Arrays;
|
|
/**
|
* AI 模型平台
|
*/
|
@Getter
|
@AllArgsConstructor
|
public enum AiPlatformEnum implements ArrayValuable<String> {
|
|
TONG_YI("TongYi", "通义千问"),
|
DEEP_SEEK("DeepSeek", "DeepSeek"),
|
OPENAI("OpenAI", "OpenAI"),
|
OLLAMA("Ollama", "Ollama");
|
|
private final String platform;
|
private final String name;
|
|
public static final String[] ARRAYS = Arrays.stream(values()).map(AiPlatformEnum::getPlatform).toArray(String[]::new);
|
|
public static AiPlatformEnum validatePlatform(String platform) {
|
for (AiPlatformEnum e : values()) {
|
if (e.getPlatform().equals(platform)) return e;
|
}
|
throw new IllegalArgumentException("非法平台：" + platform);
|
}
|
|
@Override
|
public String[] array() { return ARRAYS; }
|
|
}
|
