aieo 0.1.37 → 0.1.39

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,
@@ -23258,23 +23266,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);
@@ -23501,7 +23505,8 @@ function getModel(provider, opts) {
         usage: { include: true },
         ...isMoonshot ? { provider: { order: ["moonshotai"], allow_fallbacks: true } } : {}
       };
-      return openrouter(modelId, settings);
+      const gatewayModelId = baseURL ? `openrouter/${modelId}` : modelId;
+      return openrouter(gatewayModelId, settings);
     }
     case "xai": {
       if (baseURL) {
@@ -23744,6 +23749,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");
 
@@ -24560,9 +24654,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,
@@ -24572,6 +24669,7 @@ function consoleLogger(name19) {
   callGenerateObject,
   callGenerateText,
   callModel,
+  canonicalModelName,
   captureNativeFetchResults,
   captureNativeResults,
   computeSessionCost,
@@ -24599,12 +24697,16 @@ function consoleLogger(name19) {
   htmlToText,
   isPrivateAddress,
   linkifyCitations,
+  listModels,
   loadModelPricing,
+  maxOutputTokensFor,
   normalizeApiKey,
   normalizeCallerBaseURL,
   normalizeUsage,
+  parseModelName,
   resolveFetchBackend,
   resolveLLMConfig,
+  resolveModel,
   resolveSearchBackend,
   searchWeb,
   stripCitations,