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

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveBatch.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/retrieveBatch.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAEhE,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,MAAM,SAAS,CAAC;AAG3E,eAAO,MAAM,oCAAoC,EAAE,CACjD,QAAQ,EAAE,2BAA2B,GAAG,mBAAmB,EAC3D,cAAc,EAAE,MAAM,EACtB,gBAAgB,EAAE,OAAO,EACzB,sBAAsB,EAAE,OAAO,KAC5B,qBAAqB,GAAG,aAyB5B,CAAC"}

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"}

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

@@ -0,0 +1,265 @@
+export type CohereStreamState = {
+    generation_id: string;
+    lastIndex: number;
+};
+export interface CohereErrorResponse {
+    message: string;
+}
+export type CohereDatasetUploadStatus = string | 'unknown' | 'queued' | 'processing' | 'failed' | 'validated' | 'skipped';
+export interface CohereDataset {
+    id: string;
+    name: string;
+    created_at: string;
+    updated_at: string;
+    dataset_type: string | 'embed-input' | 'reranker-finetune-input' | 'single-label-classification-finetune-input' | 'chat-finetune-input' | 'multi-label-classification-finetune-input';
+    validation_status: CohereDatasetUploadStatus;
+    validation_error: string;
+    schema: string;
+    required_fields: string[];
+    preserve_fields: string[];
+    dataset_parts: {
+        id: string;
+        name: string;
+        url: string;
+        size_bytes: number;
+        num_rows: number;
+        samples: string[];
+    }[];
+    validation_warnings: string[];
+}
+export interface CohereGetFileResponse {
+    dataset: CohereDataset;
+}
+export interface CohereGetFilesResponse {
+    datasets: CohereDataset[];
+}
+interface CohereBatchMeta {
+    api_version: {
+        version: string;
+        is_deprecated: boolean;
+        is_experimental: boolean;
+    };
+    billed_units: {
+        images: number;
+        input_tokens: number;
+        output_tokens: number;
+        search_units: number;
+        classifications: number;
+    };
+    tokens: {
+        input_tokens: number;
+        output_tokens: number;
+    };
+    warnings: string[];
+}
+export interface CohereCreateBatchResponse {
+    job_id: string;
+    meta: CohereBatchMeta;
+}
+export interface CohereBatch {
+    job_id: string;
+    status: string | 'processing' | 'complete' | 'cancelling' | 'cancelled' | 'failed';
+    created_at: string;
+    input_dataset_id: string;
+    model: string;
+    truncate: string | 'START' | 'END';
+    name: string;
+    output_dataset_id: string;
+    meta: CohereBatchMeta;
+}
+export interface CohereListBatchResponse {
+    embed_jobs: CohereBatch[];
+}
+export interface CohereRetrieveBatchResponse extends CohereBatch {
+}
+export declare enum COHERE_STOP_REASON {
+    complete = "COMPLETE",
+    stop_sequence = "STOP_SEQUENCE",
+    max_tokens = "MAX_TOKENS",
+    tool_call = "TOOL_CALL",
+    error = "ERROR",
+    timeout = "TIMEOUT"
+}
+export type CohereChatCompletionStreamChunk = V2ChatStreamResponse.MessageStart | V2ChatStreamResponse.ContentStart | V2ChatStreamResponse.ContentDelta | V2ChatStreamResponse.ContentEnd | V2ChatStreamResponse.ToolPlanDelta | V2ChatStreamResponse.ToolCallStart | V2ChatStreamResponse.ToolCallDelta | V2ChatStreamResponse.ToolCallEnd | V2ChatStreamResponse.CitationStart | V2ChatStreamResponse.CitationEnd | V2ChatStreamResponse.MessageEnd | V2ChatStreamResponse.Debug;
+type ChatContentStartEventDeltaMessageContentType = 'text' | 'thinking';
+export interface LogprobItem {
+    /** The text chunk for which the log probabilities was calculated. */
+    text?: string;
+    /** The token ids of each token used to construct the text chunk. */
+    tokenIds: number[];
+    /** The log probability of each token used to construct the text chunk. */
+    logprobs?: number[];
+}
+export interface ToolCallV2Function {
+    name?: string;
+    arguments?: string;
+}
+export interface ToolCallV2 {
+    id?: string;
+    type?: 'function';
+    function?: ToolCallV2Function;
+}
+export interface UsageBilledUnits {
+    /** The number of billed input tokens. */
+    input_tokens?: number;
+    /** The number of billed output tokens. */
+    output_tokens?: number;
+    /** The number of billed search units. */
+    search_units?: number;
+    /** The number of billed classifications units. */
+    classifications_units?: number;
+}
+export interface UsageTokens {
+    /** The number of tokens used as input to the model. */
+    input_tokens?: number;
+    /** The number of tokens produced by the model. */
+    output_tokens?: number;
+}
+export interface Usage {
+    billed_units?: UsageBilledUnits;
+    tokens?: UsageTokens;
+}
+export interface Citation {
+    /** Start index of the cited snippet in the original source text. */
+    start?: number;
+    /** End index of the cited snippet in the original source text. */
+    end?: number;
+    /** Text snippet that is being cited. */
+    text?: string;
+    sources?: any;
+    /** Index of the content block in which this citation appears. */
+    content_index?: number;
+    type?: any;
+}
+declare namespace V2ChatStreamResponse {
+    interface MessageStart {
+        type: 'message-start';
+        id: string;
+        delta?: {
+            message: {
+                role: 'assistant';
+            };
+        };
+    }
+    interface ContentStart {
+        type: 'content-start';
+        index: number;
+        delta?: {
+            message: {
+                content: {
+                    thinking?: string;
+                    text?: string;
+                    type?: ChatContentStartEventDeltaMessageContentType;
+                };
+            };
+        };
+    }
+    interface ContentDelta {
+        type: 'content-delta';
+        index: number;
+        delta?: {
+            message: {
+                content: {
+                    thinking?: string;
+                    text?: string;
+                };
+            };
+        };
+        logprobs?: LogprobItem;
+    }
+    interface ContentEnd {
+        type: 'content-end';
+        index?: number;
+    }
+    interface ToolPlanDelta {
+        type: 'tool-plan-delta';
+        index: number;
+        delta: {
+            message: {
+                tool_plan: string;
+            };
+        };
+    }
+    interface ToolCallStart {
+        type: 'tool-call-start';
+        index: number;
+        delta: {
+            message: {
+                tool_calls: ToolCallV2;
+            };
+        };
+    }
+    interface ToolCallDelta {
+        type: 'tool-call-delta';
+        index: number;
+        delta: {
+            message: {
+                tool_calls: ToolCallV2;
+            };
+        };
+    }
+    interface ToolCallEnd {
+        type: 'tool-call-end';
+        index: number;
+    }
+    interface CitationStart {
+        type: 'citation-start';
+        index: number;
+        delta?: {
+            message?: {
+                citations: Citation;
+            };
+        };
+    }
+    interface CitationEnd {
+        type: 'citation-end';
+        index: number;
+    }
+    interface MessageEnd {
+        type: 'message-end';
+        id?: string;
+        delta?: {
+            error?: string;
+            finish_reason?: COHERE_STOP_REASON;
+            usage?: Usage;
+        };
+    }
+    interface Debug {
+        type: 'debug';
+        prompt?: string;
+    }
+}
+export interface CohereChatCompleteResponse {
+    id: string;
+    finish_reason: string;
+    message: {
+        role: 'assistant';
+        tool_calls: any[];
+        tool_plan: string;
+        content: {
+            type: 'text';
+            text: string;
+        }[] | {
+            thinking: string;
+            type: 'thinking';
+        }[];
+        citations: any;
+    };
+    usage: {
+        billed_units?: {
+            input_tokens?: number;
+            output_tokens?: number;
+        };
+        tokens?: {
+            input_tokens?: number;
+            output_tokens?: number;
+        };
+        cached_tokens?: number;
+    };
+}
+export interface CohereErrorResponse {
+    message: string;
+    id: string;
+}
+export {};
+//# sourceMappingURL=types.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,iBAAiB,GAAG;IAC9B,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,MAAM,yBAAyB,GACjC,MAAM,GACN,SAAS,GACT,QAAQ,GACR,YAAY,GACZ,QAAQ,GACR,WAAW,GACX,SAAS,CAAC;AAEd,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EACR,MAAM,GACN,aAAa,GACb,yBAAyB,GACzB,4CAA4C,GAC5C,qBAAqB,GACrB,2CAA2C,CAAC;IAChD,iBAAiB,EAAE,yBAAyB,CAAC;IAC7C,gBAAgB,EAAE,MAAM,CAAC;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,aAAa,EAAE;QACb,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,GAAG,EAAE,MAAM,CAAC;QACZ,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,EAAE,CAAC;KACnB,EAAE,CAAC;IACJ,mBAAmB,EAAE,MAAM,EAAE,CAAC;CAC/B;AAED,MAAM,WAAW,qBAAqB;IACpC,OAAO,EAAE,aAAa,CAAC;CACxB;AAED,MAAM,WAAW,sBAAsB;IACrC,QAAQ,EAAE,aAAa,EAAE,CAAC;CAC3B;AAED,UAAU,eAAe;IACvB,WAAW,EAAE;QACX,OAAO,EAAE,MAAM,CAAC;QAChB,aAAa,EAAE,OAAO,CAAC;QACvB,eAAe,EAAE,OAAO,CAAC;KAC1B,CAAC;IACF,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,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB;AAED,MAAM,WAAW,yBAAyB;IACxC,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,eAAe,CAAC;CACvB;AAED,MAAM,WAAW,WAAW;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,GAAG,YAAY,GAAG,UAAU,GAAG,YAAY,GAAG,WAAW,GAAG,QAAQ,CAAC;IACnF,UAAU,EAAE,MAAM,CAAC;IACnB,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,GAAG,OAAO,GAAG,KAAK,CAAC;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,iBAAiB,EAAE,MAAM,CAAC;IAC1B,IAAI,EAAE,eAAe,CAAC;CACvB;AAED,MAAM,WAAW,uBAAuB;IACtC,UAAU,EAAE,WAAW,EAAE,CAAC;CAC3B;AAED,MAAM,WAAW,2BAA4B,SAAQ,WAAW;CAAG;AAEnE,oBAAY,kBAAkB;IAC5B,QAAQ,aAAa;IACrB,aAAa,kBAAkB;IAC/B,UAAU,eAAe;IACzB,SAAS,cAAc;IACvB,KAAK,UAAU;IACf,OAAO,YAAY;CACpB;AAED,MAAM,MAAM,+BAA+B,GACvC,oBAAoB,CAAC,YAAY,GACjC,oBAAoB,CAAC,YAAY,GACjC,oBAAoB,CAAC,YAAY,GACjC,oBAAoB,CAAC,UAAU,GAC/B,oBAAoB,CAAC,aAAa,GAClC,oBAAoB,CAAC,aAAa,GAClC,oBAAoB,CAAC,aAAa,GAClC,oBAAoB,CAAC,WAAW,GAChC,oBAAoB,CAAC,aAAa,GAClC,oBAAoB,CAAC,WAAW,GAChC,oBAAoB,CAAC,UAAU,GAC/B,oBAAoB,CAAC,KAAK,CAAC;AAE/B,KAAK,4CAA4C,GAAG,MAAM,GAAG,UAAU,CAAC;AAExE,MAAM,WAAW,WAAW;IAC1B,qEAAqE;IACrE,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,oEAAoE;IACpE,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,0EAA0E;IAC1E,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;CACrB;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,UAAU;IACzB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,UAAU,CAAC;IAClB,QAAQ,CAAC,EAAE,kBAAkB,CAAC;CAC/B;AAED,MAAM,WAAW,gBAAgB;IAC/B,yCAAyC;IACzC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,0CAA0C;IAC1C,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,yCAAyC;IACzC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,kDAAkD;IAClD,qBAAqB,CAAC,EAAE,MAAM,CAAC;CAChC;AAED,MAAM,WAAW,WAAW;IAC1B,uDAAuD;IACvD,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,kDAAkD;IAClD,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,KAAK;IACpB,YAAY,CAAC,EAAE,gBAAgB,CAAC;IAChC,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED,MAAM,WAAW,QAAQ;IACvB,oEAAoE;IACpE,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,kEAAkE;IAClE,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,wCAAwC;IACxC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,GAAG,CAAC;IACd,iEAAiE;IACjE,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,IAAI,CAAC,EAAE,GAAG,CAAC;CACZ;AAED,kBAAU,oBAAoB,CAAC;IAC7B,UAAiB,YAAY;QAC3B,IAAI,EAAE,eAAe,CAAC;QACtB,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,CAAC,EAAE;YACN,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW,CAAC;aACnB,CAAC;SACH,CAAC;KACH;IAED,UAAiB,YAAY;QAC3B,IAAI,EAAE,eAAe,CAAC;QACtB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE;YACN,OAAO,EAAE;gBACP,OAAO,EAAE;oBACP,QAAQ,CAAC,EAAE,MAAM,CAAC;oBAClB,IAAI,CAAC,EAAE,MAAM,CAAC;oBACd,IAAI,CAAC,EAAE,4CAA4C,CAAC;iBACrD,CAAC;aACH,CAAC;SACH,CAAC;KACH;IAED,UAAiB,YAAY;QAC3B,IAAI,EAAE,eAAe,CAAC;QACtB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE;YACN,OAAO,EAAE;gBACP,OAAO,EAAE;oBACP,QAAQ,CAAC,EAAE,MAAM,CAAC;oBAClB,IAAI,CAAC,EAAE,MAAM,CAAC;iBACf,CAAC;aACH,CAAC;SACH,CAAC;QACF,QAAQ,CAAC,EAAE,WAAW,CAAC;KACxB;IAED,UAAiB,UAAU;QACzB,IAAI,EAAE,aAAa,CAAC;QACpB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB;IAED,UAAiB,aAAa;QAC5B,IAAI,EAAE,iBAAiB,CAAC;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,EAAE;YACL,OAAO,EAAE;gBACP,SAAS,EAAE,MAAM,CAAC;aACnB,CAAC;SACH,CAAC;KACH;IAED,UAAiB,aAAa;QAC5B,IAAI,EAAE,iBAAiB,CAAC;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,EAAE;YACL,OAAO,EAAE;gBACP,UAAU,EAAE,UAAU,CAAC;aACxB,CAAC;SACH,CAAC;KACH;IAED,UAAiB,aAAa;QAC5B,IAAI,EAAE,iBAAiB,CAAC;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,EAAE;YACL,OAAO,EAAE;gBACP,UAAU,EAAE,UAAU,CAAC;aACxB,CAAC;SACH,CAAC;KACH;IAED,UAAiB,WAAW;QAC1B,IAAI,EAAE,eAAe,CAAC;QACtB,KAAK,EAAE,MAAM,CAAC;KACf;IAED,UAAiB,aAAa;QAC5B,IAAI,EAAE,gBAAgB,CAAC;QACvB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE;YACN,OAAO,CAAC,EAAE;gBACR,SAAS,EAAE,QAAQ,CAAC;aACrB,CAAC;SACH,CAAC;KACH;IAED,UAAiB,WAAW;QAC1B,IAAI,EAAE,cAAc,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;KACf;IAED,UAAiB,UAAU;QACzB,IAAI,EAAE,aAAa,CAAC;QACpB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,KAAK,CAAC,EAAE;YACN,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,aAAa,CAAC,EAAE,kBAAkB,CAAC;YACnC,KAAK,CAAC,EAAE,KAAK,CAAC;SACf,CAAC;KACH;IAED,UAAiB,KAAK;QACpB,IAAI,EAAE,OAAO,CAAC;QACd,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB;CACF;AACD,MAAM,WAAW,0BAA0B;IACzC,EAAE,EAAE,MAAM,CAAC;IACX,aAAa,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE;QACP,IAAI,EAAE,WAAW,CAAC;QAClB,UAAU,EAAE,GAAG,EAAE,CAAC;QAClB,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EACH;YACE,IAAI,EAAE,MAAM,CAAC;YACb,IAAI,EAAE,MAAM,CAAC;SACd,EAAE,GACH;YACE,QAAQ,EAAE,MAAM,CAAC;YACjB,IAAI,EAAE,UAAU,CAAC;SAClB,EAAE,CAAC;QACR,SAAS,EAAE,GAAG,CAAC;KAChB,CAAC;IACF,KAAK,EAAE;QACL,YAAY,CAAC,EAAE;YACb,YAAY,CAAC,EAAE,MAAM,CAAC;YACtB,aAAa,CAAC,EAAE,MAAM,CAAC;SACxB,CAAC;QACF,MAAM,CAAC,EAAE;YACP,YAAY,CAAC,EAAE,MAAM,CAAC;YACtB,aAAa,CAAC,EAAE,MAAM,CAAC;SACxB,CAAC;QACF,aAAa,CAAC,EAAE,MAAM,CAAC;KACxB,CAAC;CACH;AACD,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,MAAM,CAAC;IAChB,EAAE,EAAE,MAAM,CAAC;CACZ"}

