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

package/dist/providers/cohere/chatComplete.js

@@ -0,0 +1,315 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. Add tool_calls to assistant messages and tool_call_id to tool messages in message transform
+ * 2. Convert Cohere's single ToolCallV2 object to array format and add index from chunk level
+ * 3. Separate content index from tool call index handling
+ * 4. Remove 'any' types and use proper TypeScript type guards for tool calls
+ * 5. Transform response_format for structured output: Cohere V2 uses json_object with a schema field,
+ *    so OpenAI's json_schema type is unwrapped and converted to json_object.
+ *    Ref: https://docs.cohere.com/docs/structured-outputs#json-schema-mode
+ * ================================
+ */
+import { COHERE } from '../../globals.js';
+import { calculateBudgetTokens } from '../../utils/reasoningUtils.js';
+import { generateErrorResponse, generateInvalidProviderResponseError, transformFinishReason } from '../utils.js';
+// TODOS: this configuration does not enforce the maximum token limit for the input parameter. If you want to enforce this, you might need to add a custom validation function or a max property to the ParameterConfig interface, and then use it in the input configuration. However, this might be complex because the token count is not a simple length check, but depends on the specific tokenization method used by the model.
+export const CohereChatCompleteConfig = {
+    stream: {
+        param: 'stream',
+        default: false,
+    },
+    model: {
+        param: 'model',
+        required: false,
+    },
+    messages: {
+        param: 'messages',
+        required: true,
+        transform: (params) => {
+            return params.messages?.map(message => {
+                const role = message.role === 'developer' ? 'system' : message.role;
+                const transformedMessage = {
+                    role,
+                    content: message.content,
+                };
+                if (message.role === 'assistant' && message.tool_calls) {
+                    transformedMessage.tool_calls = message.tool_calls;
+                }
+                if (message.role === 'tool' && message.tool_call_id) {
+                    transformedMessage.tool_call_id = message.tool_call_id;
+                }
+                return transformedMessage;
+            });
+        },
+    },
+    max_tokens: {
+        param: 'max_tokens',
+        required: false,
+    },
+    stop: {
+        param: 'stop_sequences',
+        required: false,
+        transform: (params) => {
+            if (typeof params.stop === 'string') {
+                return [params.stop];
+            }
+            return params.stop;
+        },
+    },
+    temperature: {
+        param: 'temperature',
+        required: false,
+    },
+    seed: {
+        param: 'seed',
+        required: false,
+    },
+    frequency_penalty: {
+        param: 'frequency_penalty',
+        required: false,
+    },
+    presence_penalty: {
+        param: 'presence_penalty',
+        required: false,
+    },
+    response_format: [
+        {
+            param: 'response_format',
+            required: false,
+            transform: (params) => {
+                if (params.response_format?.type === 'json_schema' && params.response_format?.json_schema?.schema) {
+                    return {
+                        type: 'json_object',
+                        schema: params.response_format.json_schema.schema,
+                    };
+                }
+                return params.response_format;
+            },
+        },
+        {
+            param: 'strict_tools',
+            required: false,
+            transform: (params) => {
+                if (params.response_format?.type === 'json_schema') {
+                    return params.response_format?.json_schema?.strict;
+                }
+                return null;
+            },
+        },
+    ],
+    top_p: {
+        param: 'p',
+        required: false,
+    },
+    tools: {
+        param: 'tools',
+        required: false,
+        transform: (params) => {
+            if (!params.tools)
+                return params.tools;
+            return params.tools.map(tool => {
+                if (tool.type === 'function' && tool.function) {
+                    const { strict: _, ...functionWithoutStrict } = tool.function;
+                    return { ...tool, function: functionWithoutStrict };
+                }
+                return tool;
+            });
+        },
+    },
+    parallel_tool_calls: {
+        param: 'parallel_tool_calls',
+        removeFromParams: true,
+    },
+    reasoning_effort: {
+        param: 'thinking',
+        required: false,
+        transform: (params) => {
+            const effort = params.reasoning_effort;
+            if (!effort) {
+                return params.thinking;
+            }
+            if (effort === 'none') {
+                return { type: 'disabled' };
+            }
+            if (effort === 'high') {
+                return { type: 'enabled' };
+            }
+            const maxTokens = params.max_completion_tokens || params.max_tokens || 2048;
+            const budgetTokens = calculateBudgetTokens(maxTokens, effort);
+            return { token_budget: budgetTokens };
+        },
+    },
+    tool_choice: {
+        param: 'tool_choice',
+        required: false,
+        transform: (params) => {
+            if (typeof params.tool_choice === 'string') {
+                switch (params.tool_choice) {
+                    case 'required':
+                        return 'REQUIRED';
+                    case 'auto':
+                        return null;
+                    case 'none':
+                        return 'NONE';
+                }
+            }
+            return 'REQUIRED';
+        },
+    },
+    // cohere specific parameters
+    documents: {
+        param: 'documents',
+        required: false,
+    },
+    citation_options: {
+        param: 'citation_options',
+        required: false,
+    },
+    safety_mode: {
+        param: 'safety_mode',
+        required: false,
+    },
+    k: {
+        param: 'k',
+        required: false,
+    },
+    thinking: {
+        param: 'thinking',
+        required: false,
+    },
+    stream_options: {
+        param: 'stream_options',
+        removeFromParams: true,
+    },
+};
+export const CohereChatCompleteResponseTransform = (response, responseStatus, responseHeaders, strictOpenAiCompliance, _gatewayRequestUrl, gatewayRequest) => {
+    if (responseStatus !== 200 && 'message' in response && typeof response.message === 'string') {
+        return generateErrorResponse({
+            message: response.message || '',
+            type: null,
+            code: null,
+        }, COHERE);
+    }
+    if ('message' in response && 'usage' in response) {
+        const prompt_tokens = response.usage?.tokens?.input_tokens ?? response.usage?.billed_units?.input_tokens ?? 0;
+        const completion_tokens = response.usage?.tokens?.output_tokens ?? response.usage?.billed_units?.output_tokens ?? 0;
+        const total_tokens = prompt_tokens + completion_tokens;
+        return {
+            id: response.id,
+            model: gatewayRequest.model || '',
+            object: 'chat.completion',
+            created: Math.floor(Date.now() / 1000),
+            provider: COHERE,
+            choices: [
+                {
+                    index: 0,
+                    finish_reason: transformFinishReason(response.finish_reason, strictOpenAiCompliance),
+                    message: {
+                        role: 'assistant',
+                        content: response.message?.content?.reduce((acc, item) => {
+                            if (item.type === 'text') {
+                                acc += item.text;
+                            }
+                            return acc;
+                        }, '') ?? '',
+                        tool_calls: response.message.tool_calls,
+                    },
+                },
+            ],
+            usage: {
+                completion_tokens,
+                prompt_tokens,
+                total_tokens,
+            },
+        };
+    }
+    return generateInvalidProviderResponseError(response, COHERE);
+};
+export const CohereChatCompleteStreamChunkTransform = (responseChunk, fallbackId, streamState = { generation_id: '', lastIndex: 0 }, strictOpenAiCompliance, gatewayRequest) => {
+    let chunk = responseChunk.trim();
+    chunk = chunk.replace(/^event:.*[\r\n]*/, '');
+    chunk = chunk.replace(/^data: /, '');
+    chunk = chunk.trim();
+    if (chunk === '[DONE]') {
+        return 'data: [DONE]\n\n';
+    }
+    const parsedChunk = JSON.parse(chunk);
+    if (parsedChunk.type === 'message-start') {
+        streamState.generation_id = parsedChunk.id;
+    }
+    const model = gatewayRequest.model || '';
+    if (parsedChunk.type === 'message-end') {
+        const prompt_tokens = parsedChunk.delta?.usage?.tokens?.input_tokens ?? parsedChunk.delta?.usage?.billed_units?.input_tokens ?? 0;
+        const completion_tokens = parsedChunk.delta?.usage?.tokens?.output_tokens ?? parsedChunk.delta?.usage?.billed_units?.output_tokens ?? 0;
+        const total_tokens = prompt_tokens + completion_tokens;
+        const usage = {
+            completion_tokens,
+            prompt_tokens,
+            total_tokens,
+        };
+        return (`data: ${JSON.stringify({
+            id: streamState.generation_id,
+            object: 'chat.completion.chunk',
+            created: Math.floor(Date.now() / 1000),
+            model: model,
+            provider: COHERE,
+            choices: [
+                {
+                    index: streamState.lastIndex,
+                    delta: {},
+                    logprobs: null,
+                    finish_reason: transformFinishReason(parsedChunk.delta?.finish_reason, strictOpenAiCompliance),
+                },
+            ],
+            usage,
+        })}` + '\n\n');
+    }
+    if ('index' in parsedChunk &&
+        parsedChunk.index !== streamState.lastIndex &&
+        (parsedChunk.type === 'content-start' || parsedChunk.type === 'content-delta' || parsedChunk.type === 'content-end')) {
+        streamState.lastIndex = parsedChunk.index ?? 0;
+    }
+    let toolCallsFromCohere = undefined;
+    let toolCallIndex = undefined;
+    if (parsedChunk.type === 'tool-call-start') {
+        toolCallsFromCohere = parsedChunk.delta.message.tool_calls;
+        toolCallIndex = parsedChunk.index;
+    }
+    else if (parsedChunk.type === 'tool-call-delta') {
+        toolCallsFromCohere = parsedChunk.delta.message.tool_calls;
+        toolCallIndex = parsedChunk.index;
+    }
+    let toolCallsArray = undefined;
+    if (toolCallsFromCohere && toolCallIndex !== undefined) {
+        const rawToolCalls = Array.isArray(toolCallsFromCohere) ? toolCallsFromCohere : [toolCallsFromCohere];
+        toolCallsArray = rawToolCalls.map(tc => ({
+            ...tc,
+            index: toolCallIndex,
+        }));
+    }
+    return (`data: ${JSON.stringify({
+        id: streamState.generation_id,
+        object: 'chat.completion.chunk',
+        created: Math.floor(Date.now() / 1000),
+        model: model,
+        provider: COHERE,
+        system_fingerprint: null,
+        choices: [
+            {
+                index: streamState.lastIndex,
+                delta: {
+                    role: 'assistant',
+                    content: parsedChunk.type === 'content-start' || parsedChunk.type === 'content-delta'
+                        ? (parsedChunk.delta?.message?.content?.text ?? '')
+                        : '',
+                    tool_calls: toolCallsArray,
+                },
+                logprobs: null,
+                finish_reason: null,
+            },
+        ],
+    })}` + '\n\n');
+};
+//# sourceMappingURL=chatComplete.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"chatComplete.js","sourceRoot":"","sources":["../../../src/providers/cohere/chatComplete.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAEvC,OAAO,EAAE,qBAAqB,EAAmB,MAAM,4BAA4B,CAAC;AAEpF,OAAO,EAAE,qBAAqB,EAAE,oCAAoC,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAU9G,saAAsa;AAEta,MAAM,CAAC,MAAM,wBAAwB,GAAmB;IACtD,MAAM,EAAE;QACN,KAAK,EAAE,QAAQ;QACf,OAAO,EAAE,KAAK;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;KAChB;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,IAAI;QACd,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;YAC5B,OAAO,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE;gBACpC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;gBACpE,MAAM,kBAAkB,GAAQ;oBAC9B,IAAI;oBACJ,OAAO,EAAE,OAAO,CAAC,OAAO;iBACzB,CAAC;gBACF,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;oBACvD,kBAAkB,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;gBACrD,CAAC;gBACD,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;oBACpD,kBAAkB,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;gBACzD,CAAC;gBACD,OAAO,kBAAkB,CAAC;YAC5B,CAAC,CAAC,CAAC;QACL,CAAC;KACF;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,QAAQ,EAAE,KAAK;KAChB;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,gBAAgB;QACvB,QAAQ,EAAE,KAAK;QACf,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;YAC5B,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACpC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACvB,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC;QACrB,CAAC;KACF;IACD,WAAW,EAAE;QACX,KAAK,EAAE,aAAa;QACpB,QAAQ,EAAE,KAAK;KAChB;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;QACb,QAAQ,EAAE,KAAK;KAChB;IACD,iBAAiB,EAAE;QACjB,KAAK,EAAE,mBAAmB;QAC1B,QAAQ,EAAE,KAAK;KAChB;IACD,gBAAgB,EAAE;QAChB,KAAK,EAAE,kBAAkB;QACzB,QAAQ,EAAE,KAAK;KAChB;IACD,eAAe,EAAE;QACf;YACE,KAAK,EAAE,iBAAiB;YACxB,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;gBAC5B,IAAI,MAAM,CAAC,eAAe,EAAE,IAAI,KAAK,aAAa,IAAI,MAAM,CAAC,eAAe,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;oBAClG,OAAO;wBACL,IAAI,EAAE,aAAa;wBACnB,MAAM,EAAE,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,MAAM;qBAClD,CAAC;gBACJ,CAAC;gBACD,OAAO,MAAM,CAAC,eAAe,CAAC;YAChC,CAAC;SACF;QACD;YACE,KAAK,EAAE,cAAc;YACrB,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;gBAC5B,IAAI,MAAM,CAAC,eAAe,EAAE,IAAI,KAAK,aAAa,EAAE,CAAC;oBACnD,OAAO,MAAM,CAAC,eAAe,EAAE,WAAW,EAAE,MAAM,CAAC;gBACrD,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;SACF;KACF;IACD,KAAK,EAAE;QACL,KAAK,EAAE,GAAG;QACV,QAAQ,EAAE,KAAK;KAChB;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;QACf,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;YAC5B,IAAI,CAAC,MAAM,CAAC,KAAK;gBAAE,OAAO,MAAM,CAAC,KAAK,CAAC;YACvC,OAAO,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;gBAC7B,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;oBAC9C,MAAM,EAAE,MAAM,EAAE,CAAC,EAAE,GAAG,qBAAqB,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC;oBAC9D,OAAO,EAAE,GAAG,IAAI,EAAE,QAAQ,EAAE,qBAAqB,EAAE,CAAC;gBACtD,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC,CAAC,CAAC;QACL,CAAC;KACF;IACD,mBAAmB,EAAE;QACnB,KAAK,EAAE,qBAAqB;QAC5B,gBAAgB,EAAE,IAAI;KACvB;IACD,gBAAgB,EAAE;QAChB,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,KAAK;QACf,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;YAC5B,MAAM,MAAM,GAAG,MAAM,CAAC,gBAA+C,CAAC;YACtE,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,MAAM,CAAC,QAAQ,CAAC;YACzB,CAAC;YACD,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;gBACtB,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YAC9B,CAAC;YACD,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;gBACtB,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YAC7B,CAAC;YACD,MAAM,SAAS,GAAG,MAAM,CAAC,qBAAqB,IAAI,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC;YAC5E,MAAM,YAAY,GAAG,qBAAqB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YAC9D,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC;QACxC,CAAC;KACF;IACD,WAAW,EAAE;QACX,KAAK,EAAE,aAAa;QACpB,QAAQ,EAAE,KAAK;QACf,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;YAC5B,IAAI,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;gBAC3C,QAAQ,MAAM,CAAC,WAAW,EAAE,CAAC;oBAC3B,KAAK,UAAU;wBACb,OAAO,UAAU,CAAC;oBACpB,KAAK,MAAM;wBACT,OAAO,IAAI,CAAC;oBACd,KAAK,MAAM;wBACT,OAAO,MAAM,CAAC;gBAClB,CAAC;YACH,CAAC;YACD,OAAO,UAAU,CAAC;QACpB,CAAC;KACF;IACD,6BAA6B;IAC7B,SAAS,EAAE;QACT,KAAK,EAAE,WAAW;QAClB,QAAQ,EAAE,KAAK;KAChB;IACD,gBAAgB,EAAE;QAChB,KAAK,EAAE,kBAAkB;QACzB,QAAQ,EAAE,KAAK;KAChB;IACD,WAAW,EAAE;QACX,KAAK,EAAE,aAAa;QACpB,QAAQ,EAAE,KAAK;KAChB;IACD,CAAC,EAAE;QACD,KAAK,EAAE,GAAG;QACV,QAAQ,EAAE,KAAK;KAChB;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,KAAK;KAChB;IACD,cAAc,EAAE;QACd,KAAK,EAAE,gBAAgB;QACvB,gBAAgB,EAAE,IAAI;KACvB;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,mCAAmC,GAOF,CAC5C,QAAQ,EACR,cAAc,EACd,eAAe,EACf,sBAAsB,EACtB,kBAAkB,EAClB,cAAc,EACd,EAAE;IACF,IAAI,cAAc,KAAK,GAAG,IAAI,SAAS,IAAI,QAAQ,IAAI,OAAO,QAAQ,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QAC5F,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,IAAI,SAAS,IAAI,QAAQ,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;QACjD,MAAM,aAAa,GAAG,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,YAAY,IAAI,QAAQ,CAAC,KAAK,EAAE,YAAY,EAAE,YAAY,IAAI,CAAC,CAAC;QAC9G,MAAM,iBAAiB,GAAG,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,aAAa,IAAI,QAAQ,CAAC,KAAK,EAAE,YAAY,EAAE,aAAa,IAAI,CAAC,CAAC;QACpH,MAAM,YAAY,GAAG,aAAa,GAAG,iBAAiB,CAAC;QACvD,OAAO;YACL,EAAE,EAAE,QAAQ,CAAC,EAAE;YACf,KAAK,EAAE,cAAc,CAAC,KAAK,IAAI,EAAE;YACjC,MAAM,EAAE,iBAAiB;YACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YACtC,QAAQ,EAAE,MAAM;YAChB,OAAO,EAAE;gBACP;oBACE,KAAK,EAAE,CAAC;oBACR,aAAa,EAAE,qBAAqB,CAAC,QAAQ,CAAC,aAAmC,EAAE,sBAAsB,CAAC;oBAC1G,OAAO,EAAE;wBACP,IAAI,EAAE,WAAW;wBACjB,OAAO,EACL,QAAQ,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;4BAC9C,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gCACzB,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC;4BACnB,CAAC;4BACD,OAAO,GAAG,CAAC;wBACb,CAAC,EAAE,EAAE,CAAC,IAAI,EAAE;wBACd,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,UAAU;qBACxC;iBACF;aACF;YACD,KAAK,EAAE;gBACL,iBAAiB;gBACjB,aAAa;gBACb,YAAY;aACb;SACF,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AAChE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,sCAAsC,GAMrC,CACZ,aAAa,EACb,UAAU,EACV,WAAW,GAAG,EAAE,aAAa,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,EACjD,sBAAsB,EACtB,cAAc,EACd,EAAE;IACF,IAAI,KAAK,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC;IACjC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;IAC9C,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;IACrC,KAAK,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAErB,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;QACvB,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,MAAM,WAAW,GAAoC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAEvE,IAAI,WAAW,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;QACzC,WAAW,CAAC,aAAa,GAAG,WAAW,CAAC,EAAE,CAAC;IAC7C,CAAC;IACD,MAAM,KAAK,GAAG,cAAc,CAAC,KAAK,IAAI,EAAE,CAAC;IAEzC,IAAI,WAAW,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;QACvC,MAAM,aAAa,GACjB,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,YAAY,IAAI,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,YAAY,IAAI,CAAC,CAAC;QAC9G,MAAM,iBAAiB,GACrB,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,aAAa,IAAI,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,aAAa,IAAI,CAAC,CAAC;QAChH,MAAM,YAAY,GAAG,aAAa,GAAG,iBAAiB,CAAC;QACvD,MAAM,KAAK,GAAG;YACZ,iBAAiB;YACjB,aAAa;YACb,YAAY;SACb,CAAC;QACF,OAAO,CACL,SAAS,IAAI,CAAC,SAAS,CAAC;YACtB,EAAE,EAAE,WAAW,CAAC,aAAa;YAC7B,MAAM,EAAE,uBAAuB;YAC/B,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YACtC,KAAK,EAAE,KAAK;YACZ,QAAQ,EAAE,MAAM;YAChB,OAAO,EAAE;gBACP;oBACE,KAAK,EAAE,WAAW,CAAC,SAAS;oBAC5B,KAAK,EAAE,EAAE;oBACT,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,qBAAqB,CAAC,WAAW,CAAC,KAAK,EAAE,aAAa,EAAE,sBAAsB,CAAC;iBAC/F;aACF;YACD,KAAK;SACN,CAAC,EAAE,GAAG,MAAM,CACd,CAAC;IACJ,CAAC;IAED,IACE,OAAO,IAAI,WAAW;QACtB,WAAW,CAAC,KAAK,KAAK,WAAW,CAAC,SAAS;QAC3C,CAAC,WAAW,CAAC,IAAI,KAAK,eAAe,IAAI,WAAW,CAAC,IAAI,KAAK,eAAe,IAAI,WAAW,CAAC,IAAI,KAAK,aAAa,CAAC,EACpH,CAAC;QACD,WAAW,CAAC,SAAS,GAAG,WAAW,CAAC,KAAK,IAAI,CAAC,CAAC;IACjD,CAAC;IAED,IAAI,mBAAmB,GAA2B,SAAS,CAAC;IAC5D,IAAI,aAAa,GAAuB,SAAS,CAAC;IAClD,IAAI,WAAW,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;QAC3C,mBAAmB,GAAG,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC;QAC3D,aAAa,GAAG,WAAW,CAAC,KAAK,CAAC;IACpC,CAAC;SAAM,IAAI,WAAW,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;QAClD,mBAAmB,GAAG,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC;QAC3D,aAAa,GAAG,WAAW,CAAC,KAAK,CAAC;IACpC,CAAC;IAED,IAAI,cAAc,GAAsD,SAAS,CAAC;IAClF,IAAI,mBAAmB,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;QACvD,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC;QACtG,cAAc,GAAG,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACvC,GAAG,EAAE;YACL,KAAK,EAAE,aAAa;SACrB,CAAC,CAAC,CAAC;IACN,CAAC;IAED,OAAO,CACL,SAAS,IAAI,CAAC,SAAS,CAAC;QACtB,EAAE,EAAE,WAAW,CAAC,aAAa;QAC7B,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QACtC,KAAK,EAAE,KAAK;QACZ,QAAQ,EAAE,MAAM;QAChB,kBAAkB,EAAE,IAAI;QACxB,OAAO,EAAE;YACP;gBACE,KAAK,EAAE,WAAW,CAAC,SAAS;gBAC5B,KAAK,EAAE;oBACL,IAAI,EAAE,WAAW;oBACjB,OAAO,EACL,WAAW,CAAC,IAAI,KAAK,eAAe,IAAI,WAAW,CAAC,IAAI,KAAK,eAAe;wBAC1E,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC;wBACnD,CAAC,CAAC,EAAE;oBACR,UAAU,EAAE,cAAc;iBAC3B;gBACD,QAAQ,EAAE,IAAI;gBACd,aAAa,EAAE,IAAI;aACpB;SACF;KACF,CAAC,EAAE,GAAG,MAAM,CACd,CAAC;AACJ,CAAC,CAAC"}

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

