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

package/dist/providers/azure-ai-inference/index.js

@@ -0,0 +1,69 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { AZURE_AI_INFERENCE, GITHUB } from '../../globals.js';
+import { AnthropicChatCompleteConfig, getAzureAnthropicChatCompleteResponseTransform, getAzureAnthropicStreamChunkTransform, } from '../anthropic/chatComplete.js';
+import AzureAIInferenceAPI, { resolveAzureFoundryUrl } from './api.js';
+import { AzureAIInferenceChatCompleteConfig, AzureAIInferenceChatCompleteResponseTransform } from './chatComplete.js';
+import { AzureAIInferenceCompleteConfig, AzureAIInferenceCompleteResponseTransform } from './complete.js';
+import { AzureAIInferenceCreateSpeechConfig, AzureAIInferenceCreateSpeechRequestHandler, AzureAIInferenceCreateSpeechResponseTransform, } from './createSpeech.js';
+import { AzureAIInferenceCreateTranscriptionRequestHandler, AzureAIInferenceCreateTranscriptionResponseTransform, } from './createTranscription.js';
+import { AzureAIInferenceCreateTranslationRequestHandler, AzureAIInferenceCreateTranslationResponseTransform, } from './createTranslation.js';
+import { AzureAIInferenceEmbedConfig, AzureAIInferenceEmbedResponseTransform } from './embed.js';
+import { AzureAIInferenceMessagesConfig, AzureAIInferenceMessagesResponseTransform } from './messages.js';
+import { AzureAIInferenceRerankConfig, AzureAIInferenceRerankResponseTransform } from './rerank.js';
+const AzureAIInferenceAPIConfig = {
+    complete: AzureAIInferenceCompleteConfig,
+    embed: AzureAIInferenceEmbedConfig,
+    createSpeech: AzureAIInferenceCreateSpeechConfig,
+    createTranscription: {},
+    createTranslation: {},
+    api: AzureAIInferenceAPI,
+    requestHandlers: {
+        createSpeech: AzureAIInferenceCreateSpeechRequestHandler,
+        createTranscription: AzureAIInferenceCreateTranscriptionRequestHandler,
+        createTranslation: AzureAIInferenceCreateTranslationRequestHandler,
+    },
+    getConfig: ({ headerOptions, providerOptions }) => {
+        const azureFoundryUrl = resolveAzureFoundryUrl({ providerOptions, headerOptions });
+        const isAnthropicModel = azureFoundryUrl?.includes('anthropic') ?? false;
+        const chatCompleteConfig = isAnthropicModel ? AnthropicChatCompleteConfig : AzureAIInferenceChatCompleteConfig;
+        const chatCompleteResponseTransform = isAnthropicModel
+            ? getAzureAnthropicChatCompleteResponseTransform(AZURE_AI_INFERENCE)
+            : AzureAIInferenceChatCompleteResponseTransform(AZURE_AI_INFERENCE);
+        const chatCompleteStreamResponseTransform = isAnthropicModel
+            ? getAzureAnthropicStreamChunkTransform(AZURE_AI_INFERENCE)
+            : undefined;
+        return {
+            chatComplete: chatCompleteConfig,
+            messages: AzureAIInferenceMessagesConfig,
+            rerank: AzureAIInferenceRerankConfig,
+            responseTransforms: {
+                complete: AzureAIInferenceCompleteResponseTransform(AZURE_AI_INFERENCE),
+                chatComplete: chatCompleteResponseTransform,
+                embed: AzureAIInferenceEmbedResponseTransform(AZURE_AI_INFERENCE),
+                messages: AzureAIInferenceMessagesResponseTransform,
+                'stream-chatComplete': chatCompleteStreamResponseTransform,
+                createSpeech: AzureAIInferenceCreateSpeechResponseTransform,
+                createTranscription: AzureAIInferenceCreateTranscriptionResponseTransform,
+                createTranslation: AzureAIInferenceCreateTranslationResponseTransform,
+                rerank: AzureAIInferenceRerankResponseTransform,
+            },
+        };
+    },
+};
+const GithubModelAPiConfig = {
+    complete: AzureAIInferenceCompleteConfig,
+    embed: AzureAIInferenceEmbedConfig,
+    api: AzureAIInferenceAPI,
+    chatComplete: AzureAIInferenceChatCompleteConfig,
+    responseTransforms: {
+        complete: AzureAIInferenceCompleteResponseTransform(GITHUB),
+        chatComplete: AzureAIInferenceChatCompleteResponseTransform(GITHUB),
+        embed: AzureAIInferenceEmbedResponseTransform(GITHUB),
+    },
+};
+export { AzureAIInferenceAPIConfig, GithubModelAPiConfig };
+//# sourceMappingURL=index.js.map

