@node-llm/core 1.15.2 → 1.16.0

package/dist/providers/bedrock/Capabilities.js

@@ -3,229 +3,108 @@
  */
 import { ModelRegistry } from "../../models/ModelRegistry.js";
 import { PricingRegistry } from "../../models/PricingRegistry.js";
-const MODEL_FAMILIES = [
-    [/anthropic\.claude-3-opus/, "claude3_opus"],
-    [/anthropic\.claude-3-sonnet/, "claude3_sonnet"],
-    [/anthropic\.claude-3-5-sonnet/, "claude3_sonnet"],
-    [/anthropic\.claude-3-7-sonnet/, "claude3_sonnet"],
-    [/anthropic\.claude-3-haiku/, "claude3_haiku"],
-    [/anthropic\.claude-3-5-haiku/, "claude3_5_haiku"],
-    [/anthropic\.claude-v2/, "claude2"],
-    [/anthropic\.claude-2/, "claude2"],
-    [/anthropic\.claude-instant/, "claude_instant"],
-    [/anthropic\.claude-(opus|sonnet|haiku)-4/, "claude4"],
-    [/amazon\.nova-2/, "nova2"],
-    [/amazon\.nova/, "nova"]
-];
 // ─────────────────────────────────────────────────────────────────────────────
-// Pricing (per million tokens)
-// ─────────────────────────────────────────────────────────────────────────────
-const PRICES = {
-    claude3_opus: { input: 15.0, output: 75.0 },
-    claude3_sonnet: { input: 3.0, output: 15.0 },
-    claude3_haiku: { input: 0.25, output: 1.25 },
-    claude3_5_haiku: { input: 0.8, output: 4.0 },
-    claude2: { input: 8.0, output: 24.0 },
-    claude_instant: { input: 0.8, output: 2.4 },
-    claude4: { input: 3.0, output: 15.0 }, // Assuming Sonnet 4 as baseline for the family
-    nova: { input: 0.06, output: 0.24 },
-    nova2: { input: 0.03, output: 0.12 }, // Assuming Nova 2 is cheaper as per trend
-    other: { input: 0.1, output: 0.2 }
-};
-// ─────────────────────────────────────────────────────────────────────────────
-// Capabilities Class
+// Model Family Patterns
 // ─────────────────────────────────────────────────────────────────────────────
 export class Capabilities {
-    static getModelFamily(modelId) {
-        for (const [pattern, family] of MODEL_FAMILIES) {
-            if (pattern.test(modelId))
-                return family;
-        }
-        return "other";
-    }
     static getContextWindow(modelId) {
-        const val = ModelRegistry.getContextWindow(modelId, "bedrock");
-        if (val)
-            return val;
-        // Claude 2 has 100k, others have 200k
-        if (/anthropic\.claude-2/.test(modelId))
-            return 100_000;
-        if (/anthropic\.claude/.test(modelId))
-            return 200_000;
-        // DeepSeek
-        if (/deepseek/.test(modelId))
-            return 128_000;
-        // Mistral
-        if (/mistral-large/.test(modelId))
-            return 128_000;
-        if (/mistral/.test(modelId))
-            return 32_000;
-        // Llama
-        if (/llama/.test(modelId))
-            return 128_000;
-        // Titan
-        if (/titan/.test(modelId))
-            return 32_000;
-        // Nova
-        if (/nova/.test(modelId))
-            return 300_000;
-        return null;
+        return ModelRegistry.getContextWindow(modelId, "bedrock") ?? 128_000;
     }
     static getMaxOutputTokens(modelId) {
-        const val = ModelRegistry.getMaxOutputTokens(modelId, "bedrock");
-        if (val)
-            return val;
-        return 4_096;
+        return ModelRegistry.getMaxOutputTokens(modelId, "bedrock") ?? 4_096;
     }
     static supportsChat(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("chat"))
-            return true;
-        return (/anthropic\.claude/.test(modelId) ||
-            /amazon\.nova/.test(modelId) ||
-            /mistral\.mistral/.test(modelId) ||
-            /meta\.llama/.test(modelId) ||
-            /deepseek/.test(modelId));
+        return ModelRegistry.supports(modelId, "chat", "bedrock") || !modelId.includes("titan-embed");
     }
     static supportsStreaming(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("streaming"))
