UNPKG

@ai-sdk/azure

Version:

The **[Azure provider](https://ai-sdk.dev/providers/ai-sdk-providers/azure)** for the [AI SDK](https://ai-sdk.dev/docs) contains language model support for the Azure OpenAI API.

177 lines (172 loc) 6.03 kB
"use strict"; var __defProp = Object.defineProperty; var __getOwnPropDesc = Object.getOwnPropertyDescriptor; var __getOwnPropNames = Object.getOwnPropertyNames; var __hasOwnProp = Object.prototype.hasOwnProperty; var __export = (target, all) => { for (var name in all) __defProp(target, name, { get: all[name], enumerable: true }); }; var __copyProps = (to, from, except, desc) => { if (from && typeof from === "object" || typeof from === "function") { for (let key of __getOwnPropNames(from)) if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable }); } return to; }; var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod); // src/index.ts var index_exports = {}; __export(index_exports, { VERSION: () => VERSION, azure: () => azure, createAzure: () => createAzure }); module.exports = __toCommonJS(index_exports); // src/azure-openai-provider.ts var import_internal2 = require("@ai-sdk/openai/internal"); var import_provider = require("@ai-sdk/provider"); var import_provider_utils = require("@ai-sdk/provider-utils"); // src/azure-openai-tools.ts var import_internal = require("@ai-sdk/openai/internal"); var azureOpenaiTools = { codeInterpreter: import_internal.codeInterpreter, fileSearch: import_internal.fileSearch, imageGeneration: import_internal.imageGeneration, webSearch: import_internal.webSearch, webSearchPreview: import_internal.webSearchPreview }; // src/version.ts var VERSION = true ? "3.0.70" : "0.0.0-test"; // src/azure-openai-provider.ts function createAzure(options = {}) { var _a; const tokenProvider = options.tokenProvider; if (options.apiKey && tokenProvider) { throw new import_provider.InvalidArgumentError({ argument: "apiKey/tokenProvider", message: "Both apiKey and tokenProvider were provided. Please use only one authentication method." }); } const getHeaders = () => { const authHeaders = tokenProvider ? {} : { "api-key": (0, import_provider_utils.loadApiKey)({ apiKey: options.apiKey, environmentVariableName: "AZURE_API_KEY", description: "Azure OpenAI" }) }; return (0, import_provider_utils.withUserAgentSuffix)( { ...authHeaders, ...options.headers }, `ai-sdk/azure/${VERSION}` ); }; const fetch = tokenProvider ? async (input, init) => { var _a2; const headers = (0, import_provider_utils.normalizeHeaders)(init == null ? void 0 : init.headers); if (headers.authorization == null) { headers.authorization = `Bearer ${await tokenProvider()}`; } return ((_a2 = options.fetch) != null ? _a2 : globalThis.fetch)(input, { ...init, headers }); } : options.fetch; const getResourceName = () => (0, import_provider_utils.loadSetting)({ settingValue: options.resourceName, settingName: "resourceName", environmentVariableName: "AZURE_RESOURCE_NAME", description: "Azure OpenAI resource name" }); const apiVersion = (_a = options.apiVersion) != null ? _a : "v1"; const url = ({ path, modelId }) => { var _a2; const baseUrlPrefix = (_a2 = options.baseURL) != null ? _a2 : `https://${getResourceName()}.openai.azure.com/openai`; let fullUrl; if (options.useDeploymentBasedUrls) { fullUrl = new URL(`${baseUrlPrefix}/deployments/${modelId}${path}`); } else { fullUrl = new URL(`${baseUrlPrefix}/v1${path}`); } fullUrl.searchParams.set("api-version", apiVersion); return fullUrl.toString(); }; const createChatModel = (deploymentName) => new import_internal2.OpenAIChatLanguageModel(deploymentName, { provider: "azure.chat", url, headers: getHeaders, fetch }); const createCompletionModel = (modelId) => new import_internal2.OpenAICompletionLanguageModel(modelId, { provider: "azure.completion", url, headers: getHeaders, fetch }); const createEmbeddingModel = (modelId) => new import_internal2.OpenAIEmbeddingModel(modelId, { provider: "azure.embeddings", headers: getHeaders, url, fetch }); const createResponsesModel = (modelId) => new import_internal2.OpenAIResponsesLanguageModel(modelId, { provider: "azure.responses", url, headers: getHeaders, fetch, fileIdPrefixes: ["assistant-"] }); const createImageModel = (modelId) => new import_internal2.OpenAIImageModel(modelId, { provider: "azure.image", url, headers: getHeaders, fetch }); const createTranscriptionModel = (modelId) => new import_internal2.OpenAITranscriptionModel(modelId, { provider: "azure.transcription", url, headers: getHeaders, fetch }); const createSpeechModel = (modelId) => new import_internal2.OpenAISpeechModel(modelId, { provider: "azure.speech", url, headers: getHeaders, fetch }); const provider = function(deploymentId) { if (new.target) { throw new Error( "The Azure OpenAI model function cannot be called with the new keyword." ); } return createResponsesModel(deploymentId); }; provider.specificationVersion = "v3"; provider.languageModel = createResponsesModel; provider.chat = createChatModel; provider.completion = createCompletionModel; provider.embedding = createEmbeddingModel; provider.embeddingModel = createEmbeddingModel; provider.textEmbedding = createEmbeddingModel; provider.textEmbeddingModel = createEmbeddingModel; provider.image = createImageModel; provider.imageModel = createImageModel; provider.responses = createResponsesModel; provider.transcription = createTranscriptionModel; provider.speech = createSpeechModel; provider.tools = azureOpenaiTools; return provider; } var azure = createAzure(); // Annotate the CommonJS export names for ESM import in node: 0 && (module.exports = { VERSION, azure, createAzure }); //# sourceMappingURL=index.js.map