package/dist/providers/azure-ai-inference/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/azure-ai-inference/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,kBAAkB,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC3D,OAAO,EACL,2BAA2B,EAC3B,8CAA8C,EAC9C,qCAAqC,GACtC,MAAM,2BAA2B,CAAC;AAEnC,OAAO,mBAAmB,EAAE,EAAE,sBAAsB,EAAE,MAAM,OAAO,CAAC;AACpE,OAAO,EAAE,kCAAkC,EAAE,6CAA6C,EAAE,MAAM,gBAAgB,CAAC;AACnH,OAAO,EAAE,8BAA8B,EAAE,yCAAyC,EAAE,MAAM,YAAY,CAAC;AACvG,OAAO,EACL,kCAAkC,EAClC,0CAA0C,EAC1C,6CAA6C,GAC9C,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACL,iDAAiD,EACjD,oDAAoD,GACrD,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,+CAA+C,EAC/C,kDAAkD,GACnD,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,2BAA2B,EAAE,sCAAsC,EAAE,MAAM,SAAS,CAAC;AAC9F,OAAO,EAAE,8BAA8B,EAAE,yCAAyC,EAAE,MAAM,YAAY,CAAC;AACvG,OAAO,EAAE,4BAA4B,EAAE,uCAAuC,EAAE,MAAM,UAAU,CAAC;AAEjG,MAAM,yBAAyB,GAAoB;IACjD,QAAQ,EAAE,8BAA8B;IACxC,KAAK,EAAE,2BAA2B;IAClC,YAAY,EAAE,kCAAkC;IAChD,mBAAmB,EAAE,EAAE;IACvB,iBAAiB,EAAE,EAAE;IACrB,GAAG,EAAE,mBAAmB;IACxB,eAAe,EAAE;QACf,YAAY,EAAE,0CAA0C;QACxD,mBAAmB,EAAE,iDAAiD;QACtE,iBAAiB,EAAE,+CAA+C;KACnE;IACD,SAAS,EAAE,CAAC,EAAE,aAAa,EAAE,eAAe,EAAE,EAAE,EAAE;QAChD,MAAM,eAAe,GAAG,sBAAsB,CAAC,EAAE,eAAe,EAAE,aAAa,EAAE,CAAC,CAAC;QACnF,MAAM,gBAAgB,GAAG,eAAe,EAAE,QAAQ,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC;QAEzE,MAAM,kBAAkB,GAAG,gBAAgB,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,kCAAkC,CAAC;QAE/G,MAAM,6BAA6B,GAAG,gBAAgB;YACpD,CAAC,CAAC,8CAA8C,CAAC,kBAAkB,CAAC;YACpE,CAAC,CAAC,6CAA6C,CAAC,kBAAkB,CAAC,CAAC;QAEtE,MAAM,mCAAmC,GAAG,gBAAgB;YAC1D,CAAC,CAAC,qCAAqC,CAAC,kBAAkB,CAAC;YAC3D,CAAC,CAAC,SAAS,CAAC;QAEd,OAAO;YACL,YAAY,EAAE,kBAAkB;YAChC,QAAQ,EAAE,8BAA8B;YACxC,MAAM,EAAE,4BAA4B;YACpC,kBAAkB,EAAE;gBAClB,QAAQ,EAAE,yCAAyC,CAAC,kBAAkB,CAAC;gBACvE,YAAY,EAAE,6BAA6B;gBAC3C,KAAK,EAAE,sCAAsC,CAAC,kBAAkB,CAAC;gBACjE,QAAQ,EAAE,yCAAyC;gBACnD,qBAAqB,EAAE,mCAAmC;gBAC1D,YAAY,EAAE,6CAA6C;gBAC3D,mBAAmB,EAAE,oDAAoD;gBACzE,iBAAiB,EAAE,kDAAkD;gBACrE,MAAM,EAAE,uCAAuC;aAChD;SACF,CAAC;IACJ,CAAC;CACF,CAAC;AAEF,MAAM,oBAAoB,GAAoB;IAC5C,QAAQ,EAAE,8BAA8B;IACxC,KAAK,EAAE,2BAA2B;IAClC,GAAG,EAAE,mBAAmB;IACxB,YAAY,EAAE,kCAAkC;IAChD,kBAAkB,EAAE;QAClB,QAAQ,EAAE,yCAAyC,CAAC,MAAM,CAAC;QAC3D,YAAY,EAAE,6CAA6C,CAAC,MAAM,CAAC;QACnE,KAAK,EAAE,sCAAsC,CAAC,MAAM,CAAC;KACtD;CACF,CAAC;AAEF,OAAO,EAAE,yBAAyB,EAAE,oBAAoB,EAAE,CAAC"}

package/dist/providers/azure-ai-inference/messages.d.ts

@@ -0,0 +1,5 @@
+import { MessagesResponse } from '../../types/messagesResponse.js';
+import { AnthropicErrorResponse } from '../anthropic/types.js';
+import { ErrorResponse } from '../types.js';
+export declare const AzureAIInferenceMessagesConfig: import("../types.js").ProviderConfig;
+export declare const AzureAIInferenceMessagesResponseTransform: (response: MessagesResponse | AnthropicErrorResponse, responseStatus: number) => MessagesResponse | ErrorResponse;

package/dist/providers/azure-ai-inference/messages.js

@@ -0,0 +1,21 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { AZURE_AI_INFERENCE } from '../../globals.js';
+import { getMessagesConfig } from '../anthropic-base/messages.js';
+import { AnthropicErrorResponseTransform } from '../anthropic/utils.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+export const AzureAIInferenceMessagesConfig = getMessagesConfig({});
+export const AzureAIInferenceMessagesResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200) {
+        const errorResposne = AnthropicErrorResponseTransform(response, AZURE_AI_INFERENCE);
+        if (errorResposne)
+            return errorResposne;
+    }
+    if ('model' in response)
+        return response;
+    return generateInvalidProviderResponseError(response, AZURE_AI_INFERENCE);
+};
+//# sourceMappingURL=messages.js.map

package/dist/providers/azure-ai-inference/messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.js","sourceRoot":"","sources":["../../../src/providers/azure-ai-inference/messages.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEnD,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAE/D,OAAO,EAAE,+BAA+B,EAAE,MAAM,oBAAoB,CAAC;AAErE,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAEhE,MAAM,CAAC,MAAM,8BAA8B,GAAG,iBAAiB,CAAC,EAAE,CAAC,CAAC;AAEpE,MAAM,CAAC,MAAM,yCAAyC,GAAG,CACvD,QAAmD,EACnD,cAAsB,EACY,EAAE;IACpC,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,+BAA+B,CAAC,QAAkC,EAAE,kBAAkB,CAAC,CAAC;QAC9G,IAAI,aAAa;YAAE,OAAO,aAAa,CAAC;IAC1C,CAAC;IAED,IAAI,OAAO,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAEzC,OAAO,oCAAoC,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;AAC5E,CAAC,CAAC"}

package/dist/providers/azure-ai-inference/rerank.d.ts

