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

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

@@ -0,0 +1,95 @@
+import { EmbedResponse } from '../../types/embedRequestBody';
+import { ErrorResponse, ProviderConfig } from '../types';
+export declare const CohereEmbedConfig: ProviderConfig;
+/**
+ * The version of the API used for the Cohere Embedding request.
+ * @interface
+ */
+export interface ApiVersion {
+    /** The version number. */
+    version: string;
+}
+/**
+ * Metadata about the embedding response.
+ * @interface
+ */
+export interface EmbedMeta {
+    /** The API version used. */
+    api_version: ApiVersion;
+    billed_units: {
+        images: number;
+        input_tokens: number;
+        output_tokens: number;
+        search_units: number;
+        classifications: number;
+    };
+    tokens: {
+        input_tokens: number;
+        output_tokens: number;
+    };
+    cached_tokens: number;
+    warnings: string[];
+}
+/**
+ * The structure of the CohereEmbedResponse.
+ * @interface
+ */
+export interface CohereEmbedResponse {
+    /** A string that represents the ID of the embedding request. */
+    id: string;
+    /** An array of strings which were the input texts to be embedded. */
+    texts: string[];
+    /** A 2D array of floating point numbers representing the embeddings. */
+    embeddings: number[][] | {
+        float: number[][];
+    };
+    /** An `EmbedMeta` object which contains metadata about the response. */
+    meta: EmbedMeta;
+    message?: string;
+}
+export declare const CohereEmbedResponseTransform: (response: CohereEmbedResponse, responseStatus: number, responseHeaders: Headers, strictOpenAiCompliance: boolean, gatewayRequestUrl: string, gatewayRequest: Params) => EmbedResponse | ErrorResponse;
+interface CohereEmbedResponseBatch {
+    custom_id: string;
+    id: string;
+    text: string;
+    embeddings: {
+        float?: {
+            array: number[];
+        };
+        int8?: {
+            array: number[];
+        };
+        uint8?: {
+            array: number[];
+        };
+        binary?: {
+            array: number[];
+        };
+        ubinary?: {
+            array: number[];
+        };
+    };
+}
+export declare const CohereEmbedResponseTransformBatch: (response: CohereEmbedResponseBatch) => {
+    id: string;
+    custom_id: string;
+    response: {
+        status_code: number;
+        request_id: string;
+        body: {
+            object: string;
+            data: {
+                object: string;
+                index: number;
+                embedding: number[] | undefined;
+            }[];
+            model: string;
+            usage: {
+                prompt_tokens: number;
+                total_tokens: number;
+            };
+        };
+    };
+};
+export {};
+//# sourceMappingURL=embed.d.ts.map

