@polpo-ai/llm 0.5.2

package/dist/model-resolver.js

@@ -0,0 +1,406 @@
+/**
+ * Model resolution — resolves model specs to AI SDK LanguageModel instances
+ * with full metadata. Handles provider overrides, model allowlists,
+ * fallback chains, and provider key validation.
+ */
+import { parseModelSpec as _parseModelSpec } from "@polpo-ai/core";
+import { getCatalogSync } from "./gateway-catalog.js";
+import { createCustomProviderModel, createGatewayModel } from "./provider-factory.js";
+import { resolveApiKey, resolveApiKeyAsync, hasOAuthProfiles, PROVIDER_ENV_MAP } from "./api-keys.js";
+// ─── Provider Override Management ────────────────────
+/** Provider overrides from polpo.json — set by the orchestrator at init time. */
+let providerOverrides = {};
+export function setProviderOverrides(overrides) {
+    providerOverrides = overrides;
+}
+export function getProviderOverrides() {
+    return { ...providerOverrides };
+}
+// ─── Model Allowlist ─────────────────────────────────
+/** Model allowlist — when set, only these models can be used. Set by orchestrator at init. */
+let modelAllowlist;
+export function setModelAllowlist(allowlist) {
+    modelAllowlist = allowlist;
+}
+export function getModelAllowlist() {
+    return modelAllowlist;
+}
+/**
+ * Check if a model spec is allowed by the allowlist.
+ * Returns true if no allowlist is set (everything allowed) or the model is in the list.
+ */
+export function isModelAllowed(spec) {
+    if (!modelAllowlist)
+        return true;
+    // Check exact match
+    if (spec in modelAllowlist)
+        return true;
+    // Check without provider prefix (e.g. "claude-opus-4.6" matches "anthropic:claude-opus-4.6")
+    const { provider, modelId } = parseModelSpec(spec);
+    const fullSpec = `${provider}:${modelId}`;
+    return fullSpec in modelAllowlist;
+}
+/**
+ * Enforce model allowlist. Throws if the model is not allowed.
+ */
+export function enforceModelAllowlist(spec) {
+    if (!isModelAllowed(spec)) {
+        const allowed = Object.keys(modelAllowlist).join(", ");
+        throw new Error(`Model "${spec}" is not in the allowlist. Allowed models: ${allowed}`);
+    }
+}
+// ─── Model Spec Parsing ──────────────────────────────
+/**
+ * Parse a model spec string into provider + modelId.
+ * Falls back to POLPO_MODEL env var. Throws if no model is available.
+ */
+export function parseModelSpec(spec) {
+    return _parseModelSpec(spec, process.env.POLPO_MODEL);
+}
+// ─── Model Resolution ────────────────────────────────
+/**
+ * Resolve a model spec to a ResolvedModel with metadata + AI SDK model instance.
+ *
+ * Resolution order:
+ * 1. If provider has an override with custom baseUrl -> create OpenAI-compatible model
+ * 2. Otherwise -> use AI Gateway (provider/modelId format)
+ *
+ * This ensures custom providers (Ollama, vLLM, etc.) work without being in the gateway.
+ */
+export function resolveModel(spec) {
+    const { provider, modelId } = parseModelSpec(spec);
+    const override = providerOverrides[provider];
+    // Custom provider with baseUrl override -> use @ai-sdk/openai with custom endpoint
+    if (override?.baseUrl) {
+        const customDef = override.models?.find(m => m.id === modelId);
+        const aiModel = createCustomProviderModel(provider, modelId, override);
+        if (customDef) {
+            return {
+                id: customDef.id,
+                name: customDef.name,
+                provider,
+                reasoning: customDef.reasoning ?? false,
+                input: customDef.input ?? ["text"],
+                cost: customDef.cost ?? { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+                contextWindow: customDef.contextWindow ?? 200_000,
+                maxTokens: customDef.maxTokens ?? 8192,
+                aiModel,
+            };
+        }
+        // No custom def — construct minimal metadata
+        return {
+            id: modelId,
+            name: modelId,
+            provider,
+            reasoning: false,
+            input: ["text"],
+            cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+            contextWindow: 200_000,
+            maxTokens: 8192,
+            aiModel,
+        };
+    }
+    // Standard provider -> route through AI Gateway
+    const aiModel = createGatewayModel(provider, modelId);
+    // Try to get metadata from cached catalog
+    const catalog = getCatalogSync();
+    const gatewayModelId = `${provider}/${modelId}`;
+    const entry = catalog.find(m => m.id === gatewayModelId);
+    if (entry) {
+        const pricing = entry.pricing;
+        return {
+            id: modelId,
+            name: entry.name || modelId,
+            provider,
+            reasoning: false, // Gateway catalog doesn't expose this directly
+            input: ["text"],
+            contextWindow: 200_000, // Gateway doesn't expose this; use sensible default
+            maxTokens: 8192,
+            cost: {
+                input: pricing ? parseFloat(pricing.input) : 0,
+                output: pricing ? parseFloat(pricing.output) : 0,
+                cacheRead: pricing?.cachedInputTokens ? parseFloat(pricing.cachedInputTokens) : 0,
+                cacheWrite: pricing?.cacheCreationInputTokens ? parseFloat(pricing.cacheCreationInputTokens) : 0,
+            },
+            aiModel,
+        };
+    }
+    // Not in catalog (yet) — return with defaults. The model may still work via gateway.
+    return {
+        id: modelId,
+        name: modelId,
+        provider,
+        reasoning: false,
+        input: ["text"],
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+        contextWindow: 200_000,
+        maxTokens: 8192,
+        aiModel,
+    };
+}
+// ─── Model Catalog Queries ───────────────────────────
+/**
+ * Get detailed model info for a specific model spec.
+ */
+export function getModelInfo(spec) {
+    try {
+        const model = resolveModel(spec);
+        return {
+            id: model.id,
+            name: model.name,
+            provider: model.provider,
+            reasoning: model.reasoning,
+            input: model.input,
+            contextWindow: model.contextWindow,
+            maxTokens: model.maxTokens,
+            cost: model.cost,
+        };
+    }
+    catch {
+        return undefined;
+    }
+}
+/**
+ * List all available providers from the AI Gateway catalog.
+ * Returns synchronously from the cache; triggers background refresh if stale.
+ */
+export function listProviders() {
+    const catalog = getCatalogSync();
+    const providers = new Set();
+    for (const entry of catalog) {
+        const slashIdx = entry.id.indexOf("/");
+        if (slashIdx > 0) {
+            providers.add(entry.id.slice(0, slashIdx));
+        }
+    }
+    // Also include custom provider overrides
+    for (const p of Object.keys(providerOverrides)) {
+        providers.add(p);
+    }
+    return Array.from(providers).sort();
+}
+/**
+ * List all models for a given provider (or all providers if none specified).
+ */
+export function listModels(provider) {
+    const catalog = getCatalogSync();
+    const models = [];
+    for (const entry of catalog) {
+        const slashIdx = entry.id.indexOf("/");
+        if (slashIdx <= 0)
+            continue;
+        const entryProvider = entry.id.slice(0, slashIdx);
+        const entryModelId = entry.id.slice(slashIdx + 1);
+        if (provider && entryProvider !== provider)
+            continue;
+        const pricing = entry.pricing;
+        models.push({
+            id: entryModelId,
+            name: entry.name || entryModelId,
+            provider: entryProvider,
+            reasoning: false,
+            input: ["text"],
+            contextWindow: 200_000,
+            maxTokens: 8192,
+            cost: {
+                input: pricing ? parseFloat(pricing.input) : 0,
+                output: pricing ? parseFloat(pricing.output) : 0,
+                cacheRead: pricing?.cachedInputTokens ? parseFloat(pricing.cachedInputTokens) : 0,
+                cacheWrite: pricing?.cacheCreationInputTokens ? parseFloat(pricing.cacheCreationInputTokens) : 0,
+            },
+        });
+    }
+    // Append custom models from provider overrides
+    const overrideProviders = provider ? [provider] : Object.keys(providerOverrides);
+    for (const p of overrideProviders) {
+        const override = providerOverrides[p];
+        if (!override?.models)
+            continue;
+        for (const m of override.models) {
+            models.push({
+                id: m.id,
+                name: m.name,
+                provider: p,
+                reasoning: m.reasoning ?? false,
+                input: m.input ?? ["text"],
+                contextWindow: m.contextWindow ?? 200_000,
+                maxTokens: m.maxTokens ?? 8192,
+                cost: m.cost ?? { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+            });
+        }
+    }
+    return models;
+}
+/**
+ * Build a dynamic model listing string for system prompts.
+ * Uses the gateway catalog instead of hardcoded lists.
+ */
+export function buildModelListingForPrompt() {
+    const lines = [
+        `Format: "provider:model" (e.g. "anthropic:claude-opus-4.6") or just "model" (auto-inferred from prefix).`,
+    ];
+    const catalog = getCatalogSync();
+    // Group models by provider
+    const byProvider = new Map();
+    for (const entry of catalog) {
+        const slashIdx = entry.id.indexOf("/");
+        if (slashIdx <= 0)
+            continue;
+        const provider = entry.id.slice(0, slashIdx);
+        if (!byProvider.has(provider))
+            byProvider.set(provider, []);
+        byProvider.get(provider).push(entry);
+    }
+    // Show the most relevant providers with their top models
+    const FEATURED_PROVIDERS = [
+        { provider: "anthropic", picks: 3 },
+        { provider: "openai", picks: 4 },
+        { provider: "google", picks: 3 },
+        { provider: "mistral", picks: 2 },
+        { provider: "groq", picks: 2 },
+        { provider: "xai", picks: 1 },
+    ];
+    for (const { provider, picks } of FEATURED_PROVIDERS) {
+        const models = byProvider.get(provider);
+        if (!models || models.length === 0)
+            continue;
+        const top = models.slice(0, picks);
+        const modelStr = top.map(m => {
+            const modelId = m.id.slice(m.id.indexOf("/") + 1);
+            const tags = [];
+            if (m.pricing && parseFloat(m.pricing.input) === 0 && parseFloat(m.pricing.output) === 0) {
+                tags.push("FREE");
+            }
+            const tagStr = tags.length > 0 ? ` (${tags.join(", ")})` : "";
+            return `${modelId}${tagStr}`;
+        }).join(", ");
+        lines.push(`- ${provider}: ${modelStr}`);
+    }
+    // Count totals
+    const totalProviders = byProvider.size + Object.keys(providerOverrides).length;
+    const totalModels = catalog.length;
+    lines.push(`- ... and ${totalProviders} total providers with ${totalModels}+ models (use "provider:model" format)`);
+    lines.push(`Configure your default model in .polpo/polpo.json or via the POLPO_MODEL env var.`);
+    return lines.join("\n");
+}
+// ─── ModelConfig Helpers ─────────────────────────────
+/**
+ * Normalize a model spec that may be string or ModelConfig into a plain string.
+ * Useful for APIs that only accept a string model spec.
+ */
+export function resolveModelSpec(spec) {
+    if (spec === undefined)
+        return undefined;
+    if (typeof spec === "string")
+        return spec;
+    return spec.primary;
+}
+// ─── Model Fallback Chain ────────────────────────────
+/**
+ * Resolve a model from a fallback chain (synchronous).
+ * Tries primary first, then each fallback in order.
+ * Returns the first model that has a valid API key.
+ */
+export function resolveModelWithFallback(config) {
+    const primary = config.primary;
+    if (!primary) {
+        throw new Error("No primary model configured. Run 'polpo setup' or set POLPO_MODEL env var.");
+    }
+    const { provider: primaryProvider } = parseModelSpec(primary);
+    if (resolveApiKey(primaryProvider)) {
+        try {
+            return { model: resolveModel(primary), spec: primary };
+        }
+        catch {
+            // Primary model not found — try fallbacks
+        }
+    }
+    if (config.fallbacks) {
+        for (const fallback of config.fallbacks) {
+            const { provider: fbProvider } = parseModelSpec(fallback);
+            if (resolveApiKey(fbProvider)) {
+                try {
+                    return { model: resolveModel(fallback), spec: fallback };
+                }
+                catch {
+                    // Model not found — try next
+                }
+            }
+        }
+    }
+    // Last resort: try primary anyway (will fail at call time with a clear error)
+    return { model: resolveModel(primary), spec: primary };
+}
+/**
+ * Resolve a model from a fallback chain (async).
+ * Tries primary first, then each fallback in order.
+ * Checks the FULL API key resolution chain including OAuth profiles with auto-refresh.
+ */
+export async function resolveModelWithFallbackAsync(config) {
+    const primary = config.primary;
+    if (!primary) {
+        throw new Error("No primary model configured. Run 'polpo setup' or set POLPO_MODEL env var.");
+    }
+    const { provider: primaryProvider } = parseModelSpec(primary);
+    if (await resolveApiKeyAsync(primaryProvider)) {
+        try {
+            return { model: resolveModel(primary), spec: primary };
+        }
+        catch {
+            // Primary model not found — try fallbacks
+        }
+    }
+    if (config.fallbacks) {
+        for (const fallback of config.fallbacks) {
+            const { provider: fbProvider } = parseModelSpec(fallback);
+            if (await resolveApiKeyAsync(fbProvider)) {
+                try {
+                    return { model: resolveModel(fallback), spec: fallback };
+                }
+                catch {
+                    // Model not found — try next
+                }
+            }
+        }
+    }
+    return { model: resolveModel(primary), spec: primary };
+}
+/**
+ * Validate that all required providers have API keys available.
+ * Returns detailed validation results for all model specs.
+ */
+export function validateProviderKeys(modelSpecs) {
+    const missing = [];
+    const seen = new Set();
+    for (const spec of modelSpecs) {
+        const { provider } = parseModelSpec(spec);
+        if (seen.has(provider))
+            continue;
+        seen.add(provider);
+        if (!resolveApiKey(provider) && !hasOAuthProfiles(provider)) {
+            missing.push({ provider, modelSpec: spec });
+        }
+    }
+    return missing;
+}
+/**
+ * Get detailed validation for a set of model specs — including which env var to set.
+ */
+export function validateProviderKeysDetailed(modelSpecs) {
+    const results = [];
+    const seen = new Set();
+    for (const spec of modelSpecs) {
+        const { provider } = parseModelSpec(spec);
+        if (seen.has(provider))
+            continue;
+        seen.add(provider);
+        results.push({
+            provider,
+            modelSpec: spec,
+            hasKey: !!resolveApiKey(provider),
+            envVar: PROVIDER_ENV_MAP[provider],
+        });
+    }
+    return results;
+}
+//# sourceMappingURL=model-resolver.js.map

package/dist/model-resolver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-resolver.js","sourceRoot":"","sources":["../src/model-resolver.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,cAAc,IAAI,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAGnE,OAAO,EAAE,cAAc,EAAkD,MAAM,sBAAsB,CAAC;AACtG,OAAO,EAAE,yBAAyB,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AACtF,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AA6BtG,wDAAwD;AAExD,iFAAiF;AACjF,IAAI,iBAAiB,GAAmC,EAAE,CAAC;AAE3D,MAAM,UAAU,oBAAoB,CAAC,SAAyC;IAC5E,iBAAiB,GAAG,SAAS,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,oBAAoB;IAClC,OAAO,EAAE,GAAG,iBAAiB,EAAE,CAAC;AAClC,CAAC;AAED,wDAAwD;AAExD,8FAA8F;AAC9F,IAAI,cAA+D,CAAC;AAEpE,MAAM,UAAU,iBAAiB,CAAC,SAA0D;IAC1F,cAAc,GAAG,SAAS,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,iBAAiB;IAC/B,OAAO,cAAc,CAAC;AACxB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,IAAY;IACzC,IAAI,CAAC,cAAc;QAAE,OAAO,IAAI,CAAC;IACjC,oBAAoB;IACpB,IAAI,IAAI,IAAI,cAAc;QAAE,OAAO,IAAI,CAAC;IACxC,6FAA6F;IAC7F,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;IACnD,MAAM,QAAQ,GAAG,GAAG,QAAQ,IAAI,OAAO,EAAE,CAAC;IAC1C,OAAO,QAAQ,IAAI,cAAc,CAAC;AACpC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,IAAY;IAChD,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,cAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxD,MAAM,IAAI,KAAK,CAAC,UAAU,IAAI,8CAA8C,OAAO,EAAE,CAAC,CAAC;IACzF,CAAC;AACH,CAAC;AAED,wDAAwD;AAExD;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,IAAa;IAC1C,OAAO,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;AACxD,CAAC;AAED,wDAAwD;AAExD;;;;;;;;GAQG;AACH,MAAM,UAAU,YAAY,CAAC,IAAa;IACxC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;IACnD,MAAM,QAAQ,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IAE7C,mFAAmF;IACnF,IAAI,QAAQ,EAAE,OAAO,EAAE,CAAC;QACtB,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;QAC/D,MAAM,OAAO,GAAG,yBAAyB,CAAC,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;QAEvE,IAAI,SAAS,EAAE,CAAC;YACd,OAAO;gBACL,EAAE,EAAE,SAAS,CAAC,EAAE;gBAChB,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,QAAQ;gBACR,SAAS,EAAE,SAAS,CAAC,SAAS,IAAI,KAAK;gBACvC,KAAK,EAAE,SAAS,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC;gBAClC,IAAI,EAAE,SAAS,CAAC,IAAI,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;gBAC5E,aAAa,EAAE,SAAS,CAAC,aAAa,IAAI,OAAO;gBACjD,SAAS,EAAE,SAAS,CAAC,SAAS,IAAI,IAAI;gBACtC,OAAO;aACR,CAAC;QACJ,CAAC;QAED,6CAA6C;QAC7C,OAAO;YACL,EAAE,EAAE,OAAO;YACX,IAAI,EAAE,OAAO;YACb,QAAQ;YACR,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,CAAC,MAAM,CAAC;YACf,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;YAC1D,aAAa,EAAE,OAAO;YACtB,SAAS,EAAE,IAAI;YACf,OAAO;SACR,CAAC;IACJ,CAAC;IAED,gDAAgD;IAChD,MAAM,OAAO,GAAG,kBAAkB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAEtD,0CAA0C;IAC1C,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IACjC,MAAM,cAAc,GAAG,GAAG,QAAQ,IAAI,OAAO,EAAE,CAAC;IAChD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,cAAc,CAAC,CAAC;IAEzD,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;QAC9B,OAAO;YACL,EAAE,EAAE,OAAO;YACX,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,OAAO;YAC3B,QAAQ;YACR,SAAS,EAAE,KAAK,EAAE,+CAA+C;YACjE,KAAK,EAAE,CAAC,MAAM,CAAC;YACf,aAAa,EAAE,OAAO,EAAE,oDAAoD;YAC5E,SAAS,EAAE,IAAI;YACf,IAAI,EAAE;gBACJ,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9C,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;gBAChD,SAAS,EAAE,OAAO,EAAE,iBAAiB,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC;gBACjF,UAAU,EAAE,OAAO,EAAE,wBAAwB,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC,CAAC;aACjG;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,qFAAqF;IACrF,OAAO;QACL,EAAE,EAAE,OAAO;QACX,IAAI,EAAE,OAAO;QACb,QAAQ;QACR,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,CAAC,MAAM,CAAC;QACf,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;QAC1D,aAAa,EAAE,OAAO;QACtB,SAAS,EAAE,IAAI;QACf,OAAO;KACR,CAAC;AACJ,CAAC;AAED,wDAAwD;AAExD;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,IAAY;IACvC,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QACjC,OAAO;YACL,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,aAAa,EAAE,KAAK,CAAC,aAAa;YAClC,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,IAAI,EAAE,KAAK,CAAC,IAAI;SACjB,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,aAAa;IAC3B,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IACjC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,MAAM,QAAQ,GAAG,KAAK,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACvC,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;YACjB,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IACD,yCAAyC;IACzC,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,EAAE,CAAC;QAC/C,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IACnB,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC;AACtC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU,CAAC,QAAiB;IAC1C,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IACjC,MAAM,MAAM,GAAgB,EAAE,CAAC;IAE/B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,MAAM,QAAQ,GAAG,KAAK,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACvC,IAAI,QAAQ,IAAI,CAAC;YAAE,SAAS;QAE5B,MAAM,aAAa,GAAG,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;QAClD,MAAM,YAAY,GAAG,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;QAElD,IAAI,QAAQ,IAAI,aAAa,KAAK,QAAQ;YAAE,SAAS;QAErD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC;YACV,EAAE,EAAE,YAAY;YAChB,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,YAAY;YAChC,QAAQ,EAAE,aAAa;YACvB,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,CAAC,MAAM,CAAC;YACf,aAAa,EAAE,OAAO;YACtB,SAAS,EAAE,IAAI;YACf,IAAI,EAAE;gBACJ,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9C,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;gBAChD,SAAS,EAAE,OAAO,EAAE,iBAAiB,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC;gBACjF,UAAU,EAAE,OAAO,EAAE,wBAAwB,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC,CAAC;aACjG;SACF,CAAC,CAAC;IACL,CAAC;IAED,+CAA+C;IAC/C,MAAM,iBAAiB,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACjF,KAAK,MAAM,CAAC,IAAI,iBAAiB,EAAE,CAAC;QAClC,MAAM,QAAQ,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,MAAM;YAAE,SAAS;QAChC,KAAK,MAAM,CAAC,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC;gBACV,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,QAAQ,EAAE,CAAC;gBACX,SAAS,EAAE,CAAC,CAAC,SAAS,IAAI,KAAK;gBAC/B,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC;gBAC1B,aAAa,EAAE,CAAC,CAAC,aAAa,IAAI,OAAO;gBACzC,SAAS,EAAE,CAAC,CAAC,SAAS,IAAI,IAAI;gBAC9B,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;aACrE,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,0BAA0B;IACxC,MAAM,KAAK,GAAa;QACtB,0GAA0G;KAC3G,CAAC;IAEF,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IAEjC,2BAA2B;IAC3B,MAAM,UAAU,GAAG,IAAI,GAAG,EAAuC,CAAC;IAClE,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,MAAM,QAAQ,GAAG,KAAK,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACvC,IAAI,QAAQ,IAAI,CAAC;YAAE,SAAS;QAC5B,MAAM,QAAQ,GAAG,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC;YAAE,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAC5D,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IAED,yDAAyD;IACzD,MAAM,kBAAkB,GAA0C;QAChE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,EAAE;QACnC,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE;QAChC,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE;QAChC,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE;QACjC,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE;QAC9B,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE;KAC9B,CAAC;IAEF,KAAK,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,kBAAkB,EAAE,CAAC;QACrD,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QAE7C,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACnC,MAAM,QAAQ,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;YAC3B,MAAM,OAAO,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;YAClD,MAAM,IAAI,GAAa,EAAE,CAAC;YAC1B,IAAI,CAAC,CAAC,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzF,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpB,CAAC;YACD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,OAAO,GAAG,OAAO,GAAG,MAAM,EAAE,CAAC;QAC/B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,KAAK,CAAC,IAAI,CAAC,KAAK,QAAQ,KAAK,QAAQ,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,eAAe;IACf,MAAM,cAAc,GAAG,UAAU,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,MAAM,CAAC;IAC/E,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;IACnC,KAAK,CAAC,IAAI,CAAC,aAAa,cAAc,yBAAyB,WAAW,wCAAwC,CAAC,CAAC;IACpH,KAAK,CAAC,IAAI,CAAC,mFAAmF,CAAC,CAAC;IAEhG,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,wDAAwD;AAExD;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAAC,IAAsC;IACrE,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IACzC,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC1C,OAAO,IAAI,CAAC,OAAO,CAAC;AACtB,CAAC;AAED,wDAAwD;AAExD;;;;GAIG;AACH,MAAM,UAAU,wBAAwB,CAAC,MAAmB;IAC1D,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;IAC/B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,4EAA4E,CAAC,CAAC;IAChG,CAAC;IACD,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IAC9D,IAAI,aAAa,CAAC,eAAe,CAAC,EAAE,CAAC;QACnC,IAAI,CAAC;YACH,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QACzD,CAAC;QAAC,MAAM,CAAC;YACP,0CAA0C;QAC5C,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QACrB,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACxC,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;YAC1D,IAAI,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC9B,IAAI,CAAC;oBACH,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gBAC3D,CAAC;gBAAC,MAAM,CAAC;oBACP,6BAA6B;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AACzD,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,6BAA6B,CAAC,MAAmB;IACrE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;IAC/B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,4EAA4E,CAAC,CAAC;IAChG,CAAC;IACD,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IAC9D,IAAI,MAAM,kBAAkB,CAAC,eAAe,CAAC,EAAE,CAAC;QAC9C,IAAI,CAAC;YACH,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QACzD,CAAC;QAAC,MAAM,CAAC;YACP,0CAA0C;QAC5C,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QACrB,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACxC,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;YAC1D,IAAI,MAAM,kBAAkB,CAAC,UAAU,CAAC,EAAE,CAAC;gBACzC,IAAI,CAAC;oBACH,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gBAC3D,CAAC;gBAAC,MAAM,CAAC;oBACP,6BAA6B;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AACzD,CAAC;AAWD;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAClC,UAAoB;IAEpB,MAAM,OAAO,GAA8C,EAAE,CAAC;IAC9D,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAE/B,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,MAAM,EAAE,QAAQ,EAAE,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC;YAAE,SAAS;QACjC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEnB,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5D,OAAO,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,4BAA4B,CAC1C,UAAoB;IAEpB,MAAM,OAAO,GAA+B,EAAE,CAAC;IAC/C,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAE/B,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,MAAM,EAAE,QAAQ,EAAE,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC;YAAE,SAAS;QACjC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEnB,OAAO,CAAC,IAAI,CAAC;YACX,QAAQ;YACR,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC;YACjC,MAAM,EAAE,gBAAgB,CAAC,QAAQ,CAAC;SACnC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/provider-factory.d.ts

@@ -0,0 +1,37 @@
+/**
+ * Provider factory — creates AI SDK LanguageModel instances.
+ *
+ * Handles both standard AI Gateway models and custom OpenAI-compatible
+ * endpoints (Ollama, vLLM, LM Studio, etc.).
+ */
+import type { LanguageModel } from "ai";
+import type { ProviderConfig, ReasoningLevel } from "@polpo-ai/core";
+/**
+ * Create an AI SDK LanguageModel for a custom (non-gateway) provider.
+ * Uses @ai-sdk/openai with a custom baseURL for OpenAI-compatible endpoints.
+ */
+export declare function createCustomProviderModel(provider: string, modelId: string, override: ProviderConfig): LanguageModel;
+/**
+ * Create an AI SDK LanguageModel via the AI Gateway.
+ * The gateway expects "provider/modelId" format.
+ */
+export declare function createGatewayModel(provider: string, modelId: string): LanguageModel;
+/**
+ * Map Polpo's ReasoningLevel to AI SDK providerOptions for reasoning/thinking.
+ *
+ * Each provider has its own way of enabling extended thinking:
+ * - Anthropic: thinking.type + thinking.budgetTokens
+ * - OpenAI: reasoningEffort
+ * - Google: thinkingConfig.thinkingBudget
+ */
+export declare function mapReasoningToProviderOptions(provider: string, level: ReasoningLevel | undefined, maxTokens: number): Record<string, Record<string, unknown>> | undefined;
+/**
+ * Build AI SDK compatible options for generateText/streamText calls.
+ *
+ * Returns an object with:
+ * - providerOptions: reasoning/thinking configuration per provider
+ * - maxTokens: max output tokens
+ * - headers: additional headers (e.g. for API key passthrough)
+ */
+export declare function buildStreamOpts(apiKey?: string, reasoning?: ReasoningLevel, maxTokens?: number): Record<string, unknown> | undefined;
+//# sourceMappingURL=provider-factory.d.ts.map

package/dist/provider-factory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-factory.d.ts","sourceRoot":"","sources":["../src/provider-factory.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAGxC,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAGrE;;;GAGG;AACH,wBAAgB,yBAAyB,CACvC,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,cAAc,GACvB,aAAa,CAaf;AAED;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,aAAa,CAGnF;AAED;;;;;;;GAOG;AACH,wBAAgB,6BAA6B,CAC3C,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,cAAc,GAAG,SAAS,EACjC,SAAS,EAAE,MAAM,GAChB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,GAAG,SAAS,CAmDrD;AAED;;;;;;;GAOG;AACH,wBAAgB,eAAe,CAC7B,MAAM,CAAC,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,cAAc,EAC1B,SAAS,CAAC,EAAE,MAAM,GACjB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAUrC"}

package/dist/provider-factory.js

@@ -0,0 +1,110 @@
+/**
+ * Provider factory — creates AI SDK LanguageModel instances.
+ *
+ * Handles both standard AI Gateway models and custom OpenAI-compatible
+ * endpoints (Ollama, vLLM, LM Studio, etc.).
+ */
+import { gateway } from "ai";
+import { createOpenAI } from "@ai-sdk/openai";
+import { resolveApiKey } from "./api-keys.js";
+/**
+ * Create an AI SDK LanguageModel for a custom (non-gateway) provider.
+ * Uses @ai-sdk/openai with a custom baseURL for OpenAI-compatible endpoints.
+ */
+export function createCustomProviderModel(provider, modelId, override) {
+    const baseURL = override.baseUrl || "http://localhost:11434/v1";
+    const apiKey = resolveApiKey(provider) || "ollama"; // Ollama doesn't need a real key
+    // For anthropic-messages API, we'd need @ai-sdk/anthropic — but custom providers
+    // are typically Ollama/vLLM/LM Studio which all speak OpenAI-compatible.
+    const openaiProvider = createOpenAI({
+        baseURL,
+        apiKey,
+        name: provider,
+    });
+    return openaiProvider(modelId);
+}
+/**
+ * Create an AI SDK LanguageModel via the AI Gateway.
+ * The gateway expects "provider/modelId" format.
+ */
+export function createGatewayModel(provider, modelId) {
+    const gatewayModelId = `${provider}/${modelId}`;
+    return gateway(gatewayModelId);
+}
+/**
+ * Map Polpo's ReasoningLevel to AI SDK providerOptions for reasoning/thinking.
+ *
+ * Each provider has its own way of enabling extended thinking:
+ * - Anthropic: thinking.type + thinking.budgetTokens
+ * - OpenAI: reasoningEffort
+ * - Google: thinkingConfig.thinkingBudget
+ */
+export function mapReasoningToProviderOptions(provider, level, maxTokens) {
+    if (!level || level === "off")
+        return undefined;
+    // Budget tokens as a fraction of maxTokens, scaling with reasoning level
+    const budgetMap = {
+        minimal: 0.1,
+        low: 0.25,
+        medium: 0.5,
+        high: 0.75,
+        xhigh: 1.0,
+    };
+    const fraction = budgetMap[level] ?? 0.5;
+    const budgetTokens = Math.round(maxTokens * fraction);
+    // OpenAI reasoning effort mapping
+    const effortMap = {
+        minimal: "low",
+        low: "low",
+        medium: "medium",
+        high: "high",
+        xhigh: "high",
+    };
+    if (provider === "anthropic") {
+        return {
+            anthropic: {
+                thinking: { type: "enabled", budgetTokens },
+            },
+        };
+    }
+    if (provider === "openai") {
+        return {
+            openai: {
+                reasoningEffort: effortMap[level] ?? "medium",
+            },
+        };
+    }
+    if (provider === "google") {
+        return {
+            google: {
+                thinkingConfig: {
+                    thinkingBudget: budgetTokens,
+                },
+            },
+        };
+    }
+    // Unknown provider — return undefined as best effort
+    return undefined;
+}
+/**
+ * Build AI SDK compatible options for generateText/streamText calls.
+ *
+ * Returns an object with:
+ * - providerOptions: reasoning/thinking configuration per provider
+ * - maxTokens: max output tokens
+ * - headers: additional headers (e.g. for API key passthrough)
+ */
+export function buildStreamOpts(apiKey, reasoning, maxTokens) {
+    const reasoningVal = reasoning && reasoning !== "off" ? reasoning : undefined;
+    if (!apiKey && !reasoningVal && !maxTokens)
+        return undefined;
+    const opts = {};
+    if (apiKey)
+        opts.apiKey = apiKey;
+    if (reasoningVal)
+        opts.reasoning = reasoningVal;
+    if (maxTokens)
+        opts.maxTokens = maxTokens;
+    return opts;
+}
+//# sourceMappingURL=provider-factory.js.map

package/dist/provider-factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-factory.js","sourceRoot":"","sources":["../src/provider-factory.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE9C,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C;;;GAGG;AACH,MAAM,UAAU,yBAAyB,CACvC,QAAgB,EAChB,OAAe,EACf,QAAwB;IAExB,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,IAAI,2BAA2B,CAAC;IAChE,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,CAAC,iCAAiC;IAErF,iFAAiF;IACjF,yEAAyE;IACzE,MAAM,cAAc,GAAG,YAAY,CAAC;QAClC,OAAO;QACP,MAAM;QACN,IAAI,EAAE,QAAQ;KACf,CAAC,CAAC;IAEH,OAAO,cAAc,CAAC,OAAO,CAA6B,CAAC;AAC7D,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAAC,QAAgB,EAAE,OAAe;IAClE,MAAM,cAAc,GAAG,GAAG,QAAQ,IAAI,OAAO,EAAE,CAAC;IAChD,OAAO,OAAO,CAAC,cAAqB,CAA6B,CAAC;AACpE,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,6BAA6B,CAC3C,QAAgB,EAChB,KAAiC,EACjC,SAAiB;IAEjB,IAAI,CAAC,KAAK,IAAI,KAAK,KAAK,KAAK;QAAE,OAAO,SAAS,CAAC;IAEhD,yEAAyE;IACzE,MAAM,SAAS,GAA2B;QACxC,OAAO,EAAE,GAAG;QACZ,GAAG,EAAE,IAAI;QACT,MAAM,EAAE,GAAG;QACX,IAAI,EAAE,IAAI;QACV,KAAK,EAAE,GAAG;KACX,CAAC;IACF,MAAM,QAAQ,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC;IACzC,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,QAAQ,CAAC,CAAC;IAEtD,kCAAkC;IAClC,MAAM,SAAS,GAA2B;QACxC,OAAO,EAAE,KAAK;QACd,GAAG,EAAE,KAAK;QACV,MAAM,EAAE,QAAQ;QAChB,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,MAAM;KACd,CAAC;IAEF,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QAC7B,OAAO;YACL,SAAS,EAAE;gBACT,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,YAAY,EAAE;aAC5C;SACF,CAAC;IACJ,CAAC;IAED,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,OAAO;YACL,MAAM,EAAE;gBACN,eAAe,EAAE,SAAS,CAAC,KAAK,CAAC,IAAI,QAAQ;aAC9C;SACF,CAAC;IACJ,CAAC;IAED,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,OAAO;YACL,MAAM,EAAE;gBACN,cAAc,EAAE;oBACd,cAAc,EAAE,YAAY;iBAC7B;aACF;SACF,CAAC;IACJ,CAAC;IAED,qDAAqD;IACrD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,eAAe,CAC7B,MAAe,EACf,SAA0B,EAC1B,SAAkB;IAElB,MAAM,YAAY,GAAG,SAAS,IAAI,SAAS,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;IAE9E,IAAI,CAAC,MAAM,IAAI,CAAC,YAAY,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IAE7D,MAAM,IAAI,GAA4B,EAAE,CAAC;IACzC,IAAI,MAAM;QAAE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACjC,IAAI,YAAY;QAAE,IAAI,CAAC,SAAS,GAAG,YAAY,CAAC;IAChD,IAAI,SAAS;QAAE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC1C,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/query.d.ts

@@ -0,0 +1,37 @@
+/**
+ * Query functions — wrappers around AI SDK generateText/streamText
+ * with cooldown integration and provider-level failover.
+ */
+import { type LanguageModelUsage } from "ai";
+import type { ReasoningLevel, ModelConfig } from "@polpo-ai/core";
+import type { ResolvedModel } from "./model-resolver.js";
+/**
+ * Simple prompt -> text completion using AI SDK generateText.
+ * Integrates with the cooldown system: marks provider cooldown on classified errors,
+ * clears cooldown on success. Uses async API key resolution (includes OAuth profiles).
+ */
+export declare function queryText(prompt: string, model?: string, reasoning?: ReasoningLevel): Promise<{
+    text: string;
+    usage?: LanguageModelUsage;
+    model: ResolvedModel;
+}>;
+/**
+ * Streaming prompt -> text with progress callback.
+ * Integrates with the cooldown system. Uses async API key resolution.
+ */
+export declare function queryStream(prompt: string, model?: string, onProgress?: (text: string) => void, reasoning?: ReasoningLevel): Promise<{
+    text: string;
+    usage?: LanguageModelUsage;
+    model: ResolvedModel;
+}>;
+/**
+ * Query with model fallback chain — tries primary model, then fallbacks.
+ * On provider-level errors, marks cooldown and tries next model.
+ */
+export declare function queryTextWithFallback(prompt: string, modelConfig: ModelConfig): Promise<{
+    text: string;
+    usage?: LanguageModelUsage;
+    model: ResolvedModel;
+    usedSpec: string;
+}>;
+//# sourceMappingURL=query.d.ts.map

package/dist/query.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query.d.ts","sourceRoot":"","sources":["../src/query.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAA4B,KAAK,kBAAkB,EAAE,MAAM,IAAI,CAAC;AACvE,OAAO,KAAK,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAGlE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAmBzD;;;;GAIG;AACH,wBAAsB,SAAS,CAC7B,MAAM,EAAE,MAAM,EACd,KAAK,CAAC,EAAE,MAAM,EACd,SAAS,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,kBAAkB,CAAC;IAAC,KAAK,EAAE,aAAa,CAAA;CAAE,CAAC,CAkC7E;AAED;;;GAGG;AACH,wBAAsB,WAAW,CAC/B,MAAM,EAAE,MAAM,EACd,KAAK,CAAC,EAAE,MAAM,EACd,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,EACnC,SAAS,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,kBAAkB,CAAC;IAAC,KAAK,EAAE,aAAa,CAAA;CAAE,CAAC,CA0C7E;AAED;;;GAGG;AACH,wBAAsB,qBAAqB,CACzC,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW,GACvB,OAAO,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,kBAAkB,CAAC;IAAC,KAAK,EAAE,aAAa,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC,CAgC/F"}