@llumiverse/drivers 1.2.0 → 1.4.0-dev.20260629.090753Z

package/lib/{esm/mistral → mistral}/index.js

@@ -1,13 +1,14 @@
-import { AbstractDriver } from "@llumiverse/core";
-import { transformSSEStream } from "@llumiverse/core/async";
-import { getJSONSafetyNotice } from "@llumiverse/core/formatters";
-import { FetchClient } from "@vertesia/api-fetch-client";
-import { formatOpenAILikeTextPrompt } from "../openai/openai_format.js";
+import { LlumiverseError, MISTRAL_DEFAULT_EMBEDDING_MODEL, normalizeEmbeddingsOptions, } from '@llumiverse/core';
+import { transformSSEStream } from '@llumiverse/core/async';
+import { AbstractDriver } from '@llumiverse/core/driver';
+import { getJSONSafetyNotice } from '@llumiverse/core/formatters';
+import { FetchClient } from '@vertesia/api-fetch-client';
+import { formatOpenAILikeTextPrompt } from '../openai/openai_format.js';
 //TODO retry on 429
 //const RETRY_STATUS_CODES = [429, 500, 502, 503, 504];
 const ENDPOINT = 'https://api.mistral.ai';
 export class MistralAIDriver extends AbstractDriver {
-    static PROVIDER = "mistralai";
+    static PROVIDER = 'mistralai';
     provider = MistralAIDriver.PROVIDER;
     apiKey;
     client;
@@ -16,8 +17,8 @@ export class MistralAIDriver extends AbstractDriver {
         super(options);
         this.apiKey = options.apiKey;
         //this.client = new MistralClient(options.apiKey, options.endpointUrl);
-        this.client = new FetchClient(options.endpoint_url || ENDPOINT).withHeaders({
-            authorization: `Bearer ${this.apiKey}`
+        this.client = new FetchClient(options.endpoint_url || ENDPOINT, this.getDriverFetch()).withHeaders({
+            authorization: `Bearer ${this.apiKey}`,
         });
     }
     getResponseFormat = (_options) => {
@@ -37,15 +38,15 @@ export class MistralAIDriver extends AbstractDriver {
         //Add JSON instruction is schema is provided
         if (opts.result_schema) {
             messages.push({
-                role: "user",
-                content: "IMPORTANT: " + getJSONSafetyNotice(opts.result_schema)
+                role: 'user',
+                content: `IMPORTANT: ${getJSONSafetyNotice(opts.result_schema)}`,
             });
         }
         return messages;
     }
     async requestTextCompletion(messages, options) {
-        if (options.model_options?._option_id !== undefined && options.model_options?._option_id !== "text-fallback") {
-            this.logger.debug({ options: options.model_options }, "Unexpected option id");
+        if (options.model_options?._option_id !== undefined && options.model_options?._option_id !== 'text-fallback') {
+            this.logger.debug({ options: options.model_options }, 'Unexpected option id');
         }
         options.model_options = options.model_options;
         const requestPayload = _makeChatCompletionRequest({
@@ -55,14 +56,14 @@ export class MistralAIDriver extends AbstractDriver {
             temperature: options.model_options?.temperature,
             responseFormat: this.getResponseFormat(options),
         });
-        this.logger.debug({ payload: JSON.stringify(requestPayload) }, "Mistral request payload");
-        const res = await this.client.post('/v1/chat/completions', {
+        this.logger.debug({ payload: JSON.stringify(requestPayload) }, 'Mistral request payload');
+        const res = (await this.client.post('/v1/chat/completions', {
             payload: requestPayload,
-        });
+        }));
         const choice = res.choices[0];
         const result = choice.message.content;
         return {
-            result: result ? [{ type: "text", value: result }] : [],
+            result: result ? [{ type: 'text', value: result }] : [],
             token_usage: {
                 prompt: res.usage.prompt_tokens,
                 result: res.usage.completion_tokens,
@@ -73,8 +74,8 @@ export class MistralAIDriver extends AbstractDriver {
         };
     }
     async requestTextCompletionStream(messages, options) {
-        if (options.model_options?._option_id !== undefined && options.model_options?._option_id !== "text-fallback") {
-            this.logger.debug({ options: options.model_options }, "Unexpected option id");
+        if (options.model_options?._option_id !== undefined && options.model_options?._option_id !== 'text-fallback') {
+            this.logger.debug({ options: options.model_options }, 'Unexpected option id');
         }
         options.model_options = options.model_options;
         const streamPayload = _makeChatCompletionRequest({
@@ -87,16 +88,16 @@ export class MistralAIDriver extends AbstractDriver {
             stream: true,
             stopSequences: options.model_options?.stop_sequence,
         });
-        this.logger.debug({ payload: JSON.stringify(streamPayload) }, "Mistral stream request payload");
-        const stream = await this.client.post('/v1/chat/completions', {
+        this.logger.debug({ payload: JSON.stringify(streamPayload) }, 'Mistral stream request payload');
+        const stream = (await this.client.post('/v1/chat/completions', {
             payload: streamPayload,
-            reader: 'sse'
-        });
+            reader: 'sse',
+        }));
         return transformSSEStream(stream, (data) => {
             const json = JSON.parse(data);
             const content = json.choices[0]?.delta.content;
             return {
-                result: content ? [{ type: "text", value: content }] : [],
+                result: content ? [{ type: 'text', value: content }] : [],
                 finish_reason: json.choices[0]?.finish_reason, //Uses expected "stop" , "length" format
                 token_usage: {
                     prompt: json.usage?.prompt_tokens,
@@ -108,7 +109,7 @@ export class MistralAIDriver extends AbstractDriver {
     }
     async listModels() {
         const models = await this.client.get('v1/models');
-        const aiModels = models.data.map(m => {
+        const aiModels = models.data.map((m) => {
             return {
                 id: m.id,
                 name: m.id,
@@ -120,21 +121,49 @@ export class MistralAIDriver extends AbstractDriver {
         return aiModels;
     }
     validateConnection() {
-        throw new Error("Method not implemented.");
+        throw new Error('Method not implemented.');
     }
-    async generateEmbeddings({ text, model = "mistral-embed" }) {
-        const r = await this.client.post('/v1/embeddings', {
-            payload: {
-                model,
-                input: [text],
-                encoding_format: "float"
-            },
+    async generateEmbeddings(options) {
+        const normalized = normalizeEmbeddingsOptions(options);
+        const model = normalized.model ?? MISTRAL_DEFAULT_EMBEDDING_MODEL;
+        const texts = normalized.inputs.map((input) => {
+            if (input.type !== 'text') {
+                throw new Error(`Provider 'mistralai' does not support '${input.type}' embeddings; only 'text' is supported.`);
+            }
+            return input.text;
         });
-        return {
-            values: r.data[0].embedding,
-            model,
-            token_count: r.usage.total_tokens || r.usage.prompt_tokens + r.usage.completion_tokens,
-        };
+        try {
+            const r = (await this.client.post('/v1/embeddings', {
+                payload: {
+                    model,
+                    input: texts,
+                    encoding_format: 'float',
+                },
+            }));
+            const ordered = [...r.data].sort((a, b) => a.index - b.index);
+            const promptTokens = r.usage?.total_tokens ??
+                (r.usage ? (r.usage.prompt_tokens ?? 0) + (r.usage.completion_tokens ?? 0) : undefined);
+            return {
+                model,
+                results: ordered.map((entry) => ({
+                    outputs: [{ values: entry.embedding, modality: 'text' }],
+                })),
+                ...(typeof promptTokens === 'number'
+                    ? { usage: { input_tokens: promptTokens, input_text_tokens: promptTokens } }
+                    : {}),
+            };
+        }
+        catch (error) {
+            if (LlumiverseError.isLlumiverseError(error))
+                throw error;
+            if (error instanceof Error && typeof error.status !== 'number')
+                throw error;
+            throw this.formatLlumiverseError(error, {
+                provider: this.provider,
+                model,
+                operation: 'execute',
+            });
+        }
     }
 }
 /**
@@ -168,5 +197,4 @@ function _makeChatCompletionRequest({ model, messages, tools, temperature, maxTo
         stop: stopSequences ?? undefined,
     };
 }
-;
 //# sourceMappingURL=index.js.map

package/lib/mistral/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/mistral/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAQH,eAAe,EACf,+BAA+B,EAC/B,0BAA0B,GAG7B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAClE,OAAO,EAAE,WAAW,EAAwB,MAAM,4BAA4B,CAAC;AAC/E,OAAO,EAAE,0BAA0B,EAA0B,MAAM,4BAA4B,CAAC;AAShG,mBAAmB;AACnB,uDAAuD;AAEvD,MAAM,QAAQ,GAAG,wBAAwB,CAAC;AAO1C,MAAM,OAAO,eAAgB,SAAQ,cAA2D;IAC5F,MAAM,CAAC,QAAQ,GAAG,WAAW,CAAC;IAE9B,QAAQ,GAAG,eAAe,CAAC,QAAQ,CAAC;IACpC,MAAM,CAAS;IACf,MAAM,CAAc;IACpB,WAAW,CAAU;IAErB,YAAY,OAA+B;QACvC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,uEAAuE;QACvE,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,YAAY,IAAI,QAAQ,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,WAAW,CAAC;YAC/F,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;SACzC,CAAC,CAAC;IACP,CAAC;IAED,iBAAiB,GAAG,CAAC,QAA0B,EAA8B,EAAE;QAC3E,+CAA+C;QAC/C,2BAA2B;QAC3B,sBAAsB;QAEtB,+CAA+C;QAC/C,oBAAoB;QACpB,uBAAuB;QAEvB,2EAA2E;QAE3E,4FAA4F;QAC5F,iHAAiH;QACjH,OAAO,SAAS,CAAC;IACrB,CAAC,CAAC;IAEQ,KAAK,CAAC,YAAY,CAAC,QAAyB,EAAE,IAAsB;QAC1E,MAAM,QAAQ,GAAG,0BAA0B,CAAC,QAAQ,CAAC,CAAC;QACtD,4CAA4C;QAC5C,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,QAAQ,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,cAAc,mBAAmB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;aACnE,CAAC,CAAC;QACP,CAAC;QACD,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,QAA6B,EAAE,OAAyB;QAChF,IAAI,OAAO,CAAC,aAAa,EAAE,UAAU,KAAK,SAAS,IAAI,OAAO,CAAC,aAAa,EAAE,UAAU,KAAK,eAAe,EAAE,CAAC;YAC3G,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,aAAa,EAAE,EAAE,sBAAsB,CAAC,CAAC;QAClF,CAAC;QACD,OAAO,CAAC,aAAa,GAAG,OAAO,CAAC,aAAoC,CAAC;QAErE,MAAM,cAAc,GAAG,0BAA0B,CAAC;YAC9C,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ,EAAE,QAAQ;YAClB,SAAS,EAAE,OAAO,CAAC,aAAa,EAAE,UAAU;YAC5C,WAAW,EAAE,OAAO,CAAC,aAAa,EAAE,WAAW;YAC/C,cAAc,EAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC;SAClD,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,EAAE,EAAE,yBAAyB,CAAC,CAAC;QAE1F,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE;YACxD,OAAO,EAAE,cAAc;SAC1B,CAAC,CAA2B,CAAC;QAE9B,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;QAEtC,OAAO;YACH,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE;YACvD,WAAW,EAAE;gBACT,MAAM,EAAE,GAAG,CAAC,KAAK,CAAC,aAAa;gBAC/B,MAAM,EAAE,GAAG,CAAC,KAAK,CAAC,iBAAiB;gBACnC,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,YAAY;aAChC;YACD,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,wCAAwC;YAC7E,iBAAiB,EAAE,OAAO,CAAC,yBAAyB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS;SACzE,CAAC;IACN,CAAC;IAED,KAAK,CAAC,2BAA2B,CAC7B,QAA6B,EAC7B,OAAyB;QAEzB,IAAI,OAAO,CAAC,aAAa,EAAE,UAAU,KAAK,SAAS,IAAI,OAAO,CAAC,aAAa,EAAE,UAAU,KAAK,eAAe,EAAE,CAAC;YAC3G,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,aAAa,EAAE,EAAE,sBAAsB,CAAC,CAAC;QAClF,CAAC;QACD,OAAO,CAAC,aAAa,GAAG,OAAO,CAAC,aAAoC,CAAC;QAErE,MAAM,aAAa,GAAG,0BAA0B,CAAC;YAC7C,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ,EAAE,QAAQ;YAClB,SAAS,EAAE,OAAO,CAAC,aAAa,EAAE,UAAU;YAC5C,WAAW,EAAE,OAAO,CAAC,aAAa,EAAE,WAAW;YAC/C,IAAI,EAAE,OAAO,CAAC,aAAa,EAAE,KAAK;YAClC,cAAc,EAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC;YAC/C,MAAM,EAAE,IAAI;YACZ,aAAa,EAAE,OAAO,CAAC,aAAa,EAAE,aAAa;SACtD,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,EAAE,EAAE,gCAAgC,CAAC,CAAC;QAEhG,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE;YAC3D,OAAO,EAAE,aAAa;YACtB,MAAM,EAAE,KAAK;SAChB,CAAC,CAAoC,CAAC;QAEvC,OAAO,kBAAkB,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;YAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC;YAC/C,OAAO;gBACH,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE;gBACzD,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,wCAAwC;gBACvF,WAAW,EAAE;oBACT,MAAM,EAAE,IAAI,CAAC,KAAK,EAAE,aAAa;oBACjC,MAAM,EAAE,IAAI,CAAC,KAAK,EAAE,iBAAiB;oBACrC,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,YAAY;iBAClC;aACJ,CAAC;QACN,CAAC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,UAAU;QACZ,MAAM,MAAM,GAAuB,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAEtE,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACnC,OAAO;gBACH,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,EAAE;gBACV,WAAW,EAAE,SAAS;gBACtB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,KAAK,EAAE,CAAC,CAAC,QAAQ;aACpB,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,kBAAkB;QACd,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,OAA0B;QAC/C,MAAM,UAAU,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,IAAI,+BAA+B,CAAC;QAClE,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YAC1C,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CACX,0CAA0C,KAAK,CAAC,IAAI,yCAAyC,CAChG,CAAC;YACN,CAAC;YACD,OAAO,KAAK,CAAC,IAAI,CAAC;QACtB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACD,MAAM,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE;gBAChD,OAAO,EAAE;oBACL,KAAK;oBACL,KAAK,EAAE,KAAK;oBACZ,eAAe,EAAE,OAAO;iBAC3B;aACJ,CAAC,CAAsB,CAAC;YACzB,MAAM,OAAO,GAA6C,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;YACxG,MAAM,YAAY,GACd,CAAC,CAAC,KAAK,EAAE,YAAY;gBACrB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;YAE5F,OAAO;gBACH,KAAK;gBACL,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;oBAC7B,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;iBAC3D,CAAC,CAAC;gBACH,GAAG,CAAC,OAAO,YAAY,KAAK,QAAQ;oBAChC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE,iBAAiB,EAAE,YAAY,EAAE,EAAE;oBAC5E,CAAC,CAAC,EAAE,CAAC;aACZ,CAAC;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,eAAe,CAAC,iBAAiB,CAAC,KAAK,CAAC;gBAAE,MAAM,KAAK,CAAC;YAC1D,IAAI,KAAK,YAAY,KAAK,IAAI,OAAQ,KAA8B,CAAC,MAAM,KAAK,QAAQ;gBAAE,MAAM,KAAK,CAAC;YACtG,MAAM,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE;gBACpC,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,KAAK;gBACL,SAAS,EAAE,SAAS;aACvB,CAAC,CAAC;QACP,CAAC;IACL,CAAC;;AAGL;;;;;;;;;;;;;;;GAeG;AACH,SAAS,0BAA0B,CAAC,EAChC,KAAK,EACL,QAAQ,EACR,KAAK,EACL,WAAW,EACX,SAAS,EACT,IAAI,EACJ,UAAU,EACV,MAAM,EACN,UAAU,EACV,cAAc,EACd,aAAa,GACS;IACtB,OAAO;QACH,KAAK;QACL,QAAQ;QACR,KAAK,EAAE,KAAK,IAAI,SAAS;QACzB,WAAW,EAAE,WAAW,IAAI,SAAS;QACrC,UAAU,EAAE,SAAS,IAAI,SAAS;QAClC,KAAK,EAAE,IAAI,IAAI,SAAS;QACxB,WAAW,EAAE,UAAU,IAAI,SAAS;QACpC,MAAM,EAAE,MAAM,IAAI,SAAS;QAC3B,WAAW,EAAE,UAAU,IAAI,SAAS;QACpC,eAAe,EAAE,cAAc,IAAI,SAAS;QAC5C,IAAI,EAAE,aAAa,IAAI,SAAS;KACnC,CAAC;AACN,CAAC"}

package/lib/{types/mistral → mistral}/types.d.ts

@@ -94,6 +94,7 @@ export interface ChatCompletionResponseChunk {
 export interface Embedding {
     id: string;
     object: 'embedding';
+    index: number;
     embedding: number[];
 }
 export interface EmbeddingResponse {

package/lib/mistral/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/mistral/types.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,eAAe;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,kBAAkB,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB,EAAE,OAAO,CAAC;IAC7B,cAAc,EAAE,OAAO,CAAC;IACxB,cAAc,EAAE,OAAO,CAAC;IACxB,oBAAoB,EAAE,OAAO,CAAC;IAC9B,UAAU,EAAE,OAAO,CAAC;IACpB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,WAAW,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,KAAK;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,OAAO,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,UAAU,EAAE,eAAe,EAAE,CAAC;CACjC;AAED,MAAM,WAAW,kBAAkB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,KAAK,EAAE,CAAC;CACjB;AAED,MAAM,WAAW,QAAQ;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,oBAAY,QAAQ;IAChB,QAAQ,aAAa;CACxB;AAED,MAAM,WAAW,YAAY;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,SAAS;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,QAAQ,CAAC;IACf,QAAQ,EAAE,YAAY,CAAC;CAC1B;AAED,oBAAY,eAAe;IACvB,IAAI,SAAS;IACb,WAAW,gBAAgB;CAC9B;AAED,oBAAY,UAAU;IAClB,IAAI,SAAS;IACb,GAAG,QAAQ;IACX,IAAI,SAAS;CAChB;AAED,MAAM,WAAW,cAAc;IAC3B,IAAI,EAAE,eAAe,CAAC;CACzB;AAED,MAAM,WAAW,UAAU;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,4BAA4B;IACzC,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE;QACL,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,iCAAiC;IAC9C,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE;QACH,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,UAAU,CAAC,EAAE,SAAS,EAAE,CAAC;KAC5B,CAAC;IACF,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,sBAAsB;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,iBAAiB,CAAC;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,4BAA4B,EAAE,CAAC;IACxC,KAAK,EAAE,UAAU,CAAC;CACrB;AAED,MAAM,WAAW,2BAA2B;IACxC,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,uBAAuB,CAAC;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,iCAAiC,EAAE,CAAC;CAChD;AAED,MAAM,WAAW,SAAS;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,WAAW,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,iBAAiB;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,SAAS,EAAE,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,UAAU,CAAC;CACrB;AAED,MAAM,WAAW,uBAAuB;IACpC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;QAAC,UAAU,CAAC,EAAE,SAAS,EAAE,CAAA;KAAE,CAAC,CAAC;IACvG,KAAK,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,QAAQ,CAAA;KAAE,CAAC,CAAC;IACpD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;CAC5B"}

package/lib/mistral/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/mistral/types.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAC/E,kBAAkB;AAsClB,MAAM,CAAN,IAAY,QAEX;AAFD,WAAY,QAAQ;IAChB,iCAAqB,CAAA;AACzB,CAAC,EAFW,QAAQ,KAAR,QAAQ,QAEnB;AAaD,MAAM,CAAN,IAAY,eAGX;AAHD,WAAY,eAAe;IACvB,gCAAa,CAAA;IACb,8CAA2B,CAAA;AAC/B,CAAC,EAHW,eAAe,KAAf,eAAe,QAG1B;AAED,MAAM,CAAN,IAAY,UAIX;AAJD,WAAY,UAAU;IAClB,2BAAa,CAAA;IACb,yBAAW,CAAA;IACX,2BAAa,CAAA;AACjB,CAAC,EAJW,UAAU,KAAV,UAAU,QAIrB;AA6ED,wBAAwB;AACxB,uDAAuD;AAEvD,wBAAwB;AACxB,0BAA0B;AAC1B,wBAAwB;AACxB,2BAA2B;AAC3B,2BAA2B;AAE3B,0CAA0C;AAC1C,yBAAyB;AACzB,mHAAmH;AACnH,gEAAgE;AAChE,gCAAgC;AAChC,8BAA8B;AAC9B,yBAAyB;AACzB,+BAA+B;AAC/B,4BAA4B;AAC5B,cAAc;AACd,gDAAgD;AAChD,cAAc;AACd,8BAA8B;AAC9B,gCAAgC;AAChC,mCAAmC;AACnC,0CAA0C;AAC1C,iBAAiB;AAEjB,iDAAiD;AAEjD,sBAAsB;AACtB,yBAAyB;AACzB,mHAAmH;AACnH,gEAAgE;AAChE,gCAAgC;AAChC,8BAA8B;AAC9B,yBAAyB;AACzB,+BAA+B;AAC/B,cAAc;AACd,gDAAgD;AAChD,cAAc;AACd,8BAA8B;AAC9B,gCAAgC;AAChC,mCAAmC;AACnC,2CAA2C;AAC3C,2CAA2C;AAE3C,4BAA4B;AAC5B,yBAAyB;AACzB,mHAAmH;AACnH,gEAAgE;AAChE,gCAAgC;AAChC,8BAA8B;AAC9B,yBAAyB;AACzB,+BAA+B;AAC/B,cAAc;AACd,gDAAgD;AAChD,cAAc;AACd,8BAA8B;AAC9B,gCAAgC;AAChC,mCAAmC;AACnC,2CAA2C;AAC3C,sEAAsE;AAEtE,4BAA4B;AAC5B,yBAAyB;AACzB,oCAAoC;AACpC,sCAAsC;AACtC,IAAI"}

package/lib/{types/openai → openai}/azure_openai.d.ts

@@ -1,11 +1,12 @@
-import { AIModel, DriverOptions, Providers } from "@llumiverse/core";
-import OpenAI, { AzureOpenAI } from "openai";
-import { BaseOpenAIDriver } from "./index.js";
+import { type AIModel, type DriverOptions, Providers } from '@llumiverse/core';
+import type OpenAI from 'openai';
+import { AzureOpenAI } from 'openai';
+import { BaseOpenAIDriver } from './index.js';
 export interface AzureOpenAIDriverOptions extends DriverOptions {
     /**
      * The credentials to use to access Azure OpenAI
      */
-    azureADTokenProvider?: any;
+    azureADTokenProvider?: (options?: unknown) => Promise<string>;
     apiKey?: string;
     endpoint?: string;
     apiVersion?: string;

package/lib/openai/azure_openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"azure_openai.d.ts","sourceRoot":"","sources":["../../src/openai/azure_openai.ts"],"names":[],"mappings":"AACA,OAAO,EACH,KAAK,OAAO,EACZ,KAAK,aAAa,EAGlB,SAAS,EACZ,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AACrC,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAE9C,MAAM,WAAW,wBAAyB,SAAQ,aAAa;IAC3D;;OAEG;IACH,oBAAoB,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IAE9D,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,UAAU,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,iBAAkB,SAAQ,gBAAgB;IACnD,OAAO,EAAE,WAAW,CAAC;IACrB,QAAQ,CAAC,QAAQ,0BAA0B;gBAG/B,aAAa,EAAE,WAAW,GAAG,wBAAwB;IAsBjE;;OAEG;IACH,+BAA+B;IAMzB,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAIhC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,KAAK,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;CAgCvF"}

package/lib/{esm/openai → openai}/azure_openai.js

@@ -1,7 +1,7 @@
-import { DefaultAzureCredential, getBearerTokenProvider } from "@azure/identity";
-import { getModelCapabilities, modelModalitiesToArray, Providers } from "@llumiverse/core";
-import { AzureOpenAI } from "openai";
-import { BaseOpenAIDriver } from "./index.js";
+import { DefaultAzureCredential, getBearerTokenProvider } from '@azure/identity';
+import { getModelCapabilities, modelModalitiesToArray, Providers, } from '@llumiverse/core';
+import { AzureOpenAI } from 'openai';
+import { BaseOpenAIDriver } from './index.js';
 export class AzureOpenAIDriver extends BaseOpenAIDriver {
     service;
     provider = Providers.azure_openai;
@@ -21,15 +21,16 @@ export class AzureOpenAIDriver extends BaseOpenAIDriver {
             apiKey: opts.apiKey,
             azureADTokenProvider: opts.azureADTokenProvider,
             endpoint: opts.endpoint,
-            apiVersion: opts.apiVersion ?? "2024-10-21",
-            deployment: opts.deployment
+            apiVersion: opts.apiVersion ?? '2024-10-21',
+            deployment: opts.deployment,
+            fetch: this.getDriverFetch(),
         });
     }
     /**
      * Get default authentication for Azure Cognitive Services API
      */
     getDefaultCognitiveServicesAuth() {
-        const scope = "https://cognitiveservices.azure.com/.default";
+        const scope = 'https://cognitiveservices.azure.com/.default';
         const azureADTokenProvider = getBearerTokenProvider(new DefaultAzureCredential(), scope);
         return azureADTokenProvider;
     }
@@ -38,31 +39,33 @@ export class AzureOpenAIDriver extends BaseOpenAIDriver {
     }
     async _listModels(_filter) {
         if (!this.service.deploymentName) {
-            throw new Error("A specific deployment is not set. Azure OpenAI cannot list deployments. Update your endpoint URL to include the deployment name, e.g., https://your-resource.openai.azure.com/openai/deployments/your-deployment/chat/completions");
+            throw new Error('A specific deployment is not set. Azure OpenAI cannot list deployments. Update your endpoint URL to include the deployment name, e.g., https://your-resource.openai.azure.com/openai/deployments/your-deployment/chat/completions');
         }
         //Do a test execution to check if the model works and to get the model ID.
         let modelID = this.service.deploymentName;
         try {
             const testResponse = await this.service.chat.completions.create({
                 model: this.service.deploymentName,
-                messages: [{ role: "user", content: "Hi" }],
+                messages: [{ role: 'user', content: 'Hi' }],
                 max_tokens: 1,
             });
             modelID = testResponse.model;
         }
         catch (error) {
-            this.logger.error({ error }, "Failed to test model for Azure OpenAI listing :");
+            this.logger.error({ error }, 'Failed to test model for Azure OpenAI listing :');
         }
-        const modelCapability = getModelCapabilities(modelID, "openai");
-        return [{
+        const modelCapability = getModelCapabilities(modelID, 'openai');
+        return [
+            {
                 id: modelID,
                 name: this.service.deploymentName,
                 provider: this.provider,
-                owner: "openai",
+                owner: 'openai',
                 input_modalities: modelModalitiesToArray(modelCapability.input),
                 output_modalities: modelModalitiesToArray(modelCapability.output),
                 tool_support: modelCapability.tool_support,
-            }];
+            },
+        ];
     }
 }
 //# sourceMappingURL=azure_openai.js.map

package/lib/openai/azure_openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"azure_openai.js","sourceRoot":"","sources":["../../src/openai/azure_openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AACjF,OAAO,EAGH,oBAAoB,EACpB,sBAAsB,EACtB,SAAS,GACZ,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AACrC,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAiB9C,MAAM,OAAO,iBAAkB,SAAQ,gBAAgB;IACnD,OAAO,CAAc;IACZ,QAAQ,GAAG,SAAS,CAAC,YAAY,CAAC;IAE3C,sEAAsE;IACtE,YAAY,aAAqD;QAC7D,IAAI,aAAa,YAAY,WAAW,EAAE,CAAC;YACvC,KAAK,CAAC,EAAE,CAAC,CAAC;YACV,IAAI,CAAC,OAAO,GAAG,aAAa,CAAC;YAC7B,OAAO;QACX,CAAC;QACD,MAAM,IAAI,GAAG,aAAa,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,CAAC;QACZ,IAAI,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YAC7C,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,+BAA+B,EAAE,CAAC;QACvE,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,IAAI,WAAW,CAAC;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,oBAAoB,EAAE,IAAI,CAAC,oBAAoB;YAC/C,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,YAAY;YAC3C,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE;SAC/B,CAAC,CAAC;IACP,CAAC;IAED;;OAEG;IACH,+BAA+B;QAC3B,MAAM,KAAK,GAAG,8CAA8C,CAAC;QAC7D,MAAM,oBAAoB,GAAG,sBAAsB,CAAC,IAAI,sBAAsB,EAAE,EAAE,KAAK,CAAC,CAAC;QACzF,OAAO,oBAAoB,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,UAAU;QACZ,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAA6C;QAC3D,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CACX,mOAAmO,CACtO,CAAC;QACN,CAAC;QAED,0EAA0E;QAC1E,IAAI,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC;YACD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBAC5D,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAClC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;gBAC3C,UAAU,EAAE,CAAC;aAChB,CAAC,CAAC;YACH,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC;QACjC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,iDAAiD,CAAC,CAAC;QACpF,CAAC;QACD,MAAM,eAAe,GAAG,oBAAoB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAChE,OAAO;YACH;gBACI,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBACjC,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,KAAK,EAAE,QAAQ;gBACf,gBAAgB,EAAE,sBAAsB,CAAC,eAAe,CAAC,KAAK,CAAC;gBAC/D,iBAAiB,EAAE,sBAAsB,CAAC,eAAe,CAAC,MAAM,CAAC;gBACjE,YAAY,EAAE,eAAe,CAAC,YAAY;aACnB;SAC9B,CAAC;IACN,CAAC;CACJ"}

package/lib/openai/drivers.d.ts

@@ -0,0 +1,4 @@
+export { AzureOpenAIDriver, type AzureOpenAIDriverOptions } from './azure_openai.js';
+export { OpenAIDriver, type OpenAIDriverOptions } from './openai.js';
+export { OpenAICompatibleDriver, type OpenAICompatibleDriverOptions } from './openai_compatible.js';
+//# sourceMappingURL=drivers.d.ts.map

package/lib/openai/drivers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"drivers.d.ts","sourceRoot":"","sources":["../../src/openai/drivers.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,iBAAiB,EAAE,KAAK,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AACrF,OAAO,EAAE,YAAY,EAAE,KAAK,mBAAmB,EAAE,MAAM,aAAa,CAAC;AACrE,OAAO,EAAE,sBAAsB,EAAE,KAAK,6BAA6B,EAAE,MAAM,wBAAwB,CAAC"}

package/lib/openai/drivers.js

@@ -0,0 +1,9 @@
+// Combined entry for the concrete OpenAI-protocol drivers, exposed as the `@llumiverse/drivers/openai`
+// subpath so consumers can lazy-load the OpenAI family per-provider without pulling the whole driver
+// barrel (e.g. via `await import('@llumiverse/drivers/openai')`). Kept separate from ./index.ts — which exports the abstract
+// BaseOpenAIDriver that openai.ts/azure_openai.ts/openai_compatible.ts extend — to avoid an ESM
+// initialization cycle (those files import BaseOpenAIDriver from ./index.js).
+export { AzureOpenAIDriver } from './azure_openai.js';
+export { OpenAIDriver } from './openai.js';
+export { OpenAICompatibleDriver } from './openai_compatible.js';
+//# sourceMappingURL=drivers.js.map

package/lib/openai/drivers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"drivers.js","sourceRoot":"","sources":["../../src/openai/drivers.ts"],"names":[],"mappings":"AAAA,uGAAuG;AACvG,qGAAqG;AACrG,6HAA6H;AAC7H,gGAAgG;AAChG,8EAA8E;AAE9E,OAAO,EAAE,iBAAiB,EAAiC,MAAM,mBAAmB,CAAC;AACrF,OAAO,EAAE,YAAY,EAA4B,MAAM,aAAa,CAAC;AACrE,OAAO,EAAE,sBAAsB,EAAsC,MAAM,wBAAwB,CAAC"}

package/lib/{types/openai → openai}/index.d.ts

@@ -1,10 +1,12 @@
-import { AbstractDriver, AIModel, Completion, CompletionChunkObject, DataSource, DriverOptions, EmbeddingsOptions, EmbeddingsResult, ExecutionOptions, LlumiverseError, LlumiverseErrorContext, Providers, ToolUse, TrainingJob, TrainingOptions, TrainingPromptOptions } from "@llumiverse/core";
-import OpenAI, { AzureOpenAI } from "openai";
+import { type AIModel, type Completion, type CompletionChunkObject, type DataSource, type DriverOptions, type EmbeddingsOptions, type EmbeddingsResult, type ExecutionOptions, LlumiverseError, type LlumiverseErrorContext, type Providers, type ToolUse, type TrainingJob, type TrainingOptions, type TrainingPromptOptions } from '@llumiverse/core';
+import { AbstractDriver } from '@llumiverse/core/driver';
+import type OpenAI from 'openai';
+import type { AzureOpenAI } from 'openai';
 type ResponseInputItem = OpenAI.Responses.ResponseInputItem;
 export interface BaseOpenAIDriverOptions extends DriverOptions {
 }
 export declare abstract class BaseOpenAIDriver extends AbstractDriver<BaseOpenAIDriverOptions, ResponseInputItem[]> {
-    abstract provider: Providers.openai | Providers.azure_openai | Providers.xai | Providers.azure_foundry | Providers.openai_compatible;
+    abstract provider: Providers.openai | Providers.azure_openai | Providers.xai | Providers.azure_foundry | Providers.togetherai | Providers.openai_compatible;
     abstract service: OpenAI | AzureOpenAI;
     constructor(opts: BaseOpenAIDriverOptions);
     extractDataFromResponse(_options: ExecutionOptions, result: OpenAI.Responses.Response): Completion;
@@ -24,7 +26,7 @@ export declare abstract class BaseOpenAIDriver extends AbstractDriver<BaseOpenAI
     listTrainableModels(): Promise<AIModel<string>[]>;
     listModels(): Promise<AIModel[]>;
     _listModels(filter?: (m: OpenAI.Models.Model) => boolean): Promise<AIModel[]>;
-    generateEmbeddings({ text, image, model }: EmbeddingsOptions): Promise<EmbeddingsResult>;
+    generateEmbeddings(options: EmbeddingsOptions): Promise<EmbeddingsResult>;
     imageModels: string[];
     /**
      * Determine if a model is specifically an image generation model (not conversational image model)
@@ -104,13 +106,14 @@ export declare abstract class BaseOpenAIDriver extends AbstractDriver<BaseOpenAI
      */
     private isOpenAIErrorRetryable;
 }
+export declare function mapResponseStream(stream: AsyncIterable<OpenAI.Responses.ResponseStreamEvent>): AsyncIterable<CompletionChunkObject>;
 /**
  * Converts function_call and function_call_output items to text messages in OpenAI conversation.
  * Preserves tool call information while removing structured items that require
  * tools to be defined in the API request.
  */
 export declare function convertOpenAIFunctionItemsToText(items: ResponseInputItem[]): ResponseInputItem[];
-export declare function collectTools(output?: OpenAI.Responses.ResponseOutputItem[]): ToolUse[] | undefined;
+export declare function collectTools(output?: OpenAI.Responses.ResponseOutputItem[]): ToolUse<unknown>[] | undefined;
 /**
  * Fix orphaned function_call items in the OpenAI Responses API conversation.
  *
@@ -122,5 +125,14 @@ export declare function collectTools(output?: OpenAI.Responses.ResponseOutputIte
  * indicating the tools were interrupted, allowing the conversation to continue.
  */
 export declare function fixOrphanedToolUse(items: ResponseInputItem[]): ResponseInputItem[];
+/**
+ * Drop function_call_output items whose call_id has no matching function_call
+ * item anywhere in the input. Mirror of {@link fixOrphanedToolUse}: that
+ * synthesizes outputs for calls left unanswered (e.g. a cancelled run); this
+ * removes outputs left dangling after their function_call was dropped (e.g. by
+ * conversation compaction/trimming). The OpenAI Responses API requires every
+ * function_call_output to reference a prior function_call's call_id.
+ */
+export declare function fixOrphanedToolResults(items: ResponseInputItem[]): ResponseInputItem[];
 export {};
 //# sourceMappingURL=index.d.ts.map

package/lib/openai/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/openai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,KAAK,OAAO,EACZ,KAAK,UAAU,EACf,KAAK,qBAAqB,EAE1B,KAAK,UAAU,EACf,KAAK,aAAa,EAElB,KAAK,iBAAiB,EACtB,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EAMrB,eAAe,EACf,KAAK,sBAAsB,EAO3B,KAAK,SAAS,EAMd,KAAK,OAAO,EACZ,KAAK,WAAW,EAEhB,KAAK,eAAe,EACpB,KAAK,qBAAqB,EAG7B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AAoB1C,KAAK,iBAAiB,GAAG,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC;AAkE5D,MAAM,WAAW,uBAAwB,SAAQ,aAAa;CAAG;AAEjE,8BAAsB,gBAAiB,SAAQ,cAAc,CAAC,uBAAuB,EAAE,iBAAiB,EAAE,CAAC;IACvG,QAAQ,CAAC,QAAQ,EACX,SAAS,CAAC,MAAM,GAChB,SAAS,CAAC,YAAY,GACtB,SAAS,CAAC,GAAG,GACb,SAAS,CAAC,aAAa,GACvB,SAAS,CAAC,UAAU,GACpB,SAAS,CAAC,iBAAiB,CAAC;IAClC,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW,CAAC;gBAE3B,IAAI,EAAE,uBAAuB;IAKzC,uBAAuB,CAAC,QAAQ,EAAE,gBAAgB,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,QAAQ,GAAG,UAAU;IAoB5F,2BAA2B,CAC7B,MAAM,EAAE,iBAAiB,EAAE,EAC3B,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;IAoE1C,qBAAqB,CAAC,MAAM,EAAE,iBAAiB,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IAiGxG,SAAS,CAAC,SAAS,CAAC,QAAQ,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC;IAkBjE;;;OAGG;IACH,0BAA0B,CACtB,MAAM,EAAE,iBAAiB,EAAE,EAC3B,MAAM,EAAE,OAAO,EAAE,EACjB,OAAO,EAAE,OAAO,EAAE,GAAG,SAAS,EAC9B,OAAO,EAAE,gBAAgB,GAC1B,iBAAiB,EAAE,GAAG,SAAS;IAkDlC,oBAAoB,CAAC,OAAO,EAAE,qBAAqB,GAAG,OAAO,CAAC,MAAM,CAAC;IAS/D,aAAa,CAAC,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,WAAW,CAAC;IAgBlF,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAKnD,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAOnD,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC;IAS5C,mBAAmB,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IAI3C,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAIhC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,KAAK,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAyD7E,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA+C/E,WAAW,WAA4C;IAEvD;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAMpC;;;OAGG;IACG,sBAAsB,CAAC,MAAM,EAAE,iBAAiB,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IA2FzG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAiCG;IACI,qBAAqB,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,sBAAsB,GAAG,eAAe;IA2D9F;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAMxB;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2BG;IACH,OAAO,CAAC,sBAAsB;CA+DjC;AAgGD,wBAAgB,iBAAiB,CAC7B,MAAM,EAAE,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC,mBAAmB,CAAC,GAC5D,aAAa,CAAC,qBAAqB,CAAC,CA4FtC;AAqDD;;;;GAIG;AACH,wBAAgB,gCAAgC,CAAC,KAAK,EAAE,iBAAiB,EAAE,GAAG,iBAAiB,EAAE,CA2BhG;AA4CD,wBAAgB,YAAY,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC,EAAE,GAAG,SAAS,CAoB3G;AAqJD;;;;;;;;;GASG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,iBAAiB,EAAE,GAAG,iBAAiB,EAAE,CAqDlF;AAED;;;;;;;GAOG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,iBAAiB,EAAE,GAAG,iBAAiB,EAAE,CActF"}