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

package/dist/providers/cohere/embed.js

@@ -0,0 +1,137 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. Default input_type to search_document
+ * ================================
+ */
+import { COHERE } from '../../globals.js';
+import { generateErrorResponse, generateInvalidProviderResponseError, validateEncodingFormatAndRemove } from '../utils.js';
+export const CohereEmbedConfig = {
+    model: {
+        param: 'model',
+        required: false,
+    },
+    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,
+    },
+    //backwards compatibility
+    embedding_types: {
+        param: 'embedding_types',
+        required: false,
+    },
+};
+export const CohereEmbedResponseTransform = (response, responseStatus, _responseHeaders, _strictOpenAiCompliance, _gatewayRequestUrl, gatewayRequest) => {
+    if (responseStatus !== 200) {
+        return generateErrorResponse({
+            message: response.message || '',
+            type: null,
+            code: null,
+        }, COHERE);
+    }
+    const model = gatewayRequest.model || '';
+    // The upstream implementation only supports float embeddings for cohere to confirm to openai signature
+    if ('embeddings' in response) {
+        let data = [];
+        if (response?.embeddings && 'float' in response.embeddings) {
+            data = response.embeddings.float.map((embedding, index) => ({
+                object: 'embedding',
+                embedding: embedding,
+                index: index,
+            }));
+        }
+        const inputTokens = response.meta?.tokens?.input_tokens ?? response.meta?.billed_units?.input_tokens ?? 0;
+        const outputTokens = response.meta?.tokens?.output_tokens ?? response.meta?.billed_units?.output_tokens ?? 0;
+        const totalTokens = inputTokens + outputTokens;
+        return {
+            object: 'list',
+            data,
+            provider: COHERE,
+            model,
+            usage: {
+                prompt_tokens: inputTokens,
+                total_tokens: totalTokens,
+            },
+        };
+    }
+    return generateInvalidProviderResponseError(response, COHERE);
+};
+export const CohereEmbedResponseTransformBatch = (response) => {
+    return {
+        id: response.id,
+        custom_id: response.custom_id,
+        response: {
+            status_code: 200,
+            request_id: response.id,
+            body: {
+                object: 'list',
+                data: [
+                    {
+                        object: 'embedding',
+                        index: 0,
+                        embedding: response.embeddings.float?.array,
+                    },
+                ],
+                model: '',
+                usage: {
+                    prompt_tokens: 0,
+                    total_tokens: 0,
+                },
+            },
+        },
+    };
+};
+//# sourceMappingURL=embed.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"embed.js","sourceRoot":"","sources":["../../../src/providers/cohere/embed.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAGvC,OAAO,EAAE,qBAAqB,EAAE,oCAAoC,EAAE,+BAA+B,EAAE,MAAM,UAAU,CAAC;AAExH,MAAM,CAAC,MAAM,iBAAiB,GAAmB;IAC/C,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;KAChB;IACD,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;IACD,yBAAyB;IACzB,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,KAAK;KAChB;CACF,CAAC;AAqDF,MAAM,CAAC,MAAM,4BAA4B,GAOJ,CACnC,QAAQ,EACR,cAAc,EACd,gBAAgB,EAChB,uBAAuB,EACvB,kBAAkB,EAClB,cAAc,EACd,EAAE;IACF,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,OAAO,qBAAqB,CAC1B;YACE,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,EAAE;YAC/B,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,IAAI;SACX,EACD,MAAM,CACP,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAI,cAAc,CAAC,KAAgB,IAAI,EAAE,CAAC;IAErD,uGAAuG;IACvG,IAAI,YAAY,IAAI,QAAQ,EAAE,CAAC;QAC7B,IAAI,IAAI,GAAwB,EAAE,CAAC;QACnC,IAAI,QAAQ,EAAE,UAAU,IAAI,OAAO,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC3D,IAAI,GAAG,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC1D,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,SAAS;gBACpB,KAAK,EAAE,KAAK;aACb,CAAC,CAAC,CAAC;QACN,CAAC;QACD,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,IAAI,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,YAAY,IAAI,CAAC,CAAC;QAC1G,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,aAAa,IAAI,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,aAAa,IAAI,CAAC,CAAC;QAC7G,MAAM,WAAW,GAAG,WAAW,GAAG,YAAY,CAAC;QAC/C,OAAO;YACL,MAAM,EAAE,MAAM;YACd,IAAI;YACJ,QAAQ,EAAE,MAAM;YAChB,KAAK;YACL,KAAK,EAAE;gBACL,aAAa,EAAE,WAAW;gBAC1B,YAAY,EAAE,WAAW;aAC1B;SACF,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AAChE,CAAC,CAAC;AAyBF,MAAM,CAAC,MAAM,iCAAiC,GAAG,CAAC,QAAkC,EAAE,EAAE;IACtF,OAAO;QACL,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,SAAS,EAAE,QAAQ,CAAC,SAAS;QAC7B,QAAQ,EAAE;YACR,WAAW,EAAE,GAAG;YAChB,UAAU,EAAE,QAAQ,CAAC,EAAE;YACvB,IAAI,EAAE;gBACJ,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE;oBACJ;wBACE,MAAM,EAAE,WAAW;wBACnB,KAAK,EAAE,CAAC;wBACR,SAAS,EAAE,QAAQ,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK;qBAC5C;iBACF;gBACD,KAAK,EAAE,EAAE;gBACT,KAAK,EAAE;oBACL,aAAa,EAAE,CAAC;oBAChB,YAAY,EAAE,CAAC;iBAChB;aACF;SACF;KACF,CAAC;AACJ,CAAC,CAAC"}

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

