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

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

@@ -0,0 +1,209 @@
+import { GITHUB } from '../../globals.js';
+import { sdkRequestUrl } from '../../utils/sdkRequestUrl.js';
+import { AZURE_FOUNDRY_BETA_KEY, resolveAnthropicBetas } from '../anthropic/anthropicBetas.js';
+import { mergeAnthropicBetasWithOverride } from '../anthropic/utils.js';
+import { getAccessTokenFromEntraId, getAccessTokenFromEntraIdUsingCertificate } from '../azure-openai/utils.js';
+import { getAccessTokenFromWorkloadIdentity } from '../azure-openai/wifAuth.js';
+const DEFAULT_ANTHROPIC_VERSION = '2023-06-01';
+function getAzureFoundryModel(providerOptions) {
+    const manifest = providerOptions.model?.manifest;
+    if (manifest?.type === 'integration/model/azure-foundry') {
+        return manifest;
+    }
+    return undefined;
+}
+export function resolveAzureFoundryUrl({ providerOptions, headerOptions, }) {
+    // Model-owned endpoint wins so request headers cannot redirect model credentials.
+    return getAzureFoundryModel(providerOptions)?.azure_endpoint ?? headerOptions.azureAiInference.azureFoundryUrl;
+}
+/**
+ * Returns anthropic-specific headers when the target model is an Anthropic model.
+ */
+function buildAnthropicHeaders({ headerOptions, callConfig, requestBody, isAnthropicModel, }) {
+    if (!isAnthropicModel)
+        return {};
+    const version = headerOptions.anthropic.anthropicVersion || DEFAULT_ANTHROPIC_VERSION;
+    const betaCsv = resolveAnthropicBetas(mergeAnthropicBetasWithOverride({
+        anthropicBeta: headerOptions.anthropic.anthropicBeta,
+        anthropicBetaOverride: headerOptions.anthropic.anthropicBetaOverride,
+        anthropicBetaSuppressed: callConfig.anthropicBetaSuppressed,
+    }, requestBody?.['anthropic_beta']), AZURE_FOUNDRY_BETA_KEY)?.join(',');
+    return {
+        'anthropic-version': version,
+        ...(betaCsv && { 'anthropic-beta': betaCsv }),
+    };
+}
+const AzureAIInferenceAPI = {
+    getBaseURL: ({ providerOptions, headerOptions }) => {
+        const provider = providerOptions.providerAccount.provider;
+        if (provider === GITHUB) {
+            return 'https://models.inference.ai.azure.com';
+        }
+        const azureFoundryUrl = resolveAzureFoundryUrl({ providerOptions, headerOptions });
+        if (azureFoundryUrl) {
+            return azureFoundryUrl;
+        }
+        // Legacy Portkey-style managed endpoint construction from request headers.
+        const azureEndpointName = headerOptions.azureAiInference.azureEndpointName;
+        if (azureEndpointName) {
+            return `https://${azureEndpointName}.inference.ml.azure.com/score`;
+        }
+        return '';
+    },
+    headers: async ({ providerOptions, headerOptions, callConfig, request }) => {
+        const model = getAzureFoundryModel(providerOptions);
+        const authData = model?.auth_data;
+        const azureFoundryUrl = resolveAzureFoundryUrl({ providerOptions, headerOptions });
+        const body = request.body;
+        const isAnthropicModel = !!azureFoundryUrl?.includes('anthropic');
+        const anthropicHeaders = buildAnthropicHeaders({
+            headerOptions,
+            callConfig,
+            requestBody: body,
+            isAnthropicModel,
+        });
+        const azureExtraParams = headerOptions.azureAiInference.azureExtraParams;
+        const baseHeaders = {
+            ...anthropicHeaders,
+            ...(!isAnthropicModel && { 'extra-parameters': azureExtraParams || 'pass-through' }),
+        };
+        const azureAuthMode = headerOptions.azure.azureAuthMode ?? authData?.type;
+        const azureEntraTenantId = headerOptions.azure.azureEntraTenantId ?? (authData && 'tenant_id' in authData ? authData.tenant_id : undefined);
+        const azureEntraClientId = headerOptions.azure.azureEntraClientId ?? (authData && 'client_id' in authData ? authData.client_id : undefined);
+        const azureEntraClientSecret = headerOptions.azure.azureEntraClientSecret ??
+            (authData?.type === 'client-secret' ? authData.client_secret : undefined);
+        if (azureAuthMode === 'certificate' && authData?.type === 'certificate') {
+            if (azureEntraTenantId && azureEntraClientId && authData.certificate && authData.certificate_thumbprint) {
+                const accessToken = await getAccessTokenFromEntraIdUsingCertificate(azureEntraTenantId, azureEntraClientId, authData.certificate, authData.certificate_thumbprint);
+                baseHeaders['Authorization'] = `Bearer ${accessToken}`;
+            }
+            return baseHeaders;
+        }
+        if (azureAuthMode === 'client-secret') {
+            if (azureEntraTenantId && azureEntraClientId && azureEntraClientSecret) {
+                const scope = 'https://cognitiveservices.azure.com/.default';
+                const accessToken = await getAccessTokenFromEntraId(azureEntraTenantId, azureEntraClientId, azureEntraClientSecret, scope);
+                baseHeaders['Authorization'] = `Bearer ${accessToken}`;
+            }
+            return baseHeaders;
+        }
+        if (azureAuthMode === 'workload-identity' && authData?.type === 'workload-identity') {
+            const azureWorkloadTenantId = authData.tenant_id;
+            const azureWorkloadClientId = authData.client_id;
+            if (azureWorkloadTenantId && azureWorkloadClientId) {
+                const accessToken = await getAccessTokenFromWorkloadIdentity(azureWorkloadTenantId, azureWorkloadClientId);
+                baseHeaders['Authorization'] = `Bearer ${accessToken}`;
+            }
+            return baseHeaders;
+        }
+        // Default: API key auth (account key or header AD token)
+        const apiKey = (authData?.type === 'api-key' ? authData.api_key : undefined) ?? headerOptions.azure.azureAdToken;
+        if (apiKey) {
+            if (isAnthropicModel) {
+                baseHeaders['x-api-key'] = apiKey;
+            }
+            else {
+                baseHeaders['Authorization'] = `Bearer ${apiKey}`;
+            }
+        }
+        return baseHeaders;
+    },
+    getEndpoint: ({ providerOptions, headerOptions, endpoint, request }) => {
+        const model = getAzureFoundryModel(providerOptions);
+        const azureFoundryUrl = resolveAzureFoundryUrl({ providerOptions, headerOptions });
+        const azureApiVersion = headerOptions.azureAiInference.azureApiVersion ?? model?.api_version;
+        const deploymentId = headerOptions.azure.deploymentId ?? model?.model_id;
+        const isAnthropicModel = !!azureFoundryUrl?.includes('anthropic');
+        let mappedFn = endpoint;
+        const ENDPOINT_MAPPING = {
+            complete: '/completions',
+            chatComplete: isAnthropicModel ? '/v1/messages' : '/chat/completions',
+            messages: '/v1/messages',
+            embed: '/embeddings',
+        };
+        const isGithub = providerOptions.providerAccount.provider === GITHUB;
+        const pathForProxy = request.path;
+        if (endpoint === 'proxy') {
+            if (pathForProxy.indexOf('/chat/completions') > -1) {
+                mappedFn = 'chatComplete';
+            }
+            else if (pathForProxy.indexOf('/completions') > -1) {
+                mappedFn = 'complete';
+            }
+            else if (pathForProxy.indexOf('/embeddings') > -1) {
+                mappedFn = 'embed';
+            }
+            else if (pathForProxy.indexOf('/audio/speech') > -1) {
+                mappedFn = 'createSpeech';
+            }
+            else if (pathForProxy.indexOf('/audio/transcriptions') > -1) {
+                mappedFn = 'createTranscription';
+            }
+            else if (pathForProxy.indexOf('/audio/translations') > -1) {
+                mappedFn = 'createTranslation';
+            }
+            else if (pathForProxy.indexOf('/rerank') > -1) {
+                mappedFn = 'rerank';
+            }
+        }
+        const urlObj = new URL(sdkRequestUrl(request));
+        const pathname = urlObj.pathname.replace('/api/llm', '');
+        const searchParams = urlObj.searchParams;
+        if (azureApiVersion) {
+            searchParams.set('api-version', azureApiVersion);
+        }
+        switch (mappedFn) {
+            case 'complete':
+            case 'chatComplete':
+            case 'embed': {
+                const endpoint = ENDPOINT_MAPPING[mappedFn];
+                if (isGithub) {
+                    return endpoint;
+                }
+                return azureApiVersion ? `${endpoint}?api-version=${azureApiVersion}` : endpoint;
+            }
+            case 'messages': {
+                return ENDPOINT_MAPPING[mappedFn];
+            }
+            case 'createSpeech': {
+                const speechPath = `/openai/deployments/${deploymentId}/audio/speech`;
+                return azureApiVersion ? `${speechPath}?api-version=${azureApiVersion}` : speechPath;
+            }
+            case 'createTranscription': {
+                const transcriptionPath = `/openai/deployments/${deploymentId}/audio/transcriptions`;
+                return azureApiVersion ? `${transcriptionPath}?api-version=${azureApiVersion}` : transcriptionPath;
+            }
+            case 'createTranslation': {
+                const translationPath = `/openai/deployments/${deploymentId}/audio/translations`;
+                return azureApiVersion ? `${translationPath}?api-version=${azureApiVersion}` : translationPath;
+            }
+            case 'realtime': {
+                if (!deploymentId) {
+                    throw new Error('deploymentId is required for realtime endpoint');
+                }
+                return `/openai/v1/realtime?model=${encodeURIComponent(deploymentId)}`;
+            }
+            case 'createModelResponse': {
+                return `/responses?${searchParams.toString()}`;
+            }
+            case 'getModelResponse': {
+                return `${pathname}?${searchParams.toString()}`;
+            }
+            case 'deleteModelResponse': {
+                return `${pathname}?${searchParams.toString()}`;
+            }
+            case 'listResponseInputItems': {
+                return `${pathname}?${searchParams.toString()}`;
+            }
+            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 '';
+        }
+    },
+};
+export default AzureAIInferenceAPI;
+//# sourceMappingURL=api.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/azure-ai-inference/api.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAEvC,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAC5F,OAAO,EAAE,+BAA+B,EAAE,MAAM,oBAAoB,CAAC;AACrE,OAAO,EAAE,yBAAyB,EAAE,yCAAyC,EAAE,MAAM,uBAAuB,CAAC;AAC7G,OAAO,EAAE,kCAAkC,EAAE,MAAM,yBAAyB,CAAC;AAG7E,MAAM,yBAAyB,GAAG,YAAY,CAAC;AAE/C,SAAS,oBAAoB,CAAC,eAAuD;IACnF,MAAM,QAAQ,GAAG,eAAe,CAAC,KAAK,EAAE,QAAQ,CAAC;IACjD,IAAI,QAAQ,EAAE,IAAI,KAAK,iCAAiC,EAAE,CAAC;QACzD,OAAO,QAA6B,CAAC;IACvC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,EACrC,eAAe,EACf,aAAa,GACkD;IAC/D,kFAAkF;IAClF,OAAO,oBAAoB,CAAC,eAAe,CAAC,EAAE,cAAc,IAAI,aAAa,CAAC,gBAAgB,CAAC,eAAe,CAAC;AACjH,CAAC;AAED;;GAEG;AACH,SAAS,qBAAqB,CAAC,EAC7B,aAAa,EACb,UAAU,EACV,WAAW,EACX,gBAAgB,GAMjB;IACC,IAAI,CAAC,gBAAgB;QAAE,OAAO,EAAE,CAAC;IACjC,MAAM,OAAO,GAAG,aAAa,CAAC,SAAS,CAAC,gBAAgB,IAAI,yBAAyB,CAAC;IACtF,MAAM,OAAO,GAAG,qBAAqB,CACnC,+BAA+B,CAC7B;QACE,aAAa,EAAE,aAAa,CAAC,SAAS,CAAC,aAAa;QACpD,qBAAqB,EAAE,aAAa,CAAC,SAAS,CAAC,qBAAqB;QACpE,uBAAuB,EAAE,UAAU,CAAC,uBAAuB;KAC5D,EACD,WAAW,EAAE,CAAC,gBAAgB,CAAkC,CACjE,EACD,sBAAsB,CACvB,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;IACb,OAAO;QACL,mBAAmB,EAAE,OAAO;QAC5B,GAAG,CAAC,OAAO,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,CAAC;KAC9C,CAAC;AACJ,CAAC;AAED,MAAM,mBAAmB,GAAsB;IAC7C,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,aAAa,EAAE,EAAE,EAAE;QACjD,MAAM,QAAQ,GAAG,eAAe,CAAC,eAAe,CAAC,QAAQ,CAAC;QAC1D,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;YACxB,OAAO,uCAAuC,CAAC;QACjD,CAAC;QAED,MAAM,eAAe,GAAG,sBAAsB,CAAC,EAAE,eAAe,EAAE,aAAa,EAAE,CAAC,CAAC;QACnF,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,eAAe,CAAC;QACzB,CAAC;QAED,2EAA2E;QAC3E,MAAM,iBAAiB,GAAG,aAAa,CAAC,gBAAgB,CAAC,iBAAiB,CAAC;QAC3E,IAAI,iBAAiB,EAAE,CAAC;YACtB,OAAO,WAAW,iBAAiB,+BAA+B,CAAC;QACrE,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,EAAE,eAAe,EAAE,aAAa,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE,EAAE;QACzE,MAAM,KAAK,GAAG,oBAAoB,CAAC,eAAe,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,KAAK,EAAE,SAAS,CAAC;QAClC,MAAM,eAAe,GAAG,sBAAsB,CAAC,EAAE,eAAe,EAAE,aAAa,EAAE,CAAC,CAAC;QACnF,MAAM,IAAI,GAAG,OAAO,CAAC,IAA0B,CAAC;QAChD,MAAM,gBAAgB,GAAG,CAAC,CAAC,eAAe,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAC;QAElE,MAAM,gBAAgB,GAAG,qBAAqB,CAAC;YAC7C,aAAa;YACb,UAAU;YACV,WAAW,EAAE,IAA2C;YACxD,gBAAgB;SACjB,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,aAAa,CAAC,gBAAgB,CAAC,gBAAgB,CAAC;QACzE,MAAM,WAAW,GAA2B;YAC1C,GAAG,gBAAgB;YACnB,GAAG,CAAC,CAAC,gBAAgB,IAAI,EAAE,kBAAkB,EAAE,gBAAgB,IAAI,cAAc,EAAE,CAAC;SACrF,CAAC;QAEF,MAAM,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,aAAa,IAAI,QAAQ,EAAE,IAAI,CAAC;QAC1E,MAAM,kBAAkB,GACtB,aAAa,CAAC,KAAK,CAAC,kBAAkB,IAAI,CAAC,QAAQ,IAAI,WAAW,IAAI,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACnH,MAAM,kBAAkB,GACtB,aAAa,CAAC,KAAK,CAAC,kBAAkB,IAAI,CAAC,QAAQ,IAAI,WAAW,IAAI,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACnH,MAAM,sBAAsB,GAC1B,aAAa,CAAC,KAAK,CAAC,sBAAsB;YAC1C,CAAC,QAAQ,EAAE,IAAI,KAAK,eAAe,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAE5E,IAAI,aAAa,KAAK,aAAa,IAAI,QAAQ,EAAE,IAAI,KAAK,aAAa,EAAE,CAAC;YACxE,IAAI,kBAAkB,IAAI,kBAAkB,IAAI,QAAQ,CAAC,WAAW,IAAI,QAAQ,CAAC,sBAAsB,EAAE,CAAC;gBACxG,MAAM,WAAW,GAAG,MAAM,yCAAyC,CACjE,kBAAkB,EAClB,kBAAkB,EAClB,QAAQ,CAAC,WAAW,EACpB,QAAQ,CAAC,sBAAsB,CAChC,CAAC;gBACF,WAAW,CAAC,eAAe,CAAC,GAAG,UAAU,WAAW,EAAE,CAAC;YACzD,CAAC;YACD,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,IAAI,aAAa,KAAK,eAAe,EAAE,CAAC;YACtC,IAAI,kBAAkB,IAAI,kBAAkB,IAAI,sBAAsB,EAAE,CAAC;gBACvE,MAAM,KAAK,GAAG,8CAA8C,CAAC;gBAC7D,MAAM,WAAW,GAAG,MAAM,yBAAyB,CACjD,kBAAkB,EAClB,kBAAkB,EAClB,sBAAsB,EACtB,KAAK,CACN,CAAC;gBACF,WAAW,CAAC,eAAe,CAAC,GAAG,UAAU,WAAW,EAAE,CAAC;YACzD,CAAC;YACD,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,IAAI,aAAa,KAAK,mBAAmB,IAAI,QAAQ,EAAE,IAAI,KAAK,mBAAmB,EAAE,CAAC;YACpF,MAAM,qBAAqB,GAAG,QAAQ,CAAC,SAAS,CAAC;YACjD,MAAM,qBAAqB,GAAG,QAAQ,CAAC,SAAS,CAAC;YACjD,IAAI,qBAAqB,IAAI,qBAAqB,EAAE,CAAC;gBACnD,MAAM,WAAW,GAAG,MAAM,kCAAkC,CAAC,qBAAqB,EAAE,qBAAqB,CAAC,CAAC;gBAC3G,WAAW,CAAC,eAAe,CAAC,GAAG,UAAU,WAAW,EAAE,CAAC;YACzD,CAAC;YACD,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,yDAAyD;QACzD,MAAM,MAAM,GAAG,CAAC,QAAQ,EAAE,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,aAAa,CAAC,KAAK,CAAC,YAAY,CAAC;QAEjH,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,gBAAgB,EAAE,CAAC;gBACrB,WAAW,CAAC,WAAW,CAAC,GAAG,MAAM,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,eAAe,CAAC,GAAG,UAAU,MAAM,EAAE,CAAC;YACpD,CAAC;QACH,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,WAAW,EAAE,CAAC,EAAE,eAAe,EAAE,aAAa,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;QACrE,MAAM,KAAK,GAAG,oBAAoB,CAAC,eAAe,CAAC,CAAC;QACpD,MAAM,eAAe,GAAG,sBAAsB,CAAC,EAAE,eAAe,EAAE,aAAa,EAAE,CAAC,CAAC;QACnF,MAAM,eAAe,GAAG,aAAa,CAAC,gBAAgB,CAAC,eAAe,IAAI,KAAK,EAAE,WAAW,CAAC;QAC7F,MAAM,YAAY,GAAG,aAAa,CAAC,KAAK,CAAC,YAAY,IAAI,KAAK,EAAE,QAAQ,CAAC;QAEzE,MAAM,gBAAgB,GAAG,CAAC,CAAC,eAAe,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAC;QAElE,IAAI,QAAQ,GAAG,QAAQ,CAAC;QAExB,MAAM,gBAAgB,GAA2B;YAC/C,QAAQ,EAAE,cAAc;YACxB,YAAY,EAAE,gBAAgB,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,mBAAmB;YACrE,QAAQ,EAAE,cAAc;YACxB,KAAK,EAAE,aAAa;SACrB,CAAC;QAEF,MAAM,QAAQ,GAAG,eAAe,CAAC,eAAe,CAAC,QAAQ,KAAK,MAAM,CAAC;QACrE,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;QAElC,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;YACzB,IAAI,YAAY,CAAC,OAAO,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBACnD,QAAQ,GAAG,cAAc,CAAC;YAC5B,CAAC;iBAAM,IAAI,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBACrD,QAAQ,GAAG,UAAU,CAAC;YACxB,CAAC;iBAAM,IAAI,YAAY,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBACpD,QAAQ,GAAG,OAAO,CAAC;YACrB,CAAC;iBAAM,IAAI,YAAY,CAAC,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBACtD,QAAQ,GAAG,cAAc,CAAC;YAC5B,CAAC;iBAAM,IAAI,YAAY,CAAC,OAAO,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBAC9D,QAAQ,GAAG,qBAAqB,CAAC;YACnC,CAAC;iBAAM,IAAI,YAAY,CAAC,OAAO,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBAC5D,QAAQ,GAAG,mBAAmB,CAAC;YACjC,CAAC;iBAAM,IAAI,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBAChD,QAAQ,GAAG,QAAQ,CAAC;YACtB,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;QACzD,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QACzC,IAAI,eAAe,EAAE,CAAC;YACpB,YAAY,CAAC,GAAG,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;QACnD,CAAC;QAED,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,UAAU,CAAC;YAChB,KAAK,cAAc,CAAC;YACpB,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,MAAM,QAAQ,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;gBAC5C,IAAI,QAAQ,EAAE,CAAC;oBACb,OAAO,QAAQ,CAAC;gBAClB,CAAC;gBACD,OAAO,eAAe,CAAC,CAAC,CAAC,GAAG,QAAQ,gBAAgB,eAAe,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;YACnF,CAAC;YACD,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,OAAO,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YACpC,CAAC;YACD,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,MAAM,UAAU,GAAG,uBAAuB,YAAY,eAAe,CAAC;gBACtE,OAAO,eAAe,CAAC,CAAC,CAAC,GAAG,UAAU,gBAAgB,eAAe,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;YACvF,CAAC;YACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,MAAM,iBAAiB,GAAG,uBAAuB,YAAY,uBAAuB,CAAC;gBACrF,OAAO,eAAe,CAAC,CAAC,CAAC,GAAG,iBAAiB,gBAAgB,eAAe,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC;YACrG,CAAC;YACD,KAAK,mBAAmB,CAAC,CAAC,CAAC;gBACzB,MAAM,eAAe,GAAG,uBAAuB,YAAY,qBAAqB,CAAC;gBACjF,OAAO,eAAe,CAAC,CAAC,CAAC,GAAG,eAAe,gBAAgB,eAAe,EAAE,CAAC,CAAC,CAAC,eAAe,CAAC;YACjG,CAAC;YACD,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;gBACpE,CAAC;gBACD,OAAO,6BAA6B,kBAAkB,CAAC,YAAY,CAAC,EAAE,CAAC;YACzE,CAAC;YACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,OAAO,cAAc,YAAY,CAAC,QAAQ,EAAE,EAAE,CAAC;YACjD,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,QAAQ,CAAC,CAAC,CAAC;gBACd,qFAAqF;gBACrF,yFAAyF;gBACzF,OAAO,SAAS,CAAC;YACnB,CAAC;YACD;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAC;AAEF,eAAe,mBAAmB,CAAC"}

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

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

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