package/dist/providers/cohere/types.js

@@ -0,0 +1,10 @@
+export var COHERE_STOP_REASON;
+(function (COHERE_STOP_REASON) {
+    COHERE_STOP_REASON["complete"] = "COMPLETE";
+    COHERE_STOP_REASON["stop_sequence"] = "STOP_SEQUENCE";
+    COHERE_STOP_REASON["max_tokens"] = "MAX_TOKENS";
+    COHERE_STOP_REASON["tool_call"] = "TOOL_CALL";
+    COHERE_STOP_REASON["error"] = "ERROR";
+    COHERE_STOP_REASON["timeout"] = "TIMEOUT";
+})(COHERE_STOP_REASON || (COHERE_STOP_REASON = {}));
+//# sourceMappingURL=types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/providers/cohere/types.ts"],"names":[],"mappings":"AAiGA,MAAM,CAAN,IAAY,kBAOX;AAPD,WAAY,kBAAkB;IAC5B,2CAAqB,CAAA;IACrB,qDAA+B,CAAA;IAC/B,+CAAyB,CAAA;IACzB,6CAAuB,CAAA;IACvB,qCAAe,CAAA;IACf,yCAAmB,CAAA;AACrB,CAAC,EAPW,kBAAkB,KAAlB,kBAAkB,QAO7B"}

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