@@ -0,0 +1,2 @@
+import type { RequestHandler } from '../types.js';
+export declare const CohereGetBatchOutputHandler: RequestHandler;

package/dist/providers/cohere/getBatchOutput.js

@@ -0,0 +1,117 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. Add avro to the imports
+ * ================================
+ */
+import * as avro from 'avsc';
+import { COHERE } from '../../globals.js';
+import { safeFetch } from '../../http/index.js';
+import { buildErrorBody, errorResponse } from '../../utils/errorHandlingUtils.js';
+import { providerApiArgsFromHandler } from '../requestHandlerApiArgs.js';
+import CohereAPIConfig from './api.js';
+import { CohereEmbedResponseTransformBatch } from './embed.js';
+export const CohereGetBatchOutputHandler = async (args) => {
+    try {
+        const apiArgs = providerApiArgsFromHandler(args, {
+            endpoint: 'retrieveBatch',
+            method: 'GET',
+            path: args.request.path.replace('/output', ''),
+        });
+        // get the base url and endpoint for retrieveBatch
+        const baseURL = await CohereAPIConfig.getBaseURL(apiArgs);
+        const endpoint = CohereAPIConfig.getEndpoint(apiArgs);
+        const headers = await CohereAPIConfig.headers({
+            ...apiArgs,
+            transformedRequestUrl: baseURL + endpoint,
+            transformedRequestBody: {},
+        });
+        // get the batch details
+        const retrieveBatchResponse = await safeFetch(baseURL + endpoint, {
+            method: 'GET',
+            headers,
+            ssrf: 'guard',
+            signal: args.signal,
+        });
+        if (!retrieveBatchResponse.ok) {
+            const errorText = await retrieveBatchResponse.text();
+            throw new Error(JSON.stringify({
+                type: 'provider_error',
+                code: retrieveBatchResponse.status,
+                message: 'cohere error: ' + errorText,
+            }));
+        }
+        const batchDetails = await retrieveBatchResponse.json();
+        const outputFileId = batchDetails.output_dataset_id;
+        // get the file details
+        const retrieveFileResponse = await safeFetch(`https://api.cohere.ai/v1/datasets/${outputFileId}`, {
+            method: 'GET',
+            headers,
+            ssrf: 'guard',
+            signal: args.signal,
+        });
+        const retrieveFileResponseJson = await retrieveFileResponse.json();
+        if (!retrieveFileResponse.ok) {
+            const errorText = await retrieveFileResponse.text();
+            throw new Error(JSON.stringify({
+                type: 'provider_error',
+                code: retrieveFileResponse.status,
+                message: 'cohere error: ' + errorText,
+            }));
+        }
+        if (!retrieveFileResponseJson.dataset.dataset_parts) {
+            throw new Error('file not found');
+        }
+        const stream = new ReadableStream({
+            start: async (controller) => {
+                const fileParts = retrieveFileResponseJson.dataset.dataset_parts;
+                for (const filePart of fileParts) {
+                    // filePart.url comes from the provider's response body, not our configuration.
+                    const filePartResponse = await safeFetch(filePart.url, {
+                        signal: args.signal,
+                        method: 'GET',
+                        headers,
+                        ssrf: 'guard',
+                    });
+                    const arrayBuffer = await filePartResponse.arrayBuffer();
+                    const buf = Buffer.from(arrayBuffer);
+                    const decoder = new avro.streams.BlockDecoder({
+                        parseHook: (schema) => {
+                            return avro.Type.forSchema(schema, { wrapUnions: true });
+                        },
+                    });
+                    const encoder = new TextEncoder();
+                    decoder.on('data', (data) => {
+                        controller.enqueue(encoder.encode(JSON.stringify(CohereEmbedResponseTransformBatch(JSON.parse(data)))));
+                        controller.enqueue(encoder.encode('\n'));
+                    });
+                    decoder.on('end', () => {
+                        controller.close();
+                    });
+                    decoder.end(buf);
+                }
+            },
+        });
+        return new Response(stream, {
+            headers: { 'Content-Type': 'application/octet-stream' },
+        });
+    }
+    catch (error) {
+        let errorBody;
+        try {
+            errorBody = JSON.parse(error.message);
+            errorBody.provider = COHERE;
+        }
+        catch {
+            errorBody = buildErrorBody({
+                message: error.message,
+                statusCode: 500,
+                type: error?.name ?? null,
+                extra: { provider: COHERE },
+            });
+        }
+        return errorResponse(errorBody, 500);
+    }
+};
+//# sourceMappingURL=getBatchOutput.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"getBatchOutput.js","sourceRoot":"","sources":["../../../src/providers/cohere/getBatchOutput.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACvC,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/E,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AAEtE,OAAO,eAAe,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,iCAAiC,EAAE,MAAM,SAAS,CAAC;AAG5D,MAAM,CAAC,MAAM,2BAA2B,GAAmB,KAAK,EAAC,IAAI,EAAC,EAAE;IACtE,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,0BAA0B,CAAC,IAAI,EAAE;YAC/C,QAAQ,EAAE,eAAe;YACzB,MAAM,EAAE,KAAK;YACb,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;SAC/C,CAAC,CAAC;QAEH,kDAAkD;QAClD,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC1D,MAAM,QAAQ,GAAG,eAAe,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC;YAC5C,GAAG,OAAO;YACV,qBAAqB,EAAE,OAAO,GAAG,QAAQ;YACzC,sBAAsB,EAAE,EAAE;SAC3B,CAAC,CAAC;QACH,wBAAwB;QACxB,MAAM,qBAAqB,GAAG,MAAM,SAAS,CAAC,OAAO,GAAG,QAAQ,EAAE;YAChE,MAAM,EAAE,KAAK;YACb,OAAO;YACP,IAAI,EAAE,OAAO;YACb,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC;QACH,IAAI,CAAC,qBAAqB,CAAC,EAAE,EAAE,CAAC;YAC9B,MAAM,SAAS,GAAG,MAAM,qBAAqB,CAAC,IAAI,EAAE,CAAC;YACrD,MAAM,IAAI,KAAK,CACb,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,qBAAqB,CAAC,MAAM;gBAClC,OAAO,EAAE,gBAAgB,GAAG,SAAS;aACtC,CAAC,CACH,CAAC;QACJ,CAAC;QACD,MAAM,YAAY,GAAgC,MAAM,qBAAqB,CAAC,IAAI,EAAE,CAAC;QACrF,MAAM,YAAY,GAAG,YAAY,CAAC,iBAAiB,CAAC;QAEpD,uBAAuB;QACvB,MAAM,oBAAoB,GAAG,MAAM,SAAS,CAAC,qCAAqC,YAAY,EAAE,EAAE;YAChG,MAAM,EAAE,KAAK;YACb,OAAO;YACP,IAAI,EAAE,OAAO;YACb,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC;QACH,MAAM,wBAAwB,GAA0B,MAAM,oBAAoB,CAAC,IAAI,EAAE,CAAC;QAC1F,IAAI,CAAC,oBAAoB,CAAC,EAAE,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,MAAM,oBAAoB,CAAC,IAAI,EAAE,CAAC;YACpD,MAAM,IAAI,KAAK,CACb,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,oBAAoB,CAAC,MAAM;gBACjC,OAAO,EAAE,gBAAgB,GAAG,SAAS;aACtC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,wBAAwB,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YACpD,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACpC,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;YAChC,KAAK,EAAE,KAAK,EAAC,UAAU,EAAC,EAAE;gBACxB,MAAM,SAAS,GAAG,wBAAwB,CAAC,OAAO,CAAC,aAAa,CAAC;gBACjE,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;oBACjC,+EAA+E;oBAC/E,MAAM,gBAAgB,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,GAAG,EAAE;wBACrD,MAAM,EAAE,IAAI,CAAC,MAAM;wBACnB,MAAM,EAAE,KAAK;wBACb,OAAO;wBACP,IAAI,EAAE,OAAO;qBACd,CAAC,CAAC;oBACH,MAAM,WAAW,GAAG,MAAM,gBAAgB,CAAC,WAAW,EAAE,CAAC;oBACzD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;oBACrC,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;wBAC5C,SAAS,EAAE,CAAC,MAAW,EAAE,EAAE;4BACzB,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;wBAC3D,CAAC;qBACF,CAAC,CAAC;oBACH,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;oBAClC,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAS,EAAE,EAAE;wBAC/B,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,iCAAiC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;wBACxG,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;oBAC3C,CAAC,CAAC,CAAC;oBACH,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;wBACrB,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,CAAC,CAAC,CAAC;oBACH,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACnB,CAAC;YACH,CAAC;SACF,CAAC,CAAC;QACH,OAAO,IAAI,QAAQ,CAAC,MAAM,EAAE;YAC1B,OAAO,EAAE,EAAE,cAAc,EAAE,0BAA0B,EAAE;SACxD,CAAC,CAAC;IACL,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,MAAM,CAAC;QAC9B,CAAC;QAAC,MAAM,CAAC;YACP,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,MAAM,EAAE;aAC5B,CAAC,CAAC;QACL,CAAC;QACD,OAAO,aAAa,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IACvC,CAAC;AACH,CAAC,CAAC"}

