@truefoundry/ai-gateway-sdk 0.0.0-stage → 0.0.1

package/dist/providers/bedrock/embed.js

@@ -0,0 +1,345 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. Add array of strings support for Titan embeddings
+ * ================================
+ */
+import { HTTPException } from 'hono/http-exception';
+import { BEDROCK, mimeTypeToExtensionMap } from '../../globals.js';
+import { generateInvalidProviderResponseError, postProcessEmbeddingResponse, validateEncodingFormatAndRemove, } from '../utils.js';
+import { parseFileData } from '../utils/fileUtils.js';
+import { BedrockErrorResponseTransform } from './chatComplete.js';
+const BASE64_MAGIC_TO_FORMAT = [
+    ['/9j/', 'jpeg'],
+    ['iVBOR', 'png'],
+    ['R0lGOD', 'gif'],
+    ['UklGR', 'webp'],
+];
+function detectImageFormat(base64Data) {
+    const { base64String, mimeType } = parseFileData(base64Data);
+    if (mimeType) {
+        const ext = mimeTypeToExtensionMap[mimeType];
+        if (ext)
+            return ext === 'jpg' ? 'jpeg' : ext;
+    }
+    for (const [prefix, fmt] of BASE64_MAGIC_TO_FORMAT) {
+        if (base64String.startsWith(prefix))
+            return fmt;
+    }
+    return 'jpeg';
+}
+export const BedrockCohereEmbedConfig = {
+    input: [
+        {
+            param: 'texts',
+            required: false,
+            transform: (params) => {
+                if (typeof params.input === 'string')
+                    return [params.input];
+                else if (Array.isArray(params.input) && params.input.length > 0) {
+                    const texts = [];
+                    params.input.forEach(item => {
+                        if (typeof item === 'string') {
+                            texts.push(item);
+                        }
+                        else if (item.text) {
+                            texts.push(item.text);
+                        }
+                    });
+                    return texts.length > 0 ? texts : undefined;
+                }
+            },
+        },
+        {
+            param: 'images',
+            required: false,
+            transform: (params) => {
+                if (Array.isArray(params.input) && params.input.length > 0) {
+                    const images = [];
+                    params.input.forEach(item => {
+                        if (typeof item === 'object' && item.image?.base64) {
+                            images.push(item.image.base64);
+                        }
+                    });
+                    return images.length > 0 ? images : undefined;
+                }
+            },
+        },
+    ],
+    input_type: {
+        param: 'input_type',
+        required: true,
+        /**
+         * ================================
+         * NOTE: Default to search_document for now
+         * ================================
+         */
+        default: 'search_document',
+    },
+    truncate: {
+        param: 'truncate',
+        required: false,
+    },
+    encoding_format: {
+        param: 'embedding_types',
+        required: false,
+        removeFromParams: validateEncodingFormatAndRemove,
+    },
+};
+export const BedrockNovaEmbedConfig = {
+    schemaVersion: {
+        param: 'schemaVersion',
+        required: true,
+        default: 'nova-multimodal-embed-v1',
+    },
+    taskType: {
+        param: 'taskType',
+        required: true,
+        default: 'SINGLE_EMBEDDING',
+    },
+    input: [
+        {
+            param: 'singleEmbeddingParams.text',
+            required: false,
+            transform: (params) => {
+                if (typeof params.input === 'string') {
+                    return { truncationMode: 'END', value: params.input };
+                }
+                if (Array.isArray(params.input)) {
+                    if (typeof params.input[0] === 'object' && params.input[0].text) {
+                        return { truncationMode: 'END', value: params.input[0].text };
+                    }
+                    if (typeof params.input[0] === 'string') {
+                        if (params.input.length === 1) {
+                            return { truncationMode: 'END', value: params.input[0] };
+                        }
+                        if (params.input.length > 1) {
+                            throw new HTTPException(400, {
+                                message: 'Nova embeddings accepts a single input per request; got multiple items in array.',
+                            });
+                        }
+                    }
+                }
+            },
+        },
+        {
+            param: 'singleEmbeddingParams.image',
+            required: false,
+            transform: (params) => {
+                if (Array.isArray(params.input) && typeof params.input[0] === 'object' && params.input[0].image?.base64) {
+                    const raw = params.input[0].image.base64;
+                    const format = detectImageFormat(raw);
+                    const { base64String } = parseFileData(raw);
+                    return {
+                        format,
+                        source: { bytes: base64String },
+                    };
+                }
+            },
+        },
+    ],
+    dimensions: {
+        param: 'singleEmbeddingParams.embeddingDimension',
+        required: false,
+    },
+    input_type: {
+        param: 'singleEmbeddingParams.embeddingPurpose',
+        required: true,
+        default: 'GENERIC_INDEX',
+        transform: (params) => {
+            const inputType = params.input_type;
+            switch (inputType) {
+                case 'search_document':
+                    return 'GENERIC_INDEX';
+                case 'search_query':
+                    return 'GENERIC_RETRIEVAL';
+                case 'classification':
+                    return 'CLASSIFICATION';
+                case 'clustering':
+                    return 'CLUSTERING';
+                default:
+                    return 'GENERIC_INDEX';
+            }
+        },
+    },
+    encoding_format: {
+        param: 'encoding_format',
+        required: false,
+        removeFromParams: validateEncodingFormatAndRemove,
+    },
+};
+export const BedrockTitanEmbedConfig = {
+    input: [
+        {
+            param: 'inputText',
+            required: false,
+            transform: (params) => {
+                if (Array.isArray(params.input) && typeof params.input[0] === 'object' && params.input[0].text) {
+                    return params.input[0].text;
+                }
+                /**
+                 * ================================
+                 * NOTE: Add array of strings support for Titan embeddings
+                 * ================================
+                 */
+                if (Array.isArray(params.input) && typeof params.input[0] === 'string') {
+                    if (params.input.length === 1)
+                        return params.input[0];
+                    else if (params.input.length > 1) {
+                        throw new HTTPException(400, {
+                            message: 'Titan embeddings accepts a single input string; got 2 items in array.',
+                        });
+                    }
+                }
+                if (typeof params.input === 'string')
+                    return params.input;
+            },
+        },
+        {
+            param: 'inputImage',
+            required: false,
+            transform: (params) => {
+                // Titan models only support one image per request
+                if (Array.isArray(params.input) && typeof params.input[0] === 'object' && params.input[0].image?.base64) {
+                    return params.input[0].image.base64;
+                }
+            },
+        },
+    ],
+    dimensions: [
+        {
+            param: 'dimensions',
+            required: false,
+            transform: (params) => {
+                if (typeof params.input === 'string')
+                    return params.dimensions;
+            },
+        },
+        {
+            param: 'embeddingConfig',
+            required: false,
+            transform: (params) => {
+                if (Array.isArray(params.input) && params.dimensions) {
+                    return {
+                        outputEmbeddingLength: params.dimensions,
+                    };
+                }
+            },
+        },
+    ],
+    encoding_format: {
+        param: 'embeddingTypes',
+        required: false,
+        removeFromParams: validateEncodingFormatAndRemove,
+    },
+    // Titan specific parameters
+    normalize: {
+        param: 'normalize',
+        required: false,
+    },
+};
+export const BedrockTitanEmbedResponseTransform = (response, responseStatus, _responseHeaders, _strictOpenAiCompliance, _gatewayRequestUrl, gatewayRequest) => {
+    if (responseStatus !== 200) {
+        const errorResponse = BedrockErrorResponseTransform(response);
+        if (errorResponse)
+            return errorResponse;
+    }
+    const model = gatewayRequest.model || '';
+    if ('embedding' in response) {
+        const openAIResponse = {
+            object: 'list',
+            data: [
+                {
+                    object: 'embedding',
+                    embedding: response.embedding,
+                    index: 0,
+                },
+            ],
+            provider: BEDROCK,
+            model,
+            usage: {
+                prompt_tokens: response.inputTextTokenCount,
+                total_tokens: response.inputTextTokenCount,
+            },
+        };
+        // Post-process to convert to base64 if requested
+        return postProcessEmbeddingResponse(openAIResponse, gatewayRequest.encoding_format);
+    }
+    return generateInvalidProviderResponseError(response, BEDROCK);
+};
+var BedrockCohereEmbedResponseType;
+(function (BedrockCohereEmbedResponseType) {
+    BedrockCohereEmbedResponseType["FLOATS"] = "embeddings_floats";
+    BedrockCohereEmbedResponseType["BY_TYPE"] = "embeddings_by_type";
+})(BedrockCohereEmbedResponseType || (BedrockCohereEmbedResponseType = {}));
+export const BedrockNovaEmbedResponseTransform = (response, responseStatus, responseHeaders, _strictOpenAiCompliance, _gatewayRequestUrl, gatewayRequest) => {
+    if (responseStatus !== 200) {
+        const errorResponse = BedrockErrorResponseTransform(response);
+        if (errorResponse)
+            return errorResponse;
+    }
+    const model = gatewayRequest.model || '';
+    if ('embeddings' in response) {
+        const openAIResponse = {
+            object: 'list',
+            data: response.embeddings.map((item, index) => ({
+                object: 'embedding',
+                embedding: item.embedding,
+                index: index,
+            })),
+            provider: BEDROCK,
+            model,
+            usage: {
+                prompt_tokens: Number(responseHeaders.get('X-Amzn-Bedrock-Input-Token-Count')) || 0,
+                total_tokens: Number(responseHeaders.get('X-Amzn-Bedrock-Input-Token-Count')) || 0,
+            },
+        };
+        return postProcessEmbeddingResponse(openAIResponse, gatewayRequest.encoding_format);
+    }
+    return generateInvalidProviderResponseError(response, BEDROCK);
+};
+export const BedrockCohereEmbedResponseTransform = (response, responseStatus, responseHeaders, _strictOpenAiCompliance, _gatewayRequestUrl, gatewayRequest) => {
+    if (responseStatus !== 200) {
+        const errorResponse = BedrockErrorResponseTransform(response);
+        if (errorResponse)
+            return errorResponse;
+    }
+    const model = gatewayRequest.model || '';
+    if ('embeddings' in response) {
+        let embeddingsList;
+        switch (response.response_type) {
+            case BedrockCohereEmbedResponseType.FLOATS:
+                // Flat array of vectors — [[0.01, ...], [0.02, ...]]
+                embeddingsList = response.embeddings;
+                break;
+            case BedrockCohereEmbedResponseType.BY_TYPE:
+                // Object keyed by type — { float: [[0.01, ...]] }
+                if (!('float' in response.embeddings)) {
+                    return generateInvalidProviderResponseError(response, BEDROCK);
+                }
+                embeddingsList = response.embeddings.float;
+                break;
+            default:
+                return generateInvalidProviderResponseError(response, BEDROCK);
+        }
+        const openAIResponse = {
+            object: 'list',
+            data: embeddingsList.map((embedding, index) => ({
+                object: 'embedding',
+                embedding: embedding,
+                index: index,
+            })),
+            provider: BEDROCK,
+            model,
+            usage: {
+                prompt_tokens: Number(responseHeaders.get('X-Amzn-Bedrock-Input-Token-Count')) || -1,
+                total_tokens: Number(responseHeaders.get('X-Amzn-Bedrock-Input-Token-Count')) || -1,
+            },
+        };
+        // Post-process to convert to base64 if requested
+        return postProcessEmbeddingResponse(openAIResponse, gatewayRequest.encoding_format);
+    }
+    return generateInvalidProviderResponseError(response, BEDROCK);
+};
+//# sourceMappingURL=embed.js.map