@@ -0,0 +1,14 @@
+import { ErrorResponse, UploadFileResponse } from '../types';
+import { CohereErrorResponse } from './types';
+interface CohereCreateDatasetResponse {
+    id: string;
+}
+export declare const CohereBatchRequestRowTransform: (row: Record<string, any>) => {
+    custom_id: any;
+    text: any;
+    id: string;
+};
+export declare const CohereUploadFileRequestTransform: (requestBody: ReadableStream, requestHeaders: Record<string, string>) => ReadableStream<any>;
+export declare const CohereUploadFileResponseTransform: (response: CohereCreateDatasetResponse | CohereErrorResponse, responseStatus: number) => UploadFileResponse | ErrorResponse;
+export {};
+//# sourceMappingURL=uploadFile.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"uploadFile.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/uploadFile.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAE7D,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAG9C,UAAU,2BAA2B;IACnC,EAAE,EAAE,MAAM,CAAC;CACZ;AAMD,eAAO,MAAM,8BAA8B,GAAI,KAAK,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;;;;CAMtE,CAAC;AAEF,eAAO,MAAM,gCAAgC,GAC3C,aAAa,cAAc,EAC3B,gBAAgB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,wBAQvC,CAAC;AAEF,eAAO,MAAM,iCAAiC,EAAE,CAC9C,QAAQ,EAAE,2BAA2B,GAAG,mBAAmB,EAC3D,cAAc,EAAE,MAAM,KACnB,kBAAkB,GAAG,aAczB,CAAC"}