-            return true;
-        return (/anthropic\.claude/.test(modelId) ||
-            /amazon\.nova/.test(modelId) ||
-            /mistral\.mistral/.test(modelId) ||
-            /meta\.llama/.test(modelId) ||
-            /deepseek/.test(modelId));
+        return (ModelRegistry.supports(modelId, "streaming", "bedrock") || !modelId.includes("titan-embed"));
     }
     static supportsVision(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.modalities?.input?.includes("image"))
-            return true;
-        if (model?.capabilities?.includes("vision"))
-            return true;
-        return /anthropic\.claude-3|anthropic\.claude-4|amazon\.nova/.test(modelId);
+        return (ModelRegistry.supports(modelId, "vision", "bedrock") ||
+            modelId.includes("vision") ||
+            modelId.includes("nova"));
     }
     static supportsTools(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("tools") || model?.capabilities?.includes("function_calling"))
-            return true;
-        return /anthropic\.claude|amazon\.nova|deepseek-chat/.test(modelId);
+        return (ModelRegistry.supports(modelId, "tools", "bedrock") ||
+            ModelRegistry.supports(modelId, "function_calling", "bedrock"));
     }
     static supportsJsonMode(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("json_mode") ||
-            model?.capabilities?.includes("structured_output"))
-            return true;
-        return /anthropic\.claude|amazon\.nova/.test(modelId);
+        return (ModelRegistry.supports(modelId, "json_mode", "bedrock") ||
+            ModelRegistry.supports(modelId, "structured_output", "bedrock"));
     }
     static supportsExtendedThinking(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("reasoning"))
-            return true;
-        return /claude-3-7|deepseek-reasoner|deepseek\.r1/.test(modelId);
+        return ModelRegistry.supports(modelId, "reasoning", "bedrock");
     }
     static supportsEmbeddings(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("embeddings"))
-            return true;
-        return /amazon\.titan-embed/.test(modelId);
+        return (ModelRegistry.supports(modelId, "embeddings", "bedrock") || modelId.includes("titan-embed"));
     }
     static supportsImageGeneration(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("image_generation"))