@@ -0,0 +1,40 @@
+import { CompletionResponse, ErrorResponse, ProviderConfig } from '../types';
+export declare const CohereCompleteConfig: ProviderConfig;
+interface CohereCompleteResponse {
+    id: string;
+    generations: {
+        id: string;
+        text: string;
+    }[];
+    prompt: string;
+    meta: {
+        api_version: {
+            version: string;
+        };
+    };
+    message?: string;
+    status?: number;
+}
+export interface CohereStreamChunk {
+    id?: string;
+    response: {
+        generations?: {
+            id: string;
+            text: string;
+            finish_reason: boolean;
+        }[];
+    };
+    prompt?: string;
+    meta?: {
+        api_version: {
+            version: string;
+        };
+    };
+    text: string;
+    is_finished: boolean;
+    index?: number;
+}
+export declare const CohereCompleteResponseTransform: (response: CohereCompleteResponse, responseStatus: number) => CompletionResponse | ErrorResponse;
+export declare const CohereCompleteStreamChunkTransform: (response: string, fallbackId: string, _streamState: Record<string, boolean>, _strictOpenAiCompliance: boolean, gatewayRequest: Params) => string;
+export {};
+//# sourceMappingURL=complete.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"complete.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/complete.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAK7E,eAAO,MAAM,oBAAoB,EAAE,cA2DlC,CAAC;AAEF,UAAU,sBAAsB;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE;QACX,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;KACd,EAAE,CAAC;IACJ,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE;QACJ,WAAW,EAAE;YACX,OAAO,EAAE,MAAM,CAAC;SACjB,CAAC;KACH,CAAC;IACF,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,iBAAiB;IAChC,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE;QACR,WAAW,CAAC,EAAE;YACZ,EAAE,EAAE,MAAM,CAAC;YACX,IAAI,EAAE,MAAM,CAAC;YACb,aAAa,EAAE,OAAO,CAAC;SACxB,EAAE,CAAC;KACL,CAAC;IACF,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE;QACL,WAAW,EAAE;YACX,OAAO,EAAE,MAAM,CAAC;SACjB,CAAC;KACH,CAAC;IACF,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,OAAO,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,eAAO,MAAM,+BAA+B,EAAE,CAC5C,QAAQ,EAAE,sBAAsB,EAChC,cAAc,EAAE,MAAM,KACnB,kBAAkB,GAAG,aAyBzB,CAAC;AAEF,eAAO,MAAM,kCAAkC,EAAE,CAC/C,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,MA4BJ,CAAC"}