package/dist/providers/cohere/uploadFile.js

@@ -0,0 +1,40 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { COHERE } from '../../globals.js';
+import { getFormdataToFormdataStreamTransformer } from '../../handlers/streamHandlerUtils.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+import { CohereErrorResponseTransform } from './utils.js';
+const CohereFileUploadFieldsMapping = {
+    file: 'file',
+};
+export const CohereBatchRequestRowTransform = (row) => {
+    return {
+        custom_id: row.custom_id,
+        text: row.body.input,
+        id: crypto.randomUUID(),
+    };
+};
+export const CohereUploadFileRequestTransform = (requestBody, requestHeaders) => {
+    const transformStream = getFormdataToFormdataStreamTransformer(requestHeaders, CohereBatchRequestRowTransform, CohereFileUploadFieldsMapping);
+    return requestBody.pipeThrough(transformStream);
+};
+export const CohereUploadFileResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200 && 'message' in response) {
+        return CohereErrorResponseTransform(response);
+    }
+    else if ('id' in response) {
+        return {
+            id: response.id,
+            object: 'file',
+            created_at: Math.floor(Date.now() / 1000),
+            filename: '',
+            purpose: '',
+            status: 'uploaded',
+        };
+    }
+    return generateInvalidProviderResponseError(response, COHERE);
+};
+//# sourceMappingURL=uploadFile.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"uploadFile.js","sourceRoot":"","sources":["../../../src/providers/cohere/uploadFile.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACvC,OAAO,EAAE,sCAAsC,EAAE,MAAM,mCAAmC,CAAC;AAE3F,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAEhE,OAAO,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAMvD,MAAM,6BAA6B,GAAG;IACpC,IAAI,EAAE,MAAM;CACb,CAAC;AAEF,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,GAAwB,EAAE,EAAE;IACzE,OAAO;QACL,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,KAAK;QACpB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;KACxB,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gCAAgC,GAAG,CAC9C,WAA2B,EAC3B,cAAsC,EACtC,EAAE;IACF,MAAM,eAAe,GAAG,sCAAsC,CAC5D,cAAc,EACd,8BAA8B,EAC9B,6BAA6B,CAC9B,CAAC;IACF,OAAO,WAAW,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;AAClD,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,iCAAiC,GAGJ,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IACrE,IAAI,cAAc,KAAK,GAAG,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;QACpD,OAAO,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;SAAM,IAAI,IAAI,IAAI,QAAQ,EAAE,CAAC;QAC5B,OAAO;YACL,EAAE,EAAE,QAAQ,CAAC,EAAE;YACf,MAAM,EAAE,MAAM;YACd,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YACzC,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE,EAAE;YACX,MAAM,EAAE,UAAU;SACnB,CAAC;IACJ,CAAC;IACD,OAAO,oCAAoC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AAChE,CAAC,CAAC"}

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

