@llumiverse/drivers/lib/cjs/openai/azure.js

LLM driver implementations. Currently supported are: openai, huggingface, bedrock, replicate.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.AzureOpenAIDriver = void 0;
const identity_1 = require("@azure/identity");
const openai_1 = require("openai");
const index_js_1 = require("./index.js");
class AzureOpenAIDriver extends index_js_1.BaseOpenAIDriver {
    service;
    provider;
    constructor(opts) {
        super(opts);
        if (!opts.azureADTokenProvider && !opts.apiKey) {
            opts.azureADTokenProvider = this.getDefaultAuth();
        }
        this.service = new openai_1.AzureOpenAI({
            apiKey: opts.apiKey,
            azureADTokenProvider: opts.azureADTokenProvider,
            endpoint: opts.endpoint,
            apiVersion: opts.apiVersion ?? "2024-10-21",
            deployment: opts.deployment
        });
        this.provider = "azure_openai";
    }
    getDefaultAuth() {
        const scope = "https://cognitiveservices.azure.com/.default";
        const azureADTokenProvider = (0, identity_1.getBearerTokenProvider)(new identity_1.DefaultAzureCredential(), scope);
        return azureADTokenProvider;
    }
}
exports.AzureOpenAIDriver = AzureOpenAIDriver;
//# sourceMappingURL=azure.js.map