@@ -0,0 +1,30 @@
+import { Params } from '../../types/requestBody.js';
+import { ErrorResponse, ParamMapTransformInput, RerankResponse } from '../types.js';
+export declare const AzureAIInferenceRerankConfig: {
+    model: {
+        param: string;
+        required: boolean;
+        transform: (_params: Params, input: ParamMapTransformInput) => string | undefined;
+    };
+    query: {
+        param: string;
+        required: boolean;
+    };
+    top_n: {
+        param: string;
+        required: boolean;
+    };
+    documents: {
+        param: string;
+        required: boolean;
+    };
+    return_documents: {
+        param: string;
+        required: boolean;
+    };
+    max_tokens_per_doc: {
+        param: string;
+        required: boolean;
+    };
+};
+export declare const AzureAIInferenceRerankResponseTransform: (response: Record<string, any>, responseStatus: number) => RerankResponse | ErrorResponse;

package/dist/providers/azure-ai-inference/rerank.js

@@ -0,0 +1,61 @@
+/**
+ * ================================
+ * TFY Managed File
+ * ================================
+ */
+import { AZURE_AI_INFERENCE } from '../../globals.js';
+import { CohereErrorResponseTransform } from '../cohere/utils.js';
+import { OpenAIErrorResponseTransform } from '../openai/utils.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+// Azure AI Inference rerank request format matches the Cohere v2 rerank format.
+export const AzureAIInferenceRerankConfig = {
+    model: {
+        param: 'model',
+        required: false,
+        transform: (_params, input) => input.headerOptions.azure?.deploymentId ??
+            input.providerOptions?.model?.manifest?.model_id,
+    },
+    query: {
+        param: 'query',
+        required: true,
+    },
+    top_n: {
+        param: 'top_n',
+        required: false,
+    },
+    documents: {
+        param: 'documents',
+        required: true,
+    },
+    return_documents: {
+        param: 'return_documents',
+        required: false,
+    },
+    max_tokens_per_doc: {
+        param: 'max_tokens_per_doc',
+        required: false,
+    },
+};
+export const AzureAIInferenceRerankResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200) {
+        // OpenAI-style error: { error: { message, code } }
+        if ('error' in response) {
+            return OpenAIErrorResponseTransform(response, AZURE_AI_INFERENCE);
+        }
+        // Cohere-style error: { message } — also covers Azure { statusCode, message }
+        if ('message' in response) {
+            return CohereErrorResponseTransform(response);
+        }
+    }
+    if ('results' in response && Array.isArray(response.results)) {
+        return {
+            id: response.id,
+            results: response.results.map(result => ({
+                index: result.index,
+                relevance_score: result.relevance_score,
+            })),
+        };
+    }
+    return generateInvalidProviderResponseError(response, AZURE_AI_INFERENCE);
+};
+//# sourceMappingURL=rerank.js.map

package/dist/providers/azure-ai-inference/rerank.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rerank.js","sourceRoot":"","sources":["../../../src/providers/azure-ai-inference/rerank.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAGnD,OAAO,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAE/D,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAEhE,gFAAgF;AAChF,MAAM,CAAC,MAAM,4BAA4B,GAAG;IAC1C,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;QACf,SAAS,EAAE,CAAC,OAAe,EAAE,KAA6B,EAAE,EAAE,CAC5D,KAAK,CAAC,aAAa,CAAC,KAAK,EAAE,YAAY;YACtC,KAAK,CAAC,eAAe,EAAE,KAAK,EAAE,QAA8C,EAAE,QAAQ;KAC1F;IACD,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;AAcF,MAAM,CAAC,MAAM,uCAAuC,GAAG,CACrD,QAA6B,EAC7B,cAAsB,EACU,EAAE;IAClC,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,mDAAmD;QACnD,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;YACxB,OAAO,4BAA4B,CAAC,QAAyB,EAAE,kBAAkB,CAAC,CAAC;QACrF,CAAC;QACD,8EAA8E;QAC9E,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;YAC1B,OAAO,4BAA4B,CAAC,QAA+B,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED,IAAI,SAAS,IAAI,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QAC7D,OAAO;YACL,EAAE,EAAG,QAA2C,CAAC,EAAE;YACnD,OAAO,EAAG,QAA2C,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC3E,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,eAAe,EAAE,MAAM,CAAC,eAAe;aACxC,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;AAC5E,CAAC,CAAC"}

package/dist/providers/azure-openai/api.d.ts

@@ -0,0 +1,3 @@
+import type { ProviderAPIConfig } from '../types.js';
+declare const AzureOpenAIAPIConfig: ProviderAPIConfig;
+export default AzureOpenAIAPIConfig;

package/dist/providers/azure-openai/api.js

