mongodb-rag-core/build/models/makeOpenAiClientFactory.js

Common elements used by MongoDB Chatbot Framework components.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.makeOpenAiClientFactory = void 0;
const openai_1 = require("openai");
const assert_1 = require("assert");
const braintrust_1 = require("braintrust");
function makeOpenAiClientFactory({ braintrust, }) {
    return {
        makeOpenAiClient(modelConfig) {
            let openAiClient;
            if (modelConfig.provider === "braintrust") {
                (0, assert_1.strict)(braintrust, "Braintrust OpenAI config must be provided");
                openAiClient = (0, braintrust_1.wrapOpenAI)(new openai_1.OpenAI({
                    apiKey: braintrust.apiKey,
                    baseURL: braintrust.endpoint,
                }));
            }
            else {
                throw new Error(`Unsupported provider: ${modelConfig.provider}`);
            }
            return openAiClient;
        },
    };
}
exports.makeOpenAiClientFactory = makeOpenAiClientFactory;
//# sourceMappingURL=makeOpenAiClientFactory.js.map