package/dist/providers/bedrock/embed.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embed.js","sourceRoot":"","sources":["../../../src/providers/bedrock/embed.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAIhE,OAAO,EACL,oCAAoC,EACpC,4BAA4B,EAC5B,+BAA+B,GAChC,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,6BAA6B,EAAE,MAAM,gBAAgB,CAAC;AAE/D,MAAM,sBAAsB,GAAuB;IACjD,CAAC,MAAM,EAAE,MAAM,CAAC;IAChB,CAAC,OAAO,EAAE,KAAK,CAAC;IAChB,CAAC,QAAQ,EAAE,KAAK,CAAC;IACjB,CAAC,OAAO,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,SAAS,iBAAiB,CAAC,UAAkB;IAC3C,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAE,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;IAC7D,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,GAAG,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,GAAG;YAAE,OAAO,GAAG,KAAK,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC;IAC/C,CAAC;IACD,KAAK,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,sBAAsB,EAAE,CAAC;QACnD,IAAI,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC;YAAE,OAAO,GAAG,CAAC;IAClD,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,MAAM,wBAAwB,GAAmB;IACtD,KAAK,EAAE;QACL;YACE,KAAK,EAAE,OAAO;YACd,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,CAAC,MAAmB,EAAwB,EAAE;gBACvD,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;oBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;qBACvD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAChE,MAAM,KAAK,GAAa,EAAE,CAAC;oBAC3B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;wBAC1B,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;4BAC7B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBACnB,CAAC;6BAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;4BACrB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBACxB,CAAC;oBACH,CAAC,CAAC,CAAC;oBACH,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC9C,CAAC;YACH,CAAC;SACF;QACD;YACE,KAAK,EAAE,QAAQ;YACf,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,CAAC,MAAmB,EAAwB,EAAE;gBACvD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC3D,MAAM,MAAM,GAAa,EAAE,CAAC;oBAC5B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;wBAC1B,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;4BACnD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;wBACjC,CAAC;oBACH,CAAC,CAAC,CAAC;oBACH,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;gBAChD,CAAC;YACH,CAAC;SACF;KACF;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,QAAQ,EAAE,IAAI;QACd;;;;WAIG;QACH,OAAO,EAAE,iBAAiB;KAC3B;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,KAAK;KAChB;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,KAAK;QACf,gBAAgB,EAAE,+BAA+B;KAClD;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,sBAAsB,GAAmB;IACpD,aAAa,EAAE;QACb,KAAK,EAAE,eAAe;QACtB,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,0BAA0B;KACpC;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,kBAAkB;KAC5B;IACD,KAAK,EAAE;QACL;YACE,KAAK,EAAE,4BAA4B;YACnC,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,CAAC,MAAmB,EAAyD,EAAE;gBACxF,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;oBACrC,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;gBACxD,CAAC;gBACD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;oBAChC,IAAI,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;wBAChE,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBAChE,CAAC;oBACD,IAAI,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;wBACxC,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;4BAC9B,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC3D,CAAC;wBACD,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BAC5B,MAAM,IAAI,aAAa,CAAC,GAAG,EAAE;gCAC3B,OAAO,EAAE,kFAAkF;6BAC5F,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;SACF;QACD;YACE,KAAK,EAAE,6BAA6B;YACpC,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,CAAC,MAAmB,EAAE,EAAE;gBACjC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;oBACxG,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;oBACzC,MAAM,MAAM,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC;oBACtC,MAAM,EAAE,YAAY,EAAE,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;oBAC5C,OAAO;wBACL,MAAM;wBACN,MAAM,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;qBAChC,CAAC;gBACJ,CAAC;YACH,CAAC;SACF;KACF;IACD,UAAU,EAAE;QACV,KAAK,EAAE,0CAA0C;QACjD,QAAQ,EAAE,KAAK;KAChB;IACD,UAAU,EAAE;QACV,KAAK,EAAE,wCAAwC;QAC/C,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,eAAe;QACxB,SAAS,EAAE,CAAC,MAAmB,EAAE,EAAE;YACjC,MAAM,SAAS,GAAI,MAAc,CAAC,UAAU,CAAC;YAC7C,QAAQ,SAAS,EAAE,CAAC;gBAClB,KAAK,iBAAiB;oBACpB,OAAO,eAAe,CAAC;gBACzB,KAAK,cAAc;oBACjB,OAAO,mBAAmB,CAAC;gBAC7B,KAAK,gBAAgB;oBACnB,OAAO,gBAAgB,CAAC;gBAC1B,KAAK,YAAY;oBACf,OAAO,YAAY,CAAC;gBACtB;oBACE,OAAO,eAAe,CAAC;YAC3B,CAAC;QACH,CAAC;KACF;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,KAAK;QACf,gBAAgB,EAAE,+BAA+B;KAClD;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,uBAAuB,GAAmB;IACrD,KAAK,EAAE;QACL;YACE,KAAK,EAAE,WAAW;YAClB,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,CAAC,MAAmB,EAAsB,EAAE;gBACrD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBAC/F,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC9B,CAAC;gBACD;;;;mBAIG;gBACH,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;oBACvE,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;wBAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;yBACjD,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACjC,MAAM,IAAI,aAAa,CAAC,GAAG,EAAE;4BAC3B,OAAO,EAAE,uEAAuE;yBACjF,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBACD,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;oBAAE,OAAO,MAAM,CAAC,KAAK,CAAC;YAC5D,CAAC;SACF;QACD;YACE,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,CAAC,MAAmB,EAAE,EAAE;gBACjC,kDAAkD;gBAClD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;oBACxG,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;gBACtC,CAAC;YACH,CAAC;SACF;KACF;IACD,UAAU,EAAE;QACV;YACE,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,CAAC,MAAmB,EAAsB,EAAE;gBACrD,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;oBAAE,OAAO,MAAM,CAAC,UAAU,CAAC;YACjE,CAAC;SACF;QACD;YACE,KAAK,EAAE,iBAAiB;YACxB,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,CAAC,MAAmB,EAAiD,EAAE;gBAChF,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;oBACrD,OAAO;wBACL,qBAAqB,EAAE,MAAM,CAAC,UAAU;qBACzC,CAAC;gBACJ,CAAC;YACH,CAAC;SACF;KACF;IACD,eAAe,EAAE;QACf,KAAK,EAAE,gBAAgB;QACvB,QAAQ,EAAE,KAAK;QACf,gBAAgB,EAAE,+BAA+B;KAClD;IACD,4BAA4B;IAC5B,SAAS,EAAE;QACT,KAAK,EAAE,WAAW;QAClB,QAAQ,EAAE,KAAK;KAChB;CACF,CAAC;AAWF,MAAM,CAAC,MAAM,kCAAkC,GAOV,CACnC,QAAQ,EACR,cAAc,EACd,gBAAgB,EAChB,uBAAuB,EACvB,kBAAkB,EAClB,cAAc,EACd,EAAE;IACF,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,6BAA6B,CAAC,QAAgC,CAAC,CAAC;QACtF,IAAI,aAAa;YAAE,OAAO,aAAa,CAAC;IAC1C,CAAC;IAED,MAAM,KAAK,GAAI,cAAc,CAAC,KAAgB,IAAI,EAAE,CAAC;IACrD,IAAI,WAAW,IAAI,QAAQ,EAAE,CAAC;QAC5B,MAAM,cAAc,GAAG;YACrB,MAAM,EAAE,MAAM;YACd,IAAI,EAAE;gBACJ;oBACE,MAAM,EAAE,WAAW;oBACnB,SAAS,EAAE,QAAQ,CAAC,SAAS;oBAC7B,KAAK,EAAE,CAAC;iBACT;aACF;YACD,QAAQ,EAAE,OAAO;YACjB,KAAK;YACL,KAAK,EAAE;gBACL,aAAa,EAAE,QAAQ,CAAC,mBAAmB;gBAC3C,YAAY,EAAE,QAAQ,CAAC,mBAAmB;aAC3C;SACF,CAAC;QAEF,iDAAiD;QACjD,OAAO,4BAA4B,CAAC,cAAc,EAAE,cAAc,CAAC,eAAyB,CAAC,CAAC;IAChG,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;AACjE,CAAC,CAAC;AAgCF,IAAK,8BAGJ;AAHD,WAAK,8BAA8B;IACjC,8DAA4B,CAAA;IAC5B,gEAA8B,CAAA;AAChC,CAAC,EAHI,8BAA8B,KAA9B,8BAA8B,QAGlC;AAYD,MAAM,CAAC,MAAM,iCAAiC,GAOT,CACnC,QAAQ,EACR,cAAc,EACd,eAAe,EACf,uBAAuB,EACvB,kBAAkB,EAClB,cAAc,EACd,EAAE;IACF,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,6BAA6B,CAAC,QAAgC,CAAC,CAAC;QACtF,IAAI,aAAa;YAAE,OAAO,aAAa,CAAC;IAC1C,CAAC;IAED,MAAM,KAAK,GAAI,cAAc,CAAC,KAAgB,IAAI,EAAE,CAAC;IAErD,IAAI,YAAY,IAAI,QAAQ,EAAE,CAAC;QAC7B,MAAM,cAAc,GAAG;YACrB,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC9C,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,KAAK,EAAE,KAAK;aACb,CAAC,CAAC;YACH,QAAQ,EAAE,OAAO;YACjB,KAAK;YACL,KAAK,EAAE;gBACL,aAAa,EAAE,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC,IAAI,CAAC;gBACnF,YAAY,EAAE,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC,IAAI,CAAC;aACnF;SACF,CAAC;QAEF,OAAO,4BAA4B,CAAC,cAAc,EAAE,cAAc,CAAC,eAAyB,CAAC,CAAC;IAChG,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;AACjE,CAAC,CAAC;AAmBF,MAAM,CAAC,MAAM,mCAAmC,GAOX,CACnC,QAAQ,EACR,cAAc,EACd,eAAe,EACf,uBAAuB,EACvB,kBAAkB,EAClB,cAAc,EACd,EAAE;IACF,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,6BAA6B,CAAC,QAAgC,CAAC,CAAC;QACtF,IAAI,aAAa;YAAE,OAAO,aAAa,CAAC;IAC1C,CAAC;IAED,MAAM,KAAK,GAAI,cAAc,CAAC,KAAgB,IAAI,EAAE,CAAC;IAErD,IAAI,YAAY,IAAI,QAAQ,EAAE,CAAC;QAC7B,IAAI,cAA0B,CAAC;QAC/B,QAAQ,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC/B,KAAK,8BAA8B,CAAC,MAAM;gBACxC,qDAAqD;gBACrD,cAAc,GAAG,QAAQ,CAAC,UAAU,CAAC;gBACrC,MAAM;YACR,KAAK,8BAA8B,CAAC,OAAO;gBACzC,kDAAkD;gBAClD,IAAI,CAAC,CAAC,OAAO,IAAI,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;oBACtC,OAAO,oCAAoC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACjE,CAAC;gBACD,cAAc,GAAG,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC;gBAC3C,MAAM;YACR;gBACE,OAAO,oCAAoC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,cAAc,GAAG;YACrB,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC9C,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,SAAS;gBACpB,KAAK,EAAE,KAAK;aACb,CAAC,CAAC;YACH,QAAQ,EAAE,OAAO;YACjB,KAAK;YACL,KAAK,EAAE;gBACL,aAAa,EAAE,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACpF,YAAY,EAAE,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC,IAAI,CAAC,CAAC;aACpF;SACF,CAAC;QAEF,iDAAiD;QACjD,OAAO,4BAA4B,CAAC,cAAc,EAAE,cAAc,CAAC,eAAyB,CAAC,CAAC;IAChG,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;AACjE,CAAC,CAAC"}