-            return true;
-        return /amazon\.titan-image-generator|stability\.stable-diffusion/.test(modelId);
-    }
-    /**
-     * Check if a model supports audio input.
-     */
-    static supportsAudio(_modelId) {
-        return false;
-    }
-    /**
-     * Get input modalities for a model.
-     */
+        return (ModelRegistry.supports(modelId, "image_generation", "bedrock") ||
+            modelId.includes("image-generator") ||
+            modelId.includes("stable-diffusion"));
+    }
+    static supportsToolChoice(_modelId) {
+        return true;
+    }
     static getInputModalities(modelId) {
+        const model = this.findModel(modelId);
+        if (model?.modalities?.input)
+            return model.modalities.input;
         const modalities = ["text"];
-        if ((/anthropic\.claude/.test(modelId) || /amazon\.nova/.test(modelId)) &&
-            this.supportsVision(modelId)) {
-            modalities.push("image");
-            modalities.push("pdf");
+        if (this.supportsVision(modelId)) {
+            modalities.push("image", "pdf");
         }
         return modalities;
     }
-    /**
-     * Get output modalities for a model.
-     */
-    static getOutputModalities(_modelId) {
-        return ["text"];
+    static getOutputModalities(modelId) {
+        const model = this.findModel(modelId);
+        return model?.modalities?.output || ["text"];
     }
-    /**
-     * Get all capabilities for a model.
-     */
     static getCapabilities(modelId) {
-        const capabilities = [];
-        if (/anthropic\.claude/.test(modelId)) {
-            capabilities.push("streaming");
+        const model = this.findModel(modelId);
+        if (model?.capabilities) {
+            const caps = [...model.capabilities];
+            if (!caps.includes("streaming"))
+                caps.push("streaming");
+            return caps;
         }
-        if (this.supportsTools(modelId)) {
+        const capabilities = ["streaming"];
+        if (this.supportsTools(modelId))
             capabilities.push("function_calling");
-        }
-        if (/claude-3-7|nova/.test(modelId)) {
+        if (this.supportsExtendedThinking(modelId))
             capabilities.push("reasoning");
-        }
-        if (/claude-3\.5|claude-3-7|nova/.test(modelId)) {
-            capabilities.push("batch");
-            capabilities.push("citations");
-        }
         return capabilities;
     }
     static getPricing(modelId) {
-        // Try registry first
-        const registryPricing = PricingRegistry.getPricing(modelId, "bedrock");
-        if (registryPricing)
-            return registryPricing;
-        // Fallback to built-in pricing
-        const family = this.getModelFamily(modelId);
-        const prices = PRICES[family];
-        return {
-            text_tokens: {
-                standard: {
-                    input_per_million: prices.input,
-                    output_per_million: prices.output
-                },
-                batch: {
-                    input_per_million: prices.input * 0.5,
-                    output_per_million: prices.output * 0.5
-                }
-            }
-        };
-    }
-    /**
-     * Get input price per million tokens.
-     */
-    static getInputPrice(modelId) {
-        const family = this.getModelFamily(modelId);
-        return PRICES[family].input;
-    }
-    /**
-     * Get output price per million tokens.
-     */
-    static getOutputPrice(modelId) {
-        const family = this.getModelFamily(modelId);
-        return PRICES[family].output;
-    }
-    /**
-     * Format model ID as display name.
-     */
+        return PricingRegistry.getPricing(modelId, "bedrock");
+    }
+    static getModelFamily(modelId) {
+        const model = this.findModel(modelId);
+        if (model?.family)
+            return model.family;
+        if (modelId.includes("claude"))
+            return "claude";
+        if (modelId.includes("mistral"))
+            return "mistral";
+        if (modelId.includes("llama"))
+            return "llama";
+        if (modelId.includes("titan"))
+            return "titan";
+        if (modelId.includes("nova"))
+            return "nova";
+        if (modelId.includes("deepseek"))
+            return "deepseek";
+        return "other";
+    }
     static formatDisplayName(modelId) {
+        const model = this.findModel(modelId);
+        if (model?.name && model.name !== modelId)
+            return model.name;
         return modelId
             .replace(/-/g, " ")
             .split(".")
             .pop()
-            .split(" ")
-            .map((word) => word.charAt(0).toUpperCase() + word.slice(1))
-            .join(" ");
+            .replace(/\b\w/g, (c) => c.toUpperCase());
     }
     static findModel(modelId) {
         return ModelRegistry.find(modelId, "bedrock");

package/dist/providers/bedrock/Chat.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock/Chat.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,WAAW,EAAE,YAAY,EAAS,MAAM,gBAAgB,CAAC;AAElE,OAAO,EAAE,aAAa,EAA6C,MAAM,aAAa,CAAC;AAavF,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAgB;IACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqB;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAErB,MAAM,EAAE,aAAa;IAMjC;;OAEG;IACG,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAwD1D;;OAEG;IACH,OAAO,CAAC,YAAY;IA6CpB;;OAEG;IACH,OAAO,CAAC,aAAa;CAiDtB"}
+{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock/Chat.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,WAAW,EAAE,YAAY,EAAS,MAAM,gBAAgB,CAAC;AAElE,OAAO,EAAE,aAAa,EAA6C,MAAM,aAAa,CAAC;AAavF,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAgB;IACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqB;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAErB,MAAM,EAAE,aAAa;IAMjC;;OAEG;IACG,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAyD1D;;OAEG;IACH,OAAO,CAAC,YAAY;IA6CpB;;OAEG;IACH,OAAO,CAAC,aAAa;CAiDtB"}

package/dist/providers/bedrock/Chat.js

@@ -40,6 +40,7 @@ export class BedrockChat {
             temperature: request.temperature,
             thinking: request.thinking,
             guardrail,
+            toolChoice: request.tool_choice,
             additionalModelRequestFields: request.additionalModelRequestFields
         });
         const bodyJson = JSON.stringify(body);

package/dist/providers/bedrock/Streaming.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Streaming.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock/Streaming.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,SAAS,EAAS,MAAM,gBAAgB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAA6C,MAAM,aAAa,CAAC;AAQvF,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAgB;IACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqB;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAErB,MAAM,EAAE,aAAa;IAM1B,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,CAAC,EAAE,eAAe,GAAG,cAAc,CAAC,SAAS,CAAC;IA4H7F;;OAEG;IACH,OAAO,CAAC,UAAU;IAyGlB,OAAO,CAAC,YAAY;CAsCrB"}
+{"version":3,"file":"Streaming.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock/Streaming.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,SAAS,EAAS,MAAM,gBAAgB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAA6C,MAAM,aAAa,CAAC;AAQvF,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAgB;IACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqB;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAErB,MAAM,EAAE,aAAa;IAM1B,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,CAAC,EAAE,eAAe,GAAG,cAAc,CAAC,SAAS,CAAC;IA6H7F;;OAEG;IACH,OAAO,CAAC,UAAU;IAyGlB,OAAO,CAAC,YAAY;CAsCrB"}

package/dist/providers/bedrock/Streaming.js

@@ -39,6 +39,7 @@ export class BedrockStreaming {
             temperature: request.temperature,
             thinking: request.thinking,
             guardrail,
+            toolChoice: request.tool_choice,
             additionalModelRequestFields: request.additionalModelRequestFields
         });
         const bodyJson = JSON.stringify(body);

package/dist/providers/bedrock/mapper.d.ts

@@ -6,8 +6,8 @@
  */
 import { Message } from "../../chat/Message.js";
 import { ToolDefinition } from "../../chat/Tool.js";
-import { BedrockMessage, BedrockContentBlock, BedrockConverseRequest, BedrockToolConfig } from "./types.js";
-import { ThinkingConfig } from "../../providers/Provider.js";
+import { BedrockToolConfig, BedrockMessage, BedrockContentBlock, BedrockConverseRequest } from "./types.js";
+import { ThinkingConfig, ToolChoice } from "../../providers/Provider.js";
 /**
  * Convert NodeLLM messages to Bedrock Converse format.
  * Extracts system messages to the top-level system field.
@@ -19,7 +19,7 @@ export declare function convertMessages(messages: Message[]): {
 /**
  * Convert NodeLLM tool definitions to Bedrock tool config.
  */
-export declare function convertTools(tools?: ToolDefinition[]): BedrockToolConfig | undefined;
+export declare function convertTools(tools?: ToolDefinition[], toolChoice?: ToolChoice): BedrockToolConfig | undefined;
 /**
  * Build a complete Bedrock Converse request from NodeLLM ChatRequest.
  */
@@ -32,6 +32,7 @@ export declare function buildConverseRequest(messages: Message[], tools?: ToolDe
         guardrailVersion: string;
         trace?: "enabled" | "disabled";
     };
+    toolChoice?: ToolChoice;
     additionalModelRequestFields?: Record<string, any>;
 }): BedrockConverseRequest;
 //# sourceMappingURL=mapper.d.ts.map

