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

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

@@ -0,0 +1,264 @@
+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 {};

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,13 @@
+import { ErrorResponse, UploadFileResponse } from '../types.js';
+import { CohereErrorResponse } from './types.js';
+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 {};

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,2 @@
+import { CohereErrorResponse } from './types.js';
+export declare const CohereErrorResponseTransform: (response: CohereErrorResponse) => import("../../utils/errorHandlingUtils.js").ErrorResponseBodyWithExtras;

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,3 @@
+import type { ProviderAPIConfig } from '../types.js';
+declare const CortexAPIConfig: ProviderAPIConfig;
+export default CortexAPIConfig;

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,3 @@
+import { ProviderConfigs } from '../types.js';
+declare const CortexConfig: ProviderConfigs;
+export default CortexConfig;

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,3 @@
+import { ProviderAPIConfig } from '../types.js';
+declare const DatabricksAPIConfig: ProviderAPIConfig;
+export default DatabricksAPIConfig;

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,17 @@
+import { ChatCompletionResponse, ErrorResponse, ProviderConfig } from '../types.js';
+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;