@node-llm/core 0.7.0 → 1.0.0

package/dist/llm.js

@@ -1,15 +1,21 @@
 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 { 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 { config } from "./config.js";
+// Provider registration map
+const PROVIDER_REGISTRARS = {
+    openai: ensureOpenAIRegistered,
+    gemini: registerGeminiProvider,
+    anthropic: registerAnthropicProvider,
+    deepseek: registerDeepSeekProvider,
+    ollama: registerOllamaProvider,
+    openrouter: registerOpenRouterProvider,
+};
 class LLMCore {
     models = ModelRegistry;
     config = config;
@@ -49,17 +55,10 @@ class LLMCore {
             };
         }
         if (typeof provider === "string") {
-            if (provider === "openai") {
-                ensureOpenAIRegistered();
-            }
-            if (provider === "gemini") {
-                registerGeminiProvider();
-            }
-            if (provider === "anthropic") {
-                registerAnthropicProvider();
-            }
-            if (provider === "deepseek") {
-                registerDeepSeekProvider();
+            // Use the provider registrars map
+            const registrar = PROVIDER_REGISTRARS[provider];
+            if (registrar) {
+                registrar();
             }
             this.provider = providerRegistry.resolve(provider);
         }
@@ -69,22 +68,25 @@ 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);
     }
     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");
@@ -93,7 +95,7 @@ class LLMCore {
             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,
@@ -108,7 +110,7 @@ class LLMCore {
             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,
@@ -136,7 +138,7 @@ class LLMCore {
             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,
@@ -157,7 +159,7 @@ 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);

package/dist/models/ModelRegistry.d.ts

@@ -1,10 +1,11 @@
 import { Model } from "./types.js";
 export declare class ModelRegistry {
     private static models;
+    static find(modelId: string, provider?: string): Model | undefined;
     /**
-     * Find a model by its ID.
+     * Add or update models in the registry.
      */
-    static find(modelId: string, provider?: string): Model | undefined;
+    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;;OAEG;IACH,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

@@ -1,12 +1,24 @@
 import { modelsData } from "./models.js";
 export class ModelRegistry {
     static models = modelsData;
-    /**
-     * Find a model by its ID.
-     */
     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.
      */