@@ -0,0 +1,227 @@
+import { sdkRequestUrl } from '../../utils/sdkRequestUrl.js';
+import { getAccessTokenFromEntraId, getAccessTokenFromEntraIdUsingCertificate, stripAzureOpenAIBaseSuffix, } from './utils.js';
+import { getAccessTokenFromWorkloadIdentity } from './wifAuth.js';
+const AzureOpenAIAPIConfig = {
+    getBaseURL: ({ providerOptions, headerOptions }) => {
+        /**
+         * ================================
+         * NOTE: Use azure_endpoint from the account when creating the base URL
+         * (formerly flattened as providerOptions.urlToFetch)
+         * ================================
+         */
+        const account = providerOptions.providerAccount.manifest;
+        if (account.azure_endpoint) {
+            return stripAzureOpenAIBaseSuffix(account.azure_endpoint) + '/openai';
+        }
+        const resourceName = headerOptions.azure.resourceName;
+        return `https://${resourceName}.openai.azure.com/openai`;
+    },
+    headers: async ({ providerOptions, headerOptions, endpoint }) => {
+        const account = providerOptions.providerAccount.manifest;
+        const authData = account.auth_data;
+        if (authData?.type === 'certificate') {
+            const { tenant_id, client_id, certificate, certificate_thumbprint } = authData;
+            if (tenant_id && client_id && certificate && certificate_thumbprint) {
+                const accessToken = await getAccessTokenFromEntraIdUsingCertificate(tenant_id, client_id, certificate, certificate_thumbprint);
+                return {
+                    Authorization: `Bearer ${accessToken}`,
+                };
+            }
+        }
+        if (authData?.type === 'client-secret') {
+            const { tenant_id, client_id, client_secret } = authData;
+            if (tenant_id && client_id && client_secret) {
+                const scope = headerOptions.azure.azureEntraScope || 'https://cognitiveservices.azure.com/.default';
+                const accessToken = await getAccessTokenFromEntraId(tenant_id, client_id, client_secret, scope);
+                return {
+                    Authorization: `Bearer ${accessToken}`,
+                };
+            }
+        }
+        if (authData?.type === 'workload-identity') {
+            const { tenant_id, client_id } = authData;
+            if (tenant_id && client_id) {
+                const accessToken = await getAccessTokenFromWorkloadIdentity(tenant_id, client_id);
+                return {
+                    Authorization: `Bearer ${accessToken}`,
+                };
+            }
+        }
+        const apiKey = authData?.type === 'api-key' ? authData.api_key : undefined;
+        const headersObj = {
+            'api-key': `${apiKey}`,
+        };
+        /**
+         * ================================
+         * NOTE: Dont add content-type for createTranscription and createTranslation
+         * ================================
+         */
+        if (endpoint === 'imageEdit')
+            headersObj['Content-Type'] = 'multipart/form-data';
+        const openaiBeta = headerOptions.azure.openaiBeta ?? headerOptions.openai.openaiBeta;
+        if (openaiBeta) {
+            headersObj['OpenAI-Beta'] = openaiBeta;
+        }
+        return headersObj;
+    },
+    getEndpoint: ({ providerOptions, headerOptions, callConfig, execution, endpoint, request }) => {
+        const account = providerOptions.providerAccount.manifest;
+        const model = providerOptions.model?.manifest;
+        const apiVersion = headerOptions.azure.apiVersion ?? model?.api_version;
+        const deploymentId = headerOptions.azure.deploymentId ?? model?.model_id;
+        const urlToFetch = account.azure_endpoint;
+        let mappedFn = endpoint;
+        if (endpoint === 'proxy' && urlToFetch) {
+            if (urlToFetch?.indexOf('/chat/completions') > -1) {
+                mappedFn = 'chatComplete';
+            }
+            else if (urlToFetch?.indexOf('/completions') > -1) {
+                mappedFn = 'complete';
+            }
+            else if (urlToFetch?.indexOf('/embeddings') > -1) {
+                mappedFn = 'embed';
+            }
+            else if (urlToFetch?.indexOf('/images/generations') > -1) {
+                mappedFn = 'imageGenerate';
+            }
+            else if (urlToFetch?.indexOf('/images/edits') > -1) {
+                mappedFn = 'imageEdit';
+            }
+            else if (urlToFetch?.indexOf('/audio/speech') > -1) {
+                mappedFn = 'createSpeech';
+            }
+            else if (urlToFetch?.indexOf('/audio/transcriptions') > -1) {
+                mappedFn = 'createTranscription';
+            }
+            else if (urlToFetch?.indexOf('/audio/translations') > -1) {
+                mappedFn = 'createTranslation';
+            }
+            else if (urlToFetch?.indexOf('/rerank') > -1) {
+                mappedFn = 'rerank';
+            }
+        }
+        const requestUrl = sdkRequestUrl(request);
+        const urlObj = new URL(requestUrl);
+        let pathname = urlObj.pathname.replace('/api/llm', '');
+        const searchParams = urlObj.searchParams;
+        // On a stateful virtual model GET/DELETE the path id is our encoded id; swap in the provider's real id decoded
+        // from it, since Azure only knows the id it minted.
+        const providerResponseId = callConfig.responsesProviderResponseId;
+        if (providerResponseId &&
+            ['getModelResponse', 'deleteModelResponse', 'listResponseInputItems'].includes(mappedFn)) {
+            const encodedInPath = requestUrl.split('/').at(mappedFn === 'listResponseInputItems' ? -2 : -1);
+            if (encodedInPath && pathname.includes(encodedInPath)) {
+                pathname = pathname.replace(encodedInPath, providerResponseId);
+            }
+        }
+        // API-version precedence: the SDK request's own `api-version` query param wins when present
+        // (e.g. the AzureOpenAI SDK always sends one); the Control Plane model configuration's
+        // `apiVersion` is used only as a fallback when the inbound request doesn't supply one.
+        const resolvedApiVersion = searchParams.get('api-version') || apiVersion;
+        if (resolvedApiVersion) {
+            searchParams.set('api-version', resolvedApiVersion);
+        }
+        switch (mappedFn) {
+            case 'complete': {
+                return `/deployments/${deploymentId}/completions?api-version=${resolvedApiVersion}`;
+            }
+            case 'chatComplete': {
+                if (execution?.openaiUseResponsesApi) {
+                    return `/responses?${searchParams.toString()}`;
+                }
+                return `/deployments/${deploymentId}/chat/completions?api-version=${resolvedApiVersion}`;
+            }
+            case 'embed': {
+                return `/deployments/${deploymentId}/embeddings?api-version=${resolvedApiVersion}`;
+            }
+            case 'imageGenerate': {
+                return `/deployments/${deploymentId}/images/generations?api-version=${resolvedApiVersion}`;
+            }
+            case 'imageEdit': {
+                return `/deployments/${deploymentId}/images/edits?api-version=${resolvedApiVersion}`;
+            }
+            case 'createSpeech': {
+                return `/deployments/${deploymentId}/audio/speech?api-version=${resolvedApiVersion}`;
+            }
+            case 'createTranscription': {
+                return `/deployments/${deploymentId}/audio/transcriptions?api-version=${resolvedApiVersion}`;
+            }
+            case 'createTranslation': {
+                return `/deployments/${deploymentId}/audio/translations?api-version=${resolvedApiVersion}`;
+            }
+            case 'realtime': {
+                return `/v1/realtime?model=${deploymentId}`;
+            }
+            case 'createModelResponse': {
+                return `${pathname}?${searchParams.toString()}`;
+            }
+            case 'createModelResponseCompact': {
+                /**
+                 * Compaction is served only on the `/openai/v1` surface, so the path is pinned instead of
+                 * taken from the inbound request: `createModelResponse` above resolves to the legacy
+                 * `/openai/responses`, which has no `/compact`.
+                 *
+                 * That surface accepts only `v1` or `preview` for `api-version` and defaults to `v1`, so an
+                 * account's dated version is not merely stale here but outside the allowed set — Azure
+                 * answers `API version not supported`. Compaction is preview-only, hence `preview` rather
+                 * than the `v1` default. Revisit if it reaches GA.
+                 * Ref - https://learn.microsoft.com/en-us/rest/api/microsoft-foundry/azureopenai/responses
+                 */
+                const compactParams = new URLSearchParams(searchParams);
+                compactParams.set('api-version', 'preview');
+                return `/v1/responses/compact?${compactParams.toString()}`;
+            }
+            case 'getModelResponse': {
+                return `${pathname}?${searchParams.toString()}`;
+            }
+            case 'deleteModelResponse': {
+                return `${pathname}?${searchParams.toString()}`;
+            }
+            case 'listResponseInputItems': {
+                return `${pathname}?${searchParams.toString()}`;
+            }
+            case 'uploadFile':
+            case 'retrieveFile':
+            case 'listFiles':
+            case 'deleteFile':
+            case 'retrieveFileContent':
+            case 'createFinetune':
+            case 'retrieveFinetune':
+            case 'listFinetunes':
+            case 'cancelFinetune':
+            case 'createBatch':
+            case 'retrieveBatch':
+            case 'cancelBatch':
+            case 'listBatches':
+                return `${pathname}?api-version=${resolvedApiVersion}`;
+            case 'rerank': {
+                // Users configure azure_endpoint as the base URL up to (but excluding) the API path,
+                // e.g. https://<hub>.services.ai.azure.com/providers/cohere/v2 — so append /rerank only.
+                return '/rerank';
+            }
+            default:
+                return '';
+        }
+    },
+    getProxyEndpoint: ({ request, providerOptions, headerOptions }) => {
+        const model = providerOptions.model?.manifest;
+        const apiVersion = headerOptions.azure.apiVersion ?? model?.api_version;
+        const reqPath = request.path;
+        const reqQuery = request.search;
+        if (!apiVersion)
+            return `${reqPath}${reqQuery}`;
+        if (!reqQuery?.includes('api-version')) {
+            let _reqQuery = reqQuery;
+            if (!reqQuery) {
+                _reqQuery = `?api-version=${apiVersion}`;
+            }
+            else {
+                _reqQuery += `&api-version=${apiVersion}`;
+            }
+            return `${reqPath}${_reqQuery}`;
+        }
+        return `${reqPath}${reqQuery}`;
+    },
+};
+export default AzureOpenAIAPIConfig;
+//# sourceMappingURL=api.js.map