package/dist/providers/cohere/complete.js

@@ -0,0 +1,116 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { COHERE } from '../../globals.js';
+import { generateErrorResponse } from '../utils.js';
+// TODOS: this configuration does not enforce the maximum token limit for the input parameter. If you want to enforce this, you might need to add a custom validation function or a max property to the ParameterConfig interface, and then use it in the input configuration. However, this might be complex because the token count is not a simple length check, but depends on the specific tokenization method used by the model.
+export const CohereCompleteConfig = {
+    model: {
+        param: 'model',
+        default: 'command',
+        required: true,
+    },
+    prompt: {
+        param: 'prompt',
+        required: true,
+    },
+    max_tokens: {
+        param: 'max_tokens',
+        min: 1,
+    },
+    temperature: {
+        param: 'temperature',
+        default: 0.75,
+        min: 0,
+        max: 5,
+    },
+    top_p: {
+        param: 'p',
+        default: 0.75,
+        min: 0,
+        max: 1,
+    },
+    top_k: {
+        param: 'k',
+        default: 0,
+        max: 500,
+    },
+    frequency_penalty: {
+        param: 'frequency_penalty',
+        default: 0,
+        min: 0,
+        max: 1,
+    },
+    presence_penalty: {
+        param: 'presence_penalty',
+        default: 0,
+        min: 0,
+        max: 1,
+    },
+    logit_bias: {
+        param: 'logit_bias',
+    },
+    n: {
+        param: 'num_generations',
+        default: 1,
+        min: 1,
+        max: 5,
+    },
+    stop: {
+        param: 'end_sequences',
+    },
+    stream: {
+        param: 'stream',
+        default: false,
+    },
+};
+export const CohereCompleteResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200) {
+        return generateErrorResponse({
+            message: response.message || '',
+            type: null,
+            code: null,
+        }, COHERE);
+    }
+    return {
+        id: response.id,
+        object: 'text_completion',
+        created: Math.floor(Date.now() / 1000),
+        model: 'Unknown',
+        provider: COHERE,
+        choices: response.generations.map((generation, index) => ({
+            text: generation.text,
+            index: index,
+            logprobs: null,
+            finish_reason: 'length',
+        })),
+    };
+};
+export const CohereCompleteStreamChunkTransform = (responseChunk, fallbackId, _streamState, _strictOpenAiCompliance, gatewayRequest) => {
+    let chunk = responseChunk.trim();
+    chunk = chunk.replace(/^data: /, '');
+    chunk = chunk.trim();
+    const parsedChunk = JSON.parse(chunk);
+    // discard the last cohere chunk as it sends the whole response combined.
+    if (parsedChunk.is_finished) {
+        return '';
+    }
+    return (`data: ${JSON.stringify({
+        id: parsedChunk.id ?? fallbackId,
+        object: 'text_completion',
+        created: Math.floor(Date.now() / 1000),
+        model: gatewayRequest.model || '',
+        provider: COHERE,
+        choices: [
+            {
+                text: parsedChunk.response?.generations?.[0]?.text ?? parsedChunk.text,
+                index: parsedChunk.index ?? 0,
+                logprobs: null,
+                finish_reason: parsedChunk.response?.generations?.[0]?.finish_reason ?? null,
+            },
+        ],
+    })}` + '\n\n');
+};
+//# sourceMappingURL=complete.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"complete.js","sourceRoot":"","sources":["../../../src/providers/cohere/complete.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAEvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAEjD,saAAsa;AAEta,MAAM,CAAC,MAAM,oBAAoB,GAAmB;IAClD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,SAAS;QAClB,QAAQ,EAAE,IAAI;KACf;IACD,MAAM,EAAE;QACN,KAAK,EAAE,QAAQ;QACf,QAAQ,EAAE,IAAI;KACf;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,GAAG,EAAE,CAAC;KACP;IACD,WAAW,EAAE;QACX,KAAK,EAAE,aAAa;QACpB,OAAO,EAAE,IAAI;QACb,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,KAAK,EAAE;QACL,KAAK,EAAE,GAAG;QACV,OAAO,EAAE,IAAI;QACb,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,KAAK,EAAE;QACL,KAAK,EAAE,GAAG;QACV,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,GAAG;KACT;IACD,iBAAiB,EAAE;QACjB,KAAK,EAAE,mBAAmB;QAC1B,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,gBAAgB,EAAE;QAChB,KAAK,EAAE,kBAAkB;QACzB,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;KACpB;IACD,CAAC,EAAE;QACD,KAAK,EAAE,iBAAiB;QACxB,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,eAAe;KACvB;IACD,MAAM,EAAE;QACN,KAAK,EAAE,QAAQ;QACf,OAAO,EAAE,KAAK;KACf;CACF,CAAC;AAsCF,MAAM,CAAC,MAAM,+BAA+B,GAGF,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IACrE,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,OAAO;QACL,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,MAAM,EAAE,iBAAiB;QACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QACtC,KAAK,EAAE,SAAS;QAChB,QAAQ,EAAE,MAAM;QAChB,OAAO,EAAE,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YACxD,IAAI,EAAE,UAAU,CAAC,IAAI;YACrB,KAAK,EAAE,KAAK;YACZ,QAAQ,EAAE,IAAI;YACd,aAAa,EAAE,QAAQ;SACxB,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,kCAAkC,GAMjC,CAAC,aAAa,EAAE,UAAU,EAAE,YAAY,EAAE,uBAAuB,EAAE,cAAc,EAAE,EAAE;IACjG,IAAI,KAAK,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC;IACjC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;IACrC,KAAK,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IACrB,MAAM,WAAW,GAAsB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAEzD,yEAAyE;IACzE,IAAI,WAAW,CAAC,WAAW,EAAE,CAAC;QAC5B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,CACL,SAAS,IAAI,CAAC,SAAS,CAAC;QACtB,EAAE,EAAE,WAAW,CAAC,EAAE,IAAI,UAAU;QAChC,MAAM,EAAE,iBAAiB;QACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QACtC,KAAK,EAAE,cAAc,CAAC,KAAK,IAAI,EAAE;QACjC,QAAQ,EAAE,MAAM;QAChB,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,WAAW,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,WAAW,CAAC,IAAI;gBACtE,KAAK,EAAE,WAAW,CAAC,KAAK,IAAI,CAAC;gBAC7B,QAAQ,EAAE,IAAI;gBACd,aAAa,EAAE,WAAW,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,IAAI;aAC7E;SACF;KACF,CAAC,EAAE,GAAG,MAAM,CACd,CAAC;AACJ,CAAC,CAAC"}

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