package/dist/providers/cohere/getFiles.d.ts

@@ -0,0 +1,4 @@
+import { ErrorResponse, GetFileResponse, GetFilesResponse } from '../types.js';
+import { CohereErrorResponse, CohereGetFileResponse, CohereGetFilesResponse } from './types.js';
+export declare const CohereGetFileResponseTransform: (response: CohereGetFileResponse | CohereErrorResponse, responseStatus: number) => GetFileResponse | ErrorResponse;
+export declare const CohereGetFilesResponseTransform: (response: CohereGetFilesResponse | CohereErrorResponse, responseStatus: number) => GetFilesResponse | ErrorResponse;

package/dist/providers/cohere/getFiles.js

@@ -0,0 +1,51 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { COHERE } from '../../globals.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+import { CohereErrorResponseTransform } from './utils.js';
+const transformDatasetUploadStatus = (status) => {
+    if (status === 'skipped' || status === 'failed') {
+        return 'error';
+    }
+    if (status === 'validated') {
+        return 'processed';
+    }
+    return 'uploaded';
+};
+const transformCohereDataset = (dataset) => {
+    return {
+        id: dataset.id,
+        object: 'file',
+        bytes: 0,
+        created_at: new Date(dataset.created_at).getTime(),
+        filename: dataset.name,
+        purpose: dataset.dataset_type, // we don't have 1:1 mapping between purpose and dataset_type
+        status: transformDatasetUploadStatus(dataset.validation_status),
+        status_details: dataset.validation_error ?? '',
+    };
+};
+export const CohereGetFileResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200 && 'message' in response) {
+        return CohereErrorResponseTransform(response);
+    }
+    if ('dataset' in response) {
+        return transformCohereDataset(response.dataset);
+    }
+    return generateInvalidProviderResponseError(response, COHERE);
+};
+export const CohereGetFilesResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200 && 'message' in response) {
+        return CohereErrorResponseTransform(response);
+    }
+    if ('datasets' in response) {
+        return {
+            data: response.datasets.map(dataset => transformCohereDataset(dataset)),
+            object: 'list',
+        };
+    }
+    return generateInvalidProviderResponseError(response, COHERE);
+};
+//# sourceMappingURL=getFiles.js.map