@@ -0,0 +1,78 @@
+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 AzureAIInferenceChatCompleteConfig = {
+    model: {
+        param: 'model',
+        required: false,
+    },
+    messages: {
+        param: 'messages',
+        default: '',
+        transform: (params) => {
+            return params.messages?.map(message => {
+                if (message.role === 'developer')
+                    return { ...message, role: 'system' };
+                return message;
+            });
+        },
+    },
+    max_tokens: {
+        param: 'max_tokens',
+        min: 0,
+    },
+    max_completion_tokens: {
+        param: 'max_tokens',
+        min: 0,
+    },
+    temperature: {
+        param: 'temperature',
+        default: 1,
+        min: 0,
+        max: 2,
+    },
+    top_p: {
+        param: 'top_p',
+        default: 1,
+        min: 0,
+        max: 1,
+    },
+    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,
+    },
+    user: {
+        param: 'user',
+    },
+    tools: {
+        param: 'tools',
+    },
+    tool_choice: {
+        param: 'tool_choice',
+    },
+    response_format: {
+        param: 'response_format',
+    },
+};
+export const AzureAIInferenceChatCompleteResponseTransform = (provider) => {
+    const transformer = (response, responseStatus) => {
+        if (responseStatus !== 200 && 'error' in response) {
+            return OpenAIErrorResponseTransform(response, provider);
+        }
+        return { ...response, provider: provider };
+    };
+    return transformer;
+};
+//# sourceMappingURL=chatComplete.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"chatComplete.js","sourceRoot":"","sources":["../../../src/providers/azure-ai-inference/chatComplete.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAG/D,saAAsa;AACta,MAAM,CAAC,MAAM,kCAAkC,GAAmB;IAChE,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;KAChB;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;QACjB,OAAO,EAAE,EAAE;QACX,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;YAC5B,OAAO,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE;gBACpC,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;oBAAE,OAAO,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gBACxE,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC;KACF;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,GAAG,EAAE,CAAC;KACP;IACD,qBAAqB,EAAE;QACrB,KAAK,EAAE,YAAY;QACnB,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,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,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;KACd;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;KACf;IACD,WAAW,EAAE;QACX,KAAK,EAAE,aAAa;KACrB;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;KACzB;CACF,CAAC;AAIF,MAAM,CAAC,MAAM,6CAA6C,GAAG,CAAC,QAAgB,EAAE,EAAE;IAChF,MAAM,WAAW,GAG6B,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;QACzE,IAAI,cAAc,KAAK,GAAG,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;YAClD,OAAO,4BAA4B,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,EAAE,GAAG,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;IAC7C,CAAC,CAAC;IACF,OAAO,WAAW,CAAC;AACrB,CAAC,CAAC"}

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

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

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