package/dist/providers/bedrock/getBatchOutput.d.ts

@@ -0,0 +1,6 @@
+import type { RequestHandler } from '../types';
+/** Drop only a trailing `/output` segment so a job id that starts with `output` stays intact. */
+export declare function stripTrailingBatchOutput(path: string): string;
+export declare const BedrockGetBatchOutputRequestHandler: RequestHandler;
+export declare const BedrockGetBatchOutputResponseTransform: (response: Response) => Response;
+//# sourceMappingURL=getBatchOutput.d.ts.map

package/dist/providers/bedrock/getBatchOutput.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"getBatchOutput.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock/getBatchOutput.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAuC/C,iGAAiG;AACjG,wBAAgB,wBAAwB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAE7D;AAED,eAAO,MAAM,mCAAmC,EAAE,cAyDjD,CAAC;AAEF,eAAO,MAAM,sCAAsC,GAAI,UAAU,QAAQ,aAExE,CAAC"}

package/dist/providers/bedrock/getBatchOutput.js

@@ -0,0 +1,112 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. Fix unused vars for linting
+ * 2. Remove the v1 from the URL for batches
+ * ================================
+ */
+import { BEDROCK } from '../../globals.js';
+import { getOctetStreamToOctetStreamTransformer } from '../../handlers/streamHandlerUtils.js';
+import { buildErrorBody, errorResponse } from '../../utils/errorHandlingUtils.js';
+import { bedrockCallRegion } from './api.js';
+import { bedrockProviderFetch } from './requestHandlerFetch.js';
+import { BedrockUploadFileResponseTransforms } from './uploadFileUtils.js';
+const getModelProvider = (modelId) => {
+    let provider = '';
+    if (modelId.includes('llama2'))
+        provider = 'llama2';
+    else if (modelId.includes('llama3'))
+        provider = 'llama3';
+    else if (modelId.includes('titan'))
+        provider = 'titan';
+    else if (modelId.includes('mistral'))
+        provider = 'mistral';
+    else if (modelId.includes('anthropic'))
+        provider = 'anthropic';
+    else if (modelId.includes('ai21'))
+        provider = 'ai21';
+    else if (modelId.includes('cohere'))
+        provider = 'cohere';
+    else if (modelId.includes('nova'))
+        provider = 'nova';
+    else
+        throw new Error('Invalid model slug');
+    return provider;
+};
+const getRowTransform = (modelId) => {
+    const provider = getModelProvider(modelId);
+    return (row) => {
+        if (!row.modelOutput && row.error)
+            return row.error;
+        const transformedResponse = BedrockUploadFileResponseTransforms[provider](row.modelOutput);
+        transformedResponse.model = modelId;
+        return {
+            id: row.modelOutput.id,
+            custom_id: row.recordId,
+            response: {
+                status_code: 200,
+                request_id: row.modelOutput.id,
+                body: transformedResponse,
+            },
+            error: null,
+        };
+    };
+};
+/** Drop only a trailing `/output` segment so a job id that starts with `output` stays intact. */
+export function stripTrailingBatchOutput(path) {
+    return path.endsWith('/output') ? path.slice(0, -'/output'.length) : path;
+}
+export const BedrockGetBatchOutputRequestHandler = async (args) => {
+    try {
+        const batchPath = stripTrailingBatchOutput(args.request.path);
+        const retrieveBatchesResponse = await bedrockProviderFetch({ ...args, request: { ...args.request, path: batchPath } }, 'retrieveBatch', { method: 'GET' });
+        const batchDetails = await retrieveBatchesResponse.json();
+        const outputFileId = batchDetails.outputDataConfig.s3OutputDataConfig.s3Uri;
+        const awsRegion = bedrockCallRegion(args.providerOptions, args.headerOptions.aws);
+        const awsS3Bucket = outputFileId.replace('s3://', '').split('/')[0];
+        const jobId = batchDetails.jobArn.split('/')[1];
+        const inputS3URIParts = batchDetails.inputDataConfig.s3InputDataConfig.s3Uri.split('/');
+        const primaryKey = outputFileId?.replace(`s3://${awsS3Bucket}/`, '') ?? '';
+        const awsS3ObjectKey = `${primaryKey}${jobId}/${inputS3URIParts[inputS3URIParts.length - 1]}.out`;
+        const awsModelProvider = batchDetails.modelId;
+        const s3FileURL = `https://${awsS3Bucket}.s3.${awsRegion}.amazonaws.com/${awsS3ObjectKey}`;
+        const s3FileResponse = await bedrockProviderFetch(args, 'retrieveFileContent', {
+            method: 'GET',
+            url: s3FileURL,
+        });
+        let responseStream;
+        if (s3FileResponse.headers.get('content-type')?.includes('octet-stream') && s3FileResponse?.body) {
+            responseStream = s3FileResponse?.body?.pipeThrough(getOctetStreamToOctetStreamTransformer(getRowTransform(awsModelProvider)));
+            return new Response(responseStream, {
+                headers: {
+                    'content-type': 'application/octet-stream',
+                },
+            });
+        }
+        else {
+            const body = await s3FileResponse.text();
+            throw new Error(body);
+        }
+    }
+    catch (error) {
+        let errorBody;
+        try {
+            errorBody = JSON.parse(error.message);
+            errorBody.provider = BEDROCK;
+        }
+        catch ( /* eslint-disable @typescript-eslint/no-unused-vars */_e) {
+            errorBody = buildErrorBody({
+                message: error.message,
+                statusCode: 500,
+                type: error?.name ?? null,
+                extra: { provider: BEDROCK },
+            });
+        }
+        return errorResponse(errorBody, 500);
+    }
+};
+export const BedrockGetBatchOutputResponseTransform = (response) => {
+    return response;
+};
+//# sourceMappingURL=getBatchOutput.js.map

