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

package/lib/vertexai/models/llama.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llama.js","sourceRoot":"","sources":["../../../src/vertexai/models/llama.ts"],"names":[],"mappings":"AAAA,OAAO,EAKH,SAAS,EAET,UAAU,GAGb,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAwD5D;;GAEG;AACH,KAAK,UAAU,cAAc,CAAC,MAAmD;IAC7E,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC/B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IACpC,CAAC;IACD,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AACnD,CAAC;AAED;;;;;GAKG;AACH,SAAS,kBAAkB,CAAC,YAA4C,EAAE,MAAmB;IACzF,MAAM,YAAY,GAAG,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;IAE/D,OAAO;QACH,QAAQ,EAAE,CAAC,GAAG,YAAY,EAAE,GAAG,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;KAC1D,CAAC;AACN,CAAC;AAED,MAAM,OAAO,oBAAoB;IAC7B,KAAK,CAAU;IAEf,YAAY,OAAe;QACvB,IAAI,CAAC,KAAK,GAAG;YACT,EAAE,EAAE,OAAO;YACX,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,UAAU;YACpB,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,UAAU,EAAE,IAAI;SACR,CAAC;IACjB,CAAC;IAED,yDAAyD;IACzD,mBAAmB,CAAC,SAAiB;QACjC,sEAAsE;QACtE,IAAI,SAAS,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAClC,OAAO,UAAU,CAAC;QACtB,CAAC;aAAM,CAAC;YACJ,OAAO,aAAa,CAAC;QACzB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CACd,OAAuB,EACvB,QAAyB,EACzB,OAAsB;QAEtB,MAAM,QAAQ,GAAmB,EAAE,CAAC;QAEpC,6DAA6D;QAC7D,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,0CAA0C;YAC1C,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,KAAK,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC;YAE1E,2CAA2C;YAC3C,IAAI,cAAc,GAAG,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;YAE3C,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5C,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;oBAC/B,IAAI,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;wBACtC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;wBAC1C,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;wBACrD,cAAc,IAAI,sBAAsB,WAAW,EAAE,CAAC;oBAC1D,CAAC;gBACL,CAAC;YACL,CAAC;YAED,QAAQ,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,IAAI;gBACV,OAAO,EAAE,cAAc;aAC1B,CAAC,CAAC;QACP,CAAC;QAED,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,QAAQ,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,sEAAsE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE;aACzH,CAAC,CAAC;QACP,CAAC;QAED,6DAA6D;QAC7D,OAAO;YACH,QAAQ,EAAE,QAAQ;SACrB,CAAC;IACN,CAAC;IAED,KAAK,CAAC,qBAAqB,CACvB,MAAsB,EACtB,MAAmB,EACnB,OAAyB;QAEzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAE5C,IAAI,YAAY,GAAG,kBAAkB,CAAC,OAAO,CAAC,YAA2B,EAAE,MAAM,CAAC,CAAC;QAEnF,MAAM,YAAY,GAAG,OAAO,CAAC,aAAoC,CAAC;QAElE,MAAM,OAAO,GAA4B;YACrC,KAAK,EAAE,QAAQ,SAAS,EAAE;YAC1B,QAAQ,EAAE,YAAY,CAAC,QAAQ;YAC/B,MAAM,EAAE,KAAK;YACb,UAAU,EAAE,YAAY,EAAE,UAAU;YACpC,WAAW,EAAE,YAAY,EAAE,WAAW;YACtC,KAAK,EAAE,YAAY,EAAE,KAAK;YAC1B,KAAK,EAAE,YAAY,EAAE,KAAK;YAC1B,sBAAsB;YACtB,UAAU,EAAE;gBACR,MAAM,EAAE;oBACJ,qBAAqB,EAAE;wBACnB,OAAO,EAAE,KAAK;wBACd,oBAAoB,EAAE,EAAE;qBAC3B;iBACJ;aACJ;SACJ,CAAC;QAEF,0CAA0C;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC7C,MAAM,cAAc,GAAG,oCAAoC,CAAC;QAC5D,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE;YAC9C,OAAO;SACV,CAAC,CAAkB,CAAC;QAErB,wBAAwB;QACxB,MAAM,gBAAgB,GAAG,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;QACrD,MAAM,IAAI,GAAG,gBAAgB,EAAE,OAAO,CAAC;QAEvC,wCAAwC;QACxC,YAAY,GAAG,kBAAkB,CAAC,YAAY,EAAE;YAC5C,QAAQ,EAAE;gBACN;oBACI,IAAI,EAAE,gBAAgB,EAAE,IAAI;oBAC5B,OAAO,EAAE,IAAI;iBAChB;aACJ;SACJ,CAAC,CAAC;QAEH,OAAO;YACH,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;YACvC,WAAW,EAAE;gBACT,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,aAAa;gBAClC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,iBAAiB;gBACtC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY;aACnC;YACD,aAAa,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa;YAC9C,YAAY;SACf,CAAC;IACN,CAAC;IAED,KAAK,CAAC,2BAA2B,CAC7B,MAAsB,EACtB,MAAmB,EACnB,OAAyB;QAEzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAE5C,MAAM,YAAY,GAAG,kBAAkB,CAAC,OAAO,CAAC,YAA2B,EAAE,MAAM,CAAC,CAAC;QAErF,MAAM,YAAY,GAAG,OAAO,CAAC,aAAoC,CAAC;QAElE,MAAM,OAAO,GAA4B;YACrC,KAAK,EAAE,QAAQ,SAAS,EAAE;YAC1B,QAAQ,EAAE,YAAY,CAAC,QAAQ;YAC/B,MAAM,EAAE,IAAI;YACZ,UAAU,EAAE,YAAY,EAAE,UAAU;YACpC,WAAW,EAAE,YAAY,EAAE,WAAW;YACtC,KAAK,EAAE,YAAY,EAAE,KAAK;YAC1B,KAAK,EAAE,YAAY,EAAE,KAAK;YAC1B,sBAAsB;YACtB,UAAU,EAAE;gBACR,MAAM,EAAE;oBACJ,qBAAqB,EAAE;wBACnB,OAAO,EAAE,KAAK;wBACd,oBAAoB,EAAE,EAAE;qBAC3B;iBACJ;aACJ;SACJ,CAAC;QAEF,0CAA0C;QAC1C,qFAAqF;QACrF,yBAAyB;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC7C,MAAM,cAAc,GAAG,oCAAoC,CAAC;QAC5D,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE;YAC9C,OAAO;YACP,MAAM,EAAE,KAAK;SAChB,CAAC,CAAoC,CAAC;QAEvC,OAAO,kBAAkB,CAAC,MAAM,EAAE,CAAC,IAAY,EAAyB,EAAE;YACtE,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAwB,CAAC;YACrD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YACjC,MAAM,OAAO,GAAG,MAAM,EAAE,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;YAC7C,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,MAAM,EAAE,aAAa;gBACpC,WAAW,EAAE,IAAI,CAAC,KAAK;oBACnB,CAAC,CAAC;wBACI,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;wBAChC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;wBACpC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;qBACjC;oBACH,CAAC,CAAC,SAAS;aAClB,CAAC;QACN,CAAC,CAAC,CAAC;IACP,CAAC;CACJ"}