@@ -0,0 +1,3 @@
+import { CohereErrorResponse } from './types';
+export declare const CohereErrorResponseTransform: (response: CohereErrorResponse) => import("../../utils/errorHandlingUtils").ErrorResponseBodyWithExtras;
+//# sourceMappingURL=utils.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/providers/cohere/utils.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAE9C,eAAO,MAAM,4BAA4B,GAAI,UAAU,mBAAmB,yEASzE,CAAC"}

package/dist/providers/cohere/utils.js

@@ -0,0 +1,15 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { COHERE } from '../../globals.js';
+import { generateErrorResponse } from '../utils.js';
+export const CohereErrorResponseTransform = (response) => {
+    return generateErrorResponse({
+        message: response.message,
+        type: null,
+        code: null,
+    }, COHERE);
+};
+//# sourceMappingURL=utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/providers/cohere/utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAGjD,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,QAA6B,EAAE,EAAE;IAC5E,OAAO,qBAAqB,CAC1B;QACE,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,IAAI,EAAE,IAAI;QACV,IAAI,EAAE,IAAI;KACX,EACD,MAAM,CACP,CAAC;AACJ,CAAC,CAAC"}

package/dist/providers/cortex/api.d.ts

@@ -0,0 +1,4 @@
+import type { ProviderAPIConfig } from '../types';
+declare const CortexAPIConfig: ProviderAPIConfig;
+export default CortexAPIConfig;
+//# sourceMappingURL=api.d.ts.map

package/dist/providers/cortex/api.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.d.ts","sourceRoot":"","sources":["../../../src/providers/cortex/api.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAElD,QAAA,MAAM,eAAe,EAAE,iBAsBtB,CAAC;AAEF,eAAe,eAAe,CAAC"}

package/dist/providers/cortex/api.js