package/dist/providers/cohere/getFiles.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getFiles.js","sourceRoot":"","sources":["../../../src/providers/cohere/getFiles.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAEvC,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAQhE,OAAO,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAEvD,MAAM,4BAA4B,GAAG,CAAC,MAAiC,EAAE,EAAE;IACzE,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;QAChD,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;QAC3B,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAEF,MAAM,sBAAsB,GAAG,CAAC,OAAsB,EAAmB,EAAE;IACzE,OAAO;QACL,EAAE,EAAE,OAAO,CAAC,EAAE;QACd,MAAM,EAAE,MAAM;QACd,KAAK,EAAE,CAAC;QACR,UAAU,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE;QAClD,QAAQ,EAAE,OAAO,CAAC,IAAI;QACtB,OAAO,EAAE,OAAO,CAAC,YAAY,EAAE,6DAA6D;QAC5F,MAAM,EAAE,4BAA4B,CAAC,OAAO,CAAC,iBAAiB,CAAC;QAC/D,cAAc,EAAE,OAAO,CAAC,gBAAgB,IAAI,EAAE;KAC/C,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAC5C,QAAqD,EACrD,cAAsB,EACW,EAAE;IACnC,IAAI,cAAc,KAAK,GAAG,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;QACpD,OAAO,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IAED,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;QAC1B,OAAO,sBAAsB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AAChE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAC7C,QAAsD,EACtD,cAAsB,EACY,EAAE;IACpC,IAAI,cAAc,KAAK,GAAG,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;QACpD,OAAO,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IAED,IAAI,UAAU,IAAI,QAAQ,EAAE,CAAC;QAC3B,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;YACvE,MAAM,EAAE,MAAM;SACf,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AAChE,CAAC,CAAC"}