package/dist/providers/azure-openai/api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/azure-openai/api.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE1D,OAAO,EACL,yBAAyB,EACzB,yCAAyC,EACzC,0BAA0B,GAC3B,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,kCAAkC,EAAE,MAAM,WAAW,CAAC;AAE/D,MAAM,oBAAoB,GAAsB;IAC9C,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,aAAa,EAAE,EAAE,EAAE;QACjD;;;;;WAKG;QACH,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAsC,CAAC;QACvF,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,OAAO,0BAA0B,CAAC,OAAO,CAAC,cAAc,CAAC,GAAG,SAAS,CAAC;QACxE,CAAC;QACD,MAAM,YAAY,GAAG,aAAa,CAAC,KAAK,CAAC,YAAY,CAAC;QACtD,OAAO,WAAW,YAAY,0BAA0B,CAAC;IAC3D,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,EAAE,eAAe,EAAE,aAAa,EAAE,QAAQ,EAAE,EAAE,EAAE;QAC9D,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAsC,CAAC;QACvF,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC;QAEnC,IAAI,QAAQ,EAAE,IAAI,KAAK,aAAa,EAAE,CAAC;YACrC,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,sBAAsB,EAAE,GAAG,QAAQ,CAAC;YAC/E,IAAI,SAAS,IAAI,SAAS,IAAI,WAAW,IAAI,sBAAsB,EAAE,CAAC;gBACpE,MAAM,WAAW,GAAG,MAAM,yCAAyC,CACjE,SAAS,EACT,SAAS,EACT,WAAW,EACX,sBAAsB,CACvB,CAAC;gBACF,OAAO;oBACL,aAAa,EAAE,UAAU,WAAW,EAAE;iBACvC,CAAC;YACJ,CAAC;QACH,CAAC;QACD,IAAI,QAAQ,EAAE,IAAI,KAAK,eAAe,EAAE,CAAC;YACvC,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,EAAE,GAAG,QAAQ,CAAC;YACzD,IAAI,SAAS,IAAI,SAAS,IAAI,aAAa,EAAE,CAAC;gBAC5C,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,eAAe,IAAI,8CAA8C,CAAC;gBACpG,MAAM,WAAW,GAAG,MAAM,yBAAyB,CAAC,SAAS,EAAE,SAAS,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC;gBAChG,OAAO;oBACL,aAAa,EAAE,UAAU,WAAW,EAAE;iBACvC,CAAC;YACJ,CAAC;QACH,CAAC;QACD,IAAI,QAAQ,EAAE,IAAI,KAAK,mBAAmB,EAAE,CAAC;YAC3C,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,QAAQ,CAAC;YAC1C,IAAI,SAAS,IAAI,SAAS,EAAE,CAAC;gBAC3B,MAAM,WAAW,GAAG,MAAM,kCAAkC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;gBACnF,OAAO;oBACL,aAAa,EAAE,UAAU,WAAW,EAAE;iBACvC,CAAC;YACJ,CAAC;QACH,CAAC;QACD,MAAM,MAAM,GAAG,QAAQ,EAAE,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3E,MAAM,UAAU,GAA2B;YACzC,SAAS,EAAE,GAAG,MAAM,EAAE;SACvB,CAAC;QACF;;;;WAIG;QACH,IAAI,QAAQ,KAAK,WAAW;YAAE,UAAU,CAAC,cAAc,CAAC,GAAG,qBAAqB,CAAC;QACjF,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC,UAAU,IAAI,aAAa,CAAC,MAAM,CAAC,UAAU,CAAC;QACrF,IAAI,UAAU,EAAE,CAAC;YACf,UAAU,CAAC,aAAa,CAAC,GAAG,UAAU,CAAC;QACzC,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,eAAe,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;QAC5F,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAsC,CAAC;QACvF,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,EAAE,QAAwC,CAAC;QAC9E,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC,UAAU,IAAI,KAAK,EAAE,WAAW,CAAC;QACxE,MAAM,YAAY,GAAG,aAAa,CAAC,KAAK,CAAC,YAAY,IAAI,KAAK,EAAE,QAAQ,CAAC;QACzE,MAAM,UAAU,GAAG,OAAO,CAAC,cAAc,CAAC;QAC1C,IAAI,QAAQ,GAAG,QAAQ,CAAC;QAExB,IAAI,QAAQ,KAAK,OAAO,IAAI,UAAU,EAAE,CAAC;YACvC,IAAI,UAAU,EAAE,OAAO,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBAClD,QAAQ,GAAG,cAAc,CAAC;YAC5B,CAAC;iBAAM,IAAI,UAAU,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBACpD,QAAQ,GAAG,UAAU,CAAC;YACxB,CAAC;iBAAM,IAAI,UAAU,EAAE,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBACnD,QAAQ,GAAG,OAAO,CAAC;YACrB,CAAC;iBAAM,IAAI,UAAU,EAAE,OAAO,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBAC3D,QAAQ,GAAG,eAAe,CAAC;YAC7B,CAAC;iBAAM,IAAI,UAAU,EAAE,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBACrD,QAAQ,GAAG,WAAW,CAAC;YACzB,CAAC;iBAAM,IAAI,UAAU,EAAE,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBACrD,QAAQ,GAAG,cAAc,CAAC;YAC5B,CAAC;iBAAM,IAAI,UAAU,EAAE,OAAO,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBAC7D,QAAQ,GAAG,qBAAqB,CAAC;YACnC,CAAC;iBAAM,IAAI,UAAU,EAAE,OAAO,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBAC3D,QAAQ,GAAG,mBAAmB,CAAC;YACjC,CAAC;iBAAM,IAAI,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBAC/C,QAAQ,GAAG,QAAQ,CAAC;YACtB,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC;QACnC,IAAI,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;QACvD,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QAEzC,+GAA+G;QAC/G,oDAAoD;QACpD,MAAM,kBAAkB,GAAG,UAAU,CAAC,2BAA2B,CAAC;QAClE,IACE,kBAAkB;YAClB,CAAC,kBAAkB,EAAE,qBAAqB,EAAE,wBAAwB,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EACxF,CAAC;YACD,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,QAAQ,KAAK,wBAAwB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAChG,IAAI,aAAa,IAAI,QAAQ,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;gBACtD,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,aAAa,EAAE,kBAAkB,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QACD,4FAA4F;QAC5F,uFAAuF;QACvF,uFAAuF;QACvF,MAAM,kBAAkB,GAAG,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,UAAU,CAAC;QACzE,IAAI,kBAAkB,EAAE,CAAC;YACvB,YAAY,CAAC,GAAG,CAAC,aAAa,EAAE,kBAAkB,CAAC,CAAC;QACtD,CAAC;QAED,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,OAAO,gBAAgB,YAAY,4BAA4B,kBAAkB,EAAE,CAAC;YACtF,CAAC;YACD,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,IAAI,SAAS,EAAE,qBAAqB,EAAE,CAAC;oBACrC,OAAO,cAAc,YAAY,CAAC,QAAQ,EAAE,EAAE,CAAC;gBACjD,CAAC;gBACD,OAAO,gBAAgB,YAAY,iCAAiC,kBAAkB,EAAE,CAAC;YAC3F,CAAC;YACD,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,OAAO,gBAAgB,YAAY,2BAA2B,kBAAkB,EAAE,CAAC;YACrF,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,OAAO,gBAAgB,YAAY,mCAAmC,kBAAkB,EAAE,CAAC;YAC7F,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,OAAO,gBAAgB,YAAY,6BAA6B,kBAAkB,EAAE,CAAC;YACvF,CAAC;YACD,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,OAAO,gBAAgB,YAAY,6BAA6B,kBAAkB,EAAE,CAAC;YACvF,CAAC;YACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,OAAO,gBAAgB,YAAY,qCAAqC,kBAAkB,EAAE,CAAC;YAC/F,CAAC;YACD,KAAK,mBAAmB,CAAC,CAAC,CAAC;gBACzB,OAAO,gBAAgB,YAAY,mCAAmC,kBAAkB,EAAE,CAAC;YAC7F,CAAC;YACD,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,OAAO,sBAAsB,YAAY,EAAE,CAAC;YAC9C,CAAC;YACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,OAAO,GAAG,QAAQ,IAAI,YAAY,CAAC,QAAQ,EAAE,EAAE,CAAC;YAClD,CAAC;YACD,KAAK,4BAA4B,CAAC,CAAC,CAAC;gBAClC;;;;;;;;;;mBAUG;gBACH,MAAM,aAAa,GAAG,IAAI,eAAe,CAAC,YAAY,CAAC,CAAC;gBACxD,aAAa,CAAC,GAAG,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;gBAC5C,OAAO,yBAAyB,aAAa,CAAC,QAAQ,EAAE,EAAE,CAAC;YAC7D,CAAC;YACD,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBACxB,OAAO,GAAG,QAAQ,IAAI,YAAY,CAAC,QAAQ,EAAE,EAAE,CAAC;YAClD,CAAC;YACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,OAAO,GAAG,QAAQ,IAAI,YAAY,CAAC,QAAQ,EAAE,EAAE,CAAC;YAClD,CAAC;YACD,KAAK,wBAAwB,CAAC,CAAC,CAAC;gBAC9B,OAAO,GAAG,QAAQ,IAAI,YAAY,CAAC,QAAQ,EAAE,EAAE,CAAC;YAClD,CAAC;YACD,KAAK,YAAY,CAAC;YAClB,KAAK,cAAc,CAAC;YACpB,KAAK,WAAW,CAAC;YACjB,KAAK,YAAY,CAAC;YAClB,KAAK,qBAAqB,CAAC;YAC3B,KAAK,gBAAgB,CAAC;YACtB,KAAK,kBAAkB,CAAC;YACxB,KAAK,eAAe,CAAC;YACrB,KAAK,gBAAgB,CAAC;YACtB,KAAK,aAAa,CAAC;YACnB,KAAK,eAAe,CAAC;YACrB,KAAK,aAAa,CAAC;YACnB,KAAK,aAAa;gBAChB,OAAO,GAAG,QAAQ,gBAAgB,kBAAkB,EAAE,CAAC;YACzD,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,qFAAqF;gBACrF,yFAAyF;gBACzF,OAAO,SAAS,CAAC;YACnB,CAAC;YACD;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;IACD,gBAAgB,EAAE,CAAC,EAAE,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,EAAE,EAAE;QAChE,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,EAAE,QAAwC,CAAC;QAC9E,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC,UAAU,IAAI,KAAK,EAAE,WAAW,CAAC;QACxE,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;QAC7B,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC;QAChC,IAAI,CAAC,UAAU;YAAE,OAAO,GAAG,OAAO,GAAG,QAAQ,EAAE,CAAC;QAChD,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACvC,IAAI,SAAS,GAAG,QAAQ,CAAC;YACzB,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,SAAS,GAAG,gBAAgB,UAAU,EAAE,CAAC;YAC3C,CAAC;iBAAM,CAAC;gBACN,SAAS,IAAI,gBAAgB,UAAU,EAAE,CAAC;YAC5C,CAAC;YACD,OAAO,GAAG,OAAO,GAAG,SAAS,EAAE,CAAC;QAClC,CAAC;QACD,OAAO,GAAG,OAAO,GAAG,QAAQ,EAAE,CAAC;IACjC,CAAC;CACF,CAAC;AAEF,eAAe,oBAAoB,CAAC"}

