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

package/lib/bedrock/embeddings.js

@@ -0,0 +1,359 @@
+import { BEDROCK_DEFAULT_EMBEDDING_MODEL } from '@llumiverse/common';
+import { Base64DataSource, buildEmbeddingsResult, dataSourceToBase64, LlumiverseError, normalizeEmbeddingsOptions, } from '@llumiverse/core';
+/** Matches s3[.region].amazonaws.com or bucket.s3[.region].amazonaws.com — anchored to prevent subdomain spoofing. */
+const S3_HOSTNAME_RE = /^(?:[^.]+\.)?s3(?:\.[a-z0-9-]+)?\.amazonaws\.com$/;
+function isS3Url(url) {
+    if (url.startsWith('s3://'))
+        return true;
+    try {
+        const parsed = new URL(url);
+        return S3_HOSTNAME_RE.test(parsed.hostname);
+    }
+    catch {
+        return false;
+    }
+}
+/**
+ * Converts an S3 HTTPS URL to s3:// URI form. Handles both addressing styles:
+ *   virtual-hosted: https://bucket.s3[.region].amazonaws.com/key
+ *   path-style:     https://s3[.region].amazonaws.com/bucket/key
+ */
+function toS3Uri(url) {
+    if (url.startsWith('s3://'))
+        return url;
+    const parsed = new URL(url);
+    // Virtual-hosted style: bucket.s3[.region].amazonaws.com
+    const virtualHostedMatch = parsed.hostname.match(/^([^.]+)\.s3(?:\.[a-z0-9-]+)?\.amazonaws\.com$/);
+    if (virtualHostedMatch) {
+        const bucket = virtualHostedMatch[1];
+        const key = parsed.pathname.replace(/^\/+/, '');
+        return `s3://${bucket}/${key}`;
+    }
+    // Path-style: s3[.region].amazonaws.com/bucket/key
+    const pathParts = parsed.pathname.replace(/^\/+/, '').split('/');
+    const bucket = pathParts[0];
+    const key = pathParts.slice(1).join('/');
+    return `s3://${bucket}/${key}`;
+}
+async function dataSourceToBedrockMediaSource(ds) {
+    const url = await ds.getURL().catch(() => undefined);
+    if (url && isS3Url(url)) {
+        return { s3Location: { uri: toS3Uri(url) } };
+    }
+    if (ds instanceof Base64DataSource) {
+        return { base64String: ds.getBase64() };
+    }
+    return { base64String: await dataSourceToBase64(ds) };
+}
+async function invokeJson(driver, modelId, body) {
+    const executor = driver.getExecutor();
+    const res = await executor.invokeModel({
+        modelId,
+        contentType: 'application/json',
+        accept: 'application/json',
+        body: JSON.stringify(body),
+    });
+    if (!res.body)
+        throw new Error(`InvokeModel returned empty response body for model: ${modelId}`);
+    return JSON.parse(new TextDecoder().decode(res.body));
+}
+// ── Shared format lookup maps ───────────────────────────────────────────────
+const MIME_TO_IMAGE_FORMAT = {
+    'image/jpeg': 'jpeg',
+    'image/jpg': 'jpeg',
+    'image/png': 'png',
+    'image/gif': 'gif',
+    'image/webp': 'webp',
+};
+const MIME_TO_AUDIO_FORMAT = {
+    'audio/mpeg': 'mp3',
+    'audio/mp3': 'mp3',
+    'audio/wav': 'wav',
+    'audio/wave': 'wav',
+    'audio/x-wav': 'wav',
+    'audio/ogg': 'ogg',
+};
+const MIME_TO_VIDEO_FORMAT = {
+    'video/mp4': 'mp4',
+    'video/quicktime': 'mov',
+    'video/x-matroska': 'mkv',
+    'video/webm': 'webm',
+    'video/x-flv': 'flv',
+    'video/mpeg': 'mpeg',
+    'video/mpg': 'mpg',
+    'video/x-ms-wmv': 'wmv',
+    'video/3gpp': '3gp',
+};
+function requireFormat(map, mime, label) {
+    const fmt = map[mime.toLowerCase()];
+    if (!fmt)
+        throw new Error(`Unsupported ${label} MIME type for Bedrock: '${mime}'`);
+    return fmt;
+}
+function toNovaEmbeddingPurpose(taskType) {
+    return taskType === 'query' ? 'GENERIC_RETRIEVAL' : 'GENERIC_INDEX';
+}
+async function toNovaSource(ds) {
+    const url = await ds.getURL().catch(() => undefined);
+    if (url && isS3Url(url))
+        return { s3Location: { uri: toS3Uri(url) } };
+    const bytes = ds instanceof Base64DataSource ? ds.getBase64() : await dataSourceToBase64(ds);
+    return { bytes };
+}
+async function buildNovaParams(input) {
+    switch (input.type) {
+        case 'text':
+            return { text: { truncationMode: 'END', value: input.text } };
+        case 'image': {
+            const source = await toNovaSource(input.source);
+            return { image: { format: requireFormat(MIME_TO_IMAGE_FORMAT, input.source.mime_type, 'image'), source } };
+        }
+        case 'audio': {
+            const source = await toNovaSource(input.source);
+            return { audio: { format: requireFormat(MIME_TO_AUDIO_FORMAT, input.source.mime_type, 'audio'), source } };
+        }
+        case 'video': {
+            // Sync path supports video up to 30 s / 100 MB via S3 (or 25 MB inline base64).
+            // Longer video requires StartAsyncInvoke with segmentationConfig — not implemented.
+            const source = await toNovaSource(input.source);
+            return {
+                video: {
+                    format: requireFormat(MIME_TO_VIDEO_FORMAT, input.source.mime_type, 'video'),
+                    source,
+                    embeddingMode: 'AUDIO_VIDEO_COMBINED',
+                },
+            };
+        }
+    }
+}
+async function generateNovaEmbeddings(driver, options, modelId) {
+    const items = [];
+    for (const input of options.inputs) {
+        const inputTaskType = input.type === 'text' ? input.task_type : undefined;
+        const purpose = toNovaEmbeddingPurpose(inputTaskType ?? options.task_type);
+        const modalParams = await buildNovaParams(input);
+        const request = {
+            taskType: 'SINGLE_EMBEDDING',
+            singleEmbeddingParams: {
+                embeddingPurpose: purpose,
+                ...(options.dimensions ? { embeddingDimension: options.dimensions } : {}),
+                ...modalParams,
+            },
+        };
+        const response = await invokeJson(driver, modelId, request);
+        const item = response.embeddings?.[0];
+        if (!item?.embedding) {
+            throw new Error(`Nova embeddings response missing 'embeddings[0].embedding' for input type '${input.type}'`);
+        }
+        items.push({ outputs: [{ values: item.embedding, modality: input.type }] });
+    }
+    return buildEmbeddingsResult(modelId, items);
+}
+async function generateTitanEmbeddings(driver, options, modelId) {
+    const isImageModel = modelId.includes('titan-embed-image');
+    const items = [];
+    let totalTokens;
+    for (const input of options.inputs) {
+        if (input.type === 'video' || input.type === 'audio') {
+            throw new Error(`Titan embeddings do not support '${input.type}' input`);
+        }
+        if (isImageModel) {
+            const body = {};
+            if (input.type === 'text')
+                body.inputText = input.text;
+            if (input.type === 'image')
+                body.inputImage = await dataSourceToBase64(input.source);
+            if (options.dimensions)
+                body.embeddingConfig = { outputEmbeddingLength: options.dimensions };
+            const res = await invokeJson(driver, modelId, body);
+            if (res.message)
+                throw new Error(`Titan image embedding error: ${res.message}`);
+            if (!res.embedding)
+                throw new Error(`Titan image embedding response missing 'embedding'`);
+            items.push({
+                outputs: [{ values: res.embedding, modality: input.type }],
+                input_tokens: res.inputTextTokenCount,
+            });
+            if (typeof res.inputTextTokenCount === 'number')
+                totalTokens = (totalTokens ?? 0) + res.inputTextTokenCount;
+        }
+        else {
+            if (input.type !== 'text') {
+                throw new Error(`Titan text embeddings model '${modelId}' only supports text input`);
+            }
+            const body = { inputText: input.text };
+            if (options.dimensions)
+                body.dimensions = options.dimensions;
+            const res = await invokeJson(driver, modelId, body);
+            if (!res.embedding)
+                throw new Error(`Titan text embedding response missing 'embedding' (model ${modelId})`);
+            items.push({
+                outputs: [{ values: res.embedding, modality: 'text' }],
+                input_tokens: res.inputTextTokenCount,
+            });
+            if (typeof res.inputTextTokenCount === 'number')
+                totalTokens = (totalTokens ?? 0) + res.inputTextTokenCount;
+        }
+    }
+    const usage = totalTokens !== undefined ? { input_tokens: totalTokens } : undefined;
+    return buildEmbeddingsResult(modelId, items, usage);
+}
+function cohereInputType(taskType) {
+    switch (taskType) {
+        case 'query':
+            return 'search_query';
+        case 'document':
+            return 'search_document';
+        default:
+            return undefined;
+    }
+}
+async function generateCohereEmbeddings(driver, options, modelId) {
+    const textInputs = [];
+    const imageInputs = [];
+    options.inputs.forEach((input, index) => {
+        if (input.type === 'text')
+            textInputs.push({ index, input });
+        else if (input.type === 'image')
+            imageInputs.push({ index, input });
+        else
+            throw new Error(`Cohere embeddings do not support '${input.type}' input`);
+    });
+    const items = new Array(options.inputs.length);
+    // Group texts by their effective input_type so per-input task_type overrides are respected.
+    const textGroups = new Map();
+    for (const entry of textInputs) {
+        const key = cohereInputType(entry.input.task_type ?? options.task_type);
+        if (!textGroups.has(key))
+            textGroups.set(key, []);
+        textGroups.get(key)?.push(entry);
+    }
+    for (const [groupInputType, group] of textGroups) {
+        const body = {
+            texts: group.map((t) => t.input.text),
+            input_type: groupInputType,
+        };
+        const res = await invokeJson(driver, modelId, body);
+        if (res.embeddings.length !== group.length) {
+            driver.logger.warn(`[Bedrock/Cohere] Embedding count mismatch: expected ${group.length}, got ${res.embeddings.length} (model ${modelId})`);
+            throw new Error(`Cohere returned ${res.embeddings.length} embeddings for ${group.length} texts (model ${modelId})`);
+        }
+        group.forEach((entry, i) => {
+            items[entry.index] = { outputs: [{ values: res.embeddings[i], modality: 'text' }] };
+        });
+    }
+    // Cohere accepts exactly one image per call; images must be data URIs.
+    for (const entry of imageInputs) {
+        const base64 = await dataSourceToBase64(entry.input.source);
+        const dataUri = `data:${entry.input.source.mime_type};base64,${base64}`;
+        const body = { images: [dataUri], input_type: 'image' };
+        const res = await invokeJson(driver, modelId, body);
+        if (!res.embeddings[0]) {
+            throw new Error(`Cohere returned no embedding for image input (model ${modelId})`);
+        }
+        items[entry.index] = { outputs: [{ values: res.embeddings[0], modality: 'image' }] };
+    }
+    return buildEmbeddingsResult(modelId, items);
+}
+function isMarengoEnvelope(r) {
+    return Array.isArray(r.embeddings);
+}
+async function generateMarengoEmbeddings(driver, options, modelId) {
+    const items = [];
+    for (const input of options.inputs) {
+        const request = await buildMarengoRequest(input);
+        const response = await invokeJson(driver, modelId, request);
+        const segments = isMarengoEnvelope(response) ? response.embeddings : [response];
+        const outputs = segments
+            .filter((seg) => Array.isArray(seg.embedding))
+            .map((seg) => ({
+            values: seg.embedding,
+            modality: input.type === 'text' ? 'text' : input.type,
+            start_sec: seg.startSec,
+            end_sec: seg.endSec,
+            embedding_option: seg.embeddingOption,
+        }));
+        if (outputs.length === 0) {
+            throw new Error(`Marengo response did not contain embedding values for input type '${input.type}'`);
+        }
+        items.push({ outputs });
+    }
+    return buildEmbeddingsResult(modelId, items);
+}
+async function buildMarengoRequest(input) {
+    switch (input.type) {
+        case 'text':
+            return { inputType: 'text', inputText: input.text };
+        case 'image': {
+            const media = await dataSourceToBedrockMediaSource(input.source);
+            return { inputType: 'image', mediaSource: media };
+        }
+        case 'video': {
+            const media = await dataSourceToBedrockMediaSource(input.source);
+            const request = { inputType: 'video', mediaSource: media };
+            if (input.start_sec !== undefined)
+                request.startSec = input.start_sec;
+            if (input.length_sec !== undefined)
+                request.lengthSec = input.length_sec;
+            if (input.use_fixed_length_sec !== undefined)
+                request.useFixedLengthSec = input.use_fixed_length_sec;
+            if (input.min_clip_sec !== undefined)
+                request.minClipSec = input.min_clip_sec;
+            // Marengo accepts at most one embedding_option per request.
+            // For multiple views (e.g. visual-text + audio), submit separate inputs — one per option.
+            if (input.embedding_option) {
+                if (input.embedding_option.length > 1) {
+                    throw new Error(`Marengo accepts only one embedding_option per request; ` +
+                        `received [${input.embedding_option.join(', ')}]. ` +
+                        `Submit separate inputs, one per option.`);
+                }
+                if (input.embedding_option.length === 1) {
+                    request.embeddingOption = input.embedding_option[0];
+                }
+            }
+            return request;
+        }
+        case 'audio': {
+            const media = await dataSourceToBedrockMediaSource(input.source);
+            const request = { inputType: 'audio', mediaSource: media };
+            if (input.start_sec !== undefined)
+                request.startSec = input.start_sec;
+            if (input.length_sec !== undefined)
+                request.lengthSec = input.length_sec;
+            return request;
+        }
+    }
+}
+// =================== Routing ===================
+export async function generateBedrockEmbeddings(driver, options) {
+    const normalized = normalizeEmbeddingsOptions(options);
+    const modelId = normalized.model ?? BEDROCK_DEFAULT_EMBEDDING_MODEL;
+    driver.logger.info(`[Bedrock] Generating embeddings with model ${modelId} for ${normalized.inputs.length} input(s)`);
+    try {
+        if (modelId.includes('twelvelabs.marengo')) {
+            return await generateMarengoEmbeddings(driver, normalized, modelId);
+        }
+        if (modelId.startsWith('cohere.embed')) {
+            return await generateCohereEmbeddings(driver, normalized, modelId);
+        }
+        if (modelId.includes('titan-embed')) {
+            return await generateTitanEmbeddings(driver, normalized, modelId);
+        }
+        return await generateNovaEmbeddings(driver, normalized, modelId);
+    }
+    catch (error) {
+        if (LlumiverseError.isLlumiverseError(error))
+            throw error;
+        // Re-throw plain validation errors (thrown before any HTTP call) as-is.
+        if (error instanceof Error &&
+            typeof error.$metadata === 'undefined' &&
+            typeof error.status !== 'number')
+            throw error;
+        throw driver.formatLlumiverseError(error, {
+            provider: 'bedrock',
+            model: modelId,
+            operation: 'execute',
+        });
+    }
+}
+//# sourceMappingURL=embeddings.js.map