@@ -0,0 +1,63 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+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 AzureAIInferenceCompleteConfig = {
+    model: {
+        param: 'model',
+        required: false,
+    },
+    prompt: {
+        param: 'prompt',
+        default: '',
+    },
+    max_tokens: {
+        param: 'max_tokens',
+        min: 0,
+    },
+    temperature: {
+        param: 'temperature',
+        default: 1,
+        min: 0,
+        max: 2,
+    },
+    top_p: {
+        param: 'top_p',
+        default: 1,
+        min: 0,
+        max: 1,
+    },
+    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,
+    },
+    user: {
+        param: 'user',
+    },
+};
+export const AzureAIInferenceCompleteResponseTransform = (provider) => {
+    const transformer = (response, responseStatus) => {
+        if (responseStatus !== 200 && 'error' in response) {
+            return OpenAIErrorResponseTransform(response, provider);
+        }
+        return { ...response, provider: provider };
+    };
+    return transformer;
+};
+//# sourceMappingURL=complete.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"complete.js","sourceRoot":"","sources":["../../../src/providers/azure-ai-inference/complete.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAG/D,saAAsa;AAEta,MAAM,CAAC,MAAM,8BAA8B,GAAmB;IAC5D,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;KAChB;IACD,MAAM,EAAE;QACN,KAAK,EAAE,QAAQ;QACf,OAAO,EAAE,EAAE;KACZ;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,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,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,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;KACd;CACF,CAAC;AAIF,MAAM,CAAC,MAAM,yCAAyC,GAAG,CAAC,QAAgB,EAAE,EAAE;IAC5E,MAAM,WAAW,GAGyB,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;QACrE,IAAI,cAAc,KAAK,GAAG,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;YAClD,OAAO,4BAA4B,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,EAAE,GAAG,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;IAC7C,CAAC,CAAC;IAEF,OAAO,WAAW,CAAC;AACrB,CAAC,CAAC"}

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