package/dist/providers/cohere/embed.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embed.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/embed.ts"],"names":[],"mappings":"AAQA,OAAO,EAAe,aAAa,EAAqB,MAAM,8BAA8B,CAAC;AAC7F,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAGzD,eAAO,MAAM,iBAAiB,EAAE,cAgE/B,CAAC;AAEF;;;GAGG;AACH,MAAM,WAAW,UAAU;IACzB,0BAA0B;IAC1B,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;;GAGG;AACH,MAAM,WAAW,SAAS;IACxB,4BAA4B;IAC5B,WAAW,EAAE,UAAU,CAAC;IACxB,YAAY,EAAE;QACZ,MAAM,EAAE,MAAM,CAAC;QACf,YAAY,EAAE,MAAM,CAAC;QACrB,aAAa,EAAE,MAAM,CAAC;QACtB,YAAY,EAAE,MAAM,CAAC;QACrB,eAAe,EAAE,MAAM,CAAC;KACzB,CAAC;IACF,MAAM,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,aAAa,EAAE,MAAM,CAAC;KACvB,CAAC;IACF,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB;AAED;;;GAGG;AACH,MAAM,WAAW,mBAAmB;IAClC,gEAAgE;IAChE,EAAE,EAAE,MAAM,CAAC;IAEX,qEAAqE;IACrE,KAAK,EAAE,MAAM,EAAE,CAAC;IAEhB,wEAAwE;IACxE,UAAU,EAAE,MAAM,EAAE,EAAE,GAAG;QAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAA;KAAE,CAAC;IAE/C,wEAAwE;IACxE,IAAI,EAAE,SAAS,CAAC;IAEhB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,eAAO,MAAM,4BAA4B,EAAE,CACzC,QAAQ,EAAE,mBAAmB,EAC7B,cAAc,EAAE,MAAM,EACtB,eAAe,EAAE,OAAO,EACxB,sBAAsB,EAAE,OAAO,EAC/B,iBAAiB,EAAE,MAAM,EACzB,cAAc,EAAE,MAAM,KACnB,aAAa,GAAG,aA+CpB,CAAC;AAEF,UAAU,wBAAwB;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE;QACV,KAAK,CAAC,EAAE;YACN,KAAK,EAAE,MAAM,EAAE,CAAC;SACjB,CAAC;QACF,IAAI,CAAC,EAAE;YACL,KAAK,EAAE,MAAM,EAAE,CAAC;SACjB,CAAC;QACF,KAAK,CAAC,EAAE;YACN,KAAK,EAAE,MAAM,EAAE,CAAC;SACjB,CAAC;QACF,MAAM,CAAC,EAAE;YACP,KAAK,EAAE,MAAM,EAAE,CAAC;SACjB,CAAC;QACF,OAAO,CAAC,EAAE;YACR,KAAK,EAAE,MAAM,EAAE,CAAC;SACjB,CAAC;KACH,CAAC;CACH;AAED,eAAO,MAAM,iCAAiC,GAAI,UAAU,wBAAwB;;;;;;;;;;;;;;;;;;;;CAwBnF,CAAC"}

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,3 @@
+import type { RequestHandler } from '../types';
+export declare const CohereGetBatchOutputHandler: RequestHandler;
+//# sourceMappingURL=getBatchOutput.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"getBatchOutput.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/getBatchOutput.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAK/C,eAAO,MAAM,2BAA2B,EAAE,cA4GzC,CAAC"}

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,5 @@
+import { ErrorResponse, GetFileResponse, GetFilesResponse } from '../types';
+import { CohereErrorResponse, CohereGetFileResponse, CohereGetFilesResponse } from './types';
+export declare const CohereGetFileResponseTransform: (response: CohereGetFileResponse | CohereErrorResponse, responseStatus: number) => GetFileResponse | ErrorResponse;
+export declare const CohereGetFilesResponseTransform: (response: CohereGetFilesResponse | CohereErrorResponse, responseStatus: number) => GetFilesResponse | ErrorResponse;
+//# sourceMappingURL=getFiles.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"getFiles.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/getFiles.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5E,OAAO,EAGL,mBAAmB,EACnB,qBAAqB,EACrB,sBAAsB,EACvB,MAAM,SAAS,CAAC;AA0BjB,eAAO,MAAM,8BAA8B,GACzC,UAAU,qBAAqB,GAAG,mBAAmB,EACrD,gBAAgB,MAAM,KACrB,eAAe,GAAG,aAUpB,CAAC;AAEF,eAAO,MAAM,+BAA+B,GAC1C,UAAU,sBAAsB,GAAG,mBAAmB,EACtD,gBAAgB,MAAM,KACrB,gBAAgB,GAAG,aAarB,CAAC"}

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,11 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. Add Rerank to the config and response transforms
+ * ================================
+ */
+import { ProviderConfigs } from '../types';
+declare const CohereConfig: ProviderConfigs;
+export default CohereConfig;
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAmB3C,QAAA,MAAM,YAAY,EAAE,eA8BnB,CAAC;AAEF,eAAe,YAAY,CAAC"}

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,4 @@
+import { ErrorResponse, ListBatchesResponse } from '../types';
+import { CohereErrorResponse, CohereListBatchResponse } from './types';
+export declare const CohereListBatchResponseTransform: (response: CohereListBatchResponse | CohereErrorResponse, responseStatus: number, _responseHeaders: Headers, strictOpenAiCompliance: boolean) => ListBatchesResponse | ErrorResponse;
+//# sourceMappingURL=listBatches.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"listBatches.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/listBatches.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AAE9D,OAAO,EAAE,mBAAmB,EAAE,uBAAuB,EAAE,MAAM,SAAS,CAAC;AAGvE,eAAO,MAAM,gCAAgC,EAAE,CAC7C,QAAQ,EAAE,uBAAuB,GAAG,mBAAmB,EACvD,cAAc,EAAE,MAAM,EACtB,gBAAgB,EAAE,OAAO,EACzB,sBAAsB,EAAE,OAAO,KAC5B,mBAAmB,GAAG,aAsB1B,CAAC"}

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,47 @@
+import { ErrorResponse, RerankResponse } from '../types';
+import { CohereErrorResponse } from './types';
+/**
+ * ================================
+ * 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 {};
+//# sourceMappingURL=rerank.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"rerank.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/rerank.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAG9C;;;;GAIG;AACH,eAAO,MAAM,kBAAkB;;;;;;;;;;;;;;;;;;;;;CAqB9B,CAAC;AAEF,UAAU,oBAAoB;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE;QACP,KAAK,EAAE,MAAM,CAAC;QACd,eAAe,EAAE,MAAM,CAAC;KACzB,EAAE,CAAC;IACJ,IAAI,EAAE;QACJ,WAAW,EAAE;YACX,OAAO,EAAE,MAAM,CAAC;SACjB,CAAC;QACF,YAAY,EAAE;YACZ,YAAY,EAAE,MAAM,CAAC;SACtB,CAAC;KACH,CAAC;CACH;AAED,eAAO,MAAM,6BAA6B,GACxC,UAAU,oBAAoB,GAAG,mBAAmB,EACpD,gBAAgB,MAAM,KACrB,cAAc,GAAG,aAYnB,CAAC"}

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