package/dist/providers/bedrock/mapper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mapper.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock/mapper.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,uBAAuB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EACL,cAAc,EACd,mBAAmB,EACnB,sBAAsB,EACtB,iBAAiB,EAClB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAiD7D;;;GAGG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG;IACpD,QAAQ,EAAE,cAAc,EAAE,CAAC;IAC3B,MAAM,CAAC,EAAE,mBAAmB,EAAE,CAAC;CAChC,CAuFA;AAMD;;GAEG;AACH,wBAAgB,YAAY,CAAC,KAAK,CAAC,EAAE,cAAc,EAAE,GAAG,iBAAiB,GAAG,SAAS,CAgBpF;AAMD;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,CAAC,EAAE,cAAc,EAAE,EACxB,OAAO,CAAC,EAAE;IACR,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,cAAc,CAAC;IAC1B,SAAS,CAAC,EAAE;QACV,mBAAmB,EAAE,MAAM,CAAC;QAC5B,gBAAgB,EAAE,MAAM,CAAC;QACzB,KAAK,CAAC,EAAE,SAAS,GAAG,UAAU,CAAC;KAChC,CAAC;IACF,4BAA4B,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACpD,GACA,sBAAsB,CAmDxB"}
+{"version":3,"file":"mapper.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock/mapper.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,uBAAuB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EACL,iBAAiB,EAEjB,cAAc,EACd,mBAAmB,EACnB,sBAAsB,EACvB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AAiDzE;;;GAGG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG;IACpD,QAAQ,EAAE,cAAc,EAAE,CAAC;IAC3B,MAAM,CAAC,EAAE,mBAAmB,EAAE,CAAC;CAChC,CAuFA;AAMD;;GAEG;AACH,wBAAgB,YAAY,CAC1B,KAAK,CAAC,EAAE,cAAc,EAAE,EACxB,UAAU,CAAC,EAAE,UAAU,GACtB,iBAAiB,GAAG,SAAS,CA8B/B;AAMD;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,CAAC,EAAE,cAAc,EAAE,EACxB,OAAO,CAAC,EAAE;IACR,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,cAAc,CAAC;IAC1B,SAAS,CAAC,EAAE;QACV,mBAAmB,EAAE,MAAM,CAAC;QAC5B,gBAAgB,EAAE,MAAM,CAAC;QACzB,KAAK,CAAC,EAAE,SAAS,GAAG,UAAU,CAAC;KAChC,CAAC;IACF,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,4BAA4B,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACpD,GACA,sBAAsB,CAmDxB"}

