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.

169 lines (166 loc) 4.85 kB
// src/azure-openai-provider.ts import { OpenAIChatLanguageModel, OpenAICompletionLanguageModel, OpenAIEmbeddingModel, OpenAIImageModel, OpenAIResponsesLanguageModel, OpenAISpeechModel, OpenAITranscriptionModel } from "@ai-sdk/openai/internal"; import { InvalidArgumentError } from "@ai-sdk/provider"; import { loadApiKey, loadSetting, normalizeHeaders, withUserAgentSuffix } from "@ai-sdk/provider-utils"; // src/azure-openai-tools.ts import { codeInterpreter, fileSearch, imageGeneration, webSearch, webSearchPreview } from "@ai-sdk/openai/internal"; var azureOpenaiTools = { codeInterpreter, fileSearch, imageGeneration, webSearch, 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 InvalidArgumentError({ argument: "apiKey/tokenProvider", message: "Both apiKey and tokenProvider were provided. Please use only one authentication method." }); } const getHeaders = () => { const authHeaders = tokenProvider ? {} : { "api-key": loadApiKey({ apiKey: options.apiKey, environmentVariableName: "AZURE_API_KEY", description: "Azure OpenAI" }) }; return withUserAgentSuffix( { ...authHeaders, ...options.headers }, `ai-sdk/azure/${VERSION}` ); }; const fetch = tokenProvider ? async (input, init) => { var _a2; const headers = 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 = () => 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 OpenAIChatLanguageModel(deploymentName, { provider: "azure.chat", url, headers: getHeaders, fetch }); const createCompletionModel = (modelId) => new OpenAICompletionLanguageModel(modelId, { provider: "azure.completion", url, headers: getHeaders, fetch }); const createEmbeddingModel = (modelId) => new OpenAIEmbeddingModel(modelId, { provider: "azure.embeddings", headers: getHeaders, url, fetch }); const createResponsesModel = (modelId) => new OpenAIResponsesLanguageModel(modelId, { provider: "azure.responses", url, headers: getHeaders, fetch, fileIdPrefixes: ["assistant-"] }); const createImageModel = (modelId) => new OpenAIImageModel(modelId, { provider: "azure.image", url, headers: getHeaders, fetch }); const createTranscriptionModel = (modelId) => new OpenAITranscriptionModel(modelId, { provider: "azure.transcription", url, headers: getHeaders, fetch }); const createSpeechModel = (modelId) => new 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(); export { VERSION, azure, createAzure }; //# sourceMappingURL=index.mjs.map