package/lib/vertexai/models.d.ts

@@ -0,0 +1,21 @@
+import type { AIModel, Completion, CompletionChunkObject, ExecutionOptions, LlumiverseError, LlumiverseErrorContext, PromptSegment } from '@llumiverse/core';
+import type { VertexAIDriver, VertexAIPrompt } from './index.js';
+export declare function trimModelName(model: string): string;
+export interface ModelDefinition<PromptT = VertexAIPrompt> {
+    model: AIModel;
+    versions?: string[];
+    createPrompt(driver: VertexAIDriver, segments: PromptSegment[], options: ExecutionOptions): Promise<PromptT>;
+    requestTextCompletion(driver: VertexAIDriver, prompt: PromptT, options: ExecutionOptions): Promise<Completion>;
+    requestTextCompletionStream(driver: VertexAIDriver, prompt: PromptT, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>;
+    preValidationProcessing?(result: Completion, options: ExecutionOptions): {
+        result: Completion;
+        options: ExecutionOptions;
+    };
+    /**
+     * Format provider-specific errors into standardized LlumiverseError.
+     * Optional - if not provided, VertexAIDriver will use default error handling.
+     */
+    formatLlumiverseError?(driver: VertexAIDriver, error: unknown, context: LlumiverseErrorContext): LlumiverseError;
+}
+export declare function getModelDefinition(model: string): ModelDefinition;
+//# sourceMappingURL=models.d.ts.map

package/lib/vertexai/models.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../src/vertexai/models.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,OAAO,EACP,UAAU,EACV,qBAAqB,EACrB,gBAAgB,EAChB,eAAe,EACf,sBAAsB,EACtB,aAAa,EAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAKjE,wBAAgB,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAGnD;AAED,MAAM,WAAW,eAAe,CAAC,OAAO,GAAG,cAAc;IACrD,KAAK,EAAE,OAAO,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,YAAY,CAAC,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC7G,qBAAqB,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;IAC/G,2BAA2B,CACvB,MAAM,EAAE,cAAc,EACtB,MAAM,EAAE,OAAO,EACf,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC,CAAC;IACjD,uBAAuB,CAAC,CACpB,MAAM,EAAE,UAAU,EAClB,OAAO,EAAE,gBAAgB,GAC1B;QAAE,MAAM,EAAE,UAAU,CAAC;QAAC,OAAO,EAAE,gBAAgB,CAAA;KAAE,CAAC;IACrD;;;OAGG;IACH,qBAAqB,CAAC,CAAC,MAAM,EAAE,cAAc,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,sBAAsB,GAAG,eAAe,CAAC;CACpH;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,eAAe,CA2BjE"}

package/lib/{esm/vertexai → vertexai}/models.js

@@ -1,13 +1,16 @@
-import { trimModelName } from "./index.js";
-import { ClaudeModelDefinition } from "./models/claude.js";
-import { GeminiModelDefinition } from "./models/gemini.js";
-import { LLamaModelDefinition } from "./models/llama.js";
+import { ClaudeModelDefinition } from './models/claude.js';
+import { GeminiModelDefinition } from './models/gemini.js';
+import { LLamaModelDefinition } from './models/llama.js';
+export function trimModelName(model) {
+    const i = model.lastIndexOf('@');
+    return i > -1 ? model.substring(0, i) : model;
+}
 export function getModelDefinition(model) {
-    const splits = model.split("/");
+    const splits = model.split('/');
     // Handle both formats: "publishers/anthropic/models/..." and "locations/.../publishers/anthropic/models/..."
     let publisher;
     let modelName;
-    const publisherIndex = splits.indexOf("publishers");
+    const publisherIndex = splits.indexOf('publishers');
     if (publisherIndex !== -1 && publisherIndex + 1 < splits.length) {
         publisher = splits[publisherIndex + 1];
         modelName = trimModelName(splits[splits.length - 1]);
@@ -17,13 +20,13 @@ export function getModelDefinition(model) {
         publisher = splits[1];
         modelName = trimModelName(splits[splits.length - 1]);
     }
-    if (publisher?.includes("anthropic")) {
+    if (publisher?.includes('anthropic')) {
         return new ClaudeModelDefinition(modelName);
     }
-    else if (publisher?.includes("google")) {
+    else if (publisher?.includes('google')) {
         return new GeminiModelDefinition(modelName);
     }
-    else if (publisher?.includes("meta")) {
+    else if (publisher?.includes('meta')) {
         return new LLamaModelDefinition(modelName);
     }
     //Fallback, assume it is Gemini.

package/lib/vertexai/models.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/vertexai/models.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAEzD,MAAM,UAAU,aAAa,CAAC,KAAa;IACvC,MAAM,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IACjC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;AAClD,CAAC;AAuBD,MAAM,UAAU,kBAAkB,CAAC,KAAa;IAC5C,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAEhC,6GAA6G;IAC7G,IAAI,SAA6B,CAAC;IAClC,IAAI,SAAiB,CAAC;IAEtB,MAAM,cAAc,GAAG,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IACpD,IAAI,cAAc,KAAK,CAAC,CAAC,IAAI,cAAc,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QAC9D,SAAS,GAAG,MAAM,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC;QACvC,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC;SAAM,CAAC;QACJ,mDAAmD;QACnD,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACtB,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC;IAED,IAAI,SAAS,EAAE,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;QACnC,OAAO,IAAI,qBAAqB,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC;SAAM,IAAI,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QACvC,OAAO,IAAI,qBAAqB,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC;SAAM,IAAI,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QACrC,OAAO,IAAI,oBAAoB,CAAC,SAAS,CAAC,CAAC;IAC/C,CAAC;IAED,gCAAgC;IAChC,OAAO,IAAI,qBAAqB,CAAC,SAAS,CAAC,CAAC;AAChD,CAAC"}

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

@@ -1,6 +1,7 @@
-import { AbstractDriver, AIModel, Completion, CompletionChunkObject, DriverOptions, EmbeddingsOptions, EmbeddingsResult, ExecutionOptions } from "@llumiverse/core";
-import { FetchClient } from "@vertesia/api-fetch-client";
-import { WatsonAuthToken } from "./interfaces.js";
+import { type AIModel, type Completion, type CompletionChunkObject, type DriverOptions, type EmbeddingsOptions, type EmbeddingsResult, type ExecutionOptions } from '@llumiverse/core';
+import { AbstractDriver } from '@llumiverse/core/driver';
+import { FetchClient } from '@vertesia/api-fetch-client';
+import type { WatsonAuthToken } from './interfaces.js';
 interface WatsonxDriverOptions extends DriverOptions {
     apiKey: string;
     projectId: string;

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/watsonx/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,KAAK,OAAO,EACZ,KAAK,UAAU,EACf,KAAK,qBAAqB,EAC1B,KAAK,aAAa,EAClB,KAAK,iBAAiB,EACtB,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EAKxB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAwB,MAAM,4BAA4B,CAAC;AAC/E,OAAO,KAAK,EAGR,eAAe,EAKlB,MAAM,iBAAiB,CAAC;AAEzB,UAAU,oBAAqB,SAAQ,aAAa;IAChD,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;CACvB;AAID,qBAAa,aAAc,SAAQ,cAAc,CAAC,oBAAoB,EAAE,MAAM,CAAC;IAC3E,MAAM,CAAC,QAAQ,SAAa;IAC5B,QAAQ,SAA0B;IAClC,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,eAAe,CAAC;IAC5B,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,WAAW,EAAE,WAAW,CAAC;gBAEb,OAAO,EAAE,oBAAoB;IAUnC,qBAAqB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IAqCrF,2BAA2B,CAC7B,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;IAuC1C,UAAU,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IAiBxC,YAAY,IAAI,OAAO,CAAC,MAAM,CAAC;IAqB/B,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC;IAStC,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAsClF"}

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

@@ -1,9 +1,10 @@
-import { AbstractDriver } from "@llumiverse/core";
-import { transformSSEStream } from "@llumiverse/core/async";
-import { FetchClient } from "@vertesia/api-fetch-client";
-const API_VERSION = "2024-03-14";
+import { LlumiverseError, normalizeEmbeddingsOptions, WATSONX_DEFAULT_EMBEDDING_MODEL, } from '@llumiverse/core';
+import { transformSSEStream } from '@llumiverse/core/async';
+import { AbstractDriver } from '@llumiverse/core/driver';
+import { FetchClient } from '@vertesia/api-fetch-client';
+const API_VERSION = '2024-03-14';
 export class WatsonxDriver extends AbstractDriver {
-    static PROVIDER = "watsonx";
+    static PROVIDER = 'watsonx';
     provider = WatsonxDriver.PROVIDER;
     apiKey;
     endpoint_url;
@@ -16,16 +17,16 @@ export class WatsonxDriver extends AbstractDriver {
         this.apiKey = options.apiKey;
         this.projectId = options.projectId;
         this.endpoint_url = options.endpointUrl;
-        this.fetchClient = new FetchClient(this.endpoint_url).withAuthCallback(async () => this.getAuthToken().then(token => `Bearer ${token}`));
+        this.fetchClient = new FetchClient(this.endpoint_url, this.getDriverFetch()).withAuthCallback(async () => this.getAuthToken().then((token) => `Bearer ${token}`));
     }
     async requestTextCompletion(prompt, 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 payload = {
             model_id: options.model,
-            input: prompt + "\n",
+            input: `${prompt}\n`,
             parameters: {
                 max_new_tokens: options.model_options?.max_tokens,
                 temperature: options.model_options?.temperature,
@@ -35,10 +36,12 @@ export class WatsonxDriver extends AbstractDriver {
             },
             project_id: this.projectId,
         };
-        const res = await this.fetchClient.post(`/ml/v1/text/generation?version=${API_VERSION}`, { payload });
+        const res = (await this.fetchClient.post(`/ml/v1/text/generation?version=${API_VERSION}`, {
+            payload,
+        }));
         const result = res.results[0];
         return {
-            result: [{ type: "text", value: result.generated_text }],
+            result: [{ type: 'text', value: result.generated_text }],
             token_usage: {
                 prompt: result.input_token_count,
                 result: result.generated_token_count,
@@ -49,13 +52,13 @@ export class WatsonxDriver extends AbstractDriver {
         };
     }
     async requestTextCompletionStream(prompt, 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 payload = {
             model_id: options.model,
-            input: prompt + "\n",
+            input: `${prompt}\n`,
             parameters: {
                 max_new_tokens: options.model_options?.max_tokens,
                 temperature: options.model_options?.temperature,
@@ -65,14 +68,16 @@ export class WatsonxDriver extends AbstractDriver {
             },
             project_id: this.projectId,
         };
-        const stream = await this.fetchClient.post(`/ml/v1/text/generation_stream?version=${API_VERSION}`, {
+        const stream = (await this.fetchClient.post(`/ml/v1/text/generation_stream?version=${API_VERSION}`, {
             payload: payload,
-            reader: 'sse'
-        });
+            reader: 'sse',
+        }));
         return transformSSEStream(stream, (data) => {
             const json = JSON.parse(data);
             return {
-                result: json.results[0]?.generated_text ? [{ type: "text", value: json.results[0].generated_text }] : [],
+                result: json.results[0]?.generated_text
+                    ? [{ type: 'text', value: json.results[0].generated_text }]
+                    : [],
                 finish_reason: watsonFinishReason(json.results[0]?.stop_reason),
                 token_usage: {
                     prompt: json.results[0].input_token_count,
@@ -83,8 +88,9 @@ export class WatsonxDriver extends AbstractDriver {
         });
     }
     async listModels() {
-        const res = await this.fetchClient.get(`/ml/v1/foundation_model_specs?version=${API_VERSION}`)
-            .catch(err => this.logger.warn("Can't list models on Watsonx: " + err));
+        const res = (await this.fetchClient
+            .get(`/ml/v1/foundation_model_specs?version=${API_VERSION}`)
+            .catch((err) => this.logger.warn(`Can't list models on Watsonx: ${err}`)));
         const aiModels = res.resources.map((m) => {
             return {
                 id: m.model_id,
@@ -102,16 +108,16 @@ export class WatsonxDriver extends AbstractDriver {
                 return this.authToken.access_token;
             }
             else {
-                this.logger.debug("Token expired, refetching");
+                this.logger.debug('Token expired, refetching');
             }
         }
-        const authToken = await fetch('https://iam.cloud.ibm.com/identity/token', {
+        const authToken = (await this.getDriverFetch()('https://iam.cloud.ibm.com/identity/token', {
             method: 'POST',
             headers: {
                 'Content-Type': 'application/x-www-form-urlencoded',
             },
             body: `grant_type=urn:ibm:params:oauth:grant-type:apikey&apikey=${this.apiKey}`,
-        }).then(response => response.json());
+        }).then((response) => response.json()));
         this.authToken = authToken;
         return this.authToken.access_token;
     }
@@ -119,36 +125,58 @@ export class WatsonxDriver extends AbstractDriver {
         return this.listModels()
             .then(() => true)
             .catch((err) => {
-            this.logger.warn({ error: err }, "Failed to connect to WatsonX");
+            this.logger.warn({ error: err }, 'Failed to connect to WatsonX');
             return false;
         });
     }
     async generateEmbeddings(options) {
-        if (options.image) {
-            throw new Error("Image embeddings not supported by Watsonx");
-        }
-        if (!options.text) {
-            throw new Error("No text provided");
-        }
+        const normalized = normalizeEmbeddingsOptions(options);
+        const texts = normalized.inputs.map((input) => {
+            if (input.type !== 'text') {
+                throw new Error(`Provider 'watsonx' does not support '${input.type}' embeddings; only 'text' is supported.`);
+            }
+            return input.text;
+        });
         const payload = {
-            inputs: [options.text],
-            model_id: options.model ?? 'ibm/slate-125m-english-rtrvr',
-            project_id: this.projectId
-        };
-        const res = await this.fetchClient.post(`/ml/v1/text/embeddings?version=${API_VERSION}`, { payload });
-        return {
-            values: res.results[0].embedding,
-            model: res.model_id
+            inputs: texts,
+            model_id: normalized.model ?? WATSONX_DEFAULT_EMBEDDING_MODEL,
+            project_id: this.projectId,
         };
+        const model = payload.model_id;
+        try {
+            const res = (await this.fetchClient.post(`/ml/v1/text/embeddings?version=${API_VERSION}`, {
+                payload,
+            }));
+            return {
+                model: res.model_id,
+                results: res.results.map((entry) => ({
+                    outputs: [{ values: entry.embedding, modality: 'text' }],
+                })),
+            };
+        }
+        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',
+            });
+        }
     }
 }
 function watsonFinishReason(reason) {
     if (!reason)
         return undefined;
     switch (reason) {
-        case 'eos_token': return "stop";
-        case 'max_tokens': return "length";
-        default: return reason;
+        case 'eos_token':
+            return 'stop';
+        case 'max_tokens':
+            return 'length';
+        default:
+            return reason;
     }
 }
 /*interface ListModelsParams extends ModelSearchPayload {

package/lib/watsonx/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/watsonx/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAQH,eAAe,EACf,0BAA0B,EAE1B,+BAA+B,GAClC,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAwB,MAAM,4BAA4B,CAAC;AAiB/E,MAAM,WAAW,GAAG,YAAY,CAAC;AAEjC,MAAM,OAAO,aAAc,SAAQ,cAA4C;IAC3E,MAAM,CAAC,QAAQ,GAAG,SAAS,CAAC;IAC5B,QAAQ,GAAG,aAAa,CAAC,QAAQ,CAAC;IAClC,MAAM,CAAS;IACf,YAAY,CAAS;IACrB,SAAS,CAAS;IAClB,SAAS,CAAmB;IAC5B,OAAO,CAAe;IACtB,WAAW,CAAc;IAEzB,YAAY,OAA6B;QACrC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;QACxC,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE,CACrG,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,UAAU,KAAK,EAAE,CAAC,CACzD,CAAC;IACN,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,MAAc,EAAE,OAAyB;QACjE,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,aAAgD,CAAC;QAEjF,MAAM,OAAO,GAAiC;YAC1C,QAAQ,EAAE,OAAO,CAAC,KAAK;YACvB,KAAK,EAAE,GAAG,MAAM,IAAI;YACpB,UAAU,EAAE;gBACR,cAAc,EAAE,OAAO,CAAC,aAAa,EAAE,UAAU;gBACjD,WAAW,EAAE,OAAO,CAAC,aAAa,EAAE,WAAW;gBAC/C,KAAK,EAAE,OAAO,CAAC,aAAa,EAAE,KAAK;gBACnC,KAAK,EAAE,OAAO,CAAC,aAAa,EAAE,KAAK;gBACnC,cAAc,EAAE,OAAO,CAAC,aAAa,EAAE,aAAa;aACvD;YACD,UAAU,EAAE,IAAI,CAAC,SAAS;SAC7B,CAAC;QAEF,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,kCAAkC,WAAW,EAAE,EAAE;YACtF,OAAO;SACV,CAAC,CAAkC,CAAC;QAErC,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAE9B,OAAO;YACH,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,cAAc,EAAE,CAAC;YACxD,WAAW,EAAE;gBACT,MAAM,EAAE,MAAM,CAAC,iBAAiB;gBAChC,MAAM,EAAE,MAAM,CAAC,qBAAqB;gBACpC,KAAK,EAAE,MAAM,CAAC,iBAAiB,GAAG,MAAM,CAAC,qBAAqB;aACjE;YACD,aAAa,EAAE,kBAAkB,CAAC,MAAM,CAAC,WAAW,CAAC;YACrD,iBAAiB,EAAE,OAAO,CAAC,yBAAyB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS;SACzE,CAAC;IACN,CAAC;IAED,KAAK,CAAC,2BAA2B,CAC7B,MAAc,EACd,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,aAAgD,CAAC;QACjF,MAAM,OAAO,GAAiC;YAC1C,QAAQ,EAAE,OAAO,CAAC,KAAK;YACvB,KAAK,EAAE,GAAG,MAAM,IAAI;YACpB,UAAU,EAAE;gBACR,cAAc,EAAE,OAAO,CAAC,aAAa,EAAE,UAAU;gBACjD,WAAW,EAAE,OAAO,CAAC,aAAa,EAAE,WAAW;gBAC/C,KAAK,EAAE,OAAO,CAAC,aAAa,EAAE,KAAK;gBACnC,KAAK,EAAE,OAAO,CAAC,aAAa,EAAE,KAAK;gBACnC,cAAc,EAAE,OAAO,CAAC,aAAa,EAAE,aAAa;aACvD;YACD,UAAU,EAAE,IAAI,CAAC,SAAS;SAC7B,CAAC;QAEF,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,yCAAyC,WAAW,EAAE,EAAE;YAChG,OAAO,EAAE,OAAO;YAChB,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,CAAkC,CAAC;YAC/D,OAAO;gBACH,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,cAAc;oBACnC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC;oBAC3D,CAAC,CAAC,EAAE;gBACR,aAAa,EAAE,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC;gBAC/D,WAAW,EAAE;oBACT,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,iBAAiB;oBACzC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,qBAAqB;oBAC7C,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,iBAAiB,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,qBAAqB;iBACnF;aACJ,CAAC;QACN,CAAC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,UAAU;QACZ,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,WAAW;aAC9B,GAAG,CAAC,yCAAyC,WAAW,EAAE,CAAC;aAC3D,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,GAAG,EAAE,CAAC,CAAC,CAA6B,CAAC;QAE3G,MAAM,QAAQ,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAmB,EAAE,EAAE;YACvD,OAAO;gBACH,EAAE,EAAE,CAAC,CAAC,QAAQ;gBACd,IAAI,EAAE,CAAC,CAAC,KAAK;gBACb,WAAW,EAAE,CAAC,CAAC,iBAAiB;gBAChC,QAAQ,EAAE,IAAI,CAAC,QAAQ;aAC1B,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,YAAY;QACd,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YAC9B,IAAI,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;gBAClC,OAAO,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;YACvC,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;YACnD,CAAC;QACL,CAAC;QACD,MAAM,SAAS,GAAG,CAAC,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC,0CAA0C,EAAE;YACvF,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACL,cAAc,EAAE,mCAAmC;aACtD;YACD,IAAI,EAAE,4DAA4D,IAAI,CAAC,MAAM,EAAE;SAClF,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAoB,CAAC;QAE3D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,OAAO,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,kBAAkB;QACpB,OAAO,IAAI,CAAC,UAAU,EAAE;aACnB,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;aAChB,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACX,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,8BAA8B,CAAC,CAAC;YACjE,OAAO,KAAK,CAAC;QACjB,CAAC,CAAC,CAAC;IACX,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,OAA0B;QAC/C,MAAM,UAAU,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;QACvD,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,wCAAwC,KAAK,CAAC,IAAI,yCAAyC,CAC9F,CAAC;YACN,CAAC;YACD,OAAO,KAAK,CAAC,IAAI,CAAC;QACtB,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAA6B;YACtC,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,UAAU,CAAC,KAAK,IAAI,+BAA+B;YAC7D,UAAU,EAAE,IAAI,CAAC,SAAS;SAC7B,CAAC;QAEF,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC;YACD,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,kCAAkC,WAAW,EAAE,EAAE;gBACtF,OAAO;aACV,CAAC,CAA8B,CAAC;YACjC,OAAO;gBACH,KAAK,EAAE,GAAG,CAAC,QAAQ;gBACnB,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;oBACjC,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;iBAC3D,CAAC,CAAC;aACN,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,SAAS,kBAAkB,CAAC,MAA0B;IAClD,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAC9B,QAAQ,MAAM,EAAE,CAAC;QACb,KAAK,WAAW;YACZ,OAAO,MAAM,CAAC;QAClB,KAAK,YAAY;YACb,OAAO,QAAQ,CAAC;QACpB;YACI,OAAO,MAAM,CAAC;IACtB,CAAC;AACL,CAAC;AAED;;GAEG"}

package/lib/watsonx/interfaces.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"interfaces.d.ts","sourceRoot":"","sources":["../../src/watsonx/interfaces.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,4BAA4B;IACzC,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE;QACR,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;QAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,KAAK,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,6BAA6B;IAC1C,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE;QACL,cAAc,EAAE,MAAM,CAAC;QACvB,qBAAqB,EAAE,MAAM,CAAC;QAC9B,iBAAiB,EAAE,MAAM,CAAC;QAC1B,WAAW,EAAE,MAAM,CAAC;KACvB,EAAE,CAAC;CACP;AAED,MAAM,WAAW,wBAAwB;IACrC,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,yBAAyB;IACtC,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE;QACL,SAAS,EAAE,MAAM,EAAE,CAAC;KACvB,EAAE,CAAC;IACJ,iBAAiB,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,gBAAgB;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB,EAAE,MAAM,CAAC;IAC1B,KAAK,EAAE;QACH,EAAE,EAAE,MAAM,CAAC;QACX,OAAO,EAAE;YACL,OAAO,EAAE,MAAM,CAAC;SACnB,CAAC;KACL,EAAE,CAAC;IACJ,aAAa,EAAE,MAAM,CAAC;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,wBAAwB;IACrC,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,gBAAgB,EAAE,CAAC;CACjC;AAED,MAAM,WAAW,eAAe;IAC5B,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;CACtB"}

package/lib/{esm/watsonx → watsonx}/interfaces.js.map

@@ -1 +1 @@
-{"version":3,"file":"interfaces.js","sourceRoot":"","sources":["../../../src/watsonx/interfaces.ts"],"names":[],"mappings":""}
+{"version":3,"file":"interfaces.js","sourceRoot":"","sources":["../../src/watsonx/interfaces.ts"],"names":[],"mappings":""}

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

@@ -1,7 +1,7 @@
-import { AIModel, DriverOptions, PromptOptions, PromptSegment, Providers } from "@llumiverse/core";
-import { FetchClient } from "@vertesia/api-fetch-client";
-import OpenAI from "openai";
-import { BaseOpenAIDriver } from "../openai/index.js";
+import { type AIModel, type DriverOptions, type PromptOptions, type PromptSegment, Providers } from '@llumiverse/core';
+import { FetchClient } from '@vertesia/api-fetch-client';
+import OpenAI from 'openai';
+import { BaseOpenAIDriver } from '../openai/index.js';
 export interface xAiDriverOptions extends DriverOptions {
     apiKey: string;
     endpoint?: string;

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/xai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,OAAO,EAAE,KAAK,aAAa,EAAE,KAAK,aAAa,EAAE,KAAK,aAAa,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACvH,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAGtD,MAAM,WAAW,gBAAiB,SAAQ,aAAa;IACnD,MAAM,EAAE,MAAM,CAAC;IAEf,QAAQ,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,SAAU,SAAQ,gBAAgB;IAC3C,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,QAAQ,iBAAiB;IAClC,WAAW,EAAE,WAAW,CAAC;IACzB,gBAAgB,SAAyB;gBAE7B,IAAI,EAAE,gBAAgB;IAmB5B,aAAa,CACf,QAAQ,EAAE,aAAa,EAAE,EACzB,IAAI,EAAE,aAAa,GACpB,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,0BAA0B,EAAE,CAAC;IAmB1D,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;CAyBzC"}

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

@@ -1,46 +1,50 @@
-import { Providers } from "@llumiverse/core";
-import { formatOpenAILikeMultimodalPrompt } from "../openai/openai_format.js";
-import { FetchClient } from "@vertesia/api-fetch-client";
-import OpenAI from "openai";
-import { BaseOpenAIDriver } from "../openai/index.js";
+import { Providers } from '@llumiverse/core';
+import { FetchClient } from '@vertesia/api-fetch-client';
+import OpenAI from 'openai';
+import { BaseOpenAIDriver } from '../openai/index.js';
+import { formatOpenAILikeMultimodalPrompt } from '../openai/openai_format.js';
 export class xAIDriver extends BaseOpenAIDriver {
     service;
     provider = Providers.xai;
     xai_service;
-    DEFAULT_ENDPOINT = "https://api.x.ai/v1";
+    DEFAULT_ENDPOINT = 'https://api.x.ai/v1';
     constructor(opts) {
         super(opts);
         if (!opts.apiKey) {
-            throw new Error("apiKey is required");
+            throw new Error('apiKey is required');
         }
         this.service = new OpenAI({
             apiKey: opts.apiKey,
             baseURL: opts.endpoint ?? this.DEFAULT_ENDPOINT,
+            fetch: this.getDriverFetch(),
         });
-        this.xai_service = new FetchClient(opts.endpoint ?? this.DEFAULT_ENDPOINT).withAuthCallback(async () => `Bearer ${opts.apiKey}`);
+        this.xai_service = new FetchClient(opts.endpoint ?? this.DEFAULT_ENDPOINT, this.getDriverFetch()).withAuthCallback(async () => `Bearer ${opts.apiKey}`);
         //this.formatPrompt = this._formatPrompt; //TODO: fix xai prompt formatting
     }
     async _formatPrompt(segments, opts) {
         const options = {
-            multimodal: opts.model.includes("vision"),
+            multimodal: opts.model.includes('vision'),
             schema: opts.result_schema,
             useToolForFormatting: false,
         };
-        const p = await formatOpenAILikeMultimodalPrompt(segments, { ...options, ...opts });
+        const p = (await formatOpenAILikeMultimodalPrompt(segments, {
+            ...options,
+            ...opts,
+        }));
         return p;
     }
     // Note: We intentionally do NOT override extractDataFromResponse here.
     // The base class implementation properly handles tool_calls extraction.
     // xAI's API is OpenAI-compatible and returns tool_calls in the same format.
     async listModels() {
-        const [lm, em] = await Promise.all([
-            this.xai_service.get("/language-models"),
-            this.xai_service.get("/embedding-models")
-        ]);
-        em.models.forEach(m => {
-            m.output_modalities.push("vectors");
+        const [lm, em] = (await Promise.all([
+            this.xai_service.get('/language-models'),
+            this.xai_service.get('/embedding-models'),
+        ]));
+        em.models.forEach((m) => {
+            m.output_modalities.push('vectors');
         });
-        const models = [...lm.models, ...em.models].map(model => {
+        const models = [...lm.models, ...em.models].map((model) => {
             return {
                 id: model.id,
                 provider: this.provider,
@@ -49,7 +53,7 @@ export class xAIDriver extends BaseOpenAIDriver {
                 is_multimodal: model.input_modalities.length > 1,
                 input_modalities: model.input_modalities,
                 output_modalities: model.output_modalities,
-                tags: [...model.input_modalities.map(m => `i:${m}`), ...model.output_modalities.map(m => `o:${m}`)],
+                tags: [...model.input_modalities.map((m) => `i:${m}`), ...model.output_modalities.map((m) => `o:${m}`)],
             };
         });
         return models;

package/lib/xai/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/xai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAA4E,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACvH,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,gCAAgC,EAAqC,MAAM,4BAA4B,CAAC;AAQjH,MAAM,OAAO,SAAU,SAAQ,gBAAgB;IAC3C,OAAO,CAAS;IACP,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC;IAClC,WAAW,CAAc;IACzB,gBAAgB,GAAG,qBAAqB,CAAC;IAEzC,YAAY,IAAsB;QAC9B,KAAK,CAAC,IAAI,CAAC,CAAC;QAEZ,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAC1C,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,IAAI,MAAM,CAAC;YACtB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,OAAO,EAAE,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,gBAAgB;YAC/C,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE;SAC/B,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAC9B,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,gBAAgB,EACtC,IAAI,CAAC,cAAc,EAAE,CACxB,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE,CAAC,UAAU,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACxD,2EAA2E;IAC/E,CAAC;IAED,KAAK,CAAC,aAAa,CACf,QAAyB,EACzB,IAAmB;QAEnB,MAAM,OAAO,GAAiC;YAC1C,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACzC,MAAM,EAAE,IAAI,CAAC,aAAa;YAC1B,oBAAoB,EAAE,KAAK;SAC9B,CAAC;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,gCAAgC,CAAC,QAAQ,EAAE;YACxD,GAAG,OAAO;YACV,GAAG,IAAI;SACV,CAAC,CAAyD,CAAC;QAE5D,OAAO,CAAC,CAAC;IACb,CAAC;IAED,uEAAuE;IACvE,wEAAwE;IACxE,4EAA4E;IAE5E,KAAK,CAAC,UAAU;QACZ,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC;YAChC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,kBAAkB,CAAC;YACxC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,mBAAmB,CAAC;SAC5C,CAAC,CAAyC,CAAC;QAE5C,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACpB,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,CAAC,GAAG,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACtD,OAAO;gBACH,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,IAAI,EAAE,KAAK,CAAC,EAAE;gBACd,WAAW,EAAE,GAAG,KAAK,CAAC,EAAE,OAAO,KAAK,CAAC,QAAQ,EAAE;gBAC/C,aAAa,EAAE,KAAK,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC;gBAChD,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;gBACxC,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;gBAC1C,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,GAAG,KAAK,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;aACxF,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ"}

package/package.json

@@ -1,17 +1,71 @@
 {
   "name": "@llumiverse/drivers",
-  "version": "1.2.0",
+  "version": "1.4.0-dev.20260629.090753Z",
   "type": "module",
   "description": "LLM driver implementations. Currently supported are: openai, huggingface, bedrock, replicate.",
   "files": [
     "lib",
     "src"
   ],
-  "types": "./lib/types/index.d.ts",
+  "main": "./lib/index.js",
+  "types": "./lib/index.d.ts",
   "exports": {
-    "types": "./lib/types/index.d.ts",
-    "import": "./lib/esm/index.js",
-    "require": "./lib/cjs/index.js"
+    ".": {
+      "types": "./lib/index.d.ts",
+      "default": "./lib/index.js"
+    },
+    "./openai": {
+      "types": "./lib/openai/drivers.d.ts",
+      "default": "./lib/openai/drivers.js"
+    },
+    "./azure": {
+      "types": "./lib/azure/azure_foundry.d.ts",
+      "default": "./lib/azure/azure_foundry.js"
+    },
+    "./anthropic": {
+      "types": "./lib/anthropic/index.d.ts",
+      "default": "./lib/anthropic/index.js"
+    },
+    "./bedrock": {
+      "types": "./lib/bedrock/index.d.ts",
+      "default": "./lib/bedrock/index.js"
+    },
+    "./vertexai": {
+      "types": "./lib/vertexai/index.d.ts",
+      "default": "./lib/vertexai/index.js"
+    },
+    "./groq": {
+      "types": "./lib/groq/index.d.ts",
+      "default": "./lib/groq/index.js"
+    },
+    "./mistral": {
+      "types": "./lib/mistral/index.d.ts",
+      "default": "./lib/mistral/index.js"
+    },
+    "./watsonx": {
+      "types": "./lib/watsonx/index.d.ts",
+      "default": "./lib/watsonx/index.js"
+    },
+    "./togetherai": {
+      "types": "./lib/togetherai/index.d.ts",
+      "default": "./lib/togetherai/index.js"
+    },
+    "./xai": {
+      "types": "./lib/xai/index.d.ts",
+      "default": "./lib/xai/index.js"
+    },
+    "./replicate": {
+      "types": "./lib/replicate.d.ts",
+      "default": "./lib/replicate.js"
+    },
+    "./huggingface": {
+      "types": "./lib/huggingface_ie.d.ts",
+      "default": "./lib/huggingface_ie.js"
+    },
+    "./test-driver": {
+      "types": "./lib/test-driver/index.d.ts",
+      "default": "./lib/test-driver/index.js"
+    }
   },
   "author": "Llumiverse",
   "license": "Apache-2.0",
@@ -41,47 +95,45 @@
     "togetherai"
   ],
   "devDependencies": {
-    "dotenv": "^16.6.1",
-    "rimraf": "^6.1.2",
-    "ts-dual-module": "^0.6.3",
-    "typescript": "^6.0.2",
-    "vitest": "^4.0.18"
+    "dotenv": "^17.4.2",
+    "rimraf": "^6.1.3",
+    "typescript": "^6.0.3",
+    "vitest": "^4.1.8"
   },
   "dependencies": {
-    "@anthropic-ai/sdk": "^0.91.1",
-    "@anthropic-ai/vertex-sdk": "^0.16.0",
-    "@aws-sdk/client-bedrock": "^3.1033.0",
-    "@aws-sdk/client-bedrock-runtime": "^3.1033.0",
-    "@aws-sdk/client-s3": "^3.1033.0",
-    "@aws-sdk/credential-providers": "^3.1033.0",
-    "@aws-sdk/lib-storage": "^3.1033.0",
-    "@aws-sdk/types": "^3.973.8",
+    "@anthropic-ai/sdk": "^0.104.1",
+    "@anthropic-ai/vertex-sdk": "^0.18.0",
+    "@aws-sdk/client-bedrock": "^3.1068.0",
+    "@aws-sdk/client-bedrock-runtime": "^3.1068.0",
+    "@aws-sdk/client-s3": "^3.1068.0",
+    "@aws-sdk/credential-providers": "^3.1068.0",
+    "@aws-sdk/lib-storage": "^3.1068.0",
+    "@aws-sdk/types": "^3.973.12",
     "@azure-rest/ai-inference": "1.0.0-beta.6",
-    "@azure/ai-projects": "1.0.0-beta.10",
+    "@azure/ai-projects": "^2.2.0",
     "@azure/core-auth": "^1.10.1",
     "@azure/core-sse": "^2.3.0",
-    "@azure/identity": "^4.13.0",
-    "@azure/openai": "2.0.0",
-    "@google-cloud/aiplatform": "^6.5.0",
-    "@google/genai": "^1.49.0",
-    "@huggingface/inference": "4.13.11",
-    "@vertesia/api-fetch-client": "^0.82.4",
+    "@azure/identity": "^4.13.1",
+    "@azure/openai": "^2.0.0",
+    "@google-cloud/aiplatform": "^6.8.1",
+    "@google/genai": "^2.8.0",
+    "@huggingface/inference": "^4.13.18",
+    "@vertesia/api-fetch-client": "^1.3.0",
     "eventsource": "^4.1.0",
-    "google-auth-library": "^10.6.2",
-    "groq-sdk": "^0.37.0",
-    "mnemonist": "^0.40.3",
+    "google-auth-library": "10.7.0",
+    "groq-sdk": "^1.2.1",
+    "mnemonist": "^0.40.4",
     "node-web-stream-adapters": "^0.2.1",
-    "openai": "^6.33.0",
+    "openai": "^6.42.0",
     "replicate": "^1.4.0",
-    "@llumiverse/core": "1.2.0",
-    "@llumiverse/common": "1.2.0"
-  },
-  "ts_dual_module": {
-    "outDir": "lib"
+    "@llumiverse/common": "1.4.0-dev.20260629.090753Z",
+    "@llumiverse/core": "1.4.0-dev.20260629.090753Z"
   },
   "scripts": {
-    "test": "vitest run --retry 3",
-    "build": "pnpm exec tsmod build",
+    "lint": "biome lint src",
+    "test": "vitest run --retry 3 --passWithNoTests",
+    "typecheck:test": "tsc -p tsconfig.test.json --noEmit",
+    "build": "pnpm exec rimraf ./lib ./tsconfig.tsbuildinfo && tsc",
     "clean": "rimraf ./lib tsconfig.tsbuildinfo"
   }
 }