@@ -0,0 +1,5 @@
+import type { Params } from '../../types/requestBody.js';
+import { ErrorResponse, ProviderConfig, type RequestHandler } from '../types.js';
+export declare const AzureAIInferenceCreateSpeechConfig: ProviderConfig;
+export declare const AzureAIInferenceCreateSpeechRequestHandler: RequestHandler<Params>;
+export declare const AzureAIInferenceCreateSpeechResponseTransform: (response: Response | ErrorResponse, responseStatus: number) => Response | ErrorResponse;

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

@@ -0,0 +1,58 @@
+import { AZURE_AI_INFERENCE } from '../../globals.js';
+import { safeFetch } from '../../http/index.js';
+import { OpenAIErrorResponseTransform } from '../openai/utils.js';
+import { providerApiArgsFromHandler } from '../requestHandlerApiArgs.js';
+import AzureAIInferenceAPI from './api.js';
+export const AzureAIInferenceCreateSpeechConfig = {
+    model: {
+        param: 'model',
+        required: true,
+        default: 'tts-1',
+    },
+    input: {
+        param: 'input',
+        required: true,
+    },
+    voice: {
+        param: 'voice',
+        required: true,
+        default: 'alloy',
+    },
+    response_format: {
+        param: 'response_format',
+        required: false,
+        default: 'mp3',
+    },
+    speed: {
+        param: 'speed',
+        required: false,
+        default: 1,
+    },
+};
+export const AzureAIInferenceCreateSpeechRequestHandler = async (args) => {
+    const requestBody = args.request.body;
+    const apiArgs = providerApiArgsFromHandler(args);
+    const baseUrl = await AzureAIInferenceAPI.getBaseURL(apiArgs);
+    const endpoint = AzureAIInferenceAPI.getEndpoint(apiArgs);
+    const url = `${baseUrl}${endpoint}`;
+    const body = { ...requestBody, model: args.modelId };
+    const headers = await AzureAIInferenceAPI.headers({
+        ...apiArgs,
+        transformedRequestBody: body,
+        transformedRequestUrl: url,
+    });
+    return safeFetch(url, {
+        ssrf: 'guard',
+        method: 'POST',
+        headers: { ...headers, 'Content-Type': 'application/json' },
+        body: JSON.stringify(body),
+        signal: args.signal,
+    });
+};
+export const AzureAIInferenceCreateSpeechResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200 && 'error' in response) {
+        return OpenAIErrorResponseTransform(response, AZURE_AI_INFERENCE);
+    }
+    return response;
+};
+//# sourceMappingURL=createSpeech.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"createSpeech.js","sourceRoot":"","sources":["../../../src/providers/azure-ai-inference/createSpeech.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,OAAO,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AAEtE,OAAO,mBAAmB,MAAM,OAAO,CAAC;AAExC,MAAM,CAAC,MAAM,kCAAkC,GAAmB;IAChE,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,OAAO;KACjB;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,OAAO;KACjB;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,KAAK;QACf,OAAO,EAAE,KAAK;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;QACf,OAAO,EAAE,CAAC;KACX;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,0CAA0C,GAA2B,KAAK,EAAC,IAAI,EAAC,EAAE;IAC7F,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IACtC,MAAM,OAAO,GAAG,0BAA0B,CAAC,IAAI,CAAC,CAAC;IACjD,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAC9D,MAAM,QAAQ,GAAG,mBAAmB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAC1D,MAAM,GAAG,GAAG,GAAG,OAAO,GAAG,QAAQ,EAAE,CAAC;IACpC,MAAM,IAAI,GAAG,EAAE,GAAG,WAAW,EAAE,KAAK,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC;IACrD,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC,OAAO,CAAC;QAChD,GAAG,OAAO;QACV,sBAAsB,EAAE,IAAI;QAC5B,qBAAqB,EAAE,GAAG;KAC3B,CAAC,CAAC;IACH,OAAO,SAAS,CAAC,GAAG,EAAE;QACpB,IAAI,EAAE,OAAO;QACb,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE;QAC3D,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;QAC1B,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,6CAA6C,GAG1B,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IAC3D,IAAI,cAAc,KAAK,GAAG,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;QAClD,OAAO,4BAA4B,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;IACpE,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC"}

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

