@tomsun28/pizza 0.0.9 → 0.1.1

package/dist/core/model-resolver.d.ts

@@ -0,0 +1,109 @@
+/**
+ * Model resolution, scoping, and initial selection
+ */
+import type { ThinkingLevel } from "./agent/types.js";
+import { type Api, type Model } from "@earendil-works/pi-ai/compat";
+import type { ModelRegistry } from "./model-registry.js";
+/** Default model IDs for each known provider */
+export declare const defaultModelPerProvider: Record<string, string>;
+export interface ScopedModel {
+    model: Model<Api>;
+    /** Thinking level if explicitly specified in pattern (e.g., "model:high"), undefined otherwise */
+    thinkingLevel?: ThinkingLevel;
+}
+/**
+ * Find an exact model reference match.
+ * Supports either a bare model id or a canonical provider/modelId reference.
+ * When matching by bare id, ambiguous matches across providers are rejected.
+ */
+export declare function findExactModelReferenceMatch(modelReference: string, availableModels: Model<Api>[]): Model<Api> | undefined;
+export interface ParsedModelResult {
+    model: Model<Api> | undefined;
+    /** Thinking level if explicitly specified in pattern, undefined otherwise */
+    thinkingLevel?: ThinkingLevel;
+    warning: string | undefined;
+}
+/**
+ * Parse a pattern to extract model and thinking level.
+ * Handles models with colons in their IDs (e.g., OpenRouter's :exacto suffix).
+ *
+ * Algorithm:
+ * 1. Try to match full pattern as a model
+ * 2. If found, return it with "off" thinking level
+ * 3. If not found and has colons, split on last colon:
+ *    - If suffix is valid thinking level, use it and recurse on prefix
+ *    - If suffix is invalid, warn and recurse on prefix with "off"
+ *
+ * @internal Exported for testing
+ */
+export declare function parseModelPattern(pattern: string, availableModels: Model<Api>[], options?: {
+    allowInvalidThinkingLevelFallback?: boolean;
+}): ParsedModelResult;
+/**
+ * Resolve model patterns to actual Model objects with optional thinking levels
+ * Format: "pattern:level" where :level is optional
+ * For each pattern, finds all matching models and picks the best version:
+ * 1. Prefer alias (e.g., claude-sonnet-4-5) over dated versions (claude-sonnet-4-5-20250929)
+ * 2. If no alias, pick the latest dated version
+ *
+ * Supports models with colons in their IDs (e.g., OpenRouter's model:exacto).
+ * The algorithm tries to match the full pattern first, then progressively
+ * strips colon-suffixes to find a match.
+ */
+export declare function resolveModelScope(patterns: string[], modelRegistry: ModelRegistry): Promise<ScopedModel[]>;
+export interface ResolveCliModelResult {
+    model: Model<Api> | undefined;
+    thinkingLevel?: ThinkingLevel;
+    warning: string | undefined;
+    /**
+     * Error message suitable for CLI display.
+     * When set, model will be undefined.
+     */
+    error: string | undefined;
+}
+/**
+ * Resolve a single model from CLI flags.
+ *
+ * Supports:
+ * - --provider <provider> --model <pattern>
+ * - --model <provider>/<pattern>
+ * - Fuzzy matching (same rules as model scoping: exact id, then partial id/name)
+ *
+ * Note: This does not apply the thinking level by itself, but it may *parse* and
+ * return a thinking level from "<pattern>:<thinking>" so the caller can apply it.
+ */
+export declare function resolveCliModel(options: {
+    cliProvider?: string;
+    cliModel?: string;
+    modelRegistry: ModelRegistry;
+}): ResolveCliModelResult;
+export interface InitialModelResult {
+    model: Model<Api> | undefined;
+    thinkingLevel: ThinkingLevel;
+    fallbackMessage: string | undefined;
+}
+/**
+ * Find the initial model to use based on priority:
+ * 1. CLI args (provider + model)
+ * 2. First model from scoped models (if not continuing/resuming)
+ * 3. Restored from session (if continuing/resuming)
+ * 4. Saved default from settings
+ * 5. First available model with valid API key
+ */
+export declare function findInitialModel(options: {
+    cliProvider?: string;
+    cliModel?: string;
+    scopedModels: ScopedModel[];
+    isContinuing: boolean;
+    defaultProvider?: string;
+    defaultModelId?: string;
+    defaultThinkingLevel?: ThinkingLevel;
+    modelRegistry: ModelRegistry;
+}): Promise<InitialModelResult>;
+/**
+ * Restore model from session, with fallback to available models
+ */
+export declare function restoreModelFromSession(savedProvider: string, savedModelId: string, currentModel: Model<Api> | undefined, shouldPrintMessages: boolean, modelRegistry: ModelRegistry): Promise<{
+    model: Model<Api> | undefined;
+    fallbackMessage: string | undefined;
+}>;