package/dist/providers/bedrock/getBatchOutput.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getBatchOutput.js","sourceRoot":"","sources":["../../../src/providers/bedrock/getBatchOutput.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EAAE,sCAAsC,EAAE,MAAM,mCAAmC,CAAC;AAC3F,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAE/E,OAAO,EAAE,iBAAiB,EAAE,MAAM,OAAO,CAAC;AAC1C,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAE7D,OAAO,EAAE,mCAAmC,EAAE,MAAM,mBAAmB,CAAC;AAExE,MAAM,gBAAgB,GAAG,CAAC,OAAe,EAAE,EAAE;IAC3C,IAAI,QAAQ,GAAG,EAAE,CAAC;IAClB,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,QAAQ,GAAG,QAAQ,CAAC;SAC/C,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,QAAQ,GAAG,QAAQ,CAAC;SACpD,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,QAAQ,GAAG,OAAO,CAAC;SAClD,IAAI,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;QAAE,QAAQ,GAAG,SAAS,CAAC;SACtD,IAAI,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC;QAAE,QAAQ,GAAG,WAAW,CAAC;SAC1D,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,QAAQ,GAAG,MAAM,CAAC;SAChD,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,QAAQ,GAAG,QAAQ,CAAC;SACpD,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,QAAQ,GAAG,MAAM,CAAC;;QAChD,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;IAC3C,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG,CAAC,OAAe,EAAE,EAAE;IAC1C,MAAM,QAAQ,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC3C,OAAO,CAAC,GAAwB,EAAE,EAAE;QAClC,IAAI,CAAC,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,KAAK;YAAE,OAAO,GAAG,CAAC,KAAK,CAAC;QACpD,MAAM,mBAAmB,GAAG,mCAAmC,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC3F,mBAAmB,CAAC,KAAK,GAAG,OAAO,CAAC;QACpC,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,WAAW,CAAC,EAAE;YACtB,SAAS,EAAE,GAAG,CAAC,QAAQ;YACvB,QAAQ,EAAE;gBACR,WAAW,EAAE,GAAG;gBAChB,UAAU,EAAE,GAAG,CAAC,WAAW,CAAC,EAAE;gBAC9B,IAAI,EAAE,mBAAmB;aAC1B;YACD,KAAK,EAAE,IAAI;SACZ,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF,iGAAiG;AACjG,MAAM,UAAU,wBAAwB,CAAC,IAAY;IACnD,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AAC5E,CAAC;AAED,MAAM,CAAC,MAAM,mCAAmC,GAAmB,KAAK,EAAC,IAAI,EAAC,EAAE;IAC9E,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,wBAAwB,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,uBAAuB,GAAG,MAAM,oBAAoB,CACxD,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAC1D,eAAe,EACf,EAAE,MAAM,EAAE,KAAK,EAAE,CAClB,CAAC;QAEF,MAAM,YAAY,GAA4B,MAAM,uBAAuB,CAAC,IAAI,EAAE,CAAC;QACnF,MAAM,YAAY,GAAG,YAAY,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAE5E,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;QAClF,MAAM,WAAW,GAAG,YAAY,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACpE,MAAM,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAChD,MAAM,eAAe,GAAG,YAAY,CAAC,eAAe,CAAC,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAExF,MAAM,UAAU,GAAG,YAAY,EAAE,OAAO,CAAC,QAAQ,WAAW,GAAG,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;QAE3E,MAAM,cAAc,GAAG,GAAG,UAAU,GAAG,KAAK,IAAI,eAAe,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;QAClG,MAAM,gBAAgB,GAAG,YAAY,CAAC,OAAO,CAAC;QAE9C,MAAM,SAAS,GAAG,WAAW,WAAW,OAAO,SAAS,kBAAkB,cAAc,EAAE,CAAC;QAC3F,MAAM,cAAc,GAAG,MAAM,oBAAoB,CAAC,IAAI,EAAE,qBAAqB,EAAE;YAC7E,MAAM,EAAE,KAAK;YACb,GAAG,EAAE,SAAS;SACf,CAAC,CAAC;QACH,IAAI,cAA8B,CAAC;QACnC,IAAI,cAAc,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,QAAQ,CAAC,cAAc,CAAC,IAAI,cAAc,EAAE,IAAI,EAAE,CAAC;YACjG,cAAc,GAAG,cAAc,EAAE,IAAI,EAAE,WAAW,CAChD,sCAAsC,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC,CAC1E,CAAC;YACF,OAAO,IAAI,QAAQ,CAAC,cAAc,EAAE;gBAClC,OAAO,EAAE;oBACP,cAAc,EAAE,0BAA0B;iBAC3C;aACF,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,IAAI,SAAS,CAAC;QAEd,IAAI,CAAC;YACH,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACtC,SAAS,CAAC,QAAQ,GAAG,OAAO,CAAC;QAC/B,CAAC;QAAC,QAAO,sDAAuD,EAAE,EAAE,CAAC;YACnE,SAAS,GAAG,cAAc,CAAC;gBACzB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,UAAU,EAAE,GAAG;gBACf,IAAI,EAAE,KAAK,EAAE,IAAI,IAAI,IAAI;gBACzB,KAAK,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE;aAC7B,CAAC,CAAC;QACL,CAAC;QACD,OAAO,aAAa,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IACvC,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,QAAkB,EAAE,EAAE;IAC3E,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC"}

