@poolzin/pool-bot 2026.2.5 → 2026.2.7

package/dist/commands/onboard-non-interactive/local/auth-choice.js

@@ -2,13 +2,15 @@ import { upsertAuthProfile } from "../../../agents/auth-profiles.js";
 import { normalizeProviderId } from "../../../agents/model-selection.js";
 import { parseDurationMs } from "../../../cli/parse-duration.js";
 import { upsertSharedEnvVar } from "../../../infra/env-file.js";
+import { shortenHomePath } from "../../../utils.js";
 import { normalizeSecretInput } from "../../../utils/normalize-secret-input.js";
 import { buildTokenProfileId, validateAnthropicSetupToken } from "../../auth-token.js";
 import { applyGoogleGeminiModelDefault } from "../../google-gemini-model-default.js";
+import { applyAuthProfileConfig, applyCloudflareAiGatewayConfig, applyHuggingfaceConfig, applyKimiCodeConfig, applyLitellmConfig, applyMinimaxApiConfig, applyMinimaxConfig, applyMoonshotConfig, applyMoonshotConfigCn, applyNvidiaConfig, applyOpencodeZenConfig, applyOpenrouterConfig, applyQianfanConfig, applySyntheticConfig, applyTogetherConfig, applyVeniceConfig, applyVercelAiGatewayConfig, applyXaiConfig, applyXiaomiConfig, applyZaiConfig, setAnthropicApiKey, setCloudflareAiGatewayConfig, setGeminiApiKey, setHuggingfaceApiKey, setKimiCodingApiKey, setLitellmApiKey, setMinimaxApiKey, setMoonshotApiKey, setNvidiaApiKey, setOpencodeZenApiKey, setOpenrouterApiKey, setQianfanApiKey, setSyntheticApiKey, setTogetherApiKey, setVeniceApiKey, setVercelAiGatewayApiKey, setXaiApiKey, setXiaomiApiKey, setZaiApiKey, } from "../../onboard-auth.js";
+import { applyCustomApiConfig, CustomApiError, parseNonInteractiveCustomApiFlags, resolveCustomProviderId, } from "../../onboard-custom.js";
 import { applyOpenAIConfig } from "../../openai-model-default.js";
-import { applyAuthProfileConfig, applyCloudflareAiGatewayConfig, applyKimiCodeConfig, applyMinimaxApiConfig, applyMinimaxConfig, applyMoonshotConfig, applyMoonshotConfigCn, applyOpencodeZenConfig, applyOpenrouterConfig, applyQianfanConfig, applySyntheticConfig, applyTogetherConfig, applyVeniceConfig, applyVercelAiGatewayConfig, applyNvidiaConfig, applyXaiConfig, applyXiaomiConfig, applyZaiConfig, setAnthropicApiKey, setCloudflareAiGatewayConfig, setGeminiApiKey, setKimiCodingApiKey, setMinimaxApiKey, setMoonshotApiKey, setNvidiaApiKey, setOpencodeZenApiKey, setOpenrouterApiKey, setQianfanApiKey, setSyntheticApiKey, setTogetherApiKey, setVeniceApiKey, setVercelAiGatewayApiKey, setXaiApiKey, setXiaomiApiKey, setZaiApiKey, } from "../../onboard-auth.js";
+import { detectZaiEndpoint } from "../../zai-endpoint-detect.js";
 import { resolveNonInteractiveApiKey } from "../api-keys.js";
