@node-llm/core 1.15.2 → 1.16.0

package/dist/providers/mistral/Capabilities.js

@@ -2,95 +2,66 @@ import { ModelRegistry } from "../../models/ModelRegistry.js";
 import { PricingRegistry } from "../../models/PricingRegistry.js";
 export class MistralCapabilities {
     static getCapabilities(modelId) {
-        const caps = ["streaming", "chat"];
-        if (this.supportsTools(modelId)) {
-            caps.push("function_calling", "tools");
+        const model = this.findModel(modelId);
+        if (model?.capabilities) {
+            const caps = [...model.capabilities];
+            if (!caps.includes("streaming"))
+                caps.push("streaming");
+            return caps;
         }
-        if (this.supportsVision(modelId)) {
+        const caps = ["streaming", "chat"];
+        if (this.supportsTools(modelId))
+            caps.push("function_calling");
+        if (this.supportsReasoning(modelId))
+            caps.push("reasoning");
+        if (this.supportsVision(modelId))
             caps.push("vision");
-        }
-        if (this.supportsStructuredOutput(modelId)) {
-            caps.push("structured_output", "json_mode");
-        }
+        if (this.supportsStructuredOutput(modelId))
+            caps.push("structured_output");
         return caps;
     }
     static getContextWindow(modelId) {
-        const val = ModelRegistry.getContextWindow(modelId, "mistral");
-        if (val)
-            return val;
-        // Mistral models typically have 32K-128K context
-        if (/mistral-large|codestral/.test(modelId)) {
-            return 128_000;
-        }
-        if (/mistral-medium|mistral-small|pixtral/.test(modelId)) {
-            return 32_000;
-        }
-        return 32_000;
+        return ModelRegistry.getContextWindow(modelId, "mistral") ?? 32_768;
     }
     static getMaxOutputTokens(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.max_output_tokens)
-            return model.max_output_tokens;
-        // Default max output for Mistral models
-        return 8_192;
+        return ModelRegistry.getMaxOutputTokens(modelId, "mistral") ?? 8_192;
     }
     static supportsVision(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.modalities?.input?.includes("image"))
-            return true;
-        // Pixtral models support vision
-        return /pixtral/.test(modelId.toLowerCase());
+        return ModelRegistry.supports(modelId, "vision", "mistral") || modelId.includes("pixtral");
     }
     static supportsTools(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("function_calling") || model?.capabilities?.includes("tools"))
-            return true;
-        // Most Mistral models support tools except embedding models
-        return !/embed/.test(modelId.toLowerCase());
+        return (ModelRegistry.supports(modelId, "tools", "mistral") ||
+            ModelRegistry.supports(modelId, "function_calling", "mistral") ||
+            !modelId.includes("embed"));
     }
     static supportsStructuredOutput(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("structured_output"))
-            return true;
-        // Most Mistral chat models support JSON mode
-        return !/embed/.test(modelId.toLowerCase());
+        return (ModelRegistry.supports(modelId, "structured_output", "mistral") ||
+            ModelRegistry.supports(modelId, "json_mode", "mistral") ||
+            !modelId.includes("embed"));
     }
     static supportsEmbeddings(modelId) {
         const model = this.findModel(modelId);
-        if (model?.modalities?.output?.includes("embeddings"))
-            return true;
-        return /embed/.test(modelId.toLowerCase());
+        return model?.modalities?.output?.includes("embeddings") ?? modelId.includes("embed");
     }
     static supportsImageGeneration(_modelId) {
         return false;
     }
     static supportsTranscription(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.modalities?.input?.includes("audio"))
-            return true;
-        // Voxtral models support transcription
-        return /voxtral/.test(modelId.toLowerCase());
+        return (ModelRegistry.supports(modelId, "transcription", "mistral") || modelId.includes("voxtral"));
     }
     static supportsModeration(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("moderation"))
-            return true;
-        return /moderation/.test(modelId.toLowerCase());
+        return (ModelRegistry.supports(modelId, "moderation", "mistral") || modelId.includes("moderation"));
     }
     static supportsReasoning(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("reasoning"))