@@ -0,0 +1,5 @@
+import { CreateBatchResponse, ErrorResponse, ProviderConfig } from '../types';
+import { CohereCreateBatchResponse, CohereErrorResponse } from './types';
+export declare const CohereCreateBatchConfig: ProviderConfig;
+export declare const CohereCreateBatchResponseTransform: (response: CohereCreateBatchResponse | CohereErrorResponse, responseStatus: number) => CreateBatchResponse | ErrorResponse;
+//# sourceMappingURL=createBatch.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"createBatch.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/createBatch.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,mBAAmB,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE9E,OAAO,EAAE,yBAAyB,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAGzE,eAAO,MAAM,uBAAuB,EAAE,cAyBrC,CAAC;AAEF,eAAO,MAAM,kCAAkC,EAAE,CAC/C,QAAQ,EAAE,yBAAyB,GAAG,mBAAmB,EACzD,cAAc,EAAE,MAAM,KACnB,mBAAmB,GAAG,aAiB1B,CAAC"}

package/dist/providers/cohere/createBatch.js

@@ -0,0 +1,51 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { COHERE } from '../../globals.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+import { CohereErrorResponseTransform } from './utils.js';
+export const CohereCreateBatchConfig = {
+    model: {
+        param: 'model',
+        required: true,
+    },
+    input_file_id: {
+        param: 'dataset_id',
+        required: true,
+    },
+    input_type: {
+        param: 'input_type',
+        required: false,
+    },
+    name: {
+        param: 'name',
+        required: false,
+    },
+    embedding_types: {
+        param: 'embedding_types',
+        required: false,
+    },
+    truncate: {
+        param: 'truncate',
+        required: false,
+    },
+};
+export const CohereCreateBatchResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200 && 'message' in response) {
+        return CohereErrorResponseTransform(response);
+    }
+    if ('job_id' in response) {
+        return {
+            id: response.job_id,
+            object: 'batch',
+            endpoint: '/v1/embed',
+            status: 'in_progress',
+            created_at: Math.floor(Date.now() / 1000),
+            meta: response.meta,
+        };
+    }
+    return generateInvalidProviderResponseError(response, COHERE);
+};
+//# sourceMappingURL=createBatch.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"createBatch.js","sourceRoot":"","sources":["../../../src/providers/cohere/createBatch.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,uBAAuB,GAAmB;IACrD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;KACf;IACD,aAAa,EAAE;QACb,KAAK,EAAE,YAAY;QACnB,QAAQ,EAAE,IAAI;KACf;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,QAAQ,EAAE,KAAK;KAChB;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;QACb,QAAQ,EAAE,KAAK;KAChB;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,KAAK;KAChB;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,KAAK;KAChB;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,kCAAkC,GAGJ,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IACtE,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,QAAQ,EAAE,WAAW;YACrB,MAAM,EAAE,aAAa;YACrB,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YACzC,IAAI,EAAE,QAAQ,CAAC,IAAI;SACpB,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AAChE,CAAC,CAAC"}

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

