const SUPPORTED_MODELS_GROUPS = {
    openai: [
        // "o1-preview",
        // "o1-mini",
        // "gpt-4o",
        // "gpt-3.5-turbo"
    ],
    groq: [
        // "mixtral-8x7b-32768",
        // "deepseek-r1-distill-llama-70b",
        "meta-llama/llama-4-scout-17b-16e-instruct",
        "gemma2-9b-it",
        "mistral-saba-24b",
        // "qwen-2.5-32b",
        "llama-3.3-70b-versatile",
        // "llama-3.3-70b-versatile"
        // "llama-3.1-70b-versatile",
        // "llama-3.3-70b-versatile"
    ],
    cerebras: ["llama-3.3-70b"],
    claude: [
        // "claude-3-5-sonnet-20241022",
        // "claude-3-opus-20240229"
    ],
    fireworks: [
        // "llama-v3p1-405b-instruct",
        // "llama-v3p1-70b-instruct",
        // "llama-v3p2-90b-vision-instruct",
        // "mixtral-8x22b-instruct",
        // "mythomax-l2-13b",
        // "yi-large"
    ],
    google: [
        // "gemini-2.0-flash-exp",
        // "gemini-1.5-flash",
        // "gemini-exp-1206",
        // "gemini-1.5-pro"
    ],
    xai: [
        // "grok-beta",
        // "grok-2",
        // "grok-2-1212",
        // "grok-2-latest",
        // "grok-beta"
    ],
    cloudflareAI: [
        "llama-3.2-3b-instruct", // max_tokens
        "llama-3-8b-instruct", // max_tokens
        "llama-3.1-8b-instruct-fast", // max_tokens
        "deepseek-math-7b-instruct",
        "deepseek-coder-6.7b-instruct-awq",
        "hermes-2-pro-mistral-7b",
        "openhermes-2.5-mistral-7b-awq",
        "mistral-7b-instruct-v0.2",
        "neural-chat-7b-v3-1-awq",
        "openchat-3.5-0106",
        // "gemma-7b-it",
    ],
};

export type SupportedModel =
    | keyof typeof SUPPORTED_MODELS_GROUPS
    | (typeof SUPPORTED_MODELS_GROUPS)[keyof typeof SUPPORTED_MODELS_GROUPS][number];

export type ModelFamily = keyof typeof SUPPORTED_MODELS_GROUPS;

function getModelFamily(model: string): ModelFamily | undefined {
    return Object.keys(SUPPORTED_MODELS_GROUPS)
        .filter((family) => {
            return SUPPORTED_MODELS_GROUPS[
                family as keyof typeof SUPPORTED_MODELS_GROUPS
                ].includes(model.trim());
        })
        .at(0) as ModelFamily | undefined;
}

const SUPPORTED_MODELS = [
    // ...SUPPORTED_MODELS_GROUPS.xai,
    // ...SUPPORTED_MODELS_GROUPS.claude,
    // ...SUPPORTED_MODELS_GROUPS.google,
    ...SUPPORTED_MODELS_GROUPS.groq,
    // ...SUPPORTED_MODELS_GROUPS.fireworks,
    // ...SUPPORTED_MODELS_GROUPS.openai,
    // ...SUPPORTED_MODELS_GROUPS.cerebras,
    // ...SUPPORTED_MODELS_GROUPS.cloudflareAI,
];

export { SUPPORTED_MODELS, SUPPORTED_MODELS_GROUPS, getModelFamily };