package/lib/bedrock/embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.js","sourceRoot":"","sources":["../../src/bedrock/embeddings.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,+BAA+B,EAAE,MAAM,oBAAoB,CAAC;AACrE,OAAO,EACH,gBAAgB,EAChB,qBAAqB,EAErB,kBAAkB,EAQlB,eAAe,EACf,0BAA0B,GAE7B,MAAM,kBAAkB,CAAC;AAY1B,sHAAsH;AACtH,MAAM,cAAc,GAAG,mDAAmD,CAAC;AAE3E,SAAS,OAAO,CAAC,GAAW;IACxB,IAAI,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC;QAAE,OAAO,IAAI,CAAC;IACzC,IAAI,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;QAC5B,OAAO,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,KAAK,CAAC;IACjB,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,SAAS,OAAO,CAAC,GAAW;IACxB,IAAI,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC;QAAE,OAAO,GAAG,CAAC;IACxC,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IAC5B,yDAAyD;IACzD,MAAM,kBAAkB,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACnG,IAAI,kBAAkB,EAAE,CAAC;QACrB,MAAM,MAAM,GAAG,kBAAkB,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAChD,OAAO,QAAQ,MAAM,IAAI,GAAG,EAAE,CAAC;IACnC,CAAC;IACD,mDAAmD;IACnD,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACjE,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;IAC5B,MAAM,GAAG,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzC,OAAO,QAAQ,MAAM,IAAI,GAAG,EAAE,CAAC;AACnC,CAAC;AAED,KAAK,UAAU,8BAA8B,CAAC,EAAc;IACxD,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACrD,IAAI,GAAG,IAAI,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACtB,OAAO,EAAE,UAAU,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;IACjD,CAAC;IACD,IAAI,EAAE,YAAY,gBAAgB,EAAE,CAAC;QACjC,OAAO,EAAE,YAAY,EAAE,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC;IAC5C,CAAC;IACD,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC,EAAE,CAAC,EAAE,CAAC;AAC1D,CAAC;AAED,KAAK,UAAU,UAAU,CAAI,MAAqB,EAAE,OAAe,EAAE,IAAa;IAC9E,MAAM,QAAQ,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;IACtC,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC;QACnC,OAAO;QACP,WAAW,EAAE,kBAAkB;QAC/B,MAAM,EAAE,kBAAkB;QAC1B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;KAC7B,CAAC,CAAC;IACH,IAAI,CAAC,GAAG,CAAC,IAAI;QAAE,MAAM,IAAI,KAAK,CAAC,uDAAuD,OAAO,EAAE,CAAC,CAAC;IACjG,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAM,CAAC;AAC/D,CAAC;AAED,+EAA+E;AAE/E,MAAM,oBAAoB,GAA2B;IACjD,YAAY,EAAE,MAAM;IACpB,WAAW,EAAE,MAAM;IACnB,WAAW,EAAE,KAAK;IAClB,WAAW,EAAE,KAAK;IAClB,YAAY,EAAE,MAAM;CACvB,CAAC;AAEF,MAAM,oBAAoB,GAA2B;IACjD,YAAY,EAAE,KAAK;IACnB,WAAW,EAAE,KAAK;IAClB,WAAW,EAAE,KAAK;IAClB,YAAY,EAAE,KAAK;IACnB,aAAa,EAAE,KAAK;IACpB,WAAW,EAAE,KAAK;CACrB,CAAC;AAEF,MAAM,oBAAoB,GAA2B;IACjD,WAAW,EAAE,KAAK;IAClB,iBAAiB,EAAE,KAAK;IACxB,kBAAkB,EAAE,KAAK;IACzB,YAAY,EAAE,MAAM;IACpB,aAAa,EAAE,KAAK;IACpB,YAAY,EAAE,MAAM;IACpB,WAAW,EAAE,KAAK;IAClB,gBAAgB,EAAE,KAAK;IACvB,YAAY,EAAE,KAAK;CACtB,CAAC;AAEF,SAAS,aAAa,CAAC,GAA2B,EAAE,IAAY,EAAE,KAAa;IAC3E,MAAM,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IACpC,IAAI,CAAC,GAAG;QAAE,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,4BAA4B,IAAI,GAAG,CAAC,CAAC;IACnF,OAAO,GAAG,CAAC;AACf,CAAC;AAmDD,SAAS,sBAAsB,CAAC,QAAuC;IACnE,OAAO,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,eAAe,CAAC;AACxE,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,EAAc;IACtC,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACrD,IAAI,GAAG,IAAI,OAAO,CAAC,GAAG,CAAC;QAAE,OAAO,EAAE,UAAU,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;IACtE,MAAM,KAAK,GAAG,EAAE,YAAY,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,MAAM,kBAAkB,CAAC,EAAE,CAAC,CAAC;IAC7F,OAAO,EAAE,KAAK,EAAE,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,eAAe,CAC1B,KAAqB;IAErB,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACjB,KAAK,MAAM;YACP,OAAO,EAAE,IAAI,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;QAClE,KAAK,OAAO,CAAC,CAAC,CAAC;YACX,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAChD,OAAO,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,aAAa,CAAC,oBAAoB,EAAE,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;QAC/G,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACX,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAChD,OAAO,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,aAAa,CAAC,oBAAoB,EAAE,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;QAC/G,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACX,gFAAgF;YAChF,oFAAoF;YACpF,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAChD,OAAO;gBACH,KAAK,EAAE;oBACH,MAAM,EAAE,aAAa,CAAC,oBAAoB,EAAE,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC;oBAC5E,MAAM;oBACN,aAAa,EAAE,sBAAsB;iBACxC;aACJ,CAAC;QACN,CAAC;IACL,CAAC;AACL,CAAC;AAED,KAAK,UAAU,sBAAsB,CACjC,MAAqB,EACrB,OAA0B,EAC1B,OAAe;IAEf,MAAM,KAAK,GAA0B,EAAE,CAAC;IAExC,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1E,MAAM,OAAO,GAAG,sBAAsB,CAAC,aAAa,IAAI,OAAO,CAAC,SAAS,CAAC,CAAC;QAC3E,MAAM,WAAW,GAAG,MAAM,eAAe,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,OAAO,GAAyB;YAClC,QAAQ,EAAE,kBAAkB;YAC5B,qBAAqB,EAAE;gBACnB,gBAAgB,EAAE,OAAO;gBACzB,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzE,GAAG,WAAW;aACjB;SACJ,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAwB,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QACnF,MAAM,IAAI,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CACX,8EAA8E,KAAK,CAAC,IAAI,GAAG,CAC9F,CAAC;QACN,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;IAChF,CAAC;IAED,OAAO,qBAAqB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;AACjD,CAAC;AAwBD,KAAK,UAAU,uBAAuB,CAClC,MAAqB,EACrB,OAA0B,EAC1B,OAAe;IAEf,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;IAC3D,MAAM,KAAK,GAA0B,EAAE,CAAC;IACxC,IAAI,WAA+B,CAAC;IAEpC,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACjC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,oCAAoC,KAAK,CAAC,IAAI,SAAS,CAAC,CAAC;QAC7E,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACf,MAAM,IAAI,GAAsB,EAAE,CAAC;YACnC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;gBAAE,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC;YACvD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;gBAAE,IAAI,CAAC,UAAU,GAAG,MAAM,kBAAkB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACrF,IAAI,OAAO,CAAC,UAAU;gBAAE,IAAI,CAAC,eAAe,GAAG,EAAE,qBAAqB,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC;YAC7F,MAAM,GAAG,GAAG,MAAM,UAAU,CAAqB,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YACxE,IAAI,GAAG,CAAC,OAAO;gBAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YAChF,IAAI,CAAC,GAAG,CAAC,SAAS;gBAAE,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;YAC1F,KAAK,CAAC,IAAI,CAAC;gBACP,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;gBAC1D,YAAY,EAAE,GAAG,CAAC,mBAAmB;aACxC,CAAC,CAAC;YACH,IAAI,OAAO,GAAG,CAAC,mBAAmB,KAAK,QAAQ;gBAAE,WAAW,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,mBAAmB,CAAC;QAChH,CAAC;aAAM,CAAC;YACJ,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,gCAAgC,OAAO,4BAA4B,CAAC,CAAC;YACzF,CAAC;YACD,MAAM,IAAI,GAAqB,EAAE,SAAS,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;YACzD,IAAI,OAAO,CAAC,UAAU;gBAAE,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;YAC7D,MAAM,GAAG,GAAG,MAAM,UAAU,CAAoB,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YACvE,IAAI,CAAC,GAAG,CAAC,SAAS;gBAAE,MAAM,IAAI,KAAK,CAAC,4DAA4D,OAAO,GAAG,CAAC,CAAC;YAC5G,KAAK,CAAC,IAAI,CAAC;gBACP,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACtD,YAAY,EAAE,GAAG,CAAC,mBAAmB;aACxC,CAAC,CAAC;YACH,IAAI,OAAO,GAAG,CAAC,mBAAmB,KAAK,QAAQ;gBAAE,WAAW,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,mBAAmB,CAAC;QAChH,CAAC;IACL,CAAC;IAED,MAAM,KAAK,GAAG,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACpF,OAAO,qBAAqB,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;AACxD,CAAC;AAkBD,SAAS,eAAe,CAAC,QAAuC;IAC5D,QAAQ,QAAQ,EAAE,CAAC;QACf,KAAK,OAAO;YACR,OAAO,cAAc,CAAC;QAC1B,KAAK,UAAU;YACX,OAAO,iBAAiB,CAAC;QAC7B;YACI,OAAO,SAAS,CAAC;IACzB,CAAC;AACL,CAAC;AAED,KAAK,UAAU,wBAAwB,CACnC,MAAqB,EACrB,OAA0B,EAC1B,OAAe;IAEf,MAAM,UAAU,GAAmD,EAAE,CAAC;IACtE,MAAM,WAAW,GAAoD,EAAE,CAAC;IACxE,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;QACpC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;YAAE,UAAU,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;aACxD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;YAAE,WAAW,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;;YAC/D,MAAM,IAAI,KAAK,CAAC,qCAAqC,KAAK,CAAC,IAAI,SAAS,CAAC,CAAC;IACnF,CAAC,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,IAAI,KAAK,CAAsB,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAEpE,4FAA4F;IAC5F,MAAM,UAAU,GAAG,IAAI,GAAG,EAA+E,CAAC;IAC1G,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,CAAC;QACxE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAClD,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IACD,KAAK,MAAM,CAAC,cAAc,EAAE,KAAK,CAAC,IAAI,UAAU,EAAE,CAAC;QAC/C,MAAM,IAAI,GAA2B;YACjC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC;YACrC,UAAU,EAAE,cAAc;SAC7B,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,UAAU,CAA0B,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;QAC7E,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,EAAE,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,IAAI,CACd,uDAAuD,KAAK,CAAC,MAAM,SAAS,GAAG,CAAC,UAAU,CAAC,MAAM,WAAW,OAAO,GAAG,CACzH,CAAC;YACF,MAAM,IAAI,KAAK,CACX,mBAAmB,GAAG,CAAC,UAAU,CAAC,MAAM,mBAAmB,KAAK,CAAC,MAAM,iBAAiB,OAAO,GAAG,CACrG,CAAC;QACN,CAAC;QACD,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;YACvB,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;QACxF,CAAC,CAAC,CAAC;IACP,CAAC;IAED,uEAAuE;IACvE,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAG,QAAQ,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,WAAW,MAAM,EAAE,CAAC;QACxE,MAAM,IAAI,GAA2B,EAAE,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;QAChF,MAAM,GAAG,GAAG,MAAM,UAAU,CAA0B,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;QAC7E,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,uDAAuD,OAAO,GAAG,CAAC,CAAC;QACvF,CAAC;QACD,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IACzF,CAAC;IAED,OAAO,qBAAqB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;AACjD,CAAC;AAMD,SAAS,iBAAiB,CAAC,CAAqB;IAC5C,OAAO,KAAK,CAAC,OAAO,CAAE,CAA8B,CAAC,UAAU,CAAC,CAAC;AACrE,CAAC;AAED,KAAK,UAAU,yBAAyB,CACpC,MAAqB,EACrB,OAA0B,EAC1B,OAAe;IAEf,MAAM,KAAK,GAA0B,EAAE,CAAC;IAExC,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAqB,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAChF,MAAM,QAAQ,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QAEhF,MAAM,OAAO,GAAsB,QAAQ;aACtC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;aAC7C,GAAG,CACA,CAAC,GAAG,EAAE,EAAE,CACJ,CAAC;YACG,MAAM,EAAE,GAAG,CAAC,SAAS;YACrB,QAAQ,EAAE,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI;YACrD,SAAS,EAAE,GAAG,CAAC,QAAQ;YACvB,OAAO,EAAE,GAAG,CAAC,MAAM;YACnB,gBAAgB,EAAE,GAAG,CAAC,eAAe;SACxC,CAA2B,CACnC,CAAC;QAEN,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,qEAAqE,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC;QACxG,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO,qBAAqB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;AACjD,CAAC;AAED,KAAK,UAAU,mBAAmB,CAAC,KAAqB;IACpD,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACjB,KAAK,MAAM;YACP,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;QACxD,KAAK,OAAO,CAAC,CAAC,CAAC;YACX,MAAM,KAAK,GAAG,MAAM,8BAA8B,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACjE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACtD,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACX,MAAM,KAAK,GAAG,MAAM,8BAA8B,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACjE,MAAM,OAAO,GAA6B,EAAE,SAAS,EAAE,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;YACrF,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS;gBAAE,OAAO,CAAC,QAAQ,GAAG,KAAK,CAAC,SAAS,CAAC;YACtE,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS;gBAAE,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC;YACzE,IAAI,KAAK,CAAC,oBAAoB,KAAK,SAAS;gBAAE,OAAO,CAAC,iBAAiB,GAAG,KAAK,CAAC,oBAAoB,CAAC;YACrG,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS;gBAAE,OAAO,CAAC,UAAU,GAAG,KAAK,CAAC,YAAY,CAAC;YAC9E,4DAA4D;YAC5D,0FAA0F;YAC1F,IAAI,KAAK,CAAC,gBAAgB,EAAE,CAAC;gBACzB,IAAI,KAAK,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpC,MAAM,IAAI,KAAK,CACX,yDAAyD;wBACrD,aAAa,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK;wBACnD,yCAAyC,CAChD,CAAC;gBACN,CAAC;gBACD,IAAI,KAAK,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACtC,OAAO,CAAC,eAAe,GAAG,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;gBACxD,CAAC;YACL,CAAC;YACD,OAAO,OAAO,CAAC;QACnB,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACX,MAAM,KAAK,GAAG,MAAM,8BAA8B,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACjE,MAAM,OAAO,GAA6B,EAAE,SAAS,EAAE,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;YACrF,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS;gBAAE,OAAO,CAAC,QAAQ,GAAG,KAAK,CAAC,SAAS,CAAC;YACtE,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS;gBAAE,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC;YACzE,OAAO,OAAO,CAAC;QACnB,CAAC;IACL,CAAC;AACL,CAAC;AAED,kDAAkD;AAElD,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC3C,MAAqB,EACrB,OAA0B;IAE1B,MAAM,UAAU,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;IACvD,MAAM,OAAO,GAAG,UAAU,CAAC,KAAK,IAAI,+BAA+B,CAAC;IACpE,MAAM,CAAC,MAAM,CAAC,IAAI,CACd,8CAA8C,OAAO,QAAQ,UAAU,CAAC,MAAM,CAAC,MAAM,WAAW,CACnG,CAAC;IACF,IAAI,CAAC;QACD,IAAI,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;YACzC,OAAO,MAAM,yBAAyB,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;YACrC,OAAO,MAAM,wBAAwB,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QACvE,CAAC;QACD,IAAI,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YAClC,OAAO,MAAM,uBAAuB,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QACtE,CAAC;QACD,OAAO,MAAM,sBAAsB,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IACrE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,IAAI,eAAe,CAAC,iBAAiB,CAAC,KAAK,CAAC;YAAE,MAAM,KAAK,CAAC;QAC1D,wEAAwE;QACxE,IACI,KAAK,YAAY,KAAK;YACtB,OAAQ,KAAiC,CAAC,SAAS,KAAK,WAAW;YACnE,OAAQ,KAA8B,CAAC,MAAM,KAAK,QAAQ;YAE1D,MAAM,KAAK,CAAC;QAChB,MAAM,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE;YACtC,QAAQ,EAAE,SAAS;YACnB,KAAK,EAAE,OAAO;YACd,SAAS,EAAE,SAAS;SACvB,CAAC,CAAC;IACP,CAAC;AACL,CAAC"}

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

