@node-llm/core 0.8.0 → 1.1.0

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAC1C,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,YAAY,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AACjD,YAAY,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AAC3C,YAAY,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACzD,YAAY,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AACrD,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAErE,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,GAAG,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AACrE,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,YAAY,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAE3D,OAAO,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AACtE,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAC3E,OAAO,EAAE,sBAAsB,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AACrF,YAAY,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAC3E,cAAc,mBAAmB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,uBAAuB,CAAC;AACtC,cAAc,wBAAwB,CAAC;AACvC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,sBAAsB,CAAC;AACrC,cAAc,uBAAuB,CAAC;AAEtC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AACzE,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,YAAY,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC"}

package/dist/index.js

@@ -1,12 +1,13 @@
-export { Chat } from "./chat/Chat.js";
-export { Stream } from "./chat/Stream.js";
-export { GeneratedImage } from "./image/GeneratedImage.js";
+export * from "./chat/Message.js";
+export * from "./chat/Content.js";
+export * from "./chat/Tool.js";
+export * from "./chat/ChatOptions.js";
+export * from "./chat/ChatResponse.js";
+export * from "./chat/Chat.js";
+export * from "./chat/ChatStream.js";
+export * from "./streaming/Stream.js";
 export { z } from "zod";
-export { LLM, Transcription, Moderation, Embedding } from "./llm.js";
+export { NodeLLM, Transcription, Moderation, Embedding } from "./llm.js";
 export { config } from "./config.js";
 export { providerRegistry } from "./providers/registry.js";
-export { OpenAIProvider } from "./providers/openai/OpenAIProvider.js";
-export { registerOpenAIProvider } from "./providers/openai/index.js";
-export { registerAnthropicProvider } from "./providers/anthropic/index.js";
-export { registerOllamaProvider, OllamaProvider } from "./providers/ollama/index.js";
-export * from "./errors/index.js";
+export { Schema } from "./schema/Schema.js";

package/dist/llm.d.ts

@@ -18,10 +18,9 @@ type LLMConfig = {
     defaultModerationModel?: string;
     defaultEmbeddingModel?: string;
 } & Partial<NodeLLMConfig>;