@@ -0,0 +1,4 @@
+import type { RequestHandler } from '../types.js';
+import { ErrorResponse } from '../types.js';
+export declare const AzureAIInferenceCreateTranscriptionRequestHandler: RequestHandler<FormData>;
+export declare const AzureAIInferenceCreateTranscriptionResponseTransform: (response: Response | ErrorResponse, responseStatus: number) => Response | ErrorResponse;

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

@@ -0,0 +1,29 @@
+import { AZURE_AI_INFERENCE } from '../../globals.js';
+import { safeFetch } from '../../http/index.js';
+import { OpenAIErrorResponseTransform } from '../openai/utils.js';
+import { providerApiArgsFromHandler } from '../requestHandlerApiArgs.js';
+import AzureAIInferenceAPI from './api.js';
+export const AzureAIInferenceCreateTranscriptionRequestHandler = async (args) => {
+    const requestBody = args.request.body;
+    const apiArgs = providerApiArgsFromHandler(args);
+    const baseUrl = await AzureAIInferenceAPI.getBaseURL(apiArgs);
+    const endpoint = AzureAIInferenceAPI.getEndpoint(apiArgs);
+    const url = `${baseUrl}${endpoint}`;
+    if (requestBody.has('model')) {
+        requestBody.delete('model');
+    }
+    requestBody.append('model', args.modelId);
+    const headers = await AzureAIInferenceAPI.headers({
+        ...apiArgs,
+        transformedRequestBody: requestBody,
+        transformedRequestUrl: url,
+    });
+    return safeFetch(url, { ssrf: 'guard', method: 'POST', headers, body: requestBody, signal: args.signal });
+};
+export const AzureAIInferenceCreateTranscriptionResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200 && 'error' in response) {
+        return OpenAIErrorResponseTransform(response, AZURE_AI_INFERENCE);
+    }
+    return response;
+};
+//# sourceMappingURL=createTranscription.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"createTranscription.js","sourceRoot":"","sources":["../../../src/providers/azure-ai-inference/createTranscription.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AAGtE,OAAO,mBAAmB,MAAM,OAAO,CAAC;AAExC,MAAM,CAAC,MAAM,iDAAiD,GAA6B,KAAK,EAAC,IAAI,EAAC,EAAE;IACtG,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IACtC,MAAM,OAAO,GAAG,0BAA0B,CAAC,IAAI,CAAC,CAAC;IACjD,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAC9D,MAAM,QAAQ,GAAG,mBAAmB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAC1D,MAAM,GAAG,GAAG,GAAG,OAAO,GAAG,QAAQ,EAAE,CAAC;IACpC,IAAI,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QAC7B,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IACD,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1C,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC,OAAO,CAAC;QAChD,GAAG,OAAO;QACV,sBAAsB,EAAE,WAAW;QACnC,qBAAqB,EAAE,GAAG;KAC3B,CAAC,CAAC;IACH,OAAO,SAAS,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;AAC5G,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,oDAAoD,GAGjC,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IAC3D,IAAI,cAAc,KAAK,GAAG,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;QAClD,OAAO,4BAA4B,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;IACpE,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC"}

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