@@ -0,0 +1,25 @@
+const CortexAPIConfig = {
+    getBaseURL: ({ providerOptions }) => {
+        const account = providerOptions.providerAccount.manifest;
+        return `https://${account.account_identifier}.snowflakecomputing.com/api/v2`;
+    },
+    headers: ({ providerOptions }) => {
+        const account = providerOptions.providerAccount.manifest;
+        return {
+            'X-Snowflake-Authorization-Token-Type': 'PROGRAMMATIC_ACCESS_TOKEN',
+            Authorization: `Bearer ${account.auth_data.pat_token}`,
+            'Content-Type': 'application/json',
+            Accept: 'application/json, text/event-stream',
+        };
+    },
+    getEndpoint: ({ endpoint }) => {
+        switch (endpoint) {
+            case 'chatComplete':
+                return '/cortex/inference:complete';
+            default:
+                return '';
+        }
+    },
+};
+export default CortexAPIConfig;
+//# sourceMappingURL=api.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/cortex/api.ts"],"names":[],"mappings":"AAGA,MAAM,eAAe,GAAsB;IACzC,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;QAClC,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAA0C,CAAC;QAC3F,OAAO,WAAW,OAAO,CAAC,kBAAkB,gCAAgC,CAAC;IAC/E,CAAC;IACD,OAAO,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;QAC/B,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAA0C,CAAC;QAC3F,OAAO;YACL,sCAAsC,EAAE,2BAA2B;YACnE,aAAa,EAAE,UAAU,OAAO,CAAC,SAAS,CAAC,SAAS,EAAE;YACtD,cAAc,EAAE,kBAAkB;YAClC,MAAM,EAAE,qCAAqC;SAC9C,CAAC;IACJ,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;QAC5B,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,cAAc;gBACjB,OAAO,4BAA4B,CAAC;YACtC;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAC;AAEF,eAAe,eAAe,CAAC"}

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

@@ -0,0 +1,4 @@
+import { ProviderConfigs } from '../types';
+declare const CortexConfig: ProviderConfigs;
+export default CortexConfig;
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/providers/cortex/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAG3C,QAAA,MAAM,YAAY,EAAE,eAmCnB,CAAC;AAEF,eAAe,YAAY,CAAC"}

package/dist/providers/cortex/index.js

@@ -0,0 +1,37 @@
+import { CORTEX } from '../../globals.js';
+import { chatCompleteParams, completeParams, embedParams, responseTransformers } from '../open-ai-base/index.js';
+import CortexAPIConfig from './api.js';
+const CortexConfig = {
+    chatComplete: chatCompleteParams([], {}, {
+        messages: {
+            param: 'messages',
+            default: '',
+            transform: (params) => {
+                return params.messages?.map((message) => {
+                    // Convert developer role to system role
+                    if (message.role === 'developer') {
+                        message = { ...message, role: 'system' };
+                    }
+                    if (Array.isArray(message.content)) {
+                        message = {
+                            ...message,
+                            content: message.content
+                                .filter((part) => part.type === 'text' && part.text)
+                                .map((part) => part.text)
+                                .join(' '),
+                        };
+                    }
+                    return message;
+                });
+            },
+        },
+    }),
+    complete: completeParams([]),
+    embed: embedParams([]),
+    api: CortexAPIConfig,
+    responseTransforms: responseTransformers(CORTEX, {
+        chatComplete: true,
+    }),
+};
+export default CortexConfig;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/cortex/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAExG,OAAO,eAAe,MAAM,OAAO,CAAC;AAEpC,MAAM,YAAY,GAAoB;IACpC,YAAY,EAAE,kBAAkB,CAC9B,EAAE,EACF,EAAE,EACF;QACE,QAAQ,EAAE;YACR,KAAK,EAAE,UAAU;YACjB,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,CAAC,MAAW,EAAE,EAAE;gBACzB,OAAO,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,OAAY,EAAE,EAAE;oBAC3C,wCAAwC;oBACxC,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBACjC,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;oBAC3C,CAAC;oBACD,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;wBACnC,OAAO,GAAG;4BACR,GAAG,OAAO;4BACV,OAAO,EAAE,OAAO,CAAC,OAAO;iCACrB,MAAM,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC;iCACxD,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;iCAC7B,IAAI,CAAC,GAAG,CAAC;yBACb,CAAC;oBACJ,CAAC;oBACD,OAAO,OAAO,CAAC;gBACjB,CAAC,CAAC,CAAC;YACL,CAAC;SACF;KACF,CACF;IACD,QAAQ,EAAE,cAAc,CAAC,EAAE,CAAC;IAC5B,KAAK,EAAE,WAAW,CAAC,EAAE,CAAC;IACtB,GAAG,EAAE,eAAe;IACpB,kBAAkB,EAAE,oBAAoB,CAAC,MAAM,EAAE;QAC/C,YAAY,EAAE,IAAI;KACnB,CAAC;CACH,CAAC;AAEF,eAAe,YAAY,CAAC"}

package/dist/providers/databricks/api.d.ts

@@ -0,0 +1,4 @@
+import { ProviderAPIConfig } from '../types';
+declare const DatabricksAPIConfig: ProviderAPIConfig;
+export default DatabricksAPIConfig;
+//# sourceMappingURL=api.d.ts.map

package/dist/providers/databricks/api.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.d.ts","sourceRoot":"","sources":["../../../src/providers/databricks/api.ts"],"names":[],"mappings":"AAOA,OAAO,EAAY,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAkBvD,QAAA,MAAM,mBAAmB,EAAE,iBA4D1B,CAAC;AAEF,eAAe,mBAAmB,CAAC"}

package/dist/providers/databricks/api.js

