package cn.iocoder.yudao.module.ai.enums.model;

import cn.iocoder.yudao.framework.common.core.ArrayValuable;
import lombok.Getter;
import lombok.RequiredArgsConstructor;

import java.util.Arrays;

/**
 * AI 模型能力枚举
 *
 * 与 {@link AiModelTypeEnum} 组合使用：同一「类型 + 能力」下只允许一个模型处于启用状态。
 * 例如对话类型下，纯文本模型与视觉模型各可启用一个，互不覆盖。
 */
@Getter
@RequiredArgsConstructor
public enum AiModelCapabilityEnum implements ArrayValuable<String> {

    TEXT("text", "文本"),
    VISION("vision", "视觉");

    private final String capability;
    private final String name;

    public static final String[] ARRAYS = Arrays.stream(values()).map(AiModelCapabilityEnum::getCapability).toArray(String[]::new);

    @Override
    public String[] array() {
        return ARRAYS;
    }

    public static AiModelCapabilityEnum validateCapability(String capability) {
        for (AiModelCapabilityEnum e : values()) {
            if (e.getCapability().equals(capability)) return e;
        }
        throw new IllegalArgumentException("非法模型能力：" + capability);
    }

    public static String defaultCapability() {
        return TEXT.getCapability();
    }

    public static String getName(String capability) {
        for (AiModelCapabilityEnum e : values()) {
            if (e.getCapability().equals(capability)) return e.getName();
        }
        return capability;
    }

}