package/dist/providers/bedrock/mapper.js

@@ -136,11 +136,11 @@ export function convertMessages(messages) {
 /**
  * Convert NodeLLM tool definitions to Bedrock tool config.
  */
-export function convertTools(tools) {
-    if (!tools || tools.length === 0) {
+export function convertTools(tools, toolChoice) {
+    if (!tools || tools.length === 0 || toolChoice === "none") {
         return undefined;
     }
-    return {
+    const config = {
         tools: tools.map((tool) => ({
             toolSpec: {
                 name: tool.function.name,
@@ -151,6 +151,21 @@ export function convertTools(tools) {
             }
         }))
     };
+    if (toolChoice) {
+        if (toolChoice === "auto") {
+            config.toolChoice = { auto: {} };
+        }
+        else if (toolChoice === "required") {
+            config.toolChoice = { any: {} };
+        }
+        else if (typeof toolChoice === "string") {
+            config.toolChoice = { tool: { name: toolChoice } };
+        }
+        else if (typeof toolChoice === "object" && "function" in toolChoice) {
+            config.toolChoice = { tool: { name: toolChoice.function.name } };
+        }
+    }
+    return config;
 }
 // ─────────────────────────────────────────────────────────────────────────────
 // Request Builder
@@ -166,7 +181,7 @@ export function buildConverseRequest(messages, tools, options) {
     if (system) {
         request.system = system;
     }
-    const toolConfig = convertTools(tools);
+    const toolConfig = convertTools(tools, options?.toolChoice);
     if (toolConfig) {
         request.toolConfig = toolConfig;
     }

package/dist/providers/deepseek/Capabilities.d.ts

@@ -1,7 +1,7 @@
 import { ModelPricing } from "../../models/types.js";
 export declare class Capabilities {
     static getCapabilities(modelId: string): string[];
-    static getContextWindow(modelId: string): number | null;
+    static getContextWindow(modelId: string): number;
     static getMaxOutputTokens(modelId: string): number | null;
     static supportsVision(modelId: string): boolean;
     static supportsTools(modelId: string): boolean;
@@ -11,6 +11,7 @@ export declare class Capabilities {
     static supportsTranscription(_modelId: string): boolean;
     static supportsModeration(_modelId: string): boolean;
     static supportsReasoning(modelId: string): boolean;
+    static supportsToolChoice(_modelId: string): boolean;
     static getPricing(modelId: string): ModelPricing | undefined;
     private static findModel;
 }

package/dist/providers/deepseek/Capabilities.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/deepseek/Capabilities.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAErD,qBAAa,YAAY;IACvB,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAcjD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAUvD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAOzD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAK/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQ9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOzD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKnD,MAAM,CAAC,uBAAuB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIzD,MAAM,CAAC,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIvD,MAAM,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIpD,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOlD,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAI5D,OAAO,CAAC,MAAM,CAAC,SAAS;CAGzB"}
+{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/deepseek/Capabilities.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAErD,qBAAa,YAAY;IACvB,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAejD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAIhD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAIzD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAI/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQ9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAMzD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKnD,MAAM,CAAC,uBAAuB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIzD,MAAM,CAAC,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIvD,MAAM,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIpD,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIlD,MAAM,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIpD,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAI5D,OAAO,CAAC,MAAM,CAAC,SAAS;CAGzB"}

package/dist/providers/deepseek/Capabilities.js

@@ -2,49 +2,40 @@ import { ModelRegistry } from "../../models/ModelRegistry.js";
 import { PricingRegistry } from "../../models/PricingRegistry.js";
 export class Capabilities {
     static getCapabilities(modelId) {
+        const model = this.findModel(modelId);
+        if (model?.capabilities) {
+            const caps = [...model.capabilities];
+            if (!caps.includes("streaming"))
+                caps.push("streaming");
+            return caps;
+        }
         const caps = ["streaming"];
-        if (/deepseek-chat/.test(modelId)) {
+        if (this.supportsTools(modelId))
             caps.push("function_calling");
-        }
-        if (/deepseek-reasoner/.test(modelId)) {
+        if (this.supportsReasoning(modelId))
             caps.push("reasoning");
-        }
         return caps;
     }
     static getContextWindow(modelId) {
-        const val = ModelRegistry.getContextWindow(modelId, "deepseek");
-        if (val)
-            return val;
-        if (/deepseek-(?:chat|reasoner)/.test(modelId)) {
-            return 128_000;
-        }
-        return 32_768;
+        return ModelRegistry.getContextWindow(modelId, "deepseek") ?? 32_768;
     }
     static getMaxOutputTokens(modelId) {
-        if (/deepseek-(?:chat|reasoner)/.test(modelId)) {
-            return 8_192;
-        }
-        return 4_096;
+        return ModelRegistry.getMaxOutputTokens(modelId, "deepseek") ?? 8_192;
     }
     static supportsVision(modelId) {
-        const model = this.findModel(modelId);
-        return model?.modalities?.input?.includes("image") || false;
+        return ModelRegistry.supports(modelId, "vision", "deepseek");
     }
     static supportsTools(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("function_calling") || model?.capabilities?.includes("tools"))
-            return true;
-        return /deepseek-chat/.test(modelId);
+        return (ModelRegistry.supports(modelId, "tools", "deepseek") ||
+            ModelRegistry.supports(modelId, "function_calling", "deepseek") ||
+            modelId.includes("chat"));
     }
     static supportsStructuredOutput(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("structured_output"))
-            return true;
-        return /deepseek-chat/.test(modelId);
+        return (ModelRegistry.supports(modelId, "structured_output", "deepseek") || modelId.includes("chat"));
     }
     static supportsEmbeddings(modelId) {
         const model = this.findModel(modelId);
-        return model?.modalities?.output?.includes("embeddings") || false;
+        return model?.modalities?.output?.includes("embeddings") ?? false;
     }
     static supportsImageGeneration(_modelId) {
         return false;
@@ -56,10 +47,10 @@ export class Capabilities {
         return false;
     }
     static supportsReasoning(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("reasoning"))
-            return true;
-        return /deepseek-reasoner/.test(modelId);
+        return ModelRegistry.supports(modelId, "reasoning", "deepseek") || modelId.includes("reasoner");
+    }
+    static supportsToolChoice(_modelId) {
+        return true;
     }
     static getPricing(modelId) {
         return PricingRegistry.getPricing(modelId, "deepseek");

package/dist/providers/deepseek/DeepSeekProvider.d.ts

@@ -20,7 +20,7 @@ export declare class DeepSeekProvider extends BaseProvider implements Provider {
         supportsModeration: (model: string) => boolean;
         supportsReasoning: (model: string) => boolean;
         supportsDeveloperRole: (_model: string) => boolean;
-        getContextWindow: (model: string) => number | null;
+        getContextWindow: (model: string) => number;
     };
     constructor(options: DeepSeekProviderOptions);
     apiBase(): string;

package/dist/providers/gemini/Capabilities.d.ts

@@ -11,6 +11,7 @@ export declare class Capabilities {
     static supportsImageGeneration(modelId: string): boolean;
     static supportsTranscription(modelId: string): boolean;
     static supportsModeration(_modelId: string): boolean;
+    static supportsToolChoice(_modelId: string): boolean;
     static normalizeTemperature(temperature: number | undefined, _model: string): number | undefined;
     static getModalities(modelId: string): {
         input: string[];
@@ -18,7 +19,6 @@ export declare class Capabilities {
     };
     static getCapabilities(modelId: string): string[];
     static getPricing(modelId: string): ModelPricing | undefined;
-    private static normalizeModelId;
     private static findModel;
 }
 //# sourceMappingURL=Capabilities.d.ts.map

package/dist/providers/gemini/Capabilities.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/gemini/Capabilities.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAErD,qBAAa,YAAY;IACvB,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IA2BvD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAwBzD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAW/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAW9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAWzD,MAAM,CAAC,0BAA0B,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAI5D,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIjD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQnD,MAAM,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAYxD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQtD,MAAM,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIpD,MAAM,CAAC,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAIhG,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE;IAY5E,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAUjD,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAI5D,OAAO,CAAC,MAAM,CAAC,gBAAgB;IAI/B,OAAO,CAAC,MAAM,CAAC,SAAS;CAGzB"}
+{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/gemini/Capabilities.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAErD,qBAAa,YAAY;IACvB,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAIvD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAIzD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAI/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAO9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIzD,MAAM,CAAC,0BAA0B,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAI5D,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOjD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKnD,MAAM,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IASxD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOtD,MAAM,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIpD,MAAM,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIpD,MAAM,CAAC,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAIhG,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE;IAc5E,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAgBjD,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAI5D,OAAO,CAAC,MAAM,CAAC,SAAS;CAGzB"}