package/dist/providers/cohere/index.d.ts

@@ -0,0 +1,10 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. Add Rerank to the config and response transforms
+ * ================================
+ */
+import { ProviderConfigs } from '../types.js';
+declare const CohereConfig: ProviderConfigs;
+export default CohereConfig;

package/dist/providers/cohere/index.js

@@ -0,0 +1,46 @@
+import CohereAPIConfig from './api.js';
+import { CohereCancelBatchResponseTransform } from './cancelBatch.js';
+import { CohereChatCompleteConfig, CohereChatCompleteResponseTransform, CohereChatCompleteStreamChunkTransform, } from './chatComplete.js';
+import { CohereCompleteConfig, CohereCompleteResponseTransform, CohereCompleteStreamChunkTransform } from './complete.js';
+import { CohereCreateBatchConfig, CohereCreateBatchResponseTransform } from './createBatch.js';
+import { CohereDeleteFileResponseTransform } from './deleteFile.js';
+import { CohereEmbedConfig, CohereEmbedResponseTransform } from './embed.js';
+import { CohereGetBatchOutputHandler } from './getBatchOutput.js';
+import { CohereGetFileResponseTransform, CohereGetFilesResponseTransform } from './getFiles.js';
+import { CohereListBatchResponseTransform } from './listBatches.js';
+import { CohereRerankConfig, CohereRerankResponseTransform } from './rerank.js';
+import { CohereRetrieveBatchResponseTransform } from './retrieveBatch.js';
+import { CohereUploadFileRequestTransform, CohereUploadFileResponseTransform } from './uploadFile.js';
+const CohereConfig = {
+    complete: CohereCompleteConfig,
+    chatComplete: CohereChatCompleteConfig,
+    embed: CohereEmbedConfig,
+    rerank: CohereRerankConfig,
+    api: CohereAPIConfig,
+    createBatch: CohereCreateBatchConfig,
+    cancelBatch: {},
+    requestHandlers: {
+        getBatchOutput: CohereGetBatchOutputHandler,
+    },
+    requestTransforms: {
+        uploadFile: CohereUploadFileRequestTransform,
+    },
+    responseTransforms: {
+        complete: CohereCompleteResponseTransform,
+        'stream-complete': CohereCompleteStreamChunkTransform,
+        chatComplete: CohereChatCompleteResponseTransform,
+        'stream-chatComplete': CohereChatCompleteStreamChunkTransform,
+        embed: CohereEmbedResponseTransform,
+        rerank: CohereRerankResponseTransform,
+        uploadFile: CohereUploadFileResponseTransform,
+        retrieveFile: CohereGetFileResponseTransform,
+        listFiles: CohereGetFilesResponseTransform,
+        deleteFile: CohereDeleteFileResponseTransform,
+        createBatch: CohereCreateBatchResponseTransform,
+        listBatches: CohereListBatchResponseTransform,
+        retrieveBatch: CohereRetrieveBatchResponseTransform,
+        cancelBatch: CohereCancelBatchResponseTransform,
+    },
+};
+export default CohereConfig;
+//# sourceMappingURL=index.js.map

