aieo 0.1.38 → 0.1.40

package/dist/index.cjs

@@ -30,9 +30,12 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  API_KEY_ENV: () => API_KEY_ENV,
   ConversationStorage: () => ConversationStorage,
+  DEFAULT_MODELS: () => DEFAULT_MODELS,
   GATEWAY_PATHS: () => GATEWAY_PATHS,
   MAX_CONVERSATIONS: () => MAX_CONVERSATIONS,
+  MODELS: () => MODELS,
   OUTRO: () => OUTRO,
   PROVIDERS: () => PROVIDERS,
   SYSTEM: () => SYSTEM,
@@ -42,6 +45,7 @@ __export(index_exports, {
   callGenerateObject: () => callGenerateObject,
   callGenerateText: () => callGenerateText,
   callModel: () => callModel,
+  canonicalModelName: () => canonicalModelName,
   captureNativeFetchResults: () => captureNativeFetchResults,
   captureNativeResults: () => captureNativeResults,
   computeSessionCost: () => computeSessionCost,
@@ -69,12 +73,16 @@ __export(index_exports, {
   htmlToText: () => htmlToText,
   isPrivateAddress: () => isPrivateAddress,
   linkifyCitations: () => linkifyCitations,
+  listModels: () => listModels,
   loadModelPricing: () => loadModelPricing,
+  maxOutputTokensFor: () => maxOutputTokensFor,
   normalizeApiKey: () => normalizeApiKey,
   normalizeCallerBaseURL: () => normalizeCallerBaseURL,
   normalizeUsage: () => normalizeUsage,
+  parseModelName: () => parseModelName,
   resolveFetchBackend: () => resolveFetchBackend,
   resolveLLMConfig: () => resolveLLMConfig,
+  resolveModel: () => resolveModel,
   resolveSearchBackend: () => resolveSearchBackend,
   searchWeb: () => searchWeb,
   stripCitations: () => stripCitations,
@@ -23167,20 +23175,20 @@ var PROVIDERS = [
 var MODELS = {
   anthropic: {
     sonnet: "claude-sonnet-5",
-    opus: "claude-opus-4-6",
+    opus: "claude-opus-5",
     haiku: "claude-haiku-4-5"
   },
   google: {
     gemini: "gemini-3-pro-preview"
   },
   openai: {
-    gpt: "gpt-5"
+    gpt: "gpt-5.6-luna"
   },
   openrouter: {
-    kimi: "moonshotai/kimi-k2.6"
+    kimi: "moonshotai/kimi-k3"
   },
   xai: {
-    grok: "grok-4"
+    grok: "grok-4.6"
   }
 };
 var DEFAULT_MODELS = {
@@ -23234,6 +23242,7 @@ function getProviderForModel(modelName) {
       return "xai";
     // Full model IDs
     case "claude-sonnet-5":
+    case "claude-opus-5":
     case "claude-opus-4-6":
     case "claude-haiku-4-5":
       return "anthropic";
@@ -23241,6 +23250,10 @@ function getProviderForModel(modelName) {
     case "gemini-2.0-flash":
       return "google";
     case "gpt-5":
+    case "gpt-5.5":
+    case "gpt-5.6-luna":
+    case "gpt-5.6-sol":
+    case "gpt-5.6-terra":
     case "gpt-4.1-mini":
       return "openai";
     default:
@@ -23250,6 +23263,11 @@ function getProviderForModel(modelName) {
       if (process.env.LLM_PROVIDER && PROVIDERS.includes(process.env.LLM_PROVIDER)) {
         return process.env.LLM_PROVIDER;
       }
+      if (typeof modelName === "string") {
+        const m = modelName.toLowerCase();
+        if (m.startsWith("gpt-")) return "openai";
+        if (m.startsWith("gemini-")) return "google";
+      }
       return "anthropic";
   }
 }
@@ -23258,23 +23276,19 @@ function normalizeApiKey(key) {
   const trimmed = key.trim();
   return trimmed.length > 0 ? trimmed : void 0;
 }
+var API_KEY_ENV = {
+  anthropic: "ANTHROPIC_API_KEY",
+  google: "GOOGLE_API_KEY",
+  openai: "OPENAI_API_KEY",
+  openrouter: "OPENROUTER_API_KEY",
+  xai: "XAI_API_KEY"
+};
 function lookupApiKeyForProvider(provider) {
-  switch (provider) {
-    case "anthropic":
-      return normalizeApiKey(process.env.ANTHROPIC_API_KEY);
-    case "google":
-      return normalizeApiKey(process.env.GOOGLE_API_KEY);
-    case "openai":
-      return normalizeApiKey(process.env.OPENAI_API_KEY);
-    case "openrouter":
-      return normalizeApiKey(process.env.OPENROUTER_API_KEY);
-    case "xai":
-      return normalizeApiKey(process.env.XAI_API_KEY);
-    case "claude_code":
-      return normalizeApiKey(process.env.CLAUDE_CODE_API_KEY);
-    default:
-      return void 0;
+  if (provider === "claude_code") {
+    return normalizeApiKey(process.env.CLAUDE_CODE_API_KEY);
   }
+  if (!PROVIDERS.includes(provider)) return void 0;
+  return normalizeApiKey(process.env[API_KEY_ENV[provider]]);
 }
 function hasApiKeyForProvider(provider) {
   return !!lookupApiKeyForProvider(provider);
@@ -23547,23 +23561,26 @@ function getModel(provider, opts) {
   }
 }
 var MODEL_CONTEXT_LIMITS = {
-  // Anthropic
+  // Anthropic — 1M is the default and the max on the 5-series; Haiku 4.5 is 200k.
   "claude-sonnet-5": 1e6,
+  "claude-opus-5": 1e6,
   "claude-opus-4-6": 1e6,
   "claude-haiku-4-5": 2e5,
-  // Google
-  "gemini-3-pro-preview": 1e6,
-  "gemini-2.0-flash": 1e6,
-  // OpenAI
+  // Google — Gemini's "1M" is 2^20.
+  "gemini-3-pro-preview": 1048576,
+  "gemini-2.0-flash": 1048576,
+  // OpenAI — 5.4+ is 1,050,000; the 4.1 family is 1,047,576; 5 through 5.3 are 400k.
   "gpt-5": 4e5,
   "gpt-5.5": 105e4,
   "gpt-5.6-terra": 105e4,
   "gpt-5.6-luna": 105e4,
   "gpt-5.6-sol": 105e4,
-  "gpt-4.1-mini": 1e6,
+  "gpt-4.1-mini": 1047576,
   // OpenRouter — values from the OpenRouter model catalog
   // (https://openrouter.ai/api/v1/models, context_length).
   "stealth/ox-alpha": 1048576,
+  "anthropic/claude-opus-5": 1e6,
+  "anthropic/claude-sonnet-5": 1e6,
   "openai/gpt-5": 4e5,
   "openai/gpt-5.5": 105e4,
   "openai/gpt-5.6-terra": 105e4,
@@ -23579,11 +23596,17 @@ var MODEL_CONTEXT_LIMITS = {
   // right matters beyond display: contextLimit drives truncateOldToolResults,
   // and premature truncation rewrites old messages — which invalidates Grok's
   // automatic prefix cache on every subsequent step.
+  "grok-4.6": 5e5,
+  "grok-4.5": 5e5,
+  "grok-4.3": 1e6,
   "grok-4": 256e3,
   "grok-4-fast": 2e6,
   "grok-4-fast-reasoning": 2e6,
   "grok-4-fast-non-reasoning": 2e6,
   "grok-code-fast-1": 256e3,
+  "x-ai/grok-4.6": 5e5,
+  "x-ai/grok-4.5": 5e5,
+  "x-ai/grok-4.3": 1e6,
   "x-ai/grok-4": 256e3,
   "x-ai/grok-4-fast": 2e6,
   "x-ai/grok-code-fast-1": 256e3
@@ -23745,6 +23768,95 @@ function resolveLLMConfig(opts) {
   return { provider, apiKey, model, modelName: effectiveModelName };
 }
 
+// src/resolve.ts
+function listModels() {
+  const out = [];
+  for (const provider of PROVIDERS) {
+    for (const [alias, modelId] of Object.entries(MODELS[provider])) {
+      if (!modelId) continue;
+      out.push({
+        provider,
+        alias,
+        modelId,
+        default: DEFAULT_MODELS[provider] === modelId
+      });
+    }
+  }
+  return out;
+}
+function parseModelName(name19) {
+  if (!name19) return {};
+  const slash = name19.indexOf("/");
+  if (slash > 0) {
+    const head = name19.slice(0, slash);
+    if (PROVIDERS.includes(head)) {
+      return { provider: head, modelId: name19.slice(slash + 1) || void 0 };
+    }
+  }
+  return { modelId: name19 };
+}
+var ALIASES = new Set(Object.values(MODELS).flatMap((m) => Object.keys(m)));
+function envProvider() {
+  const v = process.env.LLM_PROVIDER;
+  return v && PROVIDERS.includes(v) ? v : void 0;
+}
+function canonicalModelName(model, provider) {
+  const parsed = parseModelName(model);
+  const chosen = provider ?? parsed.provider ?? getProviderForModel(model);
+  if (!PROVIDERS.includes(chosen)) {
+    throw new Error(`Unknown LLM provider: "${chosen}". Supported: ${PROVIDERS.join(", ")}`);
+  }
+  const p = chosen;
+  if (model && !provider && !parsed.provider && model.includes("/") && !envProvider()) {
+    const head = model.split("/")[0];
+    throw new Error(
+      `Model "${model}" has no provider prefix: "${head}" is not a provider (${PROVIDERS.join(", ")}). For an OpenRouter-hosted model write "openrouter/${model}"; otherwise pass the provider explicitly.`
+    );
+  }
+  let modelId = parsed.modelId;
+  if (modelId && ALIASES.has(modelId)) {
+    const hit = MODELS[p][modelId];
+    if (!hit) {
+      throw new Error(
+        `Model "${modelId}" is not available for provider "${p}". Available models: ${Object.keys(MODELS[p]).join(", ")}`
+      );
+    }
+    modelId = hit;
+  }
+  if (!modelId) modelId = DEFAULT_MODELS[p];
+  return { provider: p, modelId, name: `${p}/${modelId}` };
+}
+function maxOutputTokensFor(provider) {
+  const env = Number(process.env.MAX_OUTPUT_TOKENS);
+  if (env > 0) return env;
+  return provider === "anthropic" ? 128e3 : 64e3;
+}
+async function resolveModel(opts = {}) {
+  const ref = canonicalModelName(opts.model, opts.provider);
+  const envName = API_KEY_ENV[ref.provider];
+  const apiKey = normalizeApiKey(opts.apiKey) ?? normalizeApiKey(await opts.getSecret?.(envName)) ?? normalizeApiKey(process.env[envName]);
+  if (!apiKey) {
+    throw new Error(
+      `No API key for provider "${ref.provider}": set ${envName} (or pass apiKey / getSecret)`
+    );
+  }
+  const model = getModel(ref.provider, {
+    modelName: ref.name,
+    apiKey,
+    baseUrl: opts.baseUrl,
+    headers: opts.headers,
+    abortSignal: opts.abortSignal,
+    timeoutMs: opts.timeoutMs
+  });
+  return {
+    ...ref,
+    model,
+    apiKey,
+    contextLimit: getContextLimit(ref.modelId, ref.provider),
+    maxOutputTokens: maxOutputTokensFor(ref.provider)
+  };
+}
+
 // src/stream.ts
 var import_ai = require("ai");
 
@@ -24561,9 +24673,12 @@ function consoleLogger(name19) {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  API_KEY_ENV,
   ConversationStorage,
+  DEFAULT_MODELS,
   GATEWAY_PATHS,
   MAX_CONVERSATIONS,
+  MODELS,
   OUTRO,
   PROVIDERS,
   SYSTEM,
@@ -24573,6 +24688,7 @@ function consoleLogger(name19) {
   callGenerateObject,
   callGenerateText,
   callModel,
+  canonicalModelName,
   captureNativeFetchResults,
   captureNativeResults,
   computeSessionCost,
@@ -24600,12 +24716,16 @@ function consoleLogger(name19) {
   htmlToText,
   isPrivateAddress,
   linkifyCitations,
+  listModels,
   loadModelPricing,
+  maxOutputTokensFor,
   normalizeApiKey,
   normalizeCallerBaseURL,
   normalizeUsage,
+  parseModelName,
   resolveFetchBackend,
   resolveLLMConfig,
+  resolveModel,
   resolveSearchBackend,
   searchWeb,
   stripCitations,