@@ -0,0 +1,69 @@
+import { getDatabricksOAuthToken, isOpenAIFamilyModel, isUnityCatalogModel } from './utils.js';
+const UNITY_MLFLOW_BASE_PATH = '/ai-gateway/mlflow/v1';
+const UNITY_ANTHROPIC_BASE_PATH = '/ai-gateway/anthropic';
+const UNITY_CURSOR_BASE_PATH = '/ai-gateway/cursor/v1';
+const UNITY_SURFACES = {
+    chatComplete: { basePath: UNITY_MLFLOW_BASE_PATH, endpoint: '/chat/completions' },
+    embed: { basePath: UNITY_MLFLOW_BASE_PATH, endpoint: '/embeddings' },
+    createModelResponse: { basePath: UNITY_MLFLOW_BASE_PATH, endpoint: '/responses' },
+    messages: { basePath: UNITY_ANTHROPIC_BASE_PATH, endpoint: '/v1/messages' },
+};
+function withBasePath(workspace, basePath) {
+    return workspace.includes(basePath) ? workspace : `${workspace}${basePath}`;
+}
+const DatabricksAPIConfig = {
+    getBaseURL: ({ providerOptions, callConfig, endpoint, request }) => {
+        const manifest = providerOptions.providerAccount.manifest;
+        const baseUrl = manifest.base_url;
+        if (!baseUrl) {
+            return '';
+        }
+        const workspace = baseUrl.replace(/\/$/, '').replace(/\/serving-endpoints$/, '');
+        // The Cursor surface is a path on the workspace host itself, not a separate
+        // ai-gateway host: rewriting the host resolves to a domain that does not exist.
+        if (callConfig.isCursorEndpoint) {
+            return withBasePath(workspace, UNITY_CURSOR_BASE_PATH);
+        }
+        const model = request.body?.model ?? '';
+        const surface = isUnityCatalogModel(model) && endpoint ? UNITY_SURFACES[endpoint] : undefined;
+        if (surface) {
+            return withBasePath(workspace, surface.basePath);
+        }
+        return withBasePath(workspace, '/serving-endpoints');
+    },
+    headers: async ({ providerOptions }) => {
+        const manifest = providerOptions.providerAccount.manifest;
+        const apiKey = manifest.auth_data?.type === 'api-key' ? manifest.auth_data.api_key : undefined;
+        const databricksClientId = manifest.auth_data?.type === 'service-principal' ? manifest.auth_data.client_id : undefined;
+        const databricksOAuthSecret = manifest.auth_data?.type === 'service-principal' ? manifest.auth_data.oauth_secret : undefined;
+        const urlToFetch = manifest.base_url;
+        let authToken = apiKey;
+        if (databricksClientId && databricksOAuthSecret && urlToFetch) {
+            authToken = await getDatabricksOAuthToken(urlToFetch, databricksClientId, databricksOAuthSecret);
+        }
+        const headersObj = {
+            Authorization: `Bearer ${authToken}`,
+        };
+        return headersObj;
+    },
+    getEndpoint: ({ endpoint, callConfig, request }) => {
+        if (callConfig.isCursorEndpoint) {
+            if (endpoint === 'chatComplete' || endpoint === 'createModelResponse') {
+                return '/chat/completions';
+            }
+        }
+        const model = request.body?.model;
+        if (isUnityCatalogModel(model)) {
+            return UNITY_SURFACES[endpoint]?.endpoint ?? '';
+        }
+        if (endpoint === 'createModelResponse') {
+            return isOpenAIFamilyModel(model) ? '/responses' : '/open-responses';
+        }
+        if (endpoint === 'chatComplete' || endpoint === 'embed') {
+            return `/${model}/invocations`;
+        }
+        return '';
+    },
+};
+export default DatabricksAPIConfig;
+//# sourceMappingURL=api.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/databricks/api.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,uBAAuB,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAE5F,MAAM,sBAAsB,GAAG,uBAAuB,CAAC;AACvD,MAAM,yBAAyB,GAAG,uBAAuB,CAAC;AAC1D,MAAM,sBAAsB,GAAG,uBAAuB,CAAC;AAEvD,MAAM,cAAc,GAAsE;IACxF,YAAY,EAAE,EAAE,QAAQ,EAAE,sBAAsB,EAAE,QAAQ,EAAE,mBAAmB,EAAE;IACjF,KAAK,EAAE,EAAE,QAAQ,EAAE,sBAAsB,EAAE,QAAQ,EAAE,aAAa,EAAE;IACpE,mBAAmB,EAAE,EAAE,QAAQ,EAAE,sBAAsB,EAAE,QAAQ,EAAE,YAAY,EAAE;IACjF,QAAQ,EAAE,EAAE,QAAQ,EAAE,yBAAyB,EAAE,QAAQ,EAAE,cAAc,EAAE;CAC5E,CAAC;AAEF,SAAS,YAAY,CAAC,SAAiB,EAAE,QAAgB;IACvD,OAAO,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS,GAAG,QAAQ,EAAE,CAAC;AAC9E,CAAC;AAED,MAAM,mBAAmB,GAAsB;IAC7C,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;QACjE,MAAM,QAAQ,GAAG,eAAe,CAAC,eAAe,CAAC,QAAqC,CAAC;QACvF,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC;QAClC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC,CAAC;QAEjF,4EAA4E;QAC5E,gFAAgF;QAChF,IAAI,UAAU,CAAC,gBAAgB,EAAE,CAAC;YAChC,OAAO,YAAY,CAAC,SAAS,EAAE,sBAAsB,CAAC,CAAC;QACzD,CAAC;QAED,MAAM,KAAK,GAAI,OAAO,CAAC,IAA2B,EAAE,KAAK,IAAI,EAAE,CAAC;QAChE,MAAM,OAAO,GAAG,mBAAmB,CAAC,KAAK,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC,QAAoB,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1G,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,YAAY,CAAC,SAAS,EAAE,oBAAoB,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE;QACrC,MAAM,QAAQ,GAAG,eAAe,CAAC,eAAe,CAAC,QAAqC,CAAC;QACvF,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,EAAE,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QAC/F,MAAM,kBAAkB,GACtB,QAAQ,CAAC,SAAS,EAAE,IAAI,KAAK,mBAAmB,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAC9F,MAAM,qBAAqB,GACzB,QAAQ,CAAC,SAAS,EAAE,IAAI,KAAK,mBAAmB,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC;QACjG,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC;QACrC,IAAI,SAAS,GAAG,MAAM,CAAC;QAEvB,IAAI,kBAAkB,IAAI,qBAAqB,IAAI,UAAU,EAAE,CAAC;YAC9D,SAAS,GAAG,MAAM,uBAAuB,CAAC,UAAU,EAAE,kBAAkB,EAAE,qBAAqB,CAAC,CAAC;QACnG,CAAC;QACD,MAAM,UAAU,GAA2B;YACzC,aAAa,EAAE,UAAU,SAAS,EAAE;SACrC,CAAC;QACF,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE,EAAE;QACjD,IAAI,UAAU,CAAC,gBAAgB,EAAE,CAAC;YAChC,IAAI,QAAQ,KAAK,cAAc,IAAI,QAAQ,KAAK,qBAAqB,EAAE,CAAC;gBACtE,OAAO,mBAAmB,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,MAAM,KAAK,GAAI,OAAO,CAAC,IAA2B,EAAE,KAAK,CAAC;QAC1D,IAAI,mBAAmB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,cAAc,CAAC,QAAoB,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;QAC9D,CAAC;QACD,IAAI,QAAQ,KAAK,qBAAqB,EAAE,CAAC;YACvC,OAAO,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,iBAAiB,CAAC;QACvE,CAAC;QACD,IAAI,QAAQ,KAAK,cAAc,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;YACxD,OAAO,IAAI,KAAK,cAAc,CAAC;QACjC,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;CACF,CAAC;AAEF,eAAe,mBAAmB,CAAC"}

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