@@ -0,0 +1,9 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { DeleteFileResponse, ErrorResponse } from '../types';
+import { CohereErrorResponse } from './types';
+export declare const CohereDeleteFileResponseTransform: (response: Response | CohereErrorResponse, responseStatus: number, _responseHeaders: Record<string, string>, _strictOpenAiCompliance: boolean, gatewayRequestUrl: string) => DeleteFileResponse | ErrorResponse;
+//# sourceMappingURL=deleteFile.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"deleteFile.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/deleteFile.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAC7D,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAG9C,eAAO,MAAM,iCAAiC,EAAE,CAC9C,QAAQ,EAAE,QAAQ,GAAG,mBAAmB,EACxC,cAAc,EAAE,MAAM,EACtB,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EACxC,uBAAuB,EAAE,OAAO,EAChC,iBAAiB,EAAE,MAAM,KACtB,kBAAkB,GAAG,aAgBzB,CAAC"}

package/dist/providers/cohere/deleteFile.js

@@ -0,0 +1,13 @@
+import { CohereErrorResponseTransform } from './utils.js';
+export const CohereDeleteFileResponseTransform = (response, responseStatus, _responseHeaders, _strictOpenAiCompliance, gatewayRequestUrl) => {
+    if (responseStatus !== 200 && 'message' in response) {
+        return CohereErrorResponseTransform(response);
+    }
+    const id = gatewayRequestUrl.split('/').pop() || '';
+    return {
+        object: 'file',
+        deleted: true,
+        id,
+    };
+};
+//# sourceMappingURL=deleteFile.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"deleteFile.js","sourceRoot":"","sources":["../../../src/providers/cohere/deleteFile.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAEvD,MAAM,CAAC,MAAM,iCAAiC,GAMJ,CACxC,QAAQ,EACR,cAAc,EACd,gBAAgB,EAChB,uBAAuB,EACvB,iBAAiB,EACjB,EAAE;IACF,IAAI,cAAc,KAAK,GAAG,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;QACpD,OAAO,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IACD,MAAM,EAAE,GAAG,iBAAiB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;IACpD,OAAO;QACL,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,IAAI;QACb,EAAE;KACH,CAAC;AACJ,CAAC,CAAC"}