package/dist/providers/azure-openai/chatComplete.d.ts

@@ -0,0 +1,6 @@
+import { ChatCompletionResponse, ErrorResponse, ProviderConfig } from '../types.js';
+export declare const AzureOpenAIChatCompleteConfig: ProviderConfig;
+interface AzureOpenAIChatCompleteResponse extends ChatCompletionResponse {
+}
+export declare const AzureOpenAIResponseTransform: (response: AzureOpenAIChatCompleteResponse | ErrorResponse, responseStatus: number) => ChatCompletionResponse | ErrorResponse;
+export {};

package/dist/providers/azure-openai/chatComplete.js

@@ -0,0 +1,168 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs few custom changes for TFY
+ * 1. Update stream_options config
+ * ================================
+ */
+import { AZURE_OPEN_AI } from '../../globals.js';
+import { nestCursorFlatCustomTools } from '../open-ai-base/cursor.utils.js';
+import { OpenAIErrorResponseTransform } from '../openai/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 AzureOpenAIChatCompleteConfig = {
+    model: {
+        param: 'model',
+    },
+    messages: {
+        param: 'messages',
+        default: '',
+        transform: (params) => {
+            if (!params.messages) {
+                return params.messages;
+            }
+            return params.messages.map(msg => {
+                if (msg.content && Array.isArray(msg.content)) {
+                    return {
+                        ...msg,
+                        content: msg.content.map((item) => {
+                            if (item.type === 'file' && item.file) {
+                                const { mime_type: _mime_type, file_name, ...fileWithoutMimeType } = item.file;
+                                const filename = fileWithoutMimeType.filename ??
+                                    file_name ??
+                                    (fileWithoutMimeType.file_data ? 'my_file.pdf' : undefined);
+                                return {
+                                    ...item,
+                                    file: { ...fileWithoutMimeType, ...(filename && { filename }) },
+                                };
+                            }
+                            return item;
+                        }),
+                    };
+                }
+                return msg;
+            });
+        },
+    },
+    functions: {
+        param: 'functions',
+    },
+    function_call: {
+        param: 'function_call',
+    },
+    max_tokens: {
+        param: 'max_tokens',
+        min: 0,
+    },
+    max_completion_tokens: {
+        param: 'max_completion_tokens',
+        min: 0,
+    },
+    temperature: {
+        param: 'temperature',
+        default: 1,
+        min: 0,
+        max: 2,
+    },
+    top_p: {
+        param: 'top_p',
+        default: 1,
+        min: 0,
+        max: 1,
+    },
+    n: {
+        param: 'n',
+        default: 1,
+    },
+    logprobs: {
+        param: 'logprobs',
+        default: false,
+    },
+    top_logprobs: {
+        param: 'top_logprobs',
+    },
+    stream: {
+        param: 'stream',
+        default: false,
+    },
+    stop: {
+        param: 'stop',
+    },
+    presence_penalty: {
+        param: 'presence_penalty',
+        min: -2,
+        max: 2,
+    },
+    frequency_penalty: {
+        param: 'frequency_penalty',
+        min: -2,
+        max: 2,
+    },
+    logit_bias: {
+        param: 'logit_bias',
+    },
+    user: {
+        param: 'user',
+    },
+    tools: {
+        param: 'tools',
+        transform: (params, transformInput) => transformInput.callConfig.isCursorEndpoint ? nestCursorFlatCustomTools(params) : params.tools,
+    },
+    tool_choice: {
+        param: 'tool_choice',
+    },
+    response_format: {
+        param: 'response_format',
+    },
+    store: {
+        param: 'store',
+    },
+    metadata: {
+        param: 'metadata',
+    },
+    modalities: {
+        param: 'modalities',
+    },
+    audio: {
+        param: 'audio',
+    },
+    seed: {
+        param: 'seed',
+    },
+    prediction: {
+        param: 'prediction',
+    },
+    reasoning_effort: {
+        param: 'reasoning_effort',
+    },
+    stream_options: {
+        param: 'stream_options',
+        required: (params) => params.stream === true,
+        default: { include_usage: true },
+        transform: (params) => {
+            const existingStreamOptions = params.stream_options || {};
+            return {
+                ...existingStreamOptions,
+                include_usage: true,
+            };
+        },
+    },
+    web_search_options: {
+        param: 'web_search_options',
+    },
+    prompt_cache_key: {
+        param: 'prompt_cache_key',
+    },
+    safety_identifier: {
+        param: 'safety_identifier',
+    },
+    verbosity: {
+        param: 'verbosity',
+    },
+};
+export const AzureOpenAIResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200 && 'error' in response) {
+        return OpenAIErrorResponseTransform(response, AZURE_OPEN_AI);
+    }
+    return { ...response, provider: AZURE_OPEN_AI };
+};
+//# sourceMappingURL=chatComplete.js.map