package/dist/core/model-resolver.js

@@ -0,0 +1,486 @@
+/**
+ * Model resolution, scoping, and initial selection
+ */
+import { modelsAreEqual } from "@earendil-works/pi-ai/compat";
+import chalk from "chalk";
+import { minimatch } from "minimatch";
+import { isValidThinkingLevel } from "../cli/args.js";
+import { DEFAULT_THINKING_LEVEL } from "./defaults.js";
+/** Default model IDs for each known provider */
+export const defaultModelPerProvider = {
+    "amazon-bedrock": "us.anthropic.claude-opus-4-6-v1",
+    anthropic: "claude-opus-4-6",
+    openai: "gpt-5.4",
+    "azure-openai-responses": "gpt-5.2",
+    "openai-codex": "gpt-5.4",
+    google: "gemini-2.5-pro",
+    "google-gemini-cli": "gemini-2.5-pro",
+    "google-antigravity": "gemini-3.1-pro-high",
+    "google-vertex": "gemini-3-pro-preview",
+    "github-copilot": "gpt-4o",
+    openrouter: "openai/gpt-5.1-codex",
+    "vercel-ai-gateway": "anthropic/claude-opus-4-6",
+    xai: "grok-4-fast-non-reasoning",
+    groq: "openai/gpt-oss-120b",
+    cerebras: "zai-glm-4.7",
+    zai: "glm-5",
+    mistral: "devstral-medium-latest",
+    minimax: "MiniMax-M2.7",
+    "minimax-cn": "MiniMax-M2.7",
+    huggingface: "moonshotai/Kimi-K2.5",
+    fireworks: "accounts/fireworks/models/kimi-k2p6",
+    opencode: "claude-opus-4-6",
+    "opencode-go": "kimi-k2.5",
+    "kimi-coding": "kimi-for-coding",
+};
+/**
+ * Helper to check if a model ID looks like an alias (no date suffix)
+ * Dates are typically in format: -20241022 or -20250929
+ */
+function isAlias(id) {
+    // Check if ID ends with -latest
+    if (id.endsWith("-latest"))
+        return true;
+    // Check if ID ends with a date pattern (-YYYYMMDD)
+    const datePattern = /-\d{8}$/;
+    return !datePattern.test(id);
+}
+/**
+ * Find an exact model reference match.
+ * Supports either a bare model id or a canonical provider/modelId reference.
+ * When matching by bare id, ambiguous matches across providers are rejected.
+ */
+export function findExactModelReferenceMatch(modelReference, availableModels) {
+    const trimmedReference = modelReference.trim();
+    if (!trimmedReference) {
+        return undefined;
+    }
+    const normalizedReference = trimmedReference.toLowerCase();
+    const canonicalMatches = availableModels.filter((model) => `${model.provider}/${model.id}`.toLowerCase() === normalizedReference);
+    if (canonicalMatches.length === 1) {
+        return canonicalMatches[0];
+    }
+    if (canonicalMatches.length > 1) {
+        return undefined;
+    }
+    const slashIndex = trimmedReference.indexOf("/");
+    if (slashIndex !== -1) {
+        const provider = trimmedReference.substring(0, slashIndex).trim();
+        const modelId = trimmedReference.substring(slashIndex + 1).trim();
+        if (provider && modelId) {
+            const providerMatches = availableModels.filter((model) => model.provider.toLowerCase() === provider.toLowerCase() &&
+                model.id.toLowerCase() === modelId.toLowerCase());
+            if (providerMatches.length === 1) {
+                return providerMatches[0];
+            }
+            if (providerMatches.length > 1) {
+                return undefined;
+            }
+        }
+    }
+    const idMatches = availableModels.filter((model) => model.id.toLowerCase() === normalizedReference);
+    return idMatches.length === 1 ? idMatches[0] : undefined;
+}
+/**
+ * Try to match a pattern to a model from the available models list.
+ * Returns the matched model or undefined if no match found.
+ */
+function tryMatchModel(modelPattern, availableModels) {
+    const exactMatch = findExactModelReferenceMatch(modelPattern, availableModels);
+    if (exactMatch) {
+        return exactMatch;
+    }
+    // No exact match - fall back to partial matching
+    const matches = availableModels.filter((m) => m.id.toLowerCase().includes(modelPattern.toLowerCase()) ||
+        m.name?.toLowerCase().includes(modelPattern.toLowerCase()));
+    if (matches.length === 0) {
+        return undefined;
+    }
+    // Separate into aliases and dated versions
+    const aliases = matches.filter((m) => isAlias(m.id));
+    const datedVersions = matches.filter((m) => !isAlias(m.id));
+    if (aliases.length > 0) {
+        // Prefer alias - if multiple aliases, pick the one that sorts highest
+        aliases.sort((a, b) => b.id.localeCompare(a.id));
+        return aliases[0];
+    }
+    else {
+        // No alias found, pick latest dated version
+        datedVersions.sort((a, b) => b.id.localeCompare(a.id));
+        return datedVersions[0];
+    }
+}
+function buildFallbackModel(provider, modelId, availableModels) {
+    const providerModels = availableModels.filter((m) => m.provider === provider);
+    if (providerModels.length === 0)
+        return undefined;
+    const defaultId = defaultModelPerProvider[provider];
+    const baseModel = defaultId
+        ? (providerModels.find((m) => m.id === defaultId) ?? providerModels[0])
+        : providerModels[0];
+    return {
+        ...baseModel,
+        id: modelId,
+        name: modelId,
+    };
+}
+/**
+ * Parse a pattern to extract model and thinking level.
+ * Handles models with colons in their IDs (e.g., OpenRouter's :exacto suffix).
+ *
+ * Algorithm:
+ * 1. Try to match full pattern as a model
+ * 2. If found, return it with "off" thinking level
+ * 3. If not found and has colons, split on last colon:
+ *    - If suffix is valid thinking level, use it and recurse on prefix
+ *    - If suffix is invalid, warn and recurse on prefix with "off"
+ *
+ * @internal Exported for testing
+ */
+export function parseModelPattern(pattern, availableModels, options) {
+    // Try exact match first
+    const exactMatch = tryMatchModel(pattern, availableModels);
+    if (exactMatch) {
+        return { model: exactMatch, thinkingLevel: undefined, warning: undefined };
+    }
+    // No match - try splitting on last colon if present
+    const lastColonIndex = pattern.lastIndexOf(":");
+    if (lastColonIndex === -1) {
+        // No colons, pattern simply doesn't match any model
+        return { model: undefined, thinkingLevel: undefined, warning: undefined };
+    }
+    const prefix = pattern.substring(0, lastColonIndex);
+    const suffix = pattern.substring(lastColonIndex + 1);
+    if (isValidThinkingLevel(suffix)) {
+        // Valid thinking level - recurse on prefix and use this level
+        const result = parseModelPattern(prefix, availableModels, options);
+        if (result.model) {
+            // Only use this thinking level if no warning from inner recursion
+            return {
+                model: result.model,
+                thinkingLevel: result.warning ? undefined : suffix,
+                warning: result.warning,
+            };
+        }
+        return result;
+    }
+    else {
+        // Invalid suffix
+        const allowFallback = options?.allowInvalidThinkingLevelFallback ?? true;
+        if (!allowFallback) {
+            // In strict mode (CLI --model parsing), treat it as part of the model id and fail.
+            // This avoids accidentally resolving to a different model.
+            return { model: undefined, thinkingLevel: undefined, warning: undefined };
+        }
+        // Scope mode: recurse on prefix and warn
+        const result = parseModelPattern(prefix, availableModels, options);
+        if (result.model) {
+            return {
+                model: result.model,
+                thinkingLevel: undefined,
+                warning: `Invalid thinking level "${suffix}" in pattern "${pattern}". Using default instead.`,
+            };
+        }
+        return result;
+    }
+}
+/**
+ * Resolve model patterns to actual Model objects with optional thinking levels
+ * Format: "pattern:level" where :level is optional
+ * For each pattern, finds all matching models and picks the best version:
+ * 1. Prefer alias (e.g., claude-sonnet-4-5) over dated versions (claude-sonnet-4-5-20250929)
+ * 2. If no alias, pick the latest dated version
+ *
+ * Supports models with colons in their IDs (e.g., OpenRouter's model:exacto).
+ * The algorithm tries to match the full pattern first, then progressively
+ * strips colon-suffixes to find a match.
+ */
+export async function resolveModelScope(patterns, modelRegistry) {
+    const availableModels = await modelRegistry.getAvailable();
+    const scopedModels = [];
+    for (const pattern of patterns) {
+        // Check if pattern contains glob characters
+        if (pattern.includes("*") || pattern.includes("?") || pattern.includes("[")) {
+            // Extract optional thinking level suffix (e.g., "provider/*:high")
+            const colonIdx = pattern.lastIndexOf(":");
+            let globPattern = pattern;
+            let thinkingLevel;
+            if (colonIdx !== -1) {
+                const suffix = pattern.substring(colonIdx + 1);
+                if (isValidThinkingLevel(suffix)) {
+                    thinkingLevel = suffix;
+                    globPattern = pattern.substring(0, colonIdx);
+                }
+            }
+            // Match against "provider/modelId" format OR just model ID
+            // This allows "*sonnet*" to match without requiring "anthropic/*sonnet*"
+            const matchingModels = availableModels.filter((m) => {
+                const fullId = `${m.provider}/${m.id}`;
+                return minimatch(fullId, globPattern, { nocase: true }) || minimatch(m.id, globPattern, { nocase: true });
+            });
+            if (matchingModels.length === 0) {
+                console.warn(chalk.yellow(`Warning: No models match pattern "${pattern}"`));
+                continue;
+            }
+            for (const model of matchingModels) {
+                if (!scopedModels.find((sm) => modelsAreEqual(sm.model, model))) {
+                    scopedModels.push({ model, thinkingLevel });
+                }
+            }
+            continue;
+        }
+        const { model, thinkingLevel, warning } = parseModelPattern(pattern, availableModels);
+        if (warning) {
+            console.warn(chalk.yellow(`Warning: ${warning}`));
+        }
+        if (!model) {
+            console.warn(chalk.yellow(`Warning: No models match pattern "${pattern}"`));
+            continue;
+        }
+        // Avoid duplicates
+        if (!scopedModels.find((sm) => modelsAreEqual(sm.model, model))) {
+            scopedModels.push({ model, thinkingLevel });
+        }
+    }
+    return scopedModels;
+}
+/**
+ * Resolve a single model from CLI flags.
+ *
+ * Supports:
+ * - --provider <provider> --model <pattern>
+ * - --model <provider>/<pattern>
+ * - Fuzzy matching (same rules as model scoping: exact id, then partial id/name)
+ *
+ * Note: This does not apply the thinking level by itself, but it may *parse* and
+ * return a thinking level from "<pattern>:<thinking>" so the caller can apply it.
+ */
+export function resolveCliModel(options) {
+    const { cliProvider, cliModel, modelRegistry } = options;
+    if (!cliModel) {
+        return { model: undefined, warning: undefined, error: undefined };
+    }
+    // Important: use *all* models here, not just models with pre-configured auth.
+    // This allows "--api-key" to be used for first-time setup.
+    const availableModels = modelRegistry.getAll();
+    if (availableModels.length === 0) {
+        return {
+            model: undefined,
+            warning: undefined,
+            error: "No models available. Check your installation or add models to models.json.",
+        };
+    }
+    // Build canonical provider lookup (case-insensitive)
+    const providerMap = new Map();
+    for (const m of availableModels) {
+        providerMap.set(m.provider.toLowerCase(), m.provider);
+    }
+    let provider = cliProvider ? providerMap.get(cliProvider.toLowerCase()) : undefined;
+    if (cliProvider && !provider) {
+        return {
+            model: undefined,
+            warning: undefined,
+            error: `Unknown provider "${cliProvider}". Use --list-models to see available providers/models.`,
+        };
+    }
+    // If no explicit --provider, try to interpret "provider/model" format first.
+    // When the prefix before the first slash matches a known provider, prefer that
+    // interpretation over matching models whose IDs literally contain slashes
+    // (e.g. "zai/glm-5" should resolve to provider=zai, model=glm-5, not to a
+    // vercel-ai-gateway model with id "zai/glm-5").
+    let pattern = cliModel;
+    let inferredProvider = false;
+    if (!provider) {
+        const slashIndex = cliModel.indexOf("/");
+        if (slashIndex !== -1) {
+            const maybeProvider = cliModel.substring(0, slashIndex);
+            const canonical = providerMap.get(maybeProvider.toLowerCase());
+            if (canonical) {
+                provider = canonical;
+                pattern = cliModel.substring(slashIndex + 1);
+                inferredProvider = true;
+            }
+        }
+    }
+    // If no provider was inferred from the slash, try exact matches without provider inference.
+    // This handles models whose IDs naturally contain slashes (e.g. OpenRouter-style IDs).
+    if (!provider) {
+        const lower = cliModel.toLowerCase();
+        const exact = availableModels.find((m) => m.id.toLowerCase() === lower || `${m.provider}/${m.id}`.toLowerCase() === lower);
+        if (exact) {
+            return { model: exact, warning: undefined, thinkingLevel: undefined, error: undefined };
+        }
+    }
+    if (cliProvider && provider) {
+        // If both were provided, tolerate --model <provider>/<pattern> by stripping the provider prefix
+        const prefix = `${provider}/`;
+        if (cliModel.toLowerCase().startsWith(prefix.toLowerCase())) {
+            pattern = cliModel.substring(prefix.length);
+        }
+    }
+    const candidates = provider ? availableModels.filter((m) => m.provider === provider) : availableModels;
+    const { model, thinkingLevel, warning } = parseModelPattern(pattern, candidates, {
+        allowInvalidThinkingLevelFallback: false,
+    });
+    if (model) {
+        return { model, thinkingLevel, warning, error: undefined };
+    }
+    // If we inferred a provider from the slash but found no match within that provider,
+    // fall back to matching the full input as a raw model id across all models.
+    // This handles OpenRouter-style IDs like "openai/gpt-4o:extended" where "openai"
+    // looks like a provider but the full string is actually a model id on openrouter.
+    if (inferredProvider) {
+        const lower = cliModel.toLowerCase();
+        const exact = availableModels.find((m) => m.id.toLowerCase() === lower || `${m.provider}/${m.id}`.toLowerCase() === lower);
+        if (exact) {
+            return { model: exact, warning: undefined, thinkingLevel: undefined, error: undefined };
+        }
+        // Also try parseModelPattern on the full input against all models
+        const fallback = parseModelPattern(cliModel, availableModels, {
+            allowInvalidThinkingLevelFallback: false,
+        });
+        if (fallback.model) {
+            return {
+                model: fallback.model,
+                thinkingLevel: fallback.thinkingLevel,
+                warning: fallback.warning,
+                error: undefined,
+            };
+        }
+    }
+    if (provider) {
+        const fallbackModel = buildFallbackModel(provider, pattern, availableModels);
+        if (fallbackModel) {
+            const fallbackWarning = warning
+                ? `${warning} Model "${pattern}" not found for provider "${provider}". Using custom model id.`
+                : `Model "${pattern}" not found for provider "${provider}". Using custom model id.`;
+            return { model: fallbackModel, thinkingLevel: undefined, warning: fallbackWarning, error: undefined };
+        }
+    }
+    const display = provider ? `${provider}/${pattern}` : cliModel;
+    return {
+        model: undefined,
+        thinkingLevel: undefined,
+        warning,
+        error: `Model "${display}" not found. Use --list-models to see available models.`,
+    };
+}
+/**
+ * Find the initial model to use based on priority:
+ * 1. CLI args (provider + model)
+ * 2. First model from scoped models (if not continuing/resuming)
+ * 3. Restored from session (if continuing/resuming)
+ * 4. Saved default from settings
+ * 5. First available model with valid API key
+ */
+export async function findInitialModel(options) {
+    const { cliProvider, cliModel, scopedModels, isContinuing, defaultProvider, defaultModelId, defaultThinkingLevel, modelRegistry, } = options;
+    let model;
+    let thinkingLevel = DEFAULT_THINKING_LEVEL;
+    // 1. CLI args take priority
+    if (cliProvider && cliModel) {
+        const resolved = resolveCliModel({
+            cliProvider,
+            cliModel,
+            modelRegistry,
+        });
+        if (resolved.error) {
+            console.error(chalk.red(resolved.error));
+            process.exit(1);
+        }
+        if (resolved.model) {
+            return { model: resolved.model, thinkingLevel: DEFAULT_THINKING_LEVEL, fallbackMessage: undefined };
+        }
+    }
+    // 2. Use first model from scoped models (skip if continuing/resuming)
+    if (scopedModels.length > 0 && !isContinuing) {
+        return {
+            model: scopedModels[0].model,
+            thinkingLevel: scopedModels[0].thinkingLevel ?? defaultThinkingLevel ?? DEFAULT_THINKING_LEVEL,
+            fallbackMessage: undefined,
+        };
+    }
+    // 3. Try saved default from settings
+    if (defaultProvider && defaultModelId) {
+        const found = modelRegistry.find(defaultProvider, defaultModelId);
+        if (found) {
+            model = found;
+            if (defaultThinkingLevel) {
+                thinkingLevel = defaultThinkingLevel;
+            }
+            return { model, thinkingLevel, fallbackMessage: undefined };
+        }
+    }
+    // 4. Try first available model with valid API key
+    const availableModels = await modelRegistry.getAvailable();
+    if (availableModels.length > 0) {
+        // Try to find a default model from known providers
+        for (const provider of Object.keys(defaultModelPerProvider)) {
+            const defaultId = defaultModelPerProvider[provider];
+            const match = availableModels.find((m) => m.provider === provider && m.id === defaultId);
+            if (match) {
+                return { model: match, thinkingLevel: DEFAULT_THINKING_LEVEL, fallbackMessage: undefined };
+            }
+        }
+        // If no default found, use first available
+        return { model: availableModels[0], thinkingLevel: DEFAULT_THINKING_LEVEL, fallbackMessage: undefined };
+    }
+    // 5. No model found
+    return { model: undefined, thinkingLevel: DEFAULT_THINKING_LEVEL, fallbackMessage: undefined };
+}
+/**
+ * Restore model from session, with fallback to available models
+ */
+export async function restoreModelFromSession(savedProvider, savedModelId, currentModel, shouldPrintMessages, modelRegistry) {
+    const restoredModel = modelRegistry.find(savedProvider, savedModelId);
+    // Check if restored model exists and still has auth configured
+    const hasConfiguredAuth = restoredModel ? modelRegistry.hasConfiguredAuth(restoredModel) : false;
+    if (restoredModel && hasConfiguredAuth) {
+        if (shouldPrintMessages) {
+            console.log(chalk.dim(`Restored model: ${savedProvider}/${savedModelId}`));
+        }
+        return { model: restoredModel, fallbackMessage: undefined };
+    }
+    // Model not found or no API key - fall back
+    const reason = !restoredModel ? "model no longer exists" : "no auth configured";
+    if (shouldPrintMessages) {
+        console.error(chalk.yellow(`Warning: Could not restore model ${savedProvider}/${savedModelId} (${reason}).`));
+    }
+    // If we already have a model, use it as fallback
+    if (currentModel) {
+        if (shouldPrintMessages) {
+            console.log(chalk.dim(`Falling back to: ${currentModel.provider}/${currentModel.id}`));
+        }
+        return {
+            model: currentModel,
+            fallbackMessage: `Could not restore model ${savedProvider}/${savedModelId} (${reason}). Using ${currentModel.provider}/${currentModel.id}.`,
+        };
+    }
+    // Try to find any available model
+    const availableModels = await modelRegistry.getAvailable();
+    if (availableModels.length > 0) {
+        // Try to find a default model from known providers
+        let fallbackModel;
+        for (const provider of Object.keys(defaultModelPerProvider)) {
+            const defaultId = defaultModelPerProvider[provider];
+            const match = availableModels.find((m) => m.provider === provider && m.id === defaultId);
+            if (match) {
+                fallbackModel = match;
+                break;
+            }
+        }
+        // If no default found, use first available
+        if (!fallbackModel) {
+            fallbackModel = availableModels[0];
+        }
+        if (shouldPrintMessages) {
+            console.log(chalk.dim(`Falling back to: ${fallbackModel.provider}/${fallbackModel.id}`));
+        }
+        return {
+            model: fallbackModel,
+            fallbackMessage: `Could not restore model ${savedProvider}/${savedModelId} (${reason}). Using ${fallbackModel.provider}/${fallbackModel.id}.`,
+        };
+    }
+    // No models available
+    return { model: undefined, fallbackMessage: undefined };
+}

