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

package/dist/providers/bedrock/api.js

@@ -0,0 +1,470 @@
+import { ValidationError } from '../../errors.js';
+import { ANTHROPIC, appendApnUserAgent, BEDROCK } from '../../globals.js';
+import { sdkRequestUrl } from '../../utils/sdkRequestUrl.js';
+import { BEDROCK_ANTHROPIC_INVOKE_REQUEST_FORMAT, bedrockInvokeModels, hasNativeComplete } from './constants.js';
+import { isOpenAIModel } from './modelFamily.js';
+import { generateAWSHeaders, getBedrockModelWithoutRegion, getCredentialsFromAssumedRole, getCredentialsFromRegion, getFoundationModelFromInferenceProfile, } from './utils.js';
+/**
+ * Map legacy Options / x-tfy-aws-auth-type camelCase values onto autogen auth_data.
+ * Headers still speak the Options dialect; account auth_data uses autogen discriminants.
+ */
+const authFromHeaderAws = (headerAws) => {
+    const headerAuthType = headerAws?.awsAuthType;
+    if (headerAuthType === 'accessKey' ||
+        (!headerAuthType && headerAws?.awsAccessKeyId && headerAws?.awsSecretAccessKey && !headerAws?.awsRoleArn)) {
+        return {
+            type: 'access-key-based',
+            access_key_id: headerAws?.awsAccessKeyId || '',
+            secret_access_key: headerAws?.awsSecretAccessKey || '',
+        };
+    }
+    if (headerAuthType === 'assumedRole' || headerAws?.awsRoleArn) {
+        return {
+            type: 'assumed-role-based',
+            assumed_role_arn: headerAws?.awsRoleArn || '',
+            external_id: headerAws?.awsExternalId,
+        };
+    }
+    return undefined;
+};
+/** Options / BedrockAwsAuthFields still use camelCase awsAuthType. */
+const toOptionsAwsAuthType = (auth) => {
+    switch (auth?.type) {
+        case 'access-key-based':
+            return 'accessKey';
+        case 'assumed-role-based':
+            return 'assumedRole';
+        case 'api-key':
+            return 'apiKey';
+        default:
+            return undefined;
+    }
+};
+/** Hostname-safe AWS region (rejects path/host injection via x-tfy-aws-region). */
+const AWS_REGION_HOSTNAME_RE = /^[a-z]{2}(?:-gov)?(?:-[a-z0-9]+)+-\d+$/;
+export function resolveBedrockRegion(modelManifest, account, headerAws) {
+    // Model/account win so a caller cannot redirect account credentials off AWS.
+    const region = modelManifest?.region || account.region || headerAws?.awsRegion || 'us-east-1';
+    if (!AWS_REGION_HOSTNAME_RE.test(region)) {
+        throw new ValidationError(`Invalid AWS region: ${region}`);
+    }
+    return region;
+}
+/** Region used for SigV4 and for S3 hosts built outside `headers()`. */
+export function bedrockCallRegion(providerOptions, headerAws) {
+    const account = providerOptions.providerAccount.manifest;
+    const modelManifest = providerOptions.model?.manifest?.type === 'integration/model/bedrock'
+        ? providerOptions.model.manifest
+        : undefined;
+    return resolveBedrockRegion(modelManifest, account, headerAws);
+}
+const resolveBedrockAccount = (providerOptions, headerAws) => {
+    const account = providerOptions.providerAccount.manifest;
+    const region = bedrockCallRegion(providerOptions, headerAws);
+    // Prefer account auth_data; fall back to x-tfy-aws-* header BYOK / assumed-role.
+    const auth = account.auth_data ?? authFromHeaderAws(headerAws);
+    return {
+        baseUrl: account.base_url,
+        region,
+        auth,
+    };
+};
+const toAwsAuthFields = (account, headerAws) => {
+    const auth = account.auth;
+    return {
+        awsAuthType: toOptionsAwsAuthType(auth) ?? headerAws?.awsAuthType,
+        awsRoleArn: auth?.type === 'assumed-role-based' ? auth.assumed_role_arn : headerAws?.awsRoleArn,
+        awsRegion: account.region,
+        awsExternalId: auth?.type === 'assumed-role-based' ? auth.external_id : headerAws?.awsExternalId,
+        awsAccessKeyId: auth?.type === 'access-key-based' ? auth.access_key_id : headerAws?.awsAccessKeyId,
+        awsSecretAccessKey: auth?.type === 'access-key-based' ? auth.secret_access_key : headerAws?.awsSecretAccessKey,
+        awsSessionToken: headerAws?.awsSessionToken,
+    };
+};
+/** Sync path for getEndpoint — strip foundation-model/ ARN suffix (inference profiles stay async in headers). */
+const modelIdForRouting = (model) => {
+    if (!model)
+        return model;
+    if (model.includes('foundation-model/')) {
+        return model.split('/').pop();
+    }
+    return model;
+};
+const useBedrockAnthropicInvoke = (bedrockAnthropicUseInvoke, model) => {
+    if (!bedrockAnthropicUseInvoke)
+        return false;
+    const providerModel = getBedrockModelWithoutRegion(model || '');
+    return providerModel?.split('.')?.[0] === ANTHROPIC;
+};
+const isBedrockAnthropicInvokeRequest = (endpoint, bedrockAnthropicUseInvoke, model) => endpoint === 'messages' ||
+    (endpoint === 'chatComplete' && useBedrockAnthropicInvoke(bedrockAnthropicUseInvoke, model));
+const resolveFoundationModel = async (model, aws, session) => {
+    if (!model?.includes('arn:aws'))
+        return model;
+    if (model.includes('foundation-model/')) {
+        return model.split('/').pop();
+    }
+    return (await getFoundationModelFromInferenceProfile(model, aws, session)) || model;
+};
+/**
+ * Sets `headerOptions.aws.awsBedrockModel` from a Bedrock ARN (live `foundationModel`).
+ * Foundation-model ARNs are parsed locally; inference-profile ARNs hit GetInferenceProfile.
+ */
+export async function resolveBedrockFoundationModelFromArn(providerOptions, headerOptions, session, model) {
+    if (headerOptions.aws?.awsBedrockModel || !model)
+        return;
+    const decoded = decodeURIComponent(model);
+    if (!decoded.includes('arn:aws'))
+        return;
+    let foundationModel;
+    if (decoded.includes('foundation-model/')) {
+        foundationModel = decoded.split('/').pop();
+    }
+    else {
+        const account = resolveBedrockAccount(providerOptions, headerOptions.aws);
+        foundationModel = await getFoundationModelFromInferenceProfile(decoded, toAwsAuthFields(account, headerOptions.aws), session);
+    }
+    if (foundationModel) {
+        headerOptions.aws = { ...headerOptions.aws, awsBedrockModel: foundationModel };
+    }
+}
+const AWS_CONTROL_PLANE_ENDPOINTS = [
+    'createBatch',
+    'retrieveBatch',
+    'cancelBatch',
+    'listBatches',
+    'retrieveFileContent',
+    'getBatchOutput',
+    'cancelBatch',
+    'listFinetunes',
+    'retrieveFinetune',
+    'createFinetune',
+    'cancelFinetune',
+];
+const AWS_GET_METHODS = [
+    'listBatches',
+    'retrieveBatch',
+    'retrieveFileContent',
+    'getBatchOutput',
+    'retrieveFile',
+    'retrieveFileContent',
+    'listFinetunes',
+    'retrieveFinetune',
+];
+// Endpoints that does not require model parameter
+const BEDROCK_NO_MODEL_ENDPOINTS = [
+    'listFinetunes',
+    'retrieveFinetune',
+    'cancelFinetune',
+    'listBatches',
+    'retrieveBatch',
+    'getBatchOutput',
+    'cancelBatch',
+];
+const BEDROCK_OPENAI_RESPONSES_ENDPOINT = '/openai/v1/responses';
+const ENDPOINTS_TO_ROUTE_TO_S3 = [
+    'retrieveFileContent',
+    'getBatchOutput',
+    'retrieveFile',
+    'retrieveFileContent',
+    'uploadFile',
+    'initiateMultipartUpload',
+];
+const getMethod = (endpoint, transformedRequestUrl) => {
+    if (endpoint === 'uploadFile') {
+        const url = new URL(transformedRequestUrl);
+        if (url.searchParams.get('partNumber'))
+            return 'PUT';
+        // Multipart initiate (`?uploads`) and complete (`?uploadId`) are POST.
+        // A bare object URL is a whole-object PUT.
+        if (url.searchParams.has('uploads') || url.searchParams.has('uploadId'))
+            return 'POST';
+        return 'PUT';
+    }
+    return AWS_GET_METHODS.includes(endpoint) ? 'GET' : 'POST';
+};
+const getService = (endpoint) => {
+    return ENDPOINTS_TO_ROUTE_TO_S3.includes(endpoint) ? 's3' : 'bedrock';
+};
+const setRouteSpecificHeaders = (endpoint, headers, awsHeaders) => {
+    if (endpoint === 'retrieveFile') {
+        headers['x-amz-object-attributes'] = 'ObjectSize';
+    }
+    if (endpoint === 'initiateMultipartUpload') {
+        if (awsHeaders?.awsServerSideEncryptionKMSKeyId) {
+            headers['x-amz-server-side-encryption-aws-kms-key-id'] = awsHeaders.awsServerSideEncryptionKMSKeyId;
+            headers['x-amz-server-side-encryption'] = 'aws:kms';
+        }
+        if (awsHeaders?.awsServerSideEncryption) {
+            headers['x-amz-server-side-encryption'] = awsHeaders.awsServerSideEncryption;
+        }
+    }
+};
+const BedrockAPIConfig = {
+    getBaseURL: async ({ providerOptions, headerOptions, endpoint }) => {
+        const account = resolveBedrockAccount(providerOptions, headerOptions.aws);
+        const region = account.region || 'us-east-1';
+        /**
+         * =============================
+         * Keep the logic for retrieveFile and retrieveFileContent as is
+         * =============================
+         */
+        const awsS3Bucket = headerOptions.aws?.awsS3Bucket;
+        if (endpoint === 'retrieveFile' || endpoint === 'retrieveFileContent') {
+            return `https://${awsS3Bucket}.s3.${region}.amazonaws.com`;
+        }
+        if (endpoint === 'uploadFile')
+            return `https://${awsS3Bucket}.s3.${region}.amazonaws.com`;
+        /**
+         * =============================
+         * NOTE - Add this Base URL for rerank
+         * =============================
+         */
+        if (endpoint === 'rerank') {
+            return `https://bedrock-agent-runtime.${region}.amazonaws.com`;
+        }
+        const isAWSControlPlaneEndpoint = endpoint && AWS_CONTROL_PLANE_ENDPOINTS.includes(endpoint);
+        const urlToFetch = account.baseUrl;
+        // Control-plane calls keep the real `bedrock.` host, since a bedrock-runtime
+        // VPC endpoint won't serve control-plane operations.
+        if (urlToFetch && !isAWSControlPlaneEndpoint) {
+            return urlToFetch.replace(/\/$/, '');
+        }
+        return `https://${isAWSControlPlaneEndpoint ? 'bedrock' : 'bedrock-runtime'}.${region}.amazonaws.com`;
+    },
+    headers: async ({ endpoint, providerOptions, headerOptions, session, request, transformedRequestBody, transformedRequestUrl, }) => {
+        const account = resolveBedrockAccount(providerOptions, headerOptions.aws);
+        const region = account.region || 'us-east-1';
+        const method = getMethod(endpoint, transformedRequestUrl || '');
+        const service = getService(endpoint);
+        const headers = {
+            'content-type': 'application/json',
+        };
+        if (method === 'PUT' || method === 'GET') {
+            delete headers['content-type'];
+        }
+        setRouteSpecificHeaders(endpoint, headers, headerOptions.aws);
+        const gwBody = request.body;
+        const awsAuth = toAwsAuthFields(account, headerOptions.aws);
+        const foundationModel = await resolveFoundationModel(gwBody?.model, awsAuth, session);
+        const modelForInvoke = foundationModel || gwBody?.model;
+        // InvokeModel accepts performance settings in a header.
+        const performanceLatency = gwBody?.performance_config?.latency;
+        if (isBedrockAnthropicInvokeRequest(endpoint, headerOptions.aws?.bedrockAnthropicUseInvoke, modelForInvoke) &&
+            (performanceLatency === 'standard' || performanceLatency === 'optimized')) {
+            headers['X-Amzn-Bedrock-PerformanceConfig-Latency'] = performanceLatency;
+        }
+        const isAnthropicInvokeChatComplete = endpoint === 'chatComplete' &&
+            useBedrockAnthropicInvoke(headerOptions.aws?.bedrockAnthropicUseInvoke, modelForInvoke);
+        // InvokeModel accepts guardrail settings in headers.
+        if (isAnthropicInvokeChatComplete) {
+            const guardrailConfig = gwBody?.guardrail_config ?? gwBody?.guardrailConfig;
+            if (guardrailConfig?.guardrailIdentifier && guardrailConfig?.guardrailVersion) {
+                headers['X-Amzn-Bedrock-GuardrailIdentifier'] = guardrailConfig.guardrailIdentifier;
+                headers['X-Amzn-Bedrock-GuardrailVersion'] = guardrailConfig.guardrailVersion;
+                if (guardrailConfig.trace) {
+                    headers['X-Amzn-Bedrock-Trace'] = guardrailConfig.trace.toUpperCase();
+                }
+            }
+            // TODO[SDK-Extraction]: tracing previously set CONTEXT_KEYS.TRACING_PROVIDER_REQUEST_FORMAT
+            // to BEDROCK_ANTHROPIC_INVOKE_REQUEST_FORMAT via c.set — deferred to P5/P7.
+            void BEDROCK_ANTHROPIC_INVOKE_REQUEST_FORMAT;
+        }
+        let awsAccessKeyId = '';
+        let awsSecretAccessKey = '';
+        let awsSessionToken = headerOptions.aws?.awsSessionToken || '';
+        switch (account.auth?.type) {
+            case 'api-key': {
+                if (account.auth.api_key) {
+                    headers['Authorization'] = `Bearer ${account.auth.api_key}`;
+                }
+                // API-key auth bypasses SigV4 signing, so add the APN user-agent here too.
+                appendApnUserAgent(headers);
+                return headers;
+            }
+            case 'access-key-based':
+                awsAccessKeyId = account.auth.access_key_id;
+                awsSecretAccessKey = account.auth.secret_access_key;
+                break;
+            case 'assumed-role-based': {
+                const credentials = await getCredentialsFromAssumedRole(session, account.auth.assumed_role_arn, region, account.auth.external_id, BEDROCK);
+                awsAccessKeyId = credentials.accessKeyId;
+                awsSecretAccessKey = credentials.secretAccessKey;
+                awsSessionToken = credentials.sessionToken || '';
+                break;
+            }
+            default: {
+                // No resolved auth — ambient credential chain only.
+                const credentials = await getCredentialsFromRegion(region);
+                awsAccessKeyId = credentials.accessKeyId;
+                awsSecretAccessKey = credentials.secretAccessKey;
+                awsSessionToken = credentials.sessionToken || '';
+                break;
+            }
+        }
+        let finalRequestBody = transformedRequestBody;
+        if (['cancelFinetune', 'cancelBatch'].includes(endpoint)) {
+            // Cancel doesn't require any body, but fetch is sending empty body, to match the signature this block is required.
+            finalRequestBody = '';
+        }
+        return generateAWSHeaders(finalRequestBody, headers, transformedRequestUrl || '', method, service, region, awsAccessKeyId, awsSecretAccessKey, awsSessionToken);
+    },
+    getEndpoint: ({ endpoint: endpointKind, headerOptions, request, execution }) => {
+        const inboundUrl = sdkRequestUrl(request);
+        if (endpointKind === 'retrieveFile') {
+            /**
+             * =============================
+             * NOTE - Remove the v1 from the URL for files
+             * =============================
+             */
+            const fileId = decodeURIComponent(inboundUrl.split('/files/')[1]);
+            const s3ObjectKeyParts = fileId.replace('s3://', '').split('/');
+            const s3ObjectKey = s3ObjectKeyParts.slice(1).join('/');
+            return `/${s3ObjectKey}?attributes`;
+        }
+        if (endpointKind === 'retrieveFileContent') {
+            /**
+             * =============================
+             * NOTE - Remove the v1 from the URL for files
+             * =============================
+             */
+            const fileId = decodeURIComponent(inboundUrl.split('/files/')[1]);
+            const s3ObjectKeyParts = fileId.replace('s3://', '').replace('/content', '').split('/');
+            const s3ObjectKey = s3ObjectKeyParts.slice(1).join('/');
+            return `/${s3ObjectKey}`;
+        }
+        if (endpointKind === 'uploadFile')
+            return '';
+        if (endpointKind === 'cancelBatch') {
+            /**
+             * =============================
+             * NOTE - Remove the v1 from the URL for batches
+             * =============================
+             */
+            const batchId = inboundUrl.split('/batches/')[1].split('/')[0];
+            return `/model-invocation-job/${batchId}/stop`;
+        }
+        const body = request.body;
+        const { model, stream } = body ?? {};
+        const uriEncodedModel = encodeURIComponent(decodeURIComponent(model ?? ''));
+        if (!model && !BEDROCK_NO_MODEL_ENDPOINTS.includes(endpointKind)) {
+            throw new ValidationError('Model is required');
+        }
+        let mappedFn = endpointKind;
+        if (stream) {
+            mappedFn = `stream-${endpointKind}`;
+        }
+        let endpoint = `/model/${uriEncodedModel}/invoke`;
+        let streamEndpoint = `/model/${uriEncodedModel}/invoke-with-response-stream`;
+        // Prefer ARN-resolved foundation model (live providerOptions.foundationModel).
+        const routingModel = modelIdForRouting(headerOptions.aws?.awsBedrockModel || model);
+        const anthropicInvoke = useBedrockAnthropicInvoke(headerOptions.aws?.bedrockAnthropicUseInvoke, routingModel);
+        const useInvokeChat = bedrockInvokeModels.includes(getBedrockModelWithoutRegion(routingModel || ''));
+        if ((mappedFn === 'chatComplete' && !anthropicInvoke && !useInvokeChat) ||
+            mappedFn === 'bedrockConverse' ||
+            mappedFn === 'stream-bedrockConverse' ||
+            (mappedFn === 'stream-chatComplete' && !anthropicInvoke && !useInvokeChat) ||
+            /**
+             * ================================
+             * NOTE - Route /v1/completions to Converse for models without native
+             * legacy text-completion support. Paired with BedrockConverseCompleteConfig
+             * in `./complete`, which transforms the OpenAI text-completion request
+             * body into a Converse chat request.
+             * ================================
+             */
+            ((mappedFn === 'complete' || mappedFn === 'stream-complete') && !hasNativeComplete(model))) {
+            endpoint = `/model/${uriEncodedModel}/converse`;
+            streamEndpoint = `/model/${uriEncodedModel}/converse-stream`;
+        }
+        const jobIdIndex = endpointKind === 'cancelFinetune' ? -2 : -1;
+        const jobId = inboundUrl.split('/').at(jobIdIndex);
+        switch (mappedFn) {
+            case 'createModelResponse':
+            case 'stream-createModelResponse': {
+                // Only the openai.* family is served on /openai/v1/responses; keyed on body model id.
+                if (isOpenAIModel(model ?? '')) {
+                    return BEDROCK_OPENAI_RESPONSES_ENDPOINT;
+                }
+                return '';
+            }
+            case 'chatComplete':
+            case 'messages':
+            case 'bedrockConverse': {
+                // CC→Responses on bedrock-runtime: POST /openai/v1/responses (stream uses same path).
+                if (mappedFn === 'chatComplete' && execution?.openaiUseResponsesApi) {
+                    return BEDROCK_OPENAI_RESPONSES_ENDPOINT;
+                }
+                return endpoint;
+            }
+            case 'stream-chatComplete':
+            case 'stream-messages':
+            case 'stream-bedrockConverse': {
+                if (mappedFn === 'stream-chatComplete' && execution?.openaiUseResponsesApi) {
+                    return BEDROCK_OPENAI_RESPONSES_ENDPOINT;
+                }
+                return streamEndpoint;
+            }
+            case 'complete': {
+                return endpoint;
+            }
+            case 'stream-complete': {
+                return streamEndpoint;
+            }
+            case 'embed': {
+                return endpoint;
+            }
+            /**
+             * =============================
+             * NOTE - Add Rerank endpoint
+             */
+            case 'rerank': {
+                return '/rerank';
+            }
+            case 'imageGenerate': {
+                return endpoint;
+            }
+            case 'imageEdit': {
+                return endpoint;
+            }
+            case 'imageVariation': {
+                return endpoint;
+            }
+            case 'createBatch': {
+                return '/model-invocation-job';
+            }
+            case 'cancelBatch': {
+                return `/model-invocation-job/${inboundUrl.split('/').pop()}/stop`;
+            }
+            case 'retrieveBatch': {
+                /**
+                 * =============================
+                 * NOTE - Remove the v1 from the URL for batches
+                 * =============================
+                 */
+                return `/model-invocation-job/${inboundUrl.split('/batches/')[1]}`;
+            }
+            case 'listBatches': {
+                return '/model-invocation-jobs';
+            }
+            case 'listFinetunes': {
+                return '/model-customization-jobs';
+            }
+            case 'retrieveFinetune': {
+                return `/model-customization-jobs/${jobId}`;
+            }
+            case 'createFinetune': {
+                return '/model-customization-jobs';
+            }
+            case 'cancelFinetune': {
+                return `/model-customization-jobs/${jobId}/stop`;
+            }
+            case 'messagesCountTokens': {
+                return `/model/${uriEncodedModel}/count-tokens`;
+            }
+            default:
+                return '';
+        }
+    },
+};
+export default BedrockAPIConfig;
+//# sourceMappingURL=api.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/bedrock/api.ts"],"names":[],"mappings":"AAYA,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,kBAAkB,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAIvE,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE1D,OAAO,EAAE,uCAAuC,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAC9G,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAEL,kBAAkB,EAClB,4BAA4B,EAC5B,6BAA6B,EAC7B,wBAAwB,EACxB,sCAAsC,GACvC,MAAM,SAAS,CAAC;AAwBjB;;;GAGG;AACH,MAAM,iBAAiB,GAAG,CAAC,SAAsB,EAA+B,EAAE;IAChF,MAAM,cAAc,GAAG,SAAS,EAAE,WAAW,CAAC;IAC9C,IACE,cAAc,KAAK,WAAW;QAC9B,CAAC,CAAC,cAAc,IAAI,SAAS,EAAE,cAAc,IAAI,SAAS,EAAE,kBAAkB,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,EACzG,CAAC;QACD,OAAO;YACL,IAAI,EAAE,kBAAkB;YACxB,aAAa,EAAE,SAAS,EAAE,cAAc,IAAI,EAAE;YAC9C,iBAAiB,EAAE,SAAS,EAAE,kBAAkB,IAAI,EAAE;SACvD,CAAC;IACJ,CAAC;IACD,IAAI,cAAc,KAAK,aAAa,IAAI,SAAS,EAAE,UAAU,EAAE,CAAC;QAC9D,OAAO;YACL,IAAI,EAAE,oBAAoB;YAC1B,gBAAgB,EAAE,SAAS,EAAE,UAAU,IAAI,EAAE;YAC7C,WAAW,EAAE,SAAS,EAAE,aAAa;SACtC,CAAC;IACJ,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC;AAEF,sEAAsE;AACtE,MAAM,oBAAoB,GAAG,CAAC,IAAsB,EAAsB,EAAE;IAC1E,QAAQ,IAAI,EAAE,IAAI,EAAE,CAAC;QACnB,KAAK,kBAAkB;YACrB,OAAO,WAAW,CAAC;QACrB,KAAK,oBAAoB;YACvB,OAAO,aAAa,CAAC;QACvB,KAAK,SAAS;YACZ,OAAO,QAAQ,CAAC;QAClB;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC,CAAC;AAEF,mFAAmF;AACnF,MAAM,sBAAsB,GAAG,wCAAwC,CAAC;AAExE,MAAM,UAAU,oBAAoB,CAClC,aAAuC,EACvC,OAAkC,EAClC,SAAsB;IAEtB,6EAA6E;IAC7E,MAAM,MAAM,GAAG,aAAa,EAAE,MAAM,IAAI,OAAO,CAAC,MAAM,IAAI,SAAS,EAAE,SAAS,IAAI,WAAW,CAAC;IAC9F,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACzC,MAAM,IAAI,eAAe,CAAC,uBAAuB,MAAM,EAAE,CAAC,CAAC;IAC7D,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,wEAAwE;AACxE,MAAM,UAAU,iBAAiB,CAAC,eAAgC,EAAE,SAAsB;IACxF,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAqC,CAAC;IACtF,MAAM,aAAa,GACjB,eAAe,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,KAAK,2BAA2B;QACnE,CAAC,CAAE,eAAe,CAAC,KAAK,CAAC,QAAyB;QAClD,CAAC,CAAC,SAAS,CAAC;IAChB,OAAO,oBAAoB,CAAC,aAAa,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;AACjE,CAAC;AAED,MAAM,qBAAqB,GAAG,CAAC,eAAgC,EAAE,SAAsB,EAA0B,EAAE;IACjH,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAqC,CAAC;IACtF,MAAM,MAAM,GAAG,iBAAiB,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;IAC7D,iFAAiF;IACjF,MAAM,IAAI,GAAG,OAAO,CAAC,SAAS,IAAI,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAE/D,OAAO;QACL,OAAO,EAAE,OAAO,CAAC,QAAQ;QACzB,MAAM;QACN,IAAI;KACL,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG,CAAC,OAA+B,EAAE,SAAsB,EAAwB,EAAE;IACxG,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAC1B,OAAO;QACL,WAAW,EAAE,oBAAoB,CAAC,IAAI,CAAC,IAAI,SAAS,EAAE,WAAW;QACjE,UAAU,EAAE,IAAI,EAAE,IAAI,KAAK,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS,EAAE,UAAU;QAC/F,SAAS,EAAE,OAAO,CAAC,MAAM;QACzB,aAAa,EAAE,IAAI,EAAE,IAAI,KAAK,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,EAAE,aAAa;QAChG,cAAc,EAAE,IAAI,EAAE,IAAI,KAAK,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,EAAE,cAAc;QAClG,kBAAkB,EAAE,IAAI,EAAE,IAAI,KAAK,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS,EAAE,kBAAkB;QAC9G,eAAe,EAAE,SAAS,EAAE,eAAe;KAC5C,CAAC;AACJ,CAAC,CAAC;AAEF,iHAAiH;AACjH,MAAM,iBAAiB,GAAG,CAAC,KAAyB,EAAsB,EAAE;IAC1E,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,IAAI,KAAK,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;QACxC,OAAO,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;IAChC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF,MAAM,yBAAyB,GAAG,CAAC,yBAA8C,EAAE,KAAc,EAAW,EAAE;IAC5G,IAAI,CAAC,yBAAyB;QAAE,OAAO,KAAK,CAAC;IAC7C,MAAM,aAAa,GAAG,4BAA4B,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAChE,OAAO,aAAa,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC;AACtD,CAAC,CAAC;AAEF,MAAM,+BAA+B,GAAG,CACtC,QAAgB,EAChB,yBAA8C,EAC9C,KAAc,EACL,EAAE,CACX,QAAQ,KAAK,UAAU;IACvB,CAAC,QAAQ,KAAK,cAAc,IAAI,yBAAyB,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC,CAAC;AAE/F,MAAM,sBAAsB,GAAG,KAAK,EAClC,KAAyB,EACzB,GAAyB,EACzB,OAAuB,EACM,EAAE;IAC/B,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,SAAS,CAAC;QAAE,OAAO,KAAK,CAAC;IAC9C,IAAI,KAAK,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;QACxC,OAAO,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;IAChC,CAAC;IACD,OAAO,CAAC,MAAM,sCAAsC,CAAC,KAAK,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC,IAAI,KAAK,CAAC;AACtF,CAAC,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,oCAAoC,CACxD,eAAgC,EAChC,aAAoC,EACpC,OAAuB,EACvB,KAAc;IAEd,IAAI,aAAa,CAAC,GAAG,EAAE,eAAe,IAAI,CAAC,KAAK;QAAE,OAAO;IACzD,MAAM,OAAO,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;QAAE,OAAO;IAEzC,IAAI,eAA0C,CAAC;IAC/C,IAAI,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;QAC1C,eAAe,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;IAC7C,CAAC;SAAM,CAAC;QACN,MAAM,OAAO,GAAG,qBAAqB,CAAC,eAAe,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC;QAC1E,eAAe,GAAG,MAAM,sCAAsC,CAC5D,OAAO,EACP,eAAe,CAAC,OAAO,EAAE,aAAa,CAAC,GAAG,CAAC,EAC3C,OAAO,CACR,CAAC;IACJ,CAAC;IACD,IAAI,eAAe,EAAE,CAAC;QACpB,aAAa,CAAC,GAAG,GAAG,EAAE,GAAG,aAAa,CAAC,GAAG,EAAE,eAAe,EAAE,eAAe,EAAE,CAAC;IACjF,CAAC;AACH,CAAC;AAED,MAAM,2BAA2B,GAAe;IAC9C,aAAa;IACb,eAAe;IACf,aAAa;IACb,aAAa;IACb,qBAAqB;IACrB,gBAAgB;IAChB,aAAa;IACb,eAAe;IACf,kBAAkB;IAClB,gBAAgB;IAChB,gBAAgB;CACjB,CAAC;AAEF,MAAM,eAAe,GAAe;IAClC,aAAa;IACb,eAAe;IACf,qBAAqB;IACrB,gBAAgB;IAChB,cAAc;IACd,qBAAqB;IACrB,eAAe;IACf,kBAAkB;CACnB,CAAC;AAEF,kDAAkD;AAClD,MAAM,0BAA0B,GAAe;IAC7C,eAAe;IACf,kBAAkB;IAClB,gBAAgB;IAChB,aAAa;IACb,eAAe;IACf,gBAAgB;IAChB,aAAa;CACd,CAAC;AAEF,MAAM,iCAAiC,GAAG,sBAAsB,CAAC;AAEjE,MAAM,wBAAwB,GAAG;IAC/B,qBAAqB;IACrB,gBAAgB;IAChB,cAAc;IACd,qBAAqB;IACrB,YAAY;IACZ,yBAAyB;CAC1B,CAAC;AAEF,MAAM,SAAS,GAAG,CAAC,QAAkB,EAAE,qBAA6B,EAAE,EAAE;IACtE,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;QAC9B,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,qBAAqB,CAAC,CAAC;QAC3C,IAAI,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC;YAAE,OAAO,KAAK,CAAC;QACrD,uEAAuE;QACvE,2CAA2C;QAC3C,IAAI,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC;YAAE,OAAO,MAAM,CAAC;QACvF,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,eAAe,CAAC,QAAQ,CAAC,QAAoB,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;AACzE,CAAC,CAAC;AAEF,MAAM,UAAU,GAAG,CAAC,QAAkB,EAAE,EAAE;IACxC,OAAO,wBAAwB,CAAC,QAAQ,CAAC,QAAoB,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;AACpF,CAAC,CAAC;AAEF,MAAM,uBAAuB,GAAG,CAAC,QAAgB,EAAE,OAA+B,EAAE,UAAuB,EAAE,EAAE;IAC7G,IAAI,QAAQ,KAAK,cAAc,EAAE,CAAC;QAChC,OAAO,CAAC,yBAAyB,CAAC,GAAG,YAAY,CAAC;IACpD,CAAC;IACD,IAAI,QAAQ,KAAK,yBAAyB,EAAE,CAAC;QAC3C,IAAI,UAAU,EAAE,+BAA+B,EAAE,CAAC;YAChD,OAAO,CAAC,6CAA6C,CAAC,GAAG,UAAU,CAAC,+BAA+B,CAAC;YACpG,OAAO,CAAC,8BAA8B,CAAC,GAAG,SAAS,CAAC;QACtD,CAAC;QACD,IAAI,UAAU,EAAE,uBAAuB,EAAE,CAAC;YACxC,OAAO,CAAC,8BAA8B,CAAC,GAAG,UAAU,CAAC,uBAAuB,CAAC;QAC/E,CAAC;IACH,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,gBAAgB,GAAsB;IAC1C,UAAU,EAAE,KAAK,EAAE,EAAE,eAAe,EAAE,aAAa,EAAE,QAAQ,EAAE,EAAE,EAAE;QACjE,MAAM,OAAO,GAAG,qBAAqB,CAAC,eAAe,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC;QAC1E,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,WAAW,CAAC;QAE7C;;;;WAIG;QACH,MAAM,WAAW,GAAG,aAAa,CAAC,GAAG,EAAE,WAAW,CAAC;QACnD,IAAI,QAAQ,KAAK,cAAc,IAAI,QAAQ,KAAK,qBAAqB,EAAE,CAAC;YACtE,OAAO,WAAW,WAAW,OAAO,MAAM,gBAAgB,CAAC;QAC7D,CAAC;QAED,IAAI,QAAQ,KAAK,YAAY;YAAE,OAAO,WAAW,WAAW,OAAO,MAAM,gBAAgB,CAAC;QAE1F;;;;WAIG;QACH,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO,iCAAiC,MAAM,gBAAgB,CAAC;QACjE,CAAC;QACD,MAAM,yBAAyB,GAAG,QAAQ,IAAI,2BAA2B,CAAC,QAAQ,CAAC,QAAoB,CAAC,CAAC;QACzG,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC;QACnC,6EAA6E;QAC7E,qDAAqD;QACrD,IAAI,UAAU,IAAI,CAAC,yBAAyB,EAAE,CAAC;YAC7C,OAAO,UAAU,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,WAAW,yBAAyB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB,IAAI,MAAM,gBAAgB,CAAC;IACxG,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,EACd,QAAQ,EACR,eAAe,EACf,aAAa,EACb,OAAO,EACP,OAAO,EACP,sBAAsB,EACtB,qBAAqB,GACtB,EAAE,EAAE;QACH,MAAM,OAAO,GAAG,qBAAqB,CAAC,eAAe,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC;QAC1E,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,WAAW,CAAC;QAC7C,MAAM,MAAM,GAAG,SAAS,CAAC,QAAoB,EAAE,qBAAqB,IAAI,EAAE,CAAC,CAAC;QAC5E,MAAM,OAAO,GAAG,UAAU,CAAC,QAAoB,CAAC,CAAC;QAEjD,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,kBAAkB;SACnC,CAAC;QAEF,IAAI,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;YACzC,OAAO,OAAO,CAAC,cAAc,CAAC,CAAC;QACjC,CAAC;QAED,uBAAuB,CAAC,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC;QAE9D,MAAM,MAAM,GAAG,OAAO,CAAC,IAA6C,CAAC;QACrE,MAAM,OAAO,GAAG,eAAe,CAAC,OAAO,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC;QAC5D,MAAM,eAAe,GAAG,MAAM,sBAAsB,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QACtF,MAAM,cAAc,GAAG,eAAe,IAAI,MAAM,EAAE,KAAK,CAAC;QAExD,wDAAwD;QACxD,MAAM,kBAAkB,GAAG,MAAM,EAAE,kBAAkB,EAAE,OAAO,CAAC;QAC/D,IACE,+BAA+B,CAAC,QAAQ,EAAE,aAAa,CAAC,GAAG,EAAE,yBAAyB,EAAE,cAAc,CAAC;YACvG,CAAC,kBAAkB,KAAK,UAAU,IAAI,kBAAkB,KAAK,WAAW,CAAC,EACzE,CAAC;YACD,OAAO,CAAC,0CAA0C,CAAC,GAAG,kBAAkB,CAAC;QAC3E,CAAC;QAED,MAAM,6BAA6B,GACjC,QAAQ,KAAK,cAAc;YAC3B,yBAAyB,CAAC,aAAa,CAAC,GAAG,EAAE,yBAAyB,EAAE,cAAc,CAAC,CAAC;QAC1F,qDAAqD;QACrD,IAAI,6BAA6B,EAAE,CAAC;YAClC,MAAM,eAAe,GAAG,MAAM,EAAE,gBAAgB,IAAI,MAAM,EAAE,eAAe,CAAC;YAC5E,IAAI,eAAe,EAAE,mBAAmB,IAAI,eAAe,EAAE,gBAAgB,EAAE,CAAC;gBAC9E,OAAO,CAAC,oCAAoC,CAAC,GAAG,eAAe,CAAC,mBAAmB,CAAC;gBACpF,OAAO,CAAC,iCAAiC,CAAC,GAAG,eAAe,CAAC,gBAAgB,CAAC;gBAC9E,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC;oBAC1B,OAAO,CAAC,sBAAsB,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;gBACxE,CAAC;YACH,CAAC;YACD,4FAA4F;YAC5F,4EAA4E;YAC5E,KAAK,uCAAuC,CAAC;QAC/C,CAAC;QAED,IAAI,cAAc,GAAG,EAAE,CAAC;QACxB,IAAI,kBAAkB,GAAG,EAAE,CAAC;QAC5B,IAAI,eAAe,GAAG,aAAa,CAAC,GAAG,EAAE,eAAe,IAAI,EAAE,CAAC;QAE/D,QAAQ,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC;YAC3B,KAAK,SAAS,CAAC,CAAC,CAAC;gBACf,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;oBACzB,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC9D,CAAC;gBACD,2EAA2E;gBAC3E,kBAAkB,CAAC,OAAO,CAAC,CAAC;gBAC5B,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,KAAK,kBAAkB;gBACrB,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC;gBAC5C,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC;gBACpD,MAAM;YACR,KAAK,oBAAoB,CAAC,CAAC,CAAC;gBAC1B,MAAM,WAAW,GAAG,MAAM,6BAA6B,CACrD,OAAO,EACP,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAC7B,MAAM,EACN,OAAO,CAAC,IAAI,CAAC,WAAW,EACxB,OAAO,CACR,CAAC;gBACF,cAAc,GAAG,WAAW,CAAC,WAAW,CAAC;gBACzC,kBAAkB,GAAG,WAAW,CAAC,eAAe,CAAC;gBACjD,eAAe,GAAG,WAAW,CAAC,YAAY,IAAI,EAAE,CAAC;gBACjD,MAAM;YACR,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACR,oDAAoD;gBACpD,MAAM,WAAW,GAAG,MAAM,wBAAwB,CAAC,MAAM,CAAC,CAAC;gBAC3D,cAAc,GAAG,WAAW,CAAC,WAAW,CAAC;gBACzC,kBAAkB,GAAG,WAAW,CAAC,eAAe,CAAC;gBACjD,eAAe,GAAG,WAAW,CAAC,YAAY,IAAI,EAAE,CAAC;gBACjD,MAAM;YACR,CAAC;QACH,CAAC;QAED,IAAI,gBAAgB,GAAG,sBAAsB,CAAC;QAE9C,IAAI,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAC,QAAQ,CAAC,QAAoB,CAAC,EAAE,CAAC;YACrE,mHAAmH;YACnH,gBAAgB,GAAG,EAAE,CAAC;QACxB,CAAC;QAED,OAAO,kBAAkB,CACvB,gBAAgB,EAChB,OAAO,EACP,qBAAqB,IAAI,EAAE,EAC3B,MAAM,EACN,OAAO,EACP,MAAM,EACN,cAAc,EACd,kBAAkB,EAClB,eAAe,CAChB,CAAC;IACJ,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,aAAa,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE;QAC7E,MAAM,UAAU,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QAC1C,IAAI,YAAY,KAAK,cAAc,EAAE,CAAC;YACpC;;;;eAIG;YACH,MAAM,MAAM,GAAG,kBAAkB,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAClE,MAAM,gBAAgB,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAChE,MAAM,WAAW,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACxD,OAAO,IAAI,WAAW,aAAa,CAAC;QACtC,CAAC;QACD,IAAI,YAAY,KAAK,qBAAqB,EAAE,CAAC;YAC3C;;;;eAIG;YACH,MAAM,MAAM,GAAG,kBAAkB,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAClE,MAAM,gBAAgB,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACxF,MAAM,WAAW,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACxD,OAAO,IAAI,WAAW,EAAE,CAAC;QAC3B,CAAC;QACD,IAAI,YAAY,KAAK,YAAY;YAAE,OAAO,EAAE,CAAC;QAC7C,IAAI,YAAY,KAAK,aAAa,EAAE,CAAC;YACnC;;;;eAIG;YACH,MAAM,OAAO,GAAG,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/D,OAAO,yBAAyB,OAAO,OAAO,CAAC;QACjD,CAAC;QACD,MAAM,IAAI,GAAG,OAAO,CAAC,IAA0B,CAAC;QAChD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC;QACrC,MAAM,eAAe,GAAG,kBAAkB,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC;QAC5E,IAAI,CAAC,KAAK,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,YAAwB,CAAC,EAAE,CAAC;YAC7E,MAAM,IAAI,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACjD,CAAC;QACD,IAAI,QAAQ,GAAW,YAAY,CAAC;QACpC,IAAI,MAAM,EAAE,CAAC;YACX,QAAQ,GAAG,UAAU,YAAY,EAAE,CAAC;QACtC,CAAC;QACD,IAAI,QAAQ,GAAG,UAAU,eAAe,SAAS,CAAC;QAClD,IAAI,cAAc,GAAG,UAAU,eAAe,8BAA8B,CAAC;QAE7E,+EAA+E;QAC/E,MAAM,YAAY,GAAG,iBAAiB,CAAC,aAAa,CAAC,GAAG,EAAE,eAAe,IAAI,KAAK,CAAC,CAAC;QACpF,MAAM,eAAe,GAAG,yBAAyB,CAAC,aAAa,CAAC,GAAG,EAAE,yBAAyB,EAAE,YAAY,CAAC,CAAC;QAC9G,MAAM,aAAa,GAAG,mBAAmB,CAAC,QAAQ,CAAC,4BAA4B,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC,CAAC;QAErG,IACE,CAAC,QAAQ,KAAK,cAAc,IAAI,CAAC,eAAe,IAAI,CAAC,aAAa,CAAC;YACnE,QAAQ,KAAK,iBAAiB;YAC9B,QAAQ,KAAK,wBAAwB;YACrC,CAAC,QAAQ,KAAK,qBAAqB,IAAI,CAAC,eAAe,IAAI,CAAC,aAAa,CAAC;YAC1E;;;;;;;eAOG;YACH,CAAC,CAAC,QAAQ,KAAK,UAAU,IAAI,QAAQ,KAAK,iBAAiB,CAAC,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,EAC1F,CAAC;YACD,QAAQ,GAAG,UAAU,eAAe,WAAW,CAAC;YAChD,cAAc,GAAG,UAAU,eAAe,kBAAkB,CAAC;QAC/D,CAAC;QAED,MAAM,UAAU,GAAG,YAAY,KAAK,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;QAEnD,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,qBAAqB,CAAC;YAC3B,KAAK,4BAA4B,CAAC,CAAC,CAAC;gBAClC,sFAAsF;gBACtF,IAAI,aAAa,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC;oBAC/B,OAAO,iCAAiC,CAAC;gBAC3C,CAAC;gBACD,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,KAAK,cAAc,CAAC;YACpB,KAAK,UAAU,CAAC;YAChB,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,sFAAsF;gBACtF,IAAI,QAAQ,KAAK,cAAc,IAAI,SAAS,EAAE,qBAAqB,EAAE,CAAC;oBACpE,OAAO,iCAAiC,CAAC;gBAC3C,CAAC;gBACD,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,KAAK,qBAAqB,CAAC;YAC3B,KAAK,iBAAiB,CAAC;YACvB,KAAK,wBAAwB,CAAC,CAAC,CAAC;gBAC9B,IAAI,QAAQ,KAAK,qBAAqB,IAAI,SAAS,EAAE,qBAAqB,EAAE,CAAC;oBAC3E,OAAO,iCAAiC,CAAC;gBAC3C,CAAC;gBACD,OAAO,cAAc,CAAC;YACxB,CAAC;YACD,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,OAAO,cAAc,CAAC;YACxB,CAAC;YACD,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD;;;eAGG;YACH,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,KAAK,gBAAgB,CAAC,CAAC,CAAC;gBACtB,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,OAAO,uBAAuB,CAAC;YACjC,CAAC;YACD,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,OAAO,yBAAyB,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,OAAO,CAAC;YACrE,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB;;;;mBAIG;gBACH,OAAO,yBAAyB,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACrE,CAAC;YACD,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,OAAO,wBAAwB,CAAC;YAClC,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,OAAO,2BAA2B,CAAC;YACrC,CAAC;YACD,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBACxB,OAAO,6BAA6B,KAAK,EAAE,CAAC;YAC9C,CAAC;YACD,KAAK,gBAAgB,CAAC,CAAC,CAAC;gBACtB,OAAO,2BAA2B,CAAC;YACrC,CAAC;YACD,KAAK,gBAAgB,CAAC,CAAC,CAAC;gBACtB,OAAO,6BAA6B,KAAK,OAAO,CAAC;YACnD,CAAC;YACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,OAAO,UAAU,eAAe,eAAe,CAAC;YAClD,CAAC;YACD;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAC;AAEF,eAAe,gBAAgB,CAAC"}

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