@@ -0,0 +1,4 @@
+import type { RequestHandler } from '../types.js';
+import { ErrorResponse } from '../types.js';
+export declare const AzureAIInferenceCreateTranslationRequestHandler: RequestHandler<FormData>;
+export declare const AzureAIInferenceCreateTranslationResponseTransform: (response: Response | ErrorResponse, responseStatus: number) => Response | ErrorResponse;

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

@@ -0,0 +1,29 @@
+import { AZURE_AI_INFERENCE } from '../../globals.js';
+import { safeFetch } from '../../http/index.js';
+import { OpenAIErrorResponseTransform } from '../openai/utils.js';
+import { providerApiArgsFromHandler } from '../requestHandlerApiArgs.js';
+import AzureAIInferenceAPI from './api.js';
+export const AzureAIInferenceCreateTranslationRequestHandler = async (args) => {
+    const requestBody = args.request.body;
+    const apiArgs = providerApiArgsFromHandler(args);
+    const baseUrl = await AzureAIInferenceAPI.getBaseURL(apiArgs);
+    const endpoint = AzureAIInferenceAPI.getEndpoint(apiArgs);
+    const url = `${baseUrl}${endpoint}`;
+    if (requestBody.has('model')) {
+        requestBody.delete('model');
+    }
+    requestBody.append('model', args.modelId);
+    const headers = await AzureAIInferenceAPI.headers({
+        ...apiArgs,
+        transformedRequestBody: requestBody,
+        transformedRequestUrl: url,
+    });
+    return safeFetch(url, { ssrf: 'guard', method: 'POST', headers, body: requestBody, signal: args.signal });
+};
+export const AzureAIInferenceCreateTranslationResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200 && 'error' in response) {
+        return OpenAIErrorResponseTransform(response, AZURE_AI_INFERENCE);
+    }
+    return response;
+};
+//# sourceMappingURL=createTranslation.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"createTranslation.js","sourceRoot":"","sources":["../../../src/providers/azure-ai-inference/createTranslation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AAGtE,OAAO,mBAAmB,MAAM,OAAO,CAAC;AAExC,MAAM,CAAC,MAAM,+CAA+C,GAA6B,KAAK,EAAC,IAAI,EAAC,EAAE;IACpG,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IACtC,MAAM,OAAO,GAAG,0BAA0B,CAAC,IAAI,CAAC,CAAC;IACjD,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAC9D,MAAM,QAAQ,GAAG,mBAAmB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAC1D,MAAM,GAAG,GAAG,GAAG,OAAO,GAAG,QAAQ,EAAE,CAAC;IACpC,IAAI,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QAC7B,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IACD,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1C,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC,OAAO,CAAC;QAChD,GAAG,OAAO;QACV,sBAAsB,EAAE,WAAW;QACnC,qBAAqB,EAAE,GAAG;KAC3B,CAAC,CAAC;IACH,OAAO,SAAS,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;AAC5G,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,kDAAkD,GAG/B,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IAC3D,IAAI,cAAc,KAAK,GAAG,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;QAClD,OAAO,4BAA4B,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;IACpE,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC"}

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