@@ -1,9 +1,10 @@
-import { Bedrock, type FoundationModelSummary } from "@aws-sdk/client-bedrock";
-import { BedrockRuntime, type ConverseRequest, type ConverseResponse, type ConverseStreamOutput, type Message } from "@aws-sdk/client-bedrock-runtime";
-import type { AwsCredentialIdentity, Provider } from "@aws-sdk/types";
-import { AbstractDriver, type AIModel, type Completion, type CompletionChunkObject, type DataSource, type DriverOptions, type EmbeddingsOptions, type EmbeddingsResult, type ExecutionOptions, LlumiverseError, type LlumiverseErrorContext, type PromptSegment, type TrainingJob, type TrainingOptions } from "@llumiverse/core";
-import { type NovaMessagesPrompt } from "@llumiverse/core/formatters";
-import { type TwelvelabsPegasusRequest } from "./twelvelabs.js";
+import { Bedrock, type FoundationModelSummary } from '@aws-sdk/client-bedrock';
+import { BedrockRuntime, type ConverseRequest, type ConverseResponse, type ConverseStreamOutput, type Message } from '@aws-sdk/client-bedrock-runtime';
+import type { AwsCredentialIdentity, Provider } from '@aws-sdk/types';
+import { type AIModel, type Completion, type CompletionChunkObject, type DataSource, type DriverOptions, type EmbeddingsOptions, type EmbeddingsResult, type ExecutionOptions, type HttpTimeoutOptions, LlumiverseError, type LlumiverseErrorContext, type PromptSegment, type TrainingJob, type TrainingOptions } from '@llumiverse/core';
+import { AbstractDriver } from '@llumiverse/core/driver';
+import { type NovaMessagesPrompt } from '@llumiverse/core/formatters';
+import { type TwelvelabsPegasusRequest } from './twelvelabs.js';
 export interface BedrockModelCapabilities {
     name: string;
     canStream: boolean;
@@ -35,9 +36,19 @@ export declare class BedrockDriver extends AbstractDriver<BedrockDriverOptions,
     private _service?;
     private _service_region?;
     constructor(options: BedrockDriverOptions);
-    getExecutor(): BedrockRuntime;
+    /**
+     * Build a Smithy `requestHandler` config from the driver's
+     * `httpTimeout` so AWS SDK calls fail fast on hung upstream Bedrock
+     * endpoints instead of using the AWS default (no request timeout).
+     * Returns a partial config the SDK merges into its default handler.
+     */
+    private getBedrockRequestHandlerConfig;
+    private createExecutor;
+    getExecutor(httpTimeout?: HttpTimeoutOptions): BedrockRuntime;
+    private getScopedExecutor;
     getService(region?: string): Bedrock;
     protected formatPrompt(segments: PromptSegment[], opts: ExecutionOptions): Promise<BedrockPrompt>;
+    formatDebugPrompt(prompt: BedrockPrompt): BedrockPrompt;
     /**
      * Format AWS Bedrock errors into LlumiverseError with proper status codes and retryability.
      *
@@ -104,8 +115,7 @@ export declare class BedrockDriver extends AbstractDriver<BedrockDriverOptions,
     listTrainableModels(): Promise<AIModel<string>[]>;
     listModels(): Promise<AIModel[]>;
     _listModels(foundationFilter?: (m: FoundationModelSummary) => boolean): Promise<AIModel[]>;
-    generateEmbeddings({ text, image, model }: EmbeddingsOptions): Promise<EmbeddingsResult>;
-    private generateTwelvelabsMarengoEmbeddings;
+    generateEmbeddings(options: EmbeddingsOptions): Promise<EmbeddingsResult>;
     /**
      * Cleanup AWS SDK clients when the driver is evicted from the cache.
      */
@@ -135,4 +145,18 @@ export declare function convertToolBlocksToText(messages: Message[]): Message[];
  * the tools were interrupted, allowing the conversation to continue.
  */
 export declare function fixOrphanedToolUse(messages: Message[]): Message[];
+/**
+ * Drop toolResult blocks whose toolUseId has no matching toolUse in the
+ * immediately-preceding assistant message. Mirror of {@link fixOrphanedToolUse}:
+ * that function synthesizes results for an unanswered toolUse (e.g. a cancelled
+ * run); this one removes results left dangling after their toolUse was dropped
+ * (e.g. by conversation compaction/trimming).
+ *
+ * Without this, the AWS Converse API rejects the request because every
+ * toolResult must correspond to a toolUse in the previous message. Bedrock
+ * conversations are already role-alternating, so a compaction that drops an
+ * assistant toolUse turn leaves an orphaned toolResult (and a user/user
+ * adjacency); dropping the orphan — and the now-empty message — repairs both.
+ */
+export declare function fixOrphanedToolResults(messages: Message[]): Message[];
 //# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/bedrock/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,OAAO,EAEP,KAAK,sBAAsB,EAM9B,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACH,cAAc,EAEd,KAAK,eAAe,EACpB,KAAK,gBAAgB,EACrB,KAAK,oBAAoB,EAGzB,KAAK,OAAO,EAGf,MAAM,iCAAiC,CAAC;AAEzC,OAAO,KAAK,EAAE,qBAAqB,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AACtE,OAAO,EACH,KAAK,OAAO,EAIZ,KAAK,UAAU,EACf,KAAK,qBAAqB,EAE1B,KAAK,UAAU,EACf,KAAK,aAAa,EAElB,KAAK,iBAAiB,EACtB,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EAKrB,KAAK,kBAAkB,EAGvB,eAAe,EACf,KAAK,sBAAsB,EAI3B,KAAK,aAAa,EAOlB,KAAK,WAAW,EAEhB,KAAK,eAAe,EAEvB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAoB,KAAK,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAcxF,OAAO,EAAiC,KAAK,wBAAwB,EAAE,MAAM,iBAAiB,CAAC;AAwD/F,MAAM,WAAW,wBAAwB;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,OAAO,CAAC;CACtB;AAED,MAAM,WAAW,oBAAqB,SAAQ,aAAa;IACvD;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IAEzB;;OAEG;IACH,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAE3B;;OAEG;IACH,WAAW,CAAC,EAAE,qBAAqB,GAAG,QAAQ,CAAC,qBAAqB,CAAC,CAAC;CACzE;AAiBD,MAAM,MAAM,aAAa,GAAG,kBAAkB,GAAG,eAAe,GAAG,wBAAwB,CAAC;AA0K5F,qBAAa,aAAc,SAAQ,cAAc,CAAC,oBAAoB,EAAE,aAAa,CAAC;IAClF,MAAM,CAAC,QAAQ,SAAa;IAE5B,QAAQ,SAA0B;IAElC,OAAO,CAAC,SAAS,CAAC,CAAiB;IACnC,OAAO,CAAC,QAAQ,CAAC,CAAU;IAC3B,OAAO,CAAC,eAAe,CAAC,CAAS;gBAErB,OAAO,EAAE,oBAAoB;IAOzC;;;;;OAKG;IACH,OAAO,CAAC,8BAA8B;IAYtC,OAAO,CAAC,cAAc;IAQtB,WAAW,CAAC,WAAW,CAAC,EAAE,kBAAkB;IAU5C,OAAO,CAAC,iBAAiB;IAezB,UAAU,CAAC,MAAM,GAAE,MAA4B;cAY/B,YAAY,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,CAAC;IAUhG,iBAAiB,CAAC,MAAM,EAAE,aAAa,GAAG,aAAa;IAU9D;;;;;;;;;;;;OAYG;IACI,qBAAqB,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,sBAAsB,GAAG,eAAe;IAyD9F;;;;;;;;;;;;;;;;;;;;;;;OAuBG;IACH,OAAO,CAAC,uBAAuB;IAwC/B,qBAAqB,CACjB,MAAM,EAAE,gBAAgB,EACxB,OAAO,CAAC,EAAE,aAAa,EACvB,OAAO,CAAC,EAAE,gBAAgB,GAC3B,qBAAqB;IAgExB,kBAAkB,CACd,MAAM,EAAE,oBAAoB,EAC5B,OAAO,CAAC,EAAE,aAAa,EACvB,OAAO,CAAC,EAAE,gBAAgB,EAC1B,mBAAmB,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,GAChE,qBAAqB;IAiHxB,aAAa,CAAC,WAAW,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,MAAM;YAkBnD,YAAY;cA8CV,SAAS,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC;IAsBtE;;;OAGG;IACH,0BAA0B,CACtB,MAAM,EAAE,aAAa,EACrB,MAAM,EAAE,OAAO,EAAE,EACjB,OAAO,EAAE,OAAO,EAAE,GAAG,SAAS,EAC9B,OAAO,EAAE,gBAAgB,GAC1B,eAAe,GAAG,SAAS;IAoFxB,qBAAqB,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;YAqFpF,kCAAkC;YA0ClC,wCAAwC;IAmEhD,2BAA2B,CAC7B,MAAM,EAAE,aAAa,EACrB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;IAwChD,cAAc,CAAC,MAAM,EAAE,eAAe,EAAE,OAAO,EAAE,gBAAgB;IA0PjE,SAAS,CAAC,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAIxC,sBAAsB,CAAC,MAAM,EAAE,kBAAkB,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IAuDlG,aAAa,CAAC,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,WAAW,CAAC;IA6ClF,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAgBnD,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAanD,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC;IAOtC,mBAAmB,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IAOjD,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAShC,WAAW,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC,EAAE,sBAAsB,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAgL1F,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAI/E;;OAEG;IACH,OAAO,IAAI,IAAI;CAOlB;AAyCD;;GAEG;AACH,wBAAgB,yBAAyB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAUtE;AAED;;;;;;;GAOG;AACH,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,CA0CtE;AAyED;;;;;;;;;GASG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,CAwFjE;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,CA6BrE"}