-declare class LLMCore {
+declare class NodeLLMCore {
     readonly models: typeof ModelRegistry;
     readonly config: {
-        [key: string]: any;
         openaiApiKey?: string;
         openaiApiBase?: string;
         anthropicApiKey?: string;
@@ -31,6 +30,8 @@ declare class LLMCore {
         deepseekApiKey?: string;
         deepseekApiBase?: string;
         ollamaApiBase?: string;
+        openrouterApiKey?: string;
+        openrouterApiBase?: string;
     };
     private provider?;
     private defaultTranscriptionModelId?;
@@ -70,5 +71,5 @@ declare class LLMCore {
     }): Promise<Embedding>;
 }
 export { Transcription, Moderation, Embedding };
-export declare const LLM: LLMCore;
+export declare const NodeLLM: NodeLLMCore;
 //# sourceMappingURL=llm.d.ts.map

package/dist/llm.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llm.d.ts","sourceRoot":"","sources":["../src/llm.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EACL,QAAQ,EACR,SAAS,EAKV,MAAM,yBAAyB,CAAC;AAOjC,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AACxD,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAIrD,OAAO,EAAU,aAAa,EAAE,MAAM,aAAa,CAAC;AAEpD,MAAM,WAAW,YAAY;IAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,KAAK,SAAS,GAAG;IACf,QAAQ,CAAC,EAAE,QAAQ,GAAG,MAAM,CAAC;IAC7B,KAAK,CAAC,EAAE,YAAY,CAAC;IACrB,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,qBAAqB,CAAC,EAAE,MAAM,CAAC;CAChC,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;AAE3B,cAAM,OAAO;IACX,SAAgB,MAAM,uBAAiB;IACvC,SAAgB,MAAM;;;;;;;;;;;MAAU;IAChC,OAAO,CAAC,QAAQ,CAAC,CAAW;IAC5B,OAAO,CAAC,2BAA2B,CAAC,CAAS;IAC7C,OAAO,CAAC,wBAAwB,CAAC,CAAS;IAC1C,OAAO,CAAC,uBAAuB,CAAC,CAAS;IAEzC,OAAO,CAAC,KAAK,CAGX;IAEF,SAAS,CAAC,gBAAgB,EAAE,SAAS,GAAG,CAAC,CAAC,MAAM,EAAE,aAAa,KAAK,IAAI,CAAC;IAqEzE,OAAO,CAAC,qBAAqB;IAU7B,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,IAAI;IAQ1C,UAAU,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;IAKlC,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAC;QAAC,iBAAiB,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,cAAc,CAAC;IAkB1I,UAAU,CACd,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE;QACR,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;QAC7B,iBAAiB,CAAC,EAAE,OAAO,CAAC;KAC7B,GACA,OAAO,CAAC,aAAa,CAAC;IAmBzB,IAAI,yBAAyB,IAAI,MAAM,GAAG,SAAS,CAElD;IAED,IAAI,sBAAsB,IAAI,MAAM,GAAG,SAAS,CAE/C;IAED,IAAI,qBAAqB,IAAI,MAAM,GAAG,SAAS,CAE9C;IAED,cAAc;IAIR,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,iBAAiB,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,UAAU,CAAC;IAmBlH,KAAK,CACT,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,EACxB,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,iBAAiB,CAAC,EAAE,OAAO,CAAA;KAAE,GAC7E,OAAO,CAAC,SAAS,CAAC;CAoBtB;AAED,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;AAEhD,eAAO,MAAM,GAAG,SAAgB,CAAC"}
+{"version":3,"file":"llm.d.ts","sourceRoot":"","sources":["../src/llm.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EACL,QAAQ,EACR,SAAS,EAIV,MAAM,yBAAyB,CAAC;AAUjC,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AACxD,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AASrD,OAAO,EAAU,aAAa,EAAE,MAAM,aAAa,CAAC;AAEpD,MAAM,WAAW,YAAY;IAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,KAAK,SAAS,GAAG;IACf,QAAQ,CAAC,EAAE,QAAQ,GAAG,MAAM,CAAC;IAC7B,KAAK,CAAC,EAAE,YAAY,CAAC;IACrB,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,qBAAqB,CAAC,EAAE,MAAM,CAAC;CAChC,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;AAY3B,cAAM,WAAW;IACf,SAAgB,MAAM,uBAAiB;IACvC,SAAgB,MAAM;;;;;;;;;;;;MAAU;IAChC,OAAO,CAAC,QAAQ,CAAC,CAAW;IAC5B,OAAO,CAAC,2BAA2B,CAAC,CAAS;IAC7C,OAAO,CAAC,wBAAwB,CAAC,CAAS;IAC1C,OAAO,CAAC,uBAAuB,CAAC,CAAS;IAEzC,OAAO,CAAC,KAAK,CAGX;IAEF,SAAS,CAAC,gBAAgB,EAAE,SAAS,GAAG,CAAC,CAAC,MAAM,EAAE,aAAa,KAAK,IAAI,CAAC;IAuDzE,OAAO,CAAC,qBAAqB;IAa7B,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,IAAI;IAU1C,UAAU,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;IAUlC,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAC;QAAC,iBAAiB,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,cAAc,CAAC;IAsB1I,UAAU,CACd,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE;QACR,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;QAC7B,iBAAiB,CAAC,EAAE,OAAO,CAAC;KAC7B,GACA,OAAO,CAAC,aAAa,CAAC;IAoBzB,IAAI,yBAAyB,IAAI,MAAM,GAAG,SAAS,CAElD;IAED,IAAI,sBAAsB,IAAI,MAAM,GAAG,SAAS,CAE/C;IAED,IAAI,qBAAqB,IAAI,MAAM,GAAG,SAAS,CAE9C;IAED,cAAc;IAIR,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,iBAAiB,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,UAAU,CAAC;IAoBlH,KAAK,CACT,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,EACxB,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,iBAAiB,CAAC,EAAE,OAAO,CAAA;KAAE,GAC7E,OAAO,CAAC,SAAS,CAAC;CAqBtB;AAED,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;AAEhD,eAAO,MAAM,OAAO,aAAoB,CAAC"}

package/dist/llm.js

@@ -1,17 +1,23 @@
 import { Chat } from "./chat/Chat.js";
-import { providerRegistry } from "./providers/registry.js";
-import { ensureOpenAIRegistered } from "./providers/openai/index.js";
-import { registerGeminiProvider } from "./providers/gemini/index.js";
-import { registerAnthropicProvider } from "./providers/anthropic/index.js";
-import { registerDeepSeekProvider } from "./providers/deepseek/index.js";
-import { registerOllamaProvider } from "./providers/ollama/index.js";
+import { providerRegistry, ensureOpenAIRegistered, registerAnthropicProvider, registerGeminiProvider, registerDeepSeekProvider, registerOllamaProvider, registerOpenRouterProvider, } from "./providers/registry.js";
 import { GeneratedImage } from "./image/GeneratedImage.js";
 import { ModelRegistry } from "./models/ModelRegistry.js";
 import { Transcription } from "./transcription/Transcription.js";
 import { Moderation } from "./moderation/Moderation.js";
 import { Embedding } from "./embedding/Embedding.js";
+import { ProviderNotConfiguredError, UnsupportedFeatureError, ModelCapabilityError } from "./errors/index.js";
+import { resolveModelAlias } from "./model_aliases.js";
 import { config } from "./config.js";
-class LLMCore {
+// Provider registration map
+const PROVIDER_REGISTRARS = {
+    openai: ensureOpenAIRegistered,
+    gemini: registerGeminiProvider,
+    anthropic: registerAnthropicProvider,
+    deepseek: registerDeepSeekProvider,
+    ollama: registerOllamaProvider,
+    openrouter: registerOpenRouterProvider,
+};
+class NodeLLMCore {
     models = ModelRegistry;
     config = config;
     provider;
@@ -50,20 +56,10 @@ class LLMCore {
             };
         }
         if (typeof provider === "string") {
-            if (provider === "openai") {
-                ensureOpenAIRegistered();
-            }
-            if (provider === "gemini") {
-                registerGeminiProvider();
-            }
-            if (provider === "anthropic") {
-                registerAnthropicProvider();
-            }
-            if (provider === "deepseek") {
-                registerDeepSeekProvider();
-            }
-            if (provider === "ollama") {
-                registerOllamaProvider();
+            // Use the provider registrars map
+            const registrar = PROVIDER_REGISTRARS[provider];
+            if (registrar) {
+                registrar();
             }
             this.provider = providerRegistry.resolve(provider);
         }
@@ -73,51 +69,60 @@ class LLMCore {
     }
     ensureProviderSupport(method) {
         if (!this.provider) {
-            throw new Error("LLM provider not configured");
+            throw new ProviderNotConfiguredError();
         }
         if (!this.provider[method]) {
-            throw new Error(`Provider does not support ${method}`);
+            throw new UnsupportedFeatureError("Provider", String(method));
         }
         return this.provider;
     }
     chat(model, options) {
         if (!this.provider) {
-            throw new Error("LLM provider not configured");
+            throw new ProviderNotConfiguredError();
         }
-        return new Chat(this.provider, model, options);
+        // Resolve model alias based on the current provider
+        const resolvedModel = resolveModelAlias(model, this.provider.id);
+        return new Chat(this.provider, resolvedModel, options, this.retry);
     }
     async listModels() {
         const provider = this.ensureProviderSupport("listModels");
-        return provider.listModels();
+        const models = await provider.listModels();
+        // Dynamically update the model registry with the fetched info
+        ModelRegistry.save(models);
+        return models;
     }
     async paint(prompt, options) {
         const provider = this.ensureProviderSupport("paint");
-        const model = options?.model;
+        // Default to resolving aliases
+        const rawModel = options?.model;
+        const model = resolveModelAlias(rawModel || "", provider.id);
         if (options?.assumeModelExists) {
             console.warn(`[NodeLLM] Skipping validation for model ${model}`);
         }
         else if (model && provider.capabilities && !provider.capabilities.supportsImageGeneration(model)) {
-            throw new Error(`Model ${model} does not support image generation.`);
+            throw new ModelCapabilityError(model, "image generation");
         }
         const response = await provider.paint({
             prompt,
             ...options,
+            model,
         });
         return new GeneratedImage(response);
     }
     async transcribe(file, options) {
         const provider = this.ensureProviderSupport("transcribe");
-        const model = options?.model || this.defaultTranscriptionModelId;
+        const rawModel = options?.model || this.defaultTranscriptionModelId || "";
+        const model = resolveModelAlias(rawModel, provider.id);
         if (options?.assumeModelExists) {
             console.warn(`[NodeLLM] Skipping validation for model ${model}`);
         }
         else if (model && provider.capabilities && !provider.capabilities.supportsTranscription(model)) {
-            throw new Error(`Model ${model} does not support transcription.`);
+            throw new ModelCapabilityError(model, "transcription");
         }
         const response = await provider.transcribe({
             file,
-            model,
             ...options,
+            model,
         });
         return new Transcription(response);
     }
@@ -135,23 +140,25 @@ class LLMCore {
     }
     async moderate(input, options) {
         const provider = this.ensureProviderSupport("moderate");
-        const model = options?.model || this.defaultModerationModelId;
+        const rawModel = options?.model || this.defaultModerationModelId || "";
+        const model = resolveModelAlias(rawModel, provider.id);
         if (options?.assumeModelExists) {
             console.warn(`[NodeLLM] Skipping validation for model ${model}`);
         }
         else if (model && provider.capabilities && !provider.capabilities.supportsModeration(model)) {
-            throw new Error(`Model ${model} does not support moderation.`);
+            throw new ModelCapabilityError(model, "moderation");
         }
         const response = await provider.moderate({
             input,
-            model,
             ...options,
+            model,
         });
         return new Moderation(response);
     }
     async embed(input, options) {
         const provider = this.ensureProviderSupport("embed");
-        const model = options?.model || this.defaultEmbeddingModelId;
+        const rawModel = options?.model || this.defaultEmbeddingModelId || "";
+        const model = resolveModelAlias(rawModel, provider.id);
         const request = {
             input,
             model,
@@ -161,11 +168,11 @@ class LLMCore {
             console.warn(`[NodeLLM] Skipping validation for model ${request.model}`);
         }
         else if (request.model && provider.capabilities && !provider.capabilities.supportsEmbeddings(request.model)) {
-            throw new Error(`Model ${request.model} does not support embeddings.`);
+            throw new ModelCapabilityError(request.model, "embeddings");
         }
         const response = await provider.embed(request);
         return new Embedding(response);
     }
 }
 export { Transcription, Moderation, Embedding };
-export const LLM = new LLMCore();
+export const NodeLLM = new NodeLLMCore();

package/dist/model_aliases.d.ts

@@ -0,0 +1,3 @@
+export type ProviderName = "openai" | "anthropic" | "gemini" | "vertexai" | "openrouter" | "mistral" | "deepseek" | "bedrock" | string;
+export declare function resolveModelAlias(alias: string, provider?: ProviderName): string;
+//# sourceMappingURL=model_aliases.d.ts.map

package/dist/model_aliases.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model_aliases.d.ts","sourceRoot":"","sources":["../src/model_aliases.ts"],"names":[],"mappings":"AAKA,MAAM,MAAM,YAAY,GAAG,QAAQ,GAAG,WAAW,GAAG,QAAQ,GAAG,UAAU,GAAG,YAAY,GAAG,SAAS,GAAG,UAAU,GAAG,SAAS,GAAG,MAAM,CAAC;AAEvI,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,YAAY,GAAG,MAAM,CAkBhF"}

package/dist/model_aliases.js

@@ -0,0 +1,20 @@
+// This file is auto-generated or manually maintained to map generic model aliases
+// to provider-specific model IDs.
+// @ts-ignore
+import aliases from "./aliases.json" assert { type: "json" };
+export function resolveModelAlias(alias, provider) {
+    if (!provider) {
+        return alias;
+    }
+    // Check if the alias exists in our registry
+    const aliasEntry = aliases[alias];
+    if (aliasEntry) {
+        // Check if there is a specific mapping for this provider
+        if (aliasEntry[provider.toLowerCase()]) {
+            return aliasEntry[provider.toLowerCase()];
+        }
+    }
+    // If no alias found or no mapping for this provider, return the original string
+    // This allows users to pass raw model IDs that aren't in our alias list.
+    return alias;
+}

package/dist/models/ModelRegistry.d.ts

@@ -2,6 +2,10 @@ import { Model } from "./types.js";
 export declare class ModelRegistry {
     private static models;
     static find(modelId: string, provider?: string): Model | undefined;
+    /**
+     * Add or update models in the registry.
+     */
+    static save(models: Model | Model[]): void;
     /**
      * Get all available models.
      */

package/dist/models/ModelRegistry.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ModelRegistry.d.ts","sourceRoot":"","sources":["../../src/models/ModelRegistry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAGnC,qBAAa,aAAa;IACtB,OAAO,CAAC,MAAM,CAAC,MAAM,CAA6C;IAElE,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,KAAK,GAAG,SAAS;IAMlE;;OAEG;IACH,MAAM,CAAC,GAAG,IAAI,KAAK,EAAE;IAIrB;;OAEG;IACH,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAKhF;;OAEG;IACH,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO;IAK/E;;OAEG;IACH,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAK9E;;OAEG;IACH,MAAM,CAAC,aAAa,CAAC,KAAK,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,aAAa,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,MAAM,CAAC;QAAC,gBAAgB,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM;sBAA3I,MAAM;uBAAiB,MAAM;sBAAgB,MAAM;wBAAkB,MAAM;2BAAqB,MAAM;;;;;sBAAtG,MAAM;uBAAiB,MAAM;sBAAgB,MAAM;wBAAkB,MAAM;2BAAqB,MAAM;;CA+BrJ"}
+{"version":3,"file":"ModelRegistry.d.ts","sourceRoot":"","sources":["../../src/models/ModelRegistry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAGnC,qBAAa,aAAa;IACtB,OAAO,CAAC,MAAM,CAAC,MAAM,CAA6C;IAElE,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,KAAK,GAAG,SAAS;IAMlE;;OAEG;IACH,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,GAAG,KAAK,EAAE,GAAG,IAAI;IAa1C;;OAEG;IACH,MAAM,CAAC,GAAG,IAAI,KAAK,EAAE;IAIrB;;OAEG;IACH,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAKhF;;OAEG;IACH,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO;IAK/E;;OAEG;IACH,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAK9E;;OAEG;IACH,MAAM,CAAC,aAAa,CAAC,KAAK,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,aAAa,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,MAAM,CAAC;QAAC,gBAAgB,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM;sBAA3I,MAAM;uBAAiB,MAAM;sBAAgB,MAAM;wBAAkB,MAAM;2BAAqB,MAAM;;;;;sBAAtG,MAAM;uBAAiB,MAAM;sBAAgB,MAAM;wBAAkB,MAAM;2BAAqB,MAAM;;CA+BrJ"}

package/dist/models/ModelRegistry.js

@@ -4,6 +4,21 @@ export class ModelRegistry {
     static find(modelId, provider) {
         return this.models.find(m => (m.id === modelId || m.family === modelId) && (!provider || m.provider === provider));
     }
+    /**
+     * Add or update models in the registry.
+     */
+    static save(models) {
+        const toAdd = Array.isArray(models) ? models : [models];
+        toAdd.forEach(newModel => {
+            const index = this.models.findIndex(m => m.id === newModel.id && m.provider === newModel.provider);
+            if (index >= 0) {
+                this.models[index] = newModel;
+            }
+            else {
+                this.models.push(newModel);
+            }
+        });
+    }
     /**
      * Get all available models.
      */