-            return true;
-        // Magistral models support reasoning
-        return /magistral/.test(modelId.toLowerCase());
+        return ModelRegistry.supports(modelId, "reasoning", "mistral") || modelId.includes("magistral");
+    }
+    static supportsToolChoice(_modelId) {
+        return true;
     }
     static findModel(modelId) {
         return ModelRegistry.find(modelId, "mistral");
     }
     static getPricing(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.pricing)
-            return model.pricing;
         return PricingRegistry.getPricing(modelId, "mistral");
     }
     static getInputModalities(modelId) {

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

@@ -1 +1 @@
-{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/mistral/Chat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAS,MAAM,gBAAgB,CAAC;AAgDlE,qBAAa,WAAW;IAEpB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADN,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM;IAG3B,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;CAmK3D"}
+{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/mistral/Chat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAS,MAAM,gBAAgB,CAAC;AAgDlE,qBAAa,WAAW;IAEpB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADN,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM;IAG3B,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;CAkL3D"}

package/dist/providers/mistral/Chat.js

@@ -11,7 +11,7 @@ export class MistralChat {
         this.apiKey = apiKey;
     }
     async execute(request) {
-        const { model, messages, tools, max_tokens, response_format, thinking: _thinking, headers: _headers, requestTimeout, signal: _signal, ...rest } = request;
+        const { model, messages, tools, max_tokens, response_format, thinking: _thinking, headers: _headers, tool_choice: _choice, parallel_tool_calls: _parallel, requestTimeout, signal: _signal, ...rest } = request;
         const mappedMessages = mapSystemMessages(messages, false);
         // Sanitize messages - remove internal properties that Mistral API doesn't accept
         const sanitizedMessages = mappedMessages.map((msg) => {
@@ -35,6 +35,18 @@ export class MistralChat {
             body.max_tokens = max_tokens;
         if (tools && tools.length > 0)
             body.tools = tools;
+        if (request.tool_choice) {
+            if (request.tool_choice === "required") {
+                body.tool_choice = "any";
+            }
+            else if (typeof request.tool_choice === "string" &&
+                !["auto", "none", "any"].includes(request.tool_choice)) {
+                body.tool_choice = { type: "function", function: { name: request.tool_choice } };
+            }
+            else {
+                body.tool_choice = request.tool_choice;
+            }
+        }
         // Handle structured output for Mistral
         if (response_format) {
             if (response_format.type === "json_schema") {

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

@@ -1 +1 @@
-{"version":3,"file":"Streaming.d.ts","sourceRoot":"","sources":["../../../src/providers/mistral/Streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAOxD,qBAAa,gBAAgB;IAEzB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADN,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM;IAG1B,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,CAAC,EAAE,eAAe,GAAG,cAAc,CAAC,SAAS,CAAC;CAgO9F"}
+{"version":3,"file":"Streaming.d.ts","sourceRoot":"","sources":["../../../src/providers/mistral/Streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAOxD,qBAAa,gBAAgB;IAEzB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADN,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM;IAG1B,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,CAAC,EAAE,eAAe,GAAG,cAAc,CAAC,SAAS,CAAC;CA6O9F"}

package/dist/providers/mistral/Streaming.js

@@ -35,10 +35,22 @@ export class MistralStreaming {
         };
         if (max_tokens)
             body.max_tokens = max_tokens;
-        if (tools && tools.length > 0)
+        if (tools && tools.length > 0 && request.tool_choice !== "none")
             body.tools = tools;
         if (response_format)
             body.response_format = response_format;
+        if (request.tool_choice && request.tool_choice !== "none") {
+            if (request.tool_choice === "required") {
+                body.tool_choice = "any";
+            }
+            else if (typeof request.tool_choice === "string" &&
+                !["auto", "none", "any"].includes(request.tool_choice)) {
+                body.tool_choice = { type: "function", function: { name: request.tool_choice } };
+            }
+            else {
+                body.tool_choice = request.tool_choice;
+            }
+        }
         let done = false;
         // Track tool calls being built across chunks
         const toolCallsMap = new Map();

package/dist/providers/mistral/Transcription.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Transcription.d.ts","sourceRoot":"","sources":["../../../src/providers/mistral/Transcription.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,qBAAqB,EAAwB,MAAM,gBAAgB,CAAC;AAqBnG,qBAAa,oBAAoB;IAE7B,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADN,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM;IAG3B,OAAO,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IA2D5E,OAAO,CAAC,WAAW;CAmBpB"}
+{"version":3,"file":"Transcription.d.ts","sourceRoot":"","sources":["../../../src/providers/mistral/Transcription.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,qBAAqB,EAAwB,MAAM,gBAAgB,CAAC;AAqBnG,qBAAa,oBAAoB;IAE7B,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADN,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM;IAG3B,OAAO,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IA6D5E,OAAO,CAAC,WAAW;CAmBpB"}

package/dist/providers/mistral/Transcription.js

@@ -45,7 +45,9 @@ export class MistralTranscription {
             id: s.id,
             start: s.start,
             end: s.end,
-            text: s.text
+            text: s.text,
+            speaker: s.speaker,
+            words: s.words
         }));
         return {
             text: json.text,

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

@@ -9,5 +9,6 @@ export declare class OllamaCapabilities {
     static supportsImageGeneration(modelId: string): boolean;
     static supportsTranscription(modelId: string): boolean;
     static supportsModeration(modelId: string): boolean;
+    static supportsToolChoice(_modelId: string): boolean;
 }
 //# sourceMappingURL=Capabilities.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/ollama/Capabilities.ts"],"names":[],"mappings":"AAEA,qBAAa,kBAAkB;IAC7B,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM;IAYhC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAKvD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAc/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQ9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKzD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IASnD,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKlD,MAAM,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKxD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKtD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;CAIpD"}
+{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/ollama/Capabilities.ts"],"names":[],"mappings":"AAEA,qBAAa,kBAAkB;IAC7B,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM;IAYhC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAKvD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQ/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAO9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIzD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKnD,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQlD,MAAM,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIxD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAItD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAInD,MAAM,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;CAGrD"}

package/dist/providers/ollama/Capabilities.js

@@ -13,49 +13,39 @@ export class OllamaCapabilities {
     }
     static getContextWindow(modelId) {
         const model = this.findModel(modelId);
-        return model?.context_window || 8192;
+        return model?.context_window || 32_768;
     }
     static supportsVision(modelId) {
-        const model = this.findModel(modelId);
-        if (model) {
-            return (model.modalities?.input?.includes("image") ||
-                model.capabilities?.includes("vision") ||
-                false);
-        }
-        // Fallback for custom models not in registry
-        return /vision|llava|moondream|bakllava/.test(modelId.toLowerCase());
+        return (ModelRegistry.supports(modelId, "vision", "ollama") ||
+            modelId.includes("vision") ||
+            modelId.includes("llava"));
     }
     static supportsTools(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("tools"))
-            return true;
-        // Fallback heuristics for models not in registry
-        return /gpt-|claude|gemini|deepseek|llama/.test(modelId.toLowerCase());
+        return (ModelRegistry.supports(modelId, "tools", "ollama") ||
+            ModelRegistry.supports(modelId, "function_calling", "ollama"));
     }
     static supportsStructuredOutput(modelId) {
-        const model = this.findModel(modelId);
-        return model?.capabilities?.includes("structured_output") || false;
+        return ModelRegistry.supports(modelId, "structured_output", "ollama");
     }
     static supportsEmbeddings(modelId) {
         const model = this.findModel(modelId);
-        return (model?.modalities?.output?.includes("embeddings") ||
-            model?.capabilities?.includes("embeddings") ||
-            false);
+        return model?.modalities?.output?.includes("embeddings") ?? modelId.includes("embed");
     }
     static supportsReasoning(modelId) {
-        const model = this.findModel(modelId);
-        return model?.capabilities?.includes("reasoning") || false;
+        return (ModelRegistry.supports(modelId, "reasoning", "ollama") ||
+            modelId.includes("reason") ||
+            modelId.includes("thought"));
     }
     static supportsImageGeneration(modelId) {
-        const model = this.findModel(modelId);
-        return model?.modalities?.output?.includes("image") || false;
+        return false;
     }
     static supportsTranscription(modelId) {
-        const model = this.findModel(modelId);
-        return model?.modalities?.input?.includes("audio") || false;
+        return false;
     }
     static supportsModeration(modelId) {
-        const model = this.findModel(modelId);
-        return model?.modalities?.output?.includes("moderation") || false;
+        return false;
+    }
+    static supportsToolChoice(_modelId) {
+        return true;
     }
 }

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

@@ -12,9 +12,10 @@ export declare class Capabilities {
     static supportsModeration(modelId: string): boolean;
     static supportsReasoning(modelId: string): boolean;
     static supportsDeveloperRole(modelId: string): boolean;
+    static supportsToolChoice(_modelId: string): boolean;
     static supportsPrediction(modelId: string): boolean;
     static needsMaxCompletionTokens(modelId: string): boolean;
-    static getModelType(modelId: string): "embeddings" | "audio" | "moderation" | "image" | "chat" | "audio_transcription" | "audio_speech";
+    static getModelType(modelId: string): string;
     static getModalities(modelId: string): {
         input: string[];
         output: string[];

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

@@ -1 +1 @@
-{"version":3,"file":"Capabilities.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/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;IAUvD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IASzD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAO/C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQ9C,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOzD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIjD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOnD,MAAM,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAWxD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOtD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOnD,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAOlD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAItD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAInD,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIzD,MAAM,CAAC,YAAY,CACjB,OAAO,EAAE,MAAM,GAEb,YAAY,GACZ,OAAO,GACP,YAAY,GACZ,OAAO,GACP,MAAM,GACN,qBAAqB,GACrB,cAAc;IAWlB,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE;IAiB5E,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAsBjD,MAAM,CAAC,oBAAoB,CACzB,WAAW,EAAE,MAAM,GAAG,SAAS,EAC/B,OAAO,EAAE,MAAM,GACd,MAAM,GAAG,SAAS,GAAG,IAAI;IAM5B,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAOjD,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/openai/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,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;IAQxD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKtD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKnD,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIlD,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAItD,MAAM,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAIpD,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAInD,MAAM,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAIzD,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAU5C,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE;IAgB5E,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAkBjD,MAAM,CAAC,oBAAoB,CACzB,WAAW,EAAE,MAAM,GAAG,SAAS,EAC/B,OAAO,EAAE,MAAM,GACd,MAAM,GAAG,SAAS,GAAG,IAAI;IAM5B,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAMjD,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAI5D,OAAO,CAAC,MAAM,CAAC,SAAS;CAGzB"}

package/dist/providers/openai/Capabilities.js

@@ -2,111 +2,78 @@ import { ModelRegistry } from "../../models/ModelRegistry.js";
 import { PricingRegistry } from "../../models/PricingRegistry.js";
 export class Capabilities {
     static getContextWindow(modelId) {
-        const val = ModelRegistry.getContextWindow(modelId, "openai");
-        if (val)
-            return val;
-        if (/gpt-4.*(preview|turbo|vision|o)/.test(modelId) || /o1|o3/.test(modelId))
-            return 128_000;
-        if (/gpt-4/.test(modelId))
-            return 8_192;
-        if (/gpt-3\.5/.test(modelId))
-            return 16_385;
-        return 128_000;
+        return ModelRegistry.getContextWindow(modelId, "openai") ?? 128_000;
     }
     static getMaxOutputTokens(modelId) {
-        const val = ModelRegistry.getMaxOutputTokens(modelId, "openai");
-        if (val)
-            return val;
-        if (/o1.*(pro|mini)|o3/.test(modelId))
-            return 65_536;
-        if (/gpt-4o/.test(modelId))
-            return 16_384;
-        return 4_096;
+        return ModelRegistry.getMaxOutputTokens(modelId, "openai") ?? 4_096;
     }
     static supportsVision(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.modalities?.input?.includes("image"))
-            return true;
-        return /gpt-4(?!-3)|o1/.test(modelId) && !/audio|realtime|voice/.test(modelId);
+        return ModelRegistry.supports(modelId, "vision", "openai");
     }
     static supportsTools(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("function_calling") || model?.capabilities?.includes("tools"))
-            return true;
-        return !/embedding|moderation|dall-e|tts|whisper/.test(modelId);
+        return (ModelRegistry.supports(modelId, "tools", "openai") ||
+            ModelRegistry.supports(modelId, "function_calling", "openai"));
     }
     static supportsStructuredOutput(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("structured_output"))
-            return true;
-        return /gpt-4|o1|o3/.test(modelId);
+        return ModelRegistry.supports(modelId, "structured_output", "openai");
     }
     static supportsJsonMode(modelId) {
-        return this.supportsStructuredOutput(modelId);
+        return (ModelRegistry.supports(modelId, "json_mode", "openai") ||
+            this.supportsStructuredOutput(modelId));
     }
     static supportsEmbeddings(modelId) {
         const model = this.findModel(modelId);
-        if (model?.modalities?.output?.includes("embeddings"))
-            return true;
-        return /embedding/.test(modelId);
+        return model?.modalities?.output?.includes("embeddings") ?? modelId.includes("embedding");
     }
     static supportsImageGeneration(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("image_generation") ||
-            model?.modalities?.output?.includes("image"))
-            return true;
-        return /dall-e|image/.test(modelId);
+        return (ModelRegistry.supports(modelId, "image_generation", "openai") ||
+            modelId.includes("dall-e") ||
+            modelId === "gpt-image-1");
     }
     static supportsTranscription(modelId) {
         const model = this.findModel(modelId);
-        if (model?.modalities?.input?.includes("audio"))
-            return true;
-        return /whisper|audio|transcribe/.test(modelId);
+        return model?.modalities?.input?.includes("audio") ?? modelId.includes("whisper");
     }
     static supportsModeration(modelId) {
         const model = this.findModel(modelId);
-        if (model?.modalities?.output?.includes("moderation"))
-            return true;
-        return /moderation/.test(modelId);
+        return model?.modalities?.output?.includes("moderation") ?? modelId.includes("moderation");
     }
     static supportsReasoning(modelId) {
-        const model = this.findModel(modelId);
-        if (model?.capabilities?.includes("reasoning"))
-            return true;
-        return /o\d|gpt-5/.test(modelId);
+        return ModelRegistry.supports(modelId, "reasoning", "openai");
     }
     static supportsDeveloperRole(modelId) {
         return /gpt-4o|o1|o3|gpt-5/.test(modelId);
     }
+    static supportsToolChoice(_modelId) {
+        return true;
+    }
     static supportsPrediction(modelId) {
-        return /gpt-4o/.test(modelId) || /gpt-4-/.test(modelId);
+        return /gpt-4o|gpt-4-/.test(modelId);
     }
     static needsMaxCompletionTokens(modelId) {
-        return /o1|o3|gpt-5/.test(modelId);
+        return this.supportsReasoning(modelId);
     }
     static getModelType(modelId) {
         if (this.supportsEmbeddings(modelId))
             return "embeddings";
-        if (/moderation/.test(modelId))
+        if (this.supportsModeration(modelId))
             return "moderation";
-        if (/embedding/.test(modelId))
-            return "embeddings";
-        if (/dall-e|image/.test(modelId))
+        if (this.supportsImageGeneration(modelId))
             return "image";
-        if (/whisper|transcribe/.test(modelId))
+        if (this.supportsTranscription(modelId))
             return "audio_transcription";
-        if (/tts|speech/.test(modelId))
+        if (modelId.includes("tts") || modelId.includes("speech"))
             return "audio_speech";
-        if (/audio/.test(modelId))
+        if (modelId.includes("audio"))
             return "audio";
         return "chat";
     }
     static getModalities(modelId) {
-        const input = ["text"];
-        const output = ["text"];
         const model = this.findModel(modelId);
         if (model?.modalities)
             return model.modalities;
+        const input = ["text"];
+        const output = ["text"];
         if (this.supportsVision(modelId))
             input.push("image", "pdf");
         if (this.supportsTranscription(modelId))
@@ -120,22 +87,19 @@ export class Capabilities {
         return { input, output };
     }
     static getCapabilities(modelId) {
-        const caps = ["streaming"];
         const model = this.findModel(modelId);
-        if (model) {
-            model.capabilities.forEach((c) => {
-                if (!caps.includes(c))
-                    caps.push(c);
-            });
+        if (model?.capabilities) {
+            const caps = [...model.capabilities];
+            if (!caps.includes("streaming"))
+                caps.push("streaming");
             return caps;
         }
+        const caps = ["streaming"];
         if (this.supportsTools(modelId))
             caps.push("function_calling");
         if (this.supportsStructuredOutput(modelId))
             caps.push("structured_output");
-        if (this.supportsEmbeddings(modelId))
-            caps.push("batch");
-        if (/o\d|gpt-5/.test(modelId))
+        if (this.supportsReasoning(modelId))
             caps.push("reasoning");
         if (this.supportsImageGeneration(modelId))
             caps.push("image_generation");
@@ -144,9 +108,9 @@ export class Capabilities {
         return caps;
     }
     static normalizeTemperature(temperature, modelId) {
-        if (/^(o\d|gpt-5)/.test(modelId))
+        if (this.supportsReasoning(modelId))
             return 1.0;
-        if (/-search/.test(modelId))
+        if (modelId.includes("-search"))
             return null;
         return temperature;
     }

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

@@ -1 +1 @@
-{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Chat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAS3D,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAIrD,qBAAa,UAAU;IAInB,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,MAAM;IAJzB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAGd,aAAa,EAAE,cAAc,GAAG,MAAM,EACtC,MAAM,EAAE,MAAM;IAK3B,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;CAuJ3D"}
+{"version":3,"file":"Chat.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Chat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAS3D,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAIrD,qBAAa,UAAU;IAInB,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,MAAM;IAJzB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAGd,aAAa,EAAE,cAAc,GAAG,MAAM,EACtC,MAAM,EAAE,MAAM;IAK3B,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;CA2J3D"}

package/dist/providers/openai/Chat.js

@@ -4,7 +4,7 @@ import { ModelRegistry } from "../../models/ModelRegistry.js";
 import { buildUrl } from "./utils.js";
 import { logger } from "../../utils/logger.js";
 import { fetchWithTimeout } from "../../utils/fetch.js";
-import { mapSystemMessages } from "../utils.js";
+import { mapSystemMessages, normalizeOpenAIToolChoice } from "../utils.js";
 import { enforceStrictSchema } from "../../schema/strict.js";
 export class OpenAIChat {
     providerOrUrl;
@@ -53,6 +53,9 @@ export class OpenAIChat {
                 return tool;
             });
         }
+        if (request.tool_choice) {
+            body.tool_choice = normalizeOpenAIToolChoice(request.tool_choice);
+        }
         if (prediction) {
             body.prediction = {
                 type: "content",

package/dist/providers/openai/Image.d.ts

@@ -4,5 +4,6 @@ export declare class OpenAIImage {
     private readonly apiKey;
     constructor(baseUrl: string, apiKey: string);
     execute(request: ImageRequest): Promise<ImageResponse>;
+    private executeEdit;
 }
 //# sourceMappingURL=Image.d.ts.map

package/dist/providers/openai/Image.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Image.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Image.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAM7D,qBAAa,WAAW;IAEpB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADN,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM;IAG3B,OAAO,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;CA2C7D"}
+{"version":3,"file":"Image.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Image.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAO7D,qBAAa,WAAW;IAEpB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADN,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM;IAG3B,OAAO,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;YAqD9C,WAAW;CAgE1B"}

package/dist/providers/openai/Image.js

@@ -2,6 +2,7 @@ import { handleOpenAIError } from "./Errors.js";
 import { buildUrl } from "./utils.js";
 import { logger } from "../../utils/logger.js";
 import { fetchWithTimeout } from "../../utils/fetch.js";
+import { BinaryUtils } from "../../utils/Binary.js";
 export class OpenAIImage {
     baseUrl;
     apiKey;
@@ -10,8 +11,13 @@ export class OpenAIImage {
         this.apiKey = apiKey;
     }
     async execute(request) {
+        const isEdit = !!(request.images && request.images.length > 0);
+        const model = request.model || (isEdit ? "gpt-image-1" : "dall-e-3");
+        if (isEdit) {
+            return this.executeEdit(request, model);
+        }
         const body = {
-            model: request.model || "dall-e-3",
+            model,
             prompt: request.prompt,
             size: request.size || "1024x1024",
             quality: request.quality || "standard",
@@ -28,16 +34,69 @@ export class OpenAIImage {
             body: JSON.stringify(body)
         }, request.requestTimeout);
         if (!response.ok) {
-            await handleOpenAIError(response, request.model);
+            await handleOpenAIError(response, model);
+        }
+        const json = await response.json();
+        logger.logResponse("OpenAI", response.status, response.statusText, json);
+        const data = json.data?.[0];
+        if (!data) {
+            logger.debug("OpenAI Response missing data array:", json);
+            throw new Error("OpenAI returned empty image response");
+        }
+        return {
+            url: data.url,
+            data: data.b64_json,
+            revised_prompt: data.revised_prompt,
+            mime_type: "image/png"
+        };
+    }
+    async executeEdit(request, model) {
+        const formData = new FormData();
+        formData.append("model", model);
+        formData.append("prompt", request.prompt);
+        if (request.size)
+            formData.append("size", request.size);
+        if (request.n)
+            formData.append("n", String(request.n));
+        if (request.images && request.images[0]) {
+            const imageFile = request.images[0];
+            // Download/Read the image
+            const imageBinary = await BinaryUtils.toBase64(imageFile);
+            if (imageBinary) {
+                const imageBlob = BinaryUtils.base64ToBlob(imageBinary);
+                formData.append("image", imageBlob, "image.png");
+            }
+            if (request.mask) {
+                const maskBinary = await BinaryUtils.toBase64(request.mask);
+                if (maskBinary) {
+                    const maskBlob = BinaryUtils.base64ToBlob(maskBinary);
+                    formData.append("mask", maskBlob, "mask.png");
+                }
+            }
+        }
+        const endpoint = request.prompt ? "/images/edits" : "/images/variations";
+        const url = buildUrl(this.baseUrl, endpoint);
+        logger.logRequest("OpenAI", "POST", url, { model, prompt: request.prompt });
+        const response = await fetchWithTimeout(url, {
+            method: "POST",
+            headers: {
+                Authorization: `Bearer ${this.apiKey}`
+            },
+            body: formData
+        }, request.requestTimeout);
+        if (!response.ok) {
+            await handleOpenAIError(response, model);
         }
         const json = await response.json();
         logger.logResponse("OpenAI", response.status, response.statusText, json);
         const data = json.data?.[0];
         if (!data) {
+            logger.debug("OpenAI Response missing data array:", json);
             throw new Error("OpenAI returned empty image response");
         }
         return {
             url: data.url,
+            data: data.b64_json,
             revised_prompt: data.revised_prompt,
             mime_type: "image/png"
         };

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

@@ -1 +1 @@
-{"version":3,"file":"Streaming.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAQxD,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAIrD,qBAAa,eAAe;IAIxB,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,MAAM;IAJzB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAGd,aAAa,EAAE,cAAc,GAAG,MAAM,EACtC,MAAM,EAAE,MAAM;IAK1B,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,CAAC,EAAE,eAAe,GAAG,cAAc,CAAC,SAAS,CAAC;CAkP9F"}
+{"version":3,"file":"Streaming.d.ts","sourceRoot":"","sources":["../../../src/providers/openai/Streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAQxD,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAIrD,qBAAa,eAAe;IAIxB,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,MAAM;IAJzB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAGd,aAAa,EAAE,cAAc,GAAG,MAAM,EACtC,MAAM,EAAE,MAAM;IAK1B,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,CAAC,EAAE,eAAe,GAAG,cAAc,CAAC,SAAS,CAAC;CAyP9F"}