@@ -0,0 +1,11 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. Remove the v1 from the URL for batches
+ * ================================
+ */
+import { CancelBatchResponse, ErrorResponse } from '../types';
+import { BedrockErrorResponse } from './embed';
+export declare const BedrockCancelBatchResponseTransform: (response: Response | BedrockErrorResponse, responseStatus: number, _responseHeaders: Headers, _strictOpenAiCompliance: boolean, gatewayRequestUrl: string) => CancelBatchResponse | ErrorResponse;
+//# sourceMappingURL=cancelBatch.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cancelBatch.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock/cancelBatch.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,mBAAmB,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAE9D,OAAO,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAE/C,eAAO,MAAM,mCAAmC,GAC9C,UAAU,QAAQ,GAAG,oBAAoB,EACzC,gBAAgB,MAAM,EACtB,kBAAkB,OAAO,EACzB,yBAAyB,OAAO,EAChC,mBAAmB,MAAM,KACxB,mBAAmB,GAAG,aAiBxB,CAAC"}

package/dist/providers/bedrock/cancelBatch.js

@@ -0,0 +1,20 @@
+import { BedrockErrorResponseTransform } from './chatComplete.js';
+export const BedrockCancelBatchResponseTransform = (response, responseStatus, _responseHeaders, _strictOpenAiCompliance, gatewayRequestUrl) => {
+    if (responseStatus !== 200) {
+        const errorResponse = BedrockErrorResponseTransform(response);
+        if (errorResponse)
+            return errorResponse;
+    }
+    /**
+     * =============================
+     * NOTE - Remove the v1 from the URL for batches
+     * =============================
+     */
+    const batchId = decodeURIComponent(gatewayRequestUrl.split('/batches/')[1].split('/')[0]);
+    return {
+        status: 'success',
+        object: 'batch',
+        id: batchId,
+    };
+};
+//# sourceMappingURL=cancelBatch.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cancelBatch.js","sourceRoot":"","sources":["../../../src/providers/bedrock/cancelBatch.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,6BAA6B,EAAE,MAAM,gBAAgB,CAAC;AAG/D,MAAM,CAAC,MAAM,mCAAmC,GAAG,CACjD,QAAyC,EACzC,cAAsB,EACtB,gBAAyB,EACzB,uBAAgC,EAChC,iBAAyB,EACY,EAAE;IACvC,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,6BAA6B,CAAC,QAAgC,CAAC,CAAC;QACtF,IAAI,aAAa;YAAE,OAAO,aAAa,CAAC;IAC1C,CAAC;IACD;;;;OAIG;IACH,MAAM,OAAO,GAAG,kBAAkB,CAAC,iBAAiB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAE1F,OAAO;QACL,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,OAAO;QACf,EAAE,EAAE,OAAO;KACZ,CAAC;AACJ,CAAC,CAAC"}

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