package/dist/providers/bedrock/imageEdit.d.ts

@@ -0,0 +1,19 @@
+import { ErrorResponse, ImageGenerateResponse, ProviderConfig, RequestHandler } from '../types';
+import { BedrockErrorResponse } from './embed';
+interface BedrockAmazonNovaCanvasResponse {
+    images: string[];
+}
+interface BedrockStabilityAIImageEditResponse {
+    seeds: number[];
+    finish_reasons: string[];
+    images: string[];
+}
+export declare const BedrockAmazonNovaCanvasImageEditConfig: ProviderConfig;
+export declare const BedrockStabilityAIImageEditConfig: ProviderConfig;
+export declare function BedrockImageEditRequestHandler(args: Parameters<RequestHandler<FormData>>[0]): Promise<Response>;
+export declare const BedrockStabilityAIImageEditRequestHandler: RequestHandler<FormData>;
+export declare const BedrockAmazonNovaCanvasImageEditRequestHandler: RequestHandler<FormData>;
+export declare const BedrockAmazonNovaCanvasImageEditResponseTransform: (response: BedrockAmazonNovaCanvasResponse | BedrockErrorResponse, responseStatus: number) => ImageGenerateResponse | ErrorResponse;
+export declare const BedrockStabilityAIImageEditResponseTransform: (response: BedrockStabilityAIImageEditResponse | BedrockErrorResponse, responseStatus: number) => ImageGenerateResponse | ErrorResponse;
+export {};
+//# sourceMappingURL=imageEdit.d.ts.map

