@llumiverse/drivers 1.0.0-dev.20260202.145450Z → 1.0.0

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

@@ -0,0 +1,27 @@
+import { AIModel, AbstractDriver, Completion, CompletionChunkObject, DriverOptions, EmbeddingsOptions, EmbeddingsResult, ExecutionOptions } from "@llumiverse/core";
+import { FetchClient } from "@vertesia/api-fetch-client";
+import { WatsonAuthToken } from "./interfaces.js";
+interface WatsonxDriverOptions extends DriverOptions {
+    apiKey: string;
+    projectId: string;
+    endpointUrl: string;
+}
+export declare class WatsonxDriver extends AbstractDriver<WatsonxDriverOptions, string> {
+    static PROVIDER: string;
+    provider: string;
+    apiKey: string;
+    endpoint_url: string;
+    projectId: string;
+    authToken?: WatsonAuthToken;
+    fetcher?: FetchClient;
+    fetchClient: FetchClient;
+    constructor(options: WatsonxDriverOptions);
+    requestTextCompletion(prompt: string, options: ExecutionOptions): Promise<Completion>;
+    requestTextCompletionStream(prompt: string, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>;
+    listModels(): Promise<AIModel<string>[]>;
+    getAuthToken(): Promise<string>;
+    validateConnection(): Promise<boolean>;
+    generateEmbeddings(options: EmbeddingsOptions): Promise<EmbeddingsResult>;
+}
+export {};
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/watsonx/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,qBAAqB,EAAE,aAAa,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,gBAAgB,EAAuB,MAAM,kBAAkB,CAAC;AAEzL,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAuD,eAAe,EAA2G,MAAM,iBAAiB,CAAC;AAEhN,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,CAAA;gBAEZ,OAAO,EAAE,oBAAoB;IAQnC,qBAAqB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IAmCrF,2BAA2B,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;IAwCrH,UAAU,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IAoBxC,YAAY,IAAI,OAAO,CAAC,MAAM,CAAC;IAwB/B,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC;IAStC,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAwBlF"}

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

@@ -0,0 +1,65 @@
+export interface WatsonxTextGenerationPayload {
+    model_id: string;
+    input: string;
+    parameters: {
+        max_new_tokens?: number;
+        time_limit?: number;
+        stop_sequences?: string[];
+        temperature?: number;
+        top_k?: number;
+        top_p?: number;
+    };
+    project_id: string;
+}
+export interface WatsonxTextGenerationResponse {
+    model_id: string;
+    created_at: string;
+    results: {
+        generated_text: string;
+        generated_token_count: number;
+        input_token_count: number;
+        stop_reason: string;
+    }[];
+}
+export interface GenerateEmbeddingPayload {
+    model_id: string;
+    inputs: string[];
+    project_id: string;
+}
+export interface GenerateEmbeddingResponse {
+    model_id: string;
+    created_at: string;
+    results: {
+        embedding: number[];
+    }[];
+    input_token_count: number;
+}
+export interface WatsonxModelSpec {
+    model_id: string;
+    label: string;
+    provider: string;
+    source: string;
+    short_description: string;
+    tasks: {
+        id: string;
+        ratings: {
+            quality: number;
+        };
+    }[];
+    min_shot_size: number;
+    tier: string;
+    number_params: string;
+}
+export interface WatsonxListModelResponse {
+    total_count: number;
+    limit: number;
+    resources: WatsonxModelSpec[];
+}
+export interface WatsonAuthToken {
+    access_token: string;
+    refresh_token: string;
+    token_type: string;
+    expire_in: number;
+    expiration: number;
+}
+//# sourceMappingURL=interfaces.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"interfaces.d.ts","sourceRoot":"","sources":["../../../src/watsonx/interfaces.ts"],"names":[],"mappings":"AAEA,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,CAAA;CACN;AAED,MAAM,WAAW,wBAAwB;IACrC,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;CACtB;AAGD,MAAM,WAAW,yBAAyB;IACtC,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE;QACL,SAAS,EAAE,MAAM,EAAE,CAAC;KACvB,EAAE,CAAA;IACH,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,CAAA;KACJ,EAAE,CAAC;IACJ,aAAa,EAAE,MAAM,CAAC;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;CACzB;AAGD,MAAM,WAAW,wBAAwB;IACrC,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,gBAAgB,EAAE,CAAC;CACjC;AAGD,MAAM,WAAW,eAAe;IAC5B,YAAY,EAAE,MAAM,CAAA;IACpB,aAAa,EAAE,MAAM,CAAA;IACrB,UAAU,EAAE,MAAM,CAAA;IAClB,SAAS,EAAE,MAAM,CAAA;IACjB,UAAU,EAAE,MAAM,CAAA;CACrB"}

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

@@ -0,0 +1,18 @@
+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";
+export interface xAiDriverOptions extends DriverOptions {
+    apiKey: string;
+    endpoint?: string;
+}
+export declare class xAIDriver extends BaseOpenAIDriver {
+    service: OpenAI;
+    readonly provider = Providers.xai;
+    xai_service: FetchClient;
+    DEFAULT_ENDPOINT: string;
+    constructor(opts: xAiDriverOptions);
+    _formatPrompt(segments: PromptSegment[], opts: PromptOptions): Promise<OpenAI.Chat.Completions.ChatCompletionMessageParam[]>;
+    listModels(): Promise<AIModel[]>;
+}
+//# sourceMappingURL=index.d.ts.map

package/lib/types/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,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAEnG,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAEtD,MAAM,WAAW,gBAAiB,SAAQ,aAAa;IAEnD,MAAM,EAAE,MAAM,CAAC;IAEf,QAAQ,CAAC,EAAE,MAAM,CAAC;CAErB;AAED,qBAAa,SAAU,SAAQ,gBAAgB;IAE3C,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,QAAQ,iBAAiB;IAClC,WAAW,EAAE,WAAW,CAAC;IACzB,gBAAgB,SAAyB;gBAE7B,IAAI,EAAE,gBAAgB;IAe5B,aAAa,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,IAAI,EAAE,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,0BAA0B,EAAE,CAAC;IAkB5H,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;CA6BzC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llumiverse/drivers",
-  "version": "1.0.0-dev.20260202.145450Z",
+  "version": "1.0.0",
   "type": "module",
   "description": "LLM driver implementations. Currently supported are: openai, huggingface, bedrock, replicate.",
   "files": [
@@ -45,36 +45,36 @@
     "rimraf": "^6.1.2",
     "ts-dual-module": "^0.6.3",
     "typescript": "^5.9.3",
-    "vitest": "^4.0.16"
+    "vitest": "^4.0.18"
   },
   "dependencies": {
-    "@anthropic-ai/sdk": "^0.71.2",
-    "@anthropic-ai/vertex-sdk": "^0.14.0",
-    "@aws-sdk/client-bedrock": "^3.952.0",
-    "@aws-sdk/client-bedrock-runtime": "^3.952.0",
-    "@aws-sdk/client-s3": "^3.952.0",
-    "@aws-sdk/credential-providers": "^3.952.0",
-    "@aws-sdk/lib-storage": "^3.952.0",
-    "@aws-sdk/types": "^3.936.0",
+    "@anthropic-ai/sdk": "^0.78.0",
+    "@anthropic-ai/vertex-sdk": "^0.14.4",
+    "@aws-sdk/client-bedrock": "^3.985.0",
+    "@aws-sdk/client-bedrock-runtime": "^3.985.0",
+    "@aws-sdk/client-s3": "^3.985.0",
+    "@aws-sdk/credential-providers": "^3.985.0",
+    "@aws-sdk/lib-storage": "^3.985.0",
+    "@aws-sdk/types": "^3.973.1",
     "@azure-rest/ai-inference": "1.0.0-beta.6",
     "@azure/ai-projects": "1.0.0-beta.10",
     "@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.0.0",
-    "@google/genai": "^1.31.0",
-    "@huggingface/inference": "4.13.4",
-    "@vertesia/api-fetch-client": "^0.79.3",
+    "@google-cloud/aiplatform": "^6.5.0",
+    "@google/genai": "^1.46.0",
+    "@huggingface/inference": "4.13.11",
+    "@vertesia/api-fetch-client": "^0.82.4",
     "eventsource": "^4.1.0",
     "google-auth-library": "^10.5.0",
     "groq-sdk": "^0.37.0",
     "mnemonist": "^0.40.3",
     "node-web-stream-adapters": "^0.2.1",
-    "openai": "^6.10.0",
-    "replicate": "^1.3.1",
-    "@llumiverse/common": "1.0.0-dev.20260202.145450Z",
-    "@llumiverse/core": "1.0.0-dev.20260202.145450Z"
+    "openai": "^6.22.0",
+    "replicate": "^1.4.0",
+    "@llumiverse/common": "1.0.0",
+    "@llumiverse/core": "1.0.0"
   },
   "ts_dual_module": {
     "outDir": "lib"

package/src/bedrock/converse.ts

@@ -1,39 +1,110 @@
-import { DataSource, ExecutionOptions, readStreamAsString, readStreamAsUint8Array } from "@llumiverse/core";
-import { PromptSegment, PromptRole } from "@llumiverse/core";
 import {
+    ContentBlock,
     ConversationRole,
     ConverseRequest,
     Message,
     SystemContentBlock,
-    ContentBlock,
     ToolResultContentBlock,
 } from "@aws-sdk/client-bedrock-runtime";
+import { DataSource, ExecutionOptions, PromptRole, PromptSegment, readStreamAsString, readStreamAsUint8Array } from "@llumiverse/core";
 import { parseS3UrlToUri } from "./s3.js";
 
 function roleConversion(role: PromptRole): ConversationRole {
     return role === PromptRole.assistant ? ConversationRole.ASSISTANT : ConversationRole.USER;
 }
 
-function mimeToImageType(mime: string): "png" | "jpeg" | "gif" | "webp" {
+type BedrockImageFormat = "png" | "jpeg" | "gif" | "webp";
+const BEDROCK_IMAGE_FORMATS = new Set<string>(["png", "jpeg", "gif", "webp"]);
+const mimeToImageMap: Record<string, BedrockImageFormat> = {
+    "image/png": "png",
+    "image/jpeg": "jpeg",
+    "image/gif": "gif",
+    "image/webp": "webp",
+};
+function mimeToImageType(mime: string): BedrockImageFormat {
+    const mapped = mimeToImageMap[mime];
+    if (mapped) return mapped;
     if (mime.startsWith("image/")) {
-        return mime.split("/")[1] as "png" | "jpeg" | "gif" | "webp";
+        const subtype = mime.split("/")[1];
+        if (subtype && BEDROCK_IMAGE_FORMATS.has(subtype)) {
+            return subtype as BedrockImageFormat;
+        }
     }
     return "png";
 }
 
-function mimeToDocType(mime: string): "pdf" | "csv" | "doc" | "docx" | "xls" | "xlsx" | "html" | "txt" | "md" {
+type BedrockDocFormat = "pdf" | "csv" | "doc" | "docx" | "xls" | "xlsx" | "html" | "txt" | "md";
+const BEDROCK_DOC_FORMATS = new Set<string>(["pdf", "csv", "doc", "docx", "xls", "xlsx", "html", "txt", "md"]);
+const mimeToDocMap: Record<string, BedrockDocFormat> = {
+    "application/pdf": "pdf",
+    "text/csv": "csv",
+    "application/msword": "doc",
+    "application/vnd.openxmlformats-officedocument.wordprocessingml.document": "docx",
+    "application/vnd.ms-excel": "xls",
+    "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet": "xlsx",
+    "text/html": "html",
+    "text/plain": "txt",
+    "text/markdown": "md",
+};
+function mimeToDocType(mime: string): BedrockDocFormat {
+    // 1. Exact map lookup (handles complex MIME types like vnd.openxmlformats-*)
+    const mapped = mimeToDocMap[mime];
+    if (mapped) return mapped;
+    // 2. Fallback: extract subtype for simple application/ or text/ MIME types
     if (mime.startsWith("application/") || mime.startsWith("text/")) {
-        return mime.split("/")[1] as "pdf" | "csv" | "doc" | "docx" | "xls" | "xlsx" | "html" | "txt" | "md";
+        const subtype = mime.split("/")[1];
+        if (subtype && BEDROCK_DOC_FORMATS.has(subtype)) {
+            return subtype as BedrockDocFormat;
+        }
     }
     return "txt";
 }
-function mimeToVideoType(mime: string): "mov" | "mkv" | "mp4" | "webm" | "flv" | "mpeg" | "mpg" | "wmv" | "three_gp" {
+type BedrockVideoFormat = "mov" | "mkv" | "mp4" | "webm" | "flv" | "mpeg" | "mpg" | "wmv" | "three_gp";
+const BEDROCK_VIDEO_FORMATS = new Set<string>(["mov", "mkv", "mp4", "webm", "flv", "mpeg", "mpg", "wmv", "three_gp"]);
+const mimeToVideoMap: Record<string, BedrockVideoFormat> = {
+    "video/quicktime": "mov",
+    "video/x-matroska": "mkv",
+    "video/mp4": "mp4",
+    "video/webm": "webm",
+    "video/x-flv": "flv",
+    "video/mpeg": "mpeg",
+    "video/x-ms-wmv": "wmv",
+    "video/3gpp": "three_gp",
+};
+function mimeToVideoType(mime: string): BedrockVideoFormat {
+    const mapped = mimeToVideoMap[mime];
+    if (mapped) return mapped;
     if (mime.startsWith("video/")) {
-        return mime.split("/")[1] as "mov" | "mkv" | "mp4" | "webm" | "flv" | "mpeg" | "mpg" | "wmv" | "three_gp";
+        const subtype = mime.split("/")[1];
+        if (subtype && BEDROCK_VIDEO_FORMATS.has(subtype)) {
+            return subtype as BedrockVideoFormat;
+        }
     }
     return "mp4";
 }
 
+/**
+ * Cleans a filename to conform to Bedrock's restrictions:
+ * - Alphanumeric characters, whitespace, hyphens, parentheses, square brackets.
+ * - No more than one consecutive whitespace character.
+ * - Decodes URI components (e.g., %20 -> space).
+ */
+function cleanBedrockFilename(name: string | undefined): string | undefined {
+    if (!name) return name;
+
+    try {
+        // Decode URI components like %20
+        name = decodeURIComponent(name);
+    } catch {
+        // Ignore decoding errors
+    }
+
+    return name
+        .replace(/[^\w\s\-()[\]]/g, " ") // Replace invalid characters with space
+        .replace(/\s+/g, " ") // Collapse consecutive whitespaces
+        .trim();
+}
+
 type FileProcessingMode = 'content' | 'tool';
 
 async function processFile<T extends FileProcessingMode>(
@@ -78,7 +149,7 @@ async function processFile<T extends FileProcessingMode>(
             const documentBlock = {
                 document: {
                     format: mimeToDocType(f.mime_type),
-                    name: f.name,
+                    name: cleanBedrockFilename(f.name),
                     source: { bytes: await readStreamAsUint8Array(source) },
                 },
             };
@@ -235,6 +306,10 @@ export async function formatConversePrompt(segments: PromptSegment[], options: E
             for (const file of segment.files ?? []) {
                 toolContentBlocks.push(await processFileToToolContentBlock(file));
             }
+            // Bedrock requires at least one content block in toolResult
+            if (toolContentBlocks.length === 0) {
+                toolContentBlocks.push({ text: '[No output]' });
+            }
             messages.push({
                 content: [{
                     toolResult: {