@@ -0,0 +1,71 @@
+import { Params } from '../../types/requestBody';
+import { AnthropicStreamState } from '../anthropic/types';
+import { ChatCompletionResponse, ErrorResponse, ProviderConfig } from '../types';
+import { BedrockAI21CompleteResponse, BedrockCohereCompleteResponse } from './complete';
+import { BedrockErrorResponse } from './embed';
+import { BedrockChatCompletionResponse, BedrockStreamState } from './types';
+export interface BedrockChatCompletionsParams extends Params {
+    additionalModelRequestFields?: Record<string, unknown>;
+    additional_model_request_fields?: Record<string, unknown>;
+    additionalModelResponseFieldPaths?: string[];
+    guardrailConfig?: {
+        guardrailIdentifier: string;
+        guardrailVersion: string;
+        trace?: string;
+    };
+    guardrail_config?: {
+        guardrailIdentifier: string;
+        guardrailVersion: string;
+        trace?: string;
+    };
+    anthropic_version?: string;
+    countPenalty?: number;
+}
+export interface BedrockConverseAnthropicChatCompletionsParams extends Omit<BedrockChatCompletionsParams, 'anthropic_beta'> {
+    anthropic_version?: string;
+    user?: string;
+    thinking?: {
+        type: string;
+        budget_tokens: number;
+    };
+    anthropic_beta?: string | string[];
+}
+export interface BedrockConverseCohereChatCompletionsParams extends BedrockChatCompletionsParams {
+    frequency_penalty?: number;
+    presence_penalty?: number;
+    logit_bias?: Record<string, number>;
+    n?: number;
+}
+export interface BedrockConverseAI21ChatCompletionsParams extends BedrockChatCompletionsParams {
+    frequency_penalty?: number;
+    presence_penalty?: number;
+    frequencyPenalty?: number;
+    presencePenalty?: number;
+    countPenalty?: number;
+}
+export declare const BedrockConverseChatCompleteConfig: ProviderConfig;
+export declare const BedrockErrorResponseTransform: (response: BedrockErrorResponse) => ErrorResponse | undefined;
+export declare const BedrockChatCompleteResponseTransform: (response: BedrockChatCompletionResponse | BedrockErrorResponse, responseStatus: number, responseHeaders: Headers, strictOpenAiCompliance: boolean, _gatewayRequestUrl: string, gatewayRequest: Params, 
+/** Set on the Cursor path only; unwraps tools converted from Cursor's freeform `custom` form. */
+isCursorEndpoint?: boolean) => ChatCompletionResponse | ErrorResponse;
+export declare const BedrockCursorChatCompleteResponseTransform: typeof BedrockChatCompleteResponseTransform;
+export declare const BedrockChatCompleteStreamChunkTransform: (response: string, fallbackId: string, streamState: BedrockStreamState, strictOpenAiCompliance: boolean, gatewayRequest: Params, isCursorEndpoint?: boolean) => string | string[];
+export declare const BedrockCursorChatCompleteStreamChunkTransform: typeof BedrockChatCompleteStreamChunkTransform;
+/**
+ * ================================
+ * NOTE - Add response_format to the request transform
+ * ================================
+ */
+export declare const BedrockConverseAnthropicChatCompleteConfig: ProviderConfig;
+/** OpenAI chat completions translated to Anthropic Messages for Bedrock InvokeModel. */
+export declare const BedrockAnthropicInvokeChatCompleteConfig: ProviderConfig;
+export declare const BedrockAnthropicInvokeChatCompleteResponseTransform: (response: BedrockChatCompletionResponse | BedrockErrorResponse | any, responseStatus: number, responseHeaders: Headers, strictOpenAiCompliance: boolean) => ChatCompletionResponse | ErrorResponse;
+export declare const BedrockAnthropicInvokeChatCompleteStreamChunkTransform: (responseChunk: string, fallbackId: string, streamState: AnthropicStreamState, strictOpenAiCompliance: boolean) => string | undefined;
+export declare const BedrockConverseCohereChatCompleteConfig: ProviderConfig;
+export declare const BedrockConverseAI21ChatCompleteConfig: ProviderConfig;
+export declare const BedrockCohereChatCompleteConfig: ProviderConfig;
+export declare const BedrockCohereChatCompleteResponseTransform: (response: BedrockCohereCompleteResponse | BedrockErrorResponse, responseStatus: number, responseHeaders: Headers, strictOpenAiCompliance: boolean, gatewayRequestUrl: string, gatewayRequest: Params) => ChatCompletionResponse | ErrorResponse;
+export declare const BedrockCohereChatCompleteStreamChunkTransform: (response: string, fallbackId: string, _streamState: Record<string, unknown>, _strictOpenAiCompliance: boolean, gatewayRequest: Params) => string | string[];
+export declare const BedrockAI21ChatCompleteConfig: ProviderConfig;
+export declare const BedrockAI21ChatCompleteResponseTransform: (response: BedrockAI21CompleteResponse | BedrockErrorResponse, responseStatus: number, responseHeaders: Headers, strictOpenAiCompliance: boolean, _gatewayRequestUrl: string, gatewayRequest: Params) => ChatCompletionResponse | ErrorResponse;
+//# sourceMappingURL=chatComplete.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"chatComplete.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock/chatComplete.ts"],"names":[],"mappings":"AAuBA,OAAO,EAAwB,MAAM,EAAkC,MAAM,yBAAyB,CAAC;AAUvG,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAQ1D,OAAO,EAAE,sBAAsB,EAAE,aAAa,EAA0B,cAAc,EAAE,MAAM,UAAU,CAAC;AAIzG,OAAO,EAAE,2BAA2B,EAAE,6BAA6B,EAA4B,MAAM,YAAY,CAAC;AAClH,OAAO,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAC/C,OAAO,EAEL,6BAA6B,EAI7B,kBAAkB,EAEnB,MAAM,SAAS,CAAC;AAUjB,MAAM,WAAW,4BAA6B,SAAQ,MAAM;IAC1D,4BAA4B,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACvD,+BAA+B,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC1D,iCAAiC,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7C,eAAe,CAAC,EAAE;QAChB,mBAAmB,EAAE,MAAM,CAAC;QAC5B,gBAAgB,EAAE,MAAM,CAAC;QACzB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,gBAAgB,CAAC,EAAE;QACjB,mBAAmB,EAAE,MAAM,CAAC;QAC5B,gBAAgB,EAAE,MAAM,CAAC;QACzB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,6CAA8C,SAAQ,IAAI,CACzE,4BAA4B,EAC5B,gBAAgB,CACjB;IACC,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE;QACT,IAAI,EAAE,MAAM,CAAC;QACb,aAAa,EAAE,MAAM,CAAC;KACvB,CAAC;IACF,cAAc,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CACpC;AAED,MAAM,WAAW,0CAA2C,SAAQ,4BAA4B;IAC9F,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpC,CAAC,CAAC,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,WAAW,wCAAyC,SAAQ,4BAA4B;IAC5F,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AA4TD,eAAO,MAAM,iCAAiC,EAAE,cA6M/C,CAAC;AAEF,eAAO,MAAM,6BAA6B,EAAE,CAC1C,QAAQ,EAAE,oBAAoB,KAC3B,aAAa,GAAG,SAMpB,CAAC;AAEF,eAAO,MAAM,oCAAoC,EAAE,CACjD,QAAQ,EAAE,6BAA6B,GAAG,oBAAoB,EAC9D,cAAc,EAAE,MAAM,EACtB,eAAe,EAAE,OAAO,EACxB,sBAAsB,EAAE,OAAO,EAC/B,kBAAkB,EAAE,MAAM,EAC1B,cAAc,EAAE,MAAM;AACtB,iGAAiG;AACjG,gBAAgB,CAAC,EAAE,OAAO,KACvB,sBAAsB,GAAG,aAyG7B,CAAC;AAEF,eAAO,MAAM,0CAA0C,EAAE,OAAO,oCACkC,CAAC;AAGnG,eAAO,MAAM,uCAAuC,EAAE,CACpD,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,WAAW,EAAE,kBAAkB,EAC/B,sBAAsB,EAAE,OAAO,EAC/B,cAAc,EAAE,MAAM,EACtB,gBAAgB,CAAC,EAAE,OAAO,KACvB,MAAM,GAAG,MAAM,EA8MnB,CAAC;AAEF,eAAO,MAAM,6CAA6C,EAAE,OAAO,uCAE4B,CAAC;AA2BhG;;;;GAIG;AACH,eAAO,MAAM,0CAA0C,EAAE,cAyIxD,CAAC;AAEF,wFAAwF;AACxF,eAAO,MAAM,wCAAwC,EAAE,cAuEtD,CAAC;AAGF,eAAO,MAAM,mDAAmD,EAAE,CAChE,QAAQ,EAAE,6BAA6B,GAAG,oBAAoB,GAAG,GAAG,EACpE,cAAc,EAAE,MAAM,EACtB,eAAe,EAAE,OAAO,EACxB,sBAAsB,EAAE,OAAO,KAC5B,sBAAsB,GAAG,aAW7B,CAAC;AAGF,eAAO,MAAM,sDAAsD,EAAE,CACnE,aAAa,EAAE,MAAM,EACrB,UAAU,EAAE,MAAM,EAClB,WAAW,EAAE,oBAAoB,EACjC,sBAAsB,EAAE,OAAO,KAC5B,MAAM,GAAG,SAWb,CAAC;AAEF,eAAO,MAAM,uCAAuC,EAAE,cAqCrD,CAAC;AAEF,eAAO,MAAM,qCAAqC,EAAE,cAkCnD,CAAC;AAEF,eAAO,MAAM,+BAA+B,EAAE,cA4E7C,CAAC;AAEF,eAAO,MAAM,0CAA0C,EAAE,CACvD,QAAQ,EAAE,6BAA6B,GAAG,oBAAoB,EAC9D,cAAc,EAAE,MAAM,EACtB,eAAe,EAAE,OAAO,EACxB,sBAAsB,EAAE,OAAO,EAC/B,iBAAiB,EAAE,MAAM,EACzB,cAAc,EAAE,MAAM,KACnB,sBAAsB,GAAG,aAyC7B,CAAC;AAEF,eAAO,MAAM,6CAA6C,EAAE,CAC1D,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,uBAAuB,EAAE,OAAO,EAChC,cAAc,EAAE,MAAM,KACnB,MAAM,GAAG,MAAM,EAoDnB,CAAC;AAEF,eAAO,MAAM,6BAA6B,EAAE,cAoE3C,CAAC;AAEF,eAAO,MAAM,wCAAwC,EAAE,CACrD,QAAQ,EAAE,2BAA2B,GAAG,oBAAoB,EAC5D,cAAc,EAAE,MAAM,EACtB,eAAe,EAAE,OAAO,EACxB,sBAAsB,EAAE,OAAO,EAC/B,kBAAkB,EAAE,MAAM,EAC1B,cAAc,EAAE,MAAM,KACnB,sBAAsB,GAAG,aAuC7B,CAAC"}