-import { shortenHomePath } from "../../../utils.js";
 export async function applyNonInteractiveAuthChoice(params) {
     const { authChoice, opts, runtime, baseConfig } = params;
     let nextConfig = params.nextConfig;
@@ -28,6 +30,14 @@ export async function applyNonInteractiveAuthChoice(params) {
         runtime.exit(1);
         return null;
     }
+    if (authChoice === "vllm") {
+        runtime.error([
+            'Auth choice "vllm" requires interactive mode.',
+            "Use interactive onboard/configure to enter base URL, API key, and model ID.",
+        ].join("\n"));
+        runtime.exit(1);
+        return null;
+    }
     if (authChoice === "apiKey") {
         const resolved = await resolveNonInteractiveApiKey({
             provider: "anthropic",
@@ -37,10 +47,12 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "ANTHROPIC_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setAnthropicApiKey(resolved.key);
+        }
         return applyAuthProfileConfig(nextConfig, {
             profileId: "anthropic:default",
             provider: "anthropic",
@@ -109,10 +121,12 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "GEMINI_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setGeminiApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
             profileId: "google:default",
             provider: "google",
@@ -120,7 +134,11 @@ export async function applyNonInteractiveAuthChoice(params) {
         });
         return applyGoogleGeminiModelDefault(nextConfig).next;
     }
-    if (authChoice === "zai-api-key") {
+    if (authChoice === "zai-api-key" ||
+        authChoice === "zai-coding-global" ||
+        authChoice === "zai-coding-cn" ||
+        authChoice === "zai-global" ||
+        authChoice === "zai-cn") {
         const resolved = await resolveNonInteractiveApiKey({
             provider: "zai",
             cfg: baseConfig,
@@ -129,16 +147,112 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "ZAI_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setZaiApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
             profileId: "zai:default",
             provider: "zai",
             mode: "api_key",
         });
-        return applyZaiConfig(nextConfig);
+        // Determine endpoint from authChoice or detect from the API key.
+        let endpoint;
+        let modelIdOverride;
+        if (authChoice === "zai-coding-global") {
+            endpoint = "coding-global";
+        }
+        else if (authChoice === "zai-coding-cn") {
+            endpoint = "coding-cn";
+        }
+        else if (authChoice === "zai-global") {
+            endpoint = "global";
+        }
+        else if (authChoice === "zai-cn") {
+            endpoint = "cn";
+        }
+        else {
+            const detected = await detectZaiEndpoint({ apiKey: resolved.key });
+            if (detected) {
+                endpoint = detected.endpoint;
+                modelIdOverride = detected.modelId;
+            }
+            else {
+                endpoint = "global";
+            }
+        }
+        return applyZaiConfig(nextConfig, {
+            endpoint,
+            ...(modelIdOverride ? { modelId: modelIdOverride } : {}),
+        });
+    }
+    if (authChoice === "xiaomi-api-key") {
+        const resolved = await resolveNonInteractiveApiKey({
+            provider: "xiaomi",
+            cfg: baseConfig,
+            flagValue: opts.xiaomiApiKey,
+            flagName: "--xiaomi-api-key",
+            envVar: "XIAOMI_API_KEY",
+            runtime,
+        });
+        if (!resolved) {
+            return null;
+        }
+        if (resolved.source !== "profile") {
+            await setXiaomiApiKey(resolved.key);
+        }
+        nextConfig = applyAuthProfileConfig(nextConfig, {
+            profileId: "xiaomi:default",
+            provider: "xiaomi",
+            mode: "api_key",
+        });
+        return applyXiaomiConfig(nextConfig);
+    }
+    if (authChoice === "xai-api-key") {
+        const resolved = await resolveNonInteractiveApiKey({
+            provider: "xai",
+            cfg: baseConfig,
+            flagValue: opts.xaiApiKey,
+            flagName: "--xai-api-key",
+            envVar: "XAI_API_KEY",
+            runtime,
+        });
+        if (!resolved) {
+            return null;
+        }
+        if (resolved.source !== "profile") {
+            setXaiApiKey(resolved.key);
+        }
+        nextConfig = applyAuthProfileConfig(nextConfig, {
+            profileId: "xai:default",
+            provider: "xai",
+            mode: "api_key",
+        });
+        return applyXaiConfig(nextConfig);
+    }
+    if (authChoice === "qianfan-api-key") {
+        const resolved = await resolveNonInteractiveApiKey({
+            provider: "qianfan",
+            cfg: baseConfig,
+            flagValue: opts.qianfanApiKey,
+            flagName: "--qianfan-api-key",
+            envVar: "QIANFAN_API_KEY",
+            runtime,
+        });
+        if (!resolved) {
+            return null;
+        }
+        if (resolved.source !== "profile") {
+            setQianfanApiKey(resolved.key);
+        }
+        nextConfig = applyAuthProfileConfig(nextConfig, {
+            profileId: "qianfan:default",
+            provider: "qianfan",
+            mode: "api_key",
+        });
+        return applyQianfanConfig(nextConfig);
     }
     if (authChoice === "openai-api-key") {
         const resolved = await resolveNonInteractiveApiKey({
@@ -150,8 +264,9 @@ export async function applyNonInteractiveAuthChoice(params) {
             runtime,
             allowProfile: false,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
+        }
         const key = resolved.key;
         const result = upsertSharedEnvVar({ key: "OPENAI_API_KEY", value: key });
         process.env.OPENAI_API_KEY = key;
@@ -167,10 +282,12 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "OPENROUTER_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setOpenrouterApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
             profileId: "openrouter:default",
             provider: "openrouter",
@@ -178,6 +295,28 @@ export async function applyNonInteractiveAuthChoice(params) {
         });
         return applyOpenrouterConfig(nextConfig);
     }
+    if (authChoice === "litellm-api-key") {
+        const resolved = await resolveNonInteractiveApiKey({
+            provider: "litellm",
+            cfg: baseConfig,
+            flagValue: opts.litellmApiKey,
+            flagName: "--litellm-api-key",
+            envVar: "LITELLM_API_KEY",
+            runtime,
+        });
+        if (!resolved) {
+            return null;
+        }
+        if (resolved.source !== "profile") {
+            await setLitellmApiKey(resolved.key);
+        }
+        nextConfig = applyAuthProfileConfig(nextConfig, {
+            profileId: "litellm:default",
+            provider: "litellm",
+            mode: "api_key",
+        });
+        return applyLitellmConfig(nextConfig);
+    }
     if (authChoice === "ai-gateway-api-key") {
         const resolved = await resolveNonInteractiveApiKey({
             provider: "vercel-ai-gateway",
@@ -187,10 +326,12 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "AI_GATEWAY_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setVercelAiGatewayApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
             profileId: "vercel-ai-gateway:default",
             provider: "vercel-ai-gateway",
@@ -198,6 +339,41 @@ export async function applyNonInteractiveAuthChoice(params) {
         });
         return applyVercelAiGatewayConfig(nextConfig);
     }
+    if (authChoice === "cloudflare-ai-gateway-api-key") {
+        const accountId = opts.cloudflareAiGatewayAccountId?.trim() ?? "";
+        const gatewayId = opts.cloudflareAiGatewayGatewayId?.trim() ?? "";
+        if (!accountId || !gatewayId) {
+            runtime.error([
+                'Auth choice "cloudflare-ai-gateway-api-key" requires Account ID and Gateway ID.',
+                "Use --cloudflare-ai-gateway-account-id and --cloudflare-ai-gateway-gateway-id.",
+            ].join("\n"));
+            runtime.exit(1);
+            return null;
+        }
+        const resolved = await resolveNonInteractiveApiKey({
+            provider: "cloudflare-ai-gateway",
+            cfg: baseConfig,
+            flagValue: opts.cloudflareAiGatewayApiKey,
+            flagName: "--cloudflare-ai-gateway-api-key",
+            envVar: "CLOUDFLARE_AI_GATEWAY_API_KEY",
+            runtime,
+        });
+        if (!resolved) {
+            return null;
+        }
+        if (resolved.source !== "profile") {
+            await setCloudflareAiGatewayConfig(accountId, gatewayId, resolved.key);
+        }
+        nextConfig = applyAuthProfileConfig(nextConfig, {
+            profileId: "cloudflare-ai-gateway:default",
+            provider: "cloudflare-ai-gateway",
+            mode: "api_key",
+        });
+        return applyCloudflareAiGatewayConfig(nextConfig, {
+            accountId,
+            gatewayId,
+        });
+    }
     if (authChoice === "moonshot-api-key") {
         const resolved = await resolveNonInteractiveApiKey({
             provider: "moonshot",
@@ -207,10 +383,12 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "MOONSHOT_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setMoonshotApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
             profileId: "moonshot:default",
             provider: "moonshot",
@@ -227,10 +405,12 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "MOONSHOT_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setMoonshotApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
             profileId: "moonshot:default",
             provider: "moonshot",
@@ -247,10 +427,12 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "KIMI_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setKimiCodingApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
             profileId: "kimi-coding:default",
             provider: "kimi-coding",
@@ -267,10 +449,12 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "SYNTHETIC_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setSyntheticApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
             profileId: "synthetic:default",
             provider: "synthetic",
@@ -287,10 +471,12 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "VENICE_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setVeniceApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
             profileId: "venice:default",
             provider: "venice",
@@ -298,119 +484,55 @@ export async function applyNonInteractiveAuthChoice(params) {
         });
         return applyVeniceConfig(nextConfig);
     }
-    if (authChoice === "xiaomi-api-key") {
-        const resolved = await resolveNonInteractiveApiKey({
-            provider: "xiaomi",
-            cfg: baseConfig,
-            flagValue: opts.xiaomiApiKey,
-            flagName: "--xiaomi-api-key",
-            envVar: "XIAOMI_API_KEY",
-            runtime,
-        });
-        if (!resolved)
-            return null;
-        if (resolved.source !== "profile")
-            await setXiaomiApiKey(resolved.key);
-        nextConfig = applyAuthProfileConfig(nextConfig, {
-            profileId: "xiaomi:default",
-            provider: "xiaomi",
-            mode: "api_key",
-        });
-        return applyXiaomiConfig(nextConfig);
-    }
-    if (authChoice === "xai-api-key") {
+    if (authChoice === "minimax-cloud" ||
+        authChoice === "minimax-api" ||
+        authChoice === "minimax-api-lightning") {
         const resolved = await resolveNonInteractiveApiKey({
-            provider: "xai",
+            provider: "minimax",
             cfg: baseConfig,
-            flagValue: opts.xaiApiKey,
-            flagName: "--xai-api-key",
-            envVar: "XAI_API_KEY",
+            flagValue: opts.minimaxApiKey,
+            flagName: "--minimax-api-key",
+            envVar: "MINIMAX_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
-            setXaiApiKey(resolved.key);
+        }
+        if (resolved.source !== "profile") {
+            await setMinimaxApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
-            profileId: "xai:default",
-            provider: "xai",
+            profileId: "minimax:default",
+            provider: "minimax",
             mode: "api_key",
         });
-        return applyXaiConfig(nextConfig);
+        const modelId = authChoice === "minimax-api-lightning" ? "MiniMax-M2.1-lightning" : "MiniMax-M2.1";
+        return applyMinimaxApiConfig(nextConfig, modelId);
     }
-    if (authChoice === "qianfan-api-key") {
-        const resolved = await resolveNonInteractiveApiKey({
-            provider: "qianfan",
-            cfg: baseConfig,
-            flagValue: opts.qianfanApiKey,
-            flagName: "--qianfan-api-key",
-            envVar: "QIANFAN_API_KEY",
-            runtime,
-        });
-        if (!resolved)
-            return null;
-        if (resolved.source !== "profile")
-            setQianfanApiKey(resolved.key);
-        nextConfig = applyAuthProfileConfig(nextConfig, {
-            profileId: "qianfan:default",
-            provider: "qianfan",
-            mode: "api_key",
-        });
-        return applyQianfanConfig(nextConfig);
+    if (authChoice === "minimax") {
+        return applyMinimaxConfig(nextConfig);
     }
-    if (authChoice === "nvidia-api-key") {
+    if (authChoice === "opencode-zen") {
         const resolved = await resolveNonInteractiveApiKey({
-            provider: "nvidia",
+            provider: "opencode",
             cfg: baseConfig,
-            flagValue: opts.nvidiaApiKey,
-            flagName: "--nvidia-api-key",
-            envVar: "NVIDIA_API_KEY",
+            flagValue: opts.opencodeZenApiKey,
+            flagName: "--opencode-zen-api-key",
+            envVar: "OPENCODE_API_KEY (or OPENCODE_ZEN_API_KEY)",
             runtime,
         });
-        if (!resolved)
-            return null;
-        if (resolved.source !== "profile")
-            setNvidiaApiKey(resolved.key);
-        nextConfig = applyAuthProfileConfig(nextConfig, {
-            profileId: "nvidia:default",
-            provider: "nvidia",
-            mode: "api_key",
-        });
-        return applyNvidiaConfig(nextConfig);
-    }
-    if (authChoice === "cloudflare-ai-gateway-api-key") {
-        const accountId = opts.cloudflareAiGatewayAccountId?.trim() ?? "";
-        const gatewayId = opts.cloudflareAiGatewayGatewayId?.trim() ?? "";
-        if (!accountId || !gatewayId) {
-            runtime.error([
-                'Auth choice "cloudflare-ai-gateway-api-key" requires Account ID and Gateway ID.',
-                "Use --cloudflare-ai-gateway-account-id and --cloudflare-ai-gateway-gateway-id.",
-            ].join("\n"));
-            runtime.exit(1);
+        if (!resolved) {
             return null;
         }
-        const resolved = await resolveNonInteractiveApiKey({
-            provider: "cloudflare-ai-gateway",
-            cfg: baseConfig,
-            flagValue: opts.cloudflareAiGatewayApiKey,
-            flagName: "--cloudflare-ai-gateway-api-key",
-            envVar: "CLOUDFLARE_AI_GATEWAY_API_KEY",
-            runtime,
-        });
-        if (!resolved)
-            return null;
         if (resolved.source !== "profile") {
-            await setCloudflareAiGatewayConfig(accountId, gatewayId, resolved.key);
+            await setOpencodeZenApiKey(resolved.key);
         }
         nextConfig = applyAuthProfileConfig(nextConfig, {
-            profileId: "cloudflare-ai-gateway:default",
-            provider: "cloudflare-ai-gateway",
+            profileId: "opencode:default",
+            provider: "opencode",
             mode: "api_key",
         });
-        return applyCloudflareAiGatewayConfig(nextConfig, {
-            accountId,
-            gatewayId,
-        });
+        return applyOpencodeZenConfig(nextConfig);
     }
     if (authChoice === "together-api-key") {
         const resolved = await resolveNonInteractiveApiKey({
@@ -421,10 +543,12 @@ export async function applyNonInteractiveAuthChoice(params) {
             envVar: "TOGETHER_API_KEY",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
+        }
+        if (resolved.source !== "profile") {
             await setTogetherApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
             profileId: "together:default",
             provider: "together",
@@ -432,50 +556,104 @@ export async function applyNonInteractiveAuthChoice(params) {
         });
         return applyTogetherConfig(nextConfig);
     }
-    if (authChoice === "minimax-cloud" ||
-        authChoice === "minimax-api" ||
-        authChoice === "minimax-api-lightning") {
+    if (authChoice === "huggingface-api-key") {
         const resolved = await resolveNonInteractiveApiKey({
-            provider: "minimax",
+            provider: "huggingface",
             cfg: baseConfig,
-            flagValue: opts.minimaxApiKey,
-            flagName: "--minimax-api-key",
-            envVar: "MINIMAX_API_KEY",
+            flagValue: opts.huggingfaceApiKey,
+            flagName: "--huggingface-api-key",
+            envVar: "HF_TOKEN",
             runtime,
         });
-        if (!resolved)
+        if (!resolved) {
             return null;
-        if (resolved.source !== "profile")
-            await setMinimaxApiKey(resolved.key);
+        }
+        if (resolved.source !== "profile") {
+            await setHuggingfaceApiKey(resolved.key);
+        }
         nextConfig = applyAuthProfileConfig(nextConfig, {
-            profileId: "minimax:default",
-            provider: "minimax",
+            profileId: "huggingface:default",
+            provider: "huggingface",
             mode: "api_key",
         });
-        const modelId = authChoice === "minimax-api-lightning" ? "MiniMax-M2.1-lightning" : "MiniMax-M2.1";
-        return applyMinimaxApiConfig(nextConfig, modelId);
+        return applyHuggingfaceConfig(nextConfig);
     }
-    if (authChoice === "minimax")
-        return applyMinimaxConfig(nextConfig);
-    if (authChoice === "opencode-zen") {
+    if (authChoice === "nvidia-api-key") {
         const resolved = await resolveNonInteractiveApiKey({
-            provider: "opencode",
+            provider: "nvidia",
             cfg: baseConfig,
-            flagValue: opts.opencodeZenApiKey,
-            flagName: "--opencode-zen-api-key",
-            envVar: "OPENCODE_API_KEY (or OPENCODE_ZEN_API_KEY)",
+            flagValue: opts.nvidiaApiKey,
+            flagName: "--nvidia-api-key",
+            envVar: "NVIDIA_API_KEY",
             runtime,
         });
         if (!resolved)
             return null;
         if (resolved.source !== "profile")
-            await setOpencodeZenApiKey(resolved.key);
+            setNvidiaApiKey(resolved.key);
         nextConfig = applyAuthProfileConfig(nextConfig, {
-            profileId: "opencode:default",
-            provider: "opencode",
+            profileId: "nvidia:default",
+            provider: "nvidia",
             mode: "api_key",
         });
-        return applyOpencodeZenConfig(nextConfig);
+        return applyNvidiaConfig(nextConfig);
+    }
+    if (authChoice === "custom-api-key") {
+        try {
+            const customAuth = parseNonInteractiveCustomApiFlags({
+                baseUrl: opts.customBaseUrl,
+                modelId: opts.customModelId,
+                compatibility: opts.customCompatibility,
+                apiKey: opts.customApiKey,
+                providerId: opts.customProviderId,
+            });
+            const resolvedProviderId = resolveCustomProviderId({
+                config: nextConfig,
+                baseUrl: customAuth.baseUrl,
+                providerId: customAuth.providerId,
+            });
+            const resolvedCustomApiKey = await resolveNonInteractiveApiKey({
+                provider: resolvedProviderId.providerId,
+                cfg: baseConfig,
+                flagValue: customAuth.apiKey,
+                flagName: "--custom-api-key",
+                envVar: "CUSTOM_API_KEY",
+                envVarName: "CUSTOM_API_KEY",
+                runtime,
+                required: false,
+            });
+            const result = applyCustomApiConfig({
+                config: nextConfig,
+                baseUrl: customAuth.baseUrl,
+                modelId: customAuth.modelId,
+                compatibility: customAuth.compatibility,
+                apiKey: resolvedCustomApiKey?.key,
+                providerId: customAuth.providerId,
+            });
+            if (result.providerIdRenamedFrom && result.providerId) {
+                runtime.log(`Custom provider ID "${result.providerIdRenamedFrom}" already exists for a different base URL. Using "${result.providerId}".`);
+            }
+            return result.config;
+        }
+        catch (err) {
+            if (err instanceof CustomApiError) {
+                switch (err.code) {
+                    case "missing_required":
+                    case "invalid_compatibility":
+                        runtime.error(err.message);
+                        break;
+                    default:
+                        runtime.error(`Invalid custom provider config: ${err.message}`);
+                        break;
+                }
+                runtime.exit(1);
+                return null;
+            }
+            const reason = err instanceof Error ? err.message : String(err);
+            runtime.error(`Invalid custom provider config: ${reason}`);
+            runtime.exit(1);
+            return null;
+        }
     }
     if (authChoice === "oauth" ||
         authChoice === "chutes" ||