package/dist/providers/cohere/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/cohere/index.ts"],"names":[],"mappings":"AAQA,OAAO,eAAe,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,kCAAkC,EAAE,MAAM,eAAe,CAAC;AACnE,OAAO,EACL,wBAAwB,EACxB,mCAAmC,EACnC,sCAAsC,GACvC,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,oBAAoB,EAAE,+BAA+B,EAAE,kCAAkC,EAAE,MAAM,YAAY,CAAC;AACvH,OAAO,EAAE,uBAAuB,EAAE,kCAAkC,EAAE,MAAM,eAAe,CAAC;AAC5F,OAAO,EAAE,iCAAiC,EAAE,MAAM,cAAc,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAC1E,OAAO,EAAE,2BAA2B,EAAE,MAAM,kBAAkB,CAAC;AAC/D,OAAO,EAAE,8BAA8B,EAAE,+BAA+B,EAAE,MAAM,YAAY,CAAC;AAC7F,OAAO,EAAE,gCAAgC,EAAE,MAAM,eAAe,CAAC;AACjE,OAAO,EAAE,kBAAkB,EAAE,6BAA6B,EAAE,MAAM,UAAU,CAAC;AAC7E,OAAO,EAAE,oCAAoC,EAAE,MAAM,iBAAiB,CAAC;AACvE,OAAO,EAAE,gCAAgC,EAAE,iCAAiC,EAAE,MAAM,cAAc,CAAC;AAEnG,MAAM,YAAY,GAAoB;IACpC,QAAQ,EAAE,oBAAoB;IAC9B,YAAY,EAAE,wBAAwB;IACtC,KAAK,EAAE,iBAAiB;IACxB,MAAM,EAAE,kBAAkB;IAC1B,GAAG,EAAE,eAAe;IACpB,WAAW,EAAE,uBAAuB;IACpC,WAAW,EAAE,EAAE;IACf,eAAe,EAAE;QACf,cAAc,EAAE,2BAA2B;KAC5C;IACD,iBAAiB,EAAE;QACjB,UAAU,EAAE,gCAAgC;KAC7C;IACD,kBAAkB,EAAE;QAClB,QAAQ,EAAE,+BAA+B;QACzC,iBAAiB,EAAE,kCAAkC;QACrD,YAAY,EAAE,mCAAmC;QACjD,qBAAqB,EAAE,sCAAsC;QAC7D,KAAK,EAAE,4BAA4B;QACnC,MAAM,EAAE,6BAA6B;QACrC,UAAU,EAAE,iCAAiC;QAC7C,YAAY,EAAE,8BAA8B;QAC5C,SAAS,EAAE,+BAA+B;QAC1C,UAAU,EAAE,iCAAiC;QAC7C,WAAW,EAAE,kCAAkC;QAC/C,WAAW,EAAE,gCAAgC;QAC7C,aAAa,EAAE,oCAAoC;QACnD,WAAW,EAAE,kCAAkC;KAChD;CACF,CAAC;AAEF,eAAe,YAAY,CAAC"}

package/dist/providers/cohere/listBatches.d.ts

@@ -0,0 +1,3 @@
+import { ErrorResponse, ListBatchesResponse } from '../types.js';
+import { CohereErrorResponse, CohereListBatchResponse } from './types.js';
+export declare const CohereListBatchResponseTransform: (response: CohereListBatchResponse | CohereErrorResponse, responseStatus: number, _responseHeaders: Headers, strictOpenAiCompliance: boolean) => ListBatchesResponse | ErrorResponse;

package/dist/providers/cohere/listBatches.js