@@ -0,0 +1,18 @@
+import { ChatCompletionResponse, ErrorResponse, ProviderConfig } from '../types';
+export interface DatabricksErrorResponse {
+    error_code: string;
+    message: string;
+    text?: string;
+}
+export declare const DatabricksAnthropicChatCompleteConfig: ProviderConfig;
+export declare const DatabricksChatCompleteConfig: ProviderConfig;
+/**
+ * Minimal config for cursor passthrough: only `stream` is explicitly mapped so
+ * the gateway honours the streaming flag. Everything else (messages, system,
+ * input, tools, …) passes through verbatim via the default pass-through logic
+ * in `transformUsingProviderConfig`.
+ */
+export declare const DatabricksCursorPassthroughConfig: ProviderConfig;
+export declare const DatabricksChatCompleteResponseTransform: (response: ChatCompletionResponse | DatabricksErrorResponse, responseStatus: number, strictOpenAiCompliance: boolean) => ChatCompletionResponse | ErrorResponse;
+export declare const DatabricksChatCompleteStreamChunkTransform: (response: string, fallbackId: string, streamState: Record<string, any>, strictOpenAiCompliance: boolean, gatewayRequest: any) => string;
+//# sourceMappingURL=chatComplete.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"chatComplete.d.ts","sourceRoot":"","sources":["../../../src/providers/databricks/chatComplete.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,sBAAsB,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAGjF,MAAM,WAAW,uBAAuB;IACtC,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,eAAO,MAAM,qCAAqC,EAAE,cAWhD,CAAC;AAEL,eAAO,MAAM,4BAA4B,EAAE,cAWvC,CAAC;AAEL;;;;;GAKG;AACH,eAAO,MAAM,iCAAiC,EAAE,cAG/C,CAAC;AAEF,eAAO,MAAM,uCAAuC,EAAE,CACpD,QAAQ,EAAE,sBAAsB,GAAG,uBAAuB,EAC1D,cAAc,EAAE,MAAM,EACtB,sBAAsB,EAAE,OAAO,KAC5B,sBAAsB,GAAG,aA8D7B,CAAC;AAEF,eAAO,MAAM,0CAA0C,EAAE,CACvD,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAChC,sBAAsB,EAAE,OAAO,EAC/B,cAAc,EAAE,GAAG,KAChB,MAsFJ,CAAC"}