@@ -0,0 +1,12 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { EmbedResponse } from '../../types/embedRequestBody.js';
+import { ErrorResponse, ProviderConfig } from '../types.js';
+export declare const AzureAIInferenceEmbedConfig: ProviderConfig;
+interface AzureAIInferenceEmbedResponse extends EmbedResponse {
+}
+export declare const AzureAIInferenceEmbedResponseTransform: (provider: string) => (response: AzureAIInferenceEmbedResponse | ErrorResponse, responseStatus: number) => EmbedResponse | ErrorResponse;
+export {};

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

@@ -0,0 +1,24 @@
+import { OpenAIErrorResponseTransform } from '../openai/utils.js';
+export const AzureAIInferenceEmbedConfig = {
+    model: {
+        param: 'model',
+        required: false,
+    },
+    input: {
+        param: 'input',
+        required: true,
+    },
+    user: {
+        param: 'user',
+    },
+};
+export const AzureAIInferenceEmbedResponseTransform = (provider) => {
+    const transformer = (response, responseStatus) => {
+        if (responseStatus !== 200 && 'error' in response) {
+            return OpenAIErrorResponseTransform(response, provider);
+        }
+        return { ...response, provider: provider };
+    };
+    return transformer;
+};
+//# sourceMappingURL=embed.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"embed.js","sourceRoot":"","sources":["../../../src/providers/azure-ai-inference/embed.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAG/D,MAAM,CAAC,MAAM,2BAA2B,GAAmB;IACzD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;KAChB;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;KACf;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;KACd;CACF,CAAC;AAIF,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,QAAgB,EAAE,EAAE;IACzE,MAAM,WAAW,GAGoB,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;QAChE,IAAI,cAAc,KAAK,GAAG,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;YAClD,OAAO,4BAA4B,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,EAAE,GAAG,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;IAC7C,CAAC,CAAC;IAEF,OAAO,WAAW,CAAC;AACrB,CAAC,CAAC"}

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

@@ -0,0 +1,4 @@
+import { ProviderConfigs } from '../types.js';
+declare const AzureAIInferenceAPIConfig: ProviderConfigs;
+declare const GithubModelAPiConfig: ProviderConfigs;
+export { AzureAIInferenceAPIConfig, GithubModelAPiConfig };