package/dist/core/output-guard.d.ts

@@ -0,0 +1,5 @@
+export declare function takeOverStdout(): void;
+export declare function restoreStdout(): void;
+export declare function isStdoutTakenOver(): boolean;
+export declare function writeRawStdout(text: string): void;
+export declare function flushRawStdout(): Promise<void>;

package/dist/core/output-guard.js

@@ -0,0 +1,58 @@
+let stdoutTakeoverState;
+export function takeOverStdout() {
+    if (stdoutTakeoverState) {
+        return;
+    }
+    const rawStdoutWrite = process.stdout.write.bind(process.stdout);
+    const rawStderrWrite = process.stderr.write.bind(process.stderr);
+    const originalStdoutWrite = process.stdout.write;
+    process.stdout.write = ((chunk, encodingOrCallback, callback) => {
+        if (typeof encodingOrCallback === "function") {
+            return rawStderrWrite(String(chunk), encodingOrCallback);
+        }
+        return rawStderrWrite(String(chunk), callback);
+    });
+    stdoutTakeoverState = {
+        rawStdoutWrite,
+        rawStderrWrite,
+        originalStdoutWrite,
+    };
+}
+export function restoreStdout() {
+    if (!stdoutTakeoverState) {
+        return;
+    }
+    process.stdout.write = stdoutTakeoverState.originalStdoutWrite;
+    stdoutTakeoverState = undefined;
+}
+export function isStdoutTakenOver() {
+    return stdoutTakeoverState !== undefined;
+}
+export function writeRawStdout(text) {
+    if (stdoutTakeoverState) {
+        stdoutTakeoverState.rawStdoutWrite(text);
+        return;
+    }
+    process.stdout.write(text);
+}
+export async function flushRawStdout() {
+    if (stdoutTakeoverState) {
+        await new Promise((resolve, reject) => {
+            stdoutTakeoverState?.rawStdoutWrite("", (err) => {
+                if (err)
+                    reject(err);
+                else
+                    resolve();
+            });
+        });
+        return;
+    }
+    await new Promise((resolve, reject) => {
+        process.stdout.write("", (err) => {
+            if (err)
+                reject(err);
+            else
+                resolve();
+        });
+    });
+}