package/dist/providers/bedrock/imageEdit.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"imageEdit.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock/imageEdit.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAIhG,OAAO,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAY/C,UAAU,+BAA+B;IACvC,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,UAAU,mCAAmC;IAC3C,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAkBD,eAAO,MAAM,sCAAsC,EAAE,cA2KpD,CAAC;AAEF,eAAO,MAAM,iCAAiC,EAAE,cAwJ/C,CAAC;AAIF,wBAAsB,8BAA8B,CAAC,IAAI,EAAE,UAAU,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC,CAUrH;AAED,eAAO,MAAM,yCAAyC,EAAE,cAAc,CAAC,QAAQ,CAwD9E,CAAC;AAGF,eAAO,MAAM,8CAA8C,EAAE,cAAc,CAAC,QAAQ,CAsEnF,CAAC;AAEF,eAAO,MAAM,iDAAiD,EAAE,CAC9D,QAAQ,EAAE,+BAA+B,GAAG,oBAAoB,EAChE,cAAc,EAAE,MAAM,KACnB,qBAAqB,GAAG,aAe5B,CAAC;AAEF,eAAO,MAAM,4CAA4C,EAAE,CACzD,QAAQ,EAAE,mCAAmC,GAAG,oBAAoB,EACpE,cAAc,EAAE,MAAM,KACnB,qBAAqB,GAAG,aAe5B,CAAC"}