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; }

}