package/dist/providers/azure-openai/chatComplete.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatComplete.js","sourceRoot":"","sources":["../../../src/providers/azure-openai/chatComplete.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C,OAAO,EAAE,yBAAyB,EAAE,MAAM,8BAA8B,CAAC;AACzE,OAAO,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAG/D,saAAsa;AAEta,MAAM,CAAC,MAAM,6BAA6B,GAAmB;IAC3D,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;KACf;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;QACjB,OAAO,EAAE,EAAE;QACX,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;YAC5B,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACrB,OAAO,MAAM,CAAC,QAAQ,CAAC;YACzB,CAAC;YACD,OAAO,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;gBAC/B,IAAI,GAAG,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC9C,OAAO;wBACL,GAAG,GAAG;wBACN,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE;4BACrC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gCACtC,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,mBAAmB,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC;gCAC/E,MAAM,QAAQ,GACZ,mBAAmB,CAAC,QAAQ;oCAC5B,SAAS;oCACT,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gCAC9D,OAAO;oCACL,GAAG,IAAI;oCACP,IAAI,EAAE,EAAE,GAAG,mBAAmB,EAAE,GAAG,CAAC,QAAQ,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE;iCAChE,CAAC;4BACJ,CAAC;4BACD,OAAO,IAAI,CAAC;wBACd,CAAC,CAAC;qBACH,CAAC;gBACJ,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;QACL,CAAC;KACF;IACD,SAAS,EAAE;QACT,KAAK,EAAE,WAAW;KACnB;IACD,aAAa,EAAE;QACb,KAAK,EAAE,eAAe;KACvB;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,GAAG,EAAE,CAAC;KACP;IACD,qBAAqB,EAAE;QACrB,KAAK,EAAE,uBAAuB;QAC9B,GAAG,EAAE,CAAC;KACP;IACD,WAAW,EAAE;QACX,KAAK,EAAE,aAAa;QACpB,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,CAAC,EAAE;QACD,KAAK,EAAE,GAAG;QACV,OAAO,EAAE,CAAC;KACX;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;QACjB,OAAO,EAAE,KAAK;KACf;IACD,YAAY,EAAE;QACZ,KAAK,EAAE,cAAc;KACtB;IACD,MAAM,EAAE;QACN,KAAK,EAAE,QAAQ;QACf,OAAO,EAAE,KAAK;KACf;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;KACd;IACD,gBAAgB,EAAE;QAChB,KAAK,EAAE,kBAAkB;QACzB,GAAG,EAAE,CAAC,CAAC;QACP,GAAG,EAAE,CAAC;KACP;IACD,iBAAiB,EAAE;QACjB,KAAK,EAAE,mBAAmB;QAC1B,GAAG,EAAE,CAAC,CAAC;QACP,GAAG,EAAE,CAAC;KACP;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;KACpB;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;KACd;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,SAAS,EAAE,CAAC,MAAc,EAAE,cAAsC,EAAE,EAAE,CACpE,cAAc,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK;KAChG;IACD,WAAW,EAAE;QACX,KAAK,EAAE,aAAa;KACrB;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;KACzB;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;KACf;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;KAClB;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;KACpB;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;KACf;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;KACd;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;KACpB;IACD,gBAAgB,EAAE;QAChB,KAAK,EAAE,kBAAkB;KAC1B;IACD,cAAc,EAAE;QACd,KAAK,EAAE,gBAAgB;QACvB,QAAQ,EAAE,CAAC,MAAc,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,IAAI;QACpD,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;QAChC,SAAS,EAAE,CAAC,MAAW,EAAE,EAAE;YACzB,MAAM,qBAAqB,GAAG,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;YAC1D,OAAO;gBACL,GAAG,qBAAqB;gBACxB,aAAa,EAAE,IAAI;aACpB,CAAC;QACJ,CAAC;KACF;IACD,kBAAkB,EAAE;QAClB,KAAK,EAAE,oBAAoB;KAC5B;IACD,gBAAgB,EAAE;QAChB,KAAK,EAAE,kBAAkB;KAC1B;IACD,iBAAiB,EAAE;QACjB,KAAK,EAAE,mBAAmB;KAC3B;IACD,SAAS,EAAE;QACT,KAAK,EAAE,WAAW;KACnB;CACF,CAAC;AAIF,MAAM,CAAC,MAAM,4BAA4B,GAGK,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IACzE,IAAI,cAAc,KAAK,GAAG,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;QAClD,OAAO,4BAA4B,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,EAAE,GAAG,QAAQ,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC;AAClD,CAAC,CAAC"}

package/dist/providers/azure-openai/complete.d.ts

@@ -0,0 +1,6 @@
+import { CompletionResponse, ErrorResponse, ProviderConfig } from '../types.js';
+export declare const AzureOpenAICompleteConfig: ProviderConfig;
+interface AzureOpenAICompleteResponse extends CompletionResponse {
+}
+export declare const AzureOpenAICompleteResponseTransform: (response: AzureOpenAICompleteResponse | ErrorResponse, responseStatus: number) => CompletionResponse | ErrorResponse;
+export {};