@@ -0,0 +1,30 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { COHERE } from '../../globals.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+import { CohereErrorResponseTransform } from './utils.js';
+export const CohereListBatchResponseTransform = (response, responseStatus, _responseHeaders, strictOpenAiCompliance) => {
+    if (responseStatus !== 200 && 'message' in response) {
+        return CohereErrorResponseTransform(response);
+    }
+    if ('embed_jobs' in response) {
+        return {
+            object: 'list',
+            data: response.embed_jobs.map(job => ({
+                id: job.job_id,
+                object: 'batch',
+                created_at: new Date(job.created_at).getTime(),
+                status: job.status,
+                input_file_id: job.input_dataset_id,
+                output_file_id: job.output_dataset_id,
+                metadata: job.meta,
+                ...(strictOpenAiCompliance ? { model: job.model, truncate: job.truncate, name: job.name } : {}),
+            })),
+        };
+    }
+    return generateInvalidProviderResponseError(response, COHERE);
+};
+//# sourceMappingURL=listBatches.js.map

package/dist/providers/cohere/listBatches.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"listBatches.js","sourceRoot":"","sources":["../../../src/providers/cohere/listBatches.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAEvC,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAEhE,OAAO,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAEvD,MAAM,CAAC,MAAM,gCAAgC,GAKF,CAAC,QAAQ,EAAE,cAAc,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,EAAE;IAChH,IAAI,cAAc,KAAK,GAAG,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;QACpD,OAAO,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IAED,IAAI,YAAY,IAAI,QAAQ,EAAE,CAAC;QAC7B,OAAO;YACL,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACpC,EAAE,EAAE,GAAG,CAAC,MAAM;gBACd,MAAM,EAAE,OAAO;gBACf,UAAU,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE;gBAC9C,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,aAAa,EAAE,GAAG,CAAC,gBAAgB;gBACnC,cAAc,EAAE,GAAG,CAAC,iBAAiB;gBACrC,QAAQ,EAAE,GAAG,CAAC,IAAI;gBAClB,GAAG,CAAC,sBAAsB,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAChG,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AAChE,CAAC,CAAC"}

package/dist/providers/cohere/rerank.d.ts

@@ -0,0 +1,46 @@
+import { ErrorResponse, RerankResponse } from '../types.js';
+import { CohereErrorResponse } from './types.js';
+/**
+ * ================================
+ * TFY Managed File
+ * ================================
+ */
+export declare const CohereRerankConfig: {
+    query: {
+        param: string;
+        required: boolean;
+    };
+    top_n: {
+        param: string;
+        required: boolean;
+    };
+    documents: {
+        param: string;
+        required: boolean;
+    };
+    return_documents: {
+        param: string;
+        required: boolean;
+    };
+    max_tokens_per_doc: {
+        param: string;
+        required: boolean;
+    };
+};
+interface CohereRerankResponse {
+    id: string;
+    results: {
+        index: number;
+        relevance_score: number;
+    }[];
+    meta: {
+        api_version: {
+            version: string;
+        };
+        billed_units: {
+            search_units: number;
+        };
+    };
+}
+export declare const CohereRerankResponseTransform: (response: CohereRerankResponse | CohereErrorResponse, responseStatus: number) => RerankResponse | ErrorResponse;
+export {};

package/dist/providers/cohere/rerank.js

@@ -0,0 +1,41 @@
+import { CohereErrorResponseTransform } from './utils.js';
+/**
+ * ================================
+ * TFY Managed File
+ * ================================
+ */
+export const CohereRerankConfig = {
+    query: {
+        param: 'query',
+        required: true,
+    },
+    top_n: {
+        param: 'top_n',
+        required: false,
+    },
+    documents: {
+        param: 'documents',
+        required: true,
+    },
+    return_documents: {
+        param: 'return_documents',
+        required: false,
+    },
+    max_tokens_per_doc: {
+        param: 'max_tokens_per_doc',
+        required: false,
+    },
+};
+export const CohereRerankResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200 && 'message' in response) {
+        return CohereErrorResponseTransform(response);
+    }
+    return {
+        id: response.id,
+        results: response.results.map(result => ({
+            index: result.index,
+            relevance_score: result.relevance_score,
+        })),
+    };
+};
+//# sourceMappingURL=rerank.js.map

package/dist/providers/cohere/rerank.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rerank.js","sourceRoot":"","sources":["../../../src/providers/cohere/rerank.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAEvD;;;;GAIG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG;IAChC,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;KAChB;IACD,SAAS,EAAE;QACT,KAAK,EAAE,WAAW;QAClB,QAAQ,EAAE,IAAI;KACf;IACD,gBAAgB,EAAE;QAChB,KAAK,EAAE,kBAAkB;QACzB,QAAQ,EAAE,KAAK;KAChB;IACD,kBAAkB,EAAE;QAClB,KAAK,EAAE,oBAAoB;QAC3B,QAAQ,EAAE,KAAK;KAChB;CACF,CAAC;AAkBF,MAAM,CAAC,MAAM,6BAA6B,GAAG,CAC3C,QAAoD,EACpD,cAAsB,EACU,EAAE;IAClC,IAAI,cAAc,KAAK,GAAG,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;QACpD,OAAO,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IAED,OAAO;QACL,EAAE,EAAG,QAAiC,CAAC,EAAE;QACzC,OAAO,EAAG,QAAiC,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACjE,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,eAAe,EAAE,MAAM,CAAC,eAAe;SACxC,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC,CAAC"}

package/dist/providers/cohere/retrieveBatch.d.ts

@@ -0,0 +1,3 @@
+import { ErrorResponse, RetrieveBatchResponse } from '../types.js';
+import { CohereErrorResponse, CohereRetrieveBatchResponse } from './types.js';
+export declare const CohereRetrieveBatchResponseTransform: (response: CohereRetrieveBatchResponse | CohereErrorResponse, responseStatus: number, _responseHeaders: Headers, strictOpenAiCompliance: boolean) => RetrieveBatchResponse | ErrorResponse;

package/dist/providers/cohere/retrieveBatch.js

@@ -0,0 +1,33 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { COHERE } from '../../globals.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+import { CohereErrorResponseTransform } from './utils.js';
+export const CohereRetrieveBatchResponseTransform = (response, responseStatus, _responseHeaders, strictOpenAiCompliance) => {
+    if (responseStatus !== 200 && 'message' in response) {
+        return CohereErrorResponseTransform(response);
+    }
+    if ('job_id' in response) {
+        return {
+            id: response.job_id,
+            object: 'batch',
+            created_at: new Date(response.created_at).getTime(),
+            status: response.status,
+            input_file_id: response.input_dataset_id,
+            output_file_id: response.output_dataset_id,
+            metadata: response.meta,
+            ...(strictOpenAiCompliance
+                ? {
+                    model: response.model,
+                    truncate: response.truncate,
+                    name: response.name,
+                }
+                : {}),
+        };
+    }
+    return generateInvalidProviderResponseError(response, COHERE);
+};
+//# sourceMappingURL=retrieveBatch.js.map

package/dist/providers/cohere/retrieveBatch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveBatch.js","sourceRoot":"","sources":["../../../src/providers/cohere/retrieveBatch.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAEvC,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAEhE,OAAO,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAEvD,MAAM,CAAC,MAAM,oCAAoC,GAKJ,CAAC,QAAQ,EAAE,cAAc,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,EAAE;IAClH,IAAI,cAAc,KAAK,GAAG,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;QACpD,OAAO,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IAED,IAAI,QAAQ,IAAI,QAAQ,EAAE,CAAC;QACzB,OAAO;YACL,EAAE,EAAE,QAAQ,CAAC,MAAM;YACnB,MAAM,EAAE,OAAO;YACf,UAAU,EAAE,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE;YACnD,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,aAAa,EAAE,QAAQ,CAAC,gBAAgB;YACxC,cAAc,EAAE,QAAQ,CAAC,iBAAiB;YAC1C,QAAQ,EAAE,QAAQ,CAAC,IAAI;YACvB,GAAG,CAAC,sBAAsB;gBACxB,CAAC,CAAC;oBACE,KAAK,EAAE,QAAQ,CAAC,KAAK;oBACrB,QAAQ,EAAE,QAAQ,CAAC,QAAQ;oBAC3B,IAAI,EAAE,QAAQ,CAAC,IAAI;iBACpB;gBACH,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AAChE,CAAC,CAAC"}