@auto-browse/auto-browse/dist/llm.js

AI-powered browser automation

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.analyticsModel = exports.analyticsProvider = void 0;
exports.createLLMModel = createLLMModel;
const openai_1 = require("@langchain/openai");
const google_genai_1 = require("@langchain/google-genai");
const anthropic_1 = require("@langchain/anthropic");
const google_vertexai_1 = require("@langchain/google-vertexai");
const ollama_1 = require("@langchain/ollama");
const dotenv_1 = __importDefault(require("dotenv"));
// Load environment variables
dotenv_1.default.config();
// Export analytics-specific variables
exports.analyticsProvider = process.env.LLM_PROVIDER || "openai";
exports.analyticsModel = process.env.AUTOBROWSE_LLM_MODEL || "gpt-4o-mini";
function createLLMModel() {
    const provider = process.env.LLM_PROVIDER || "openai";
    const model = process.env.AUTOBROWSE_LLM_MODEL || "gpt-4o-mini";
    if (provider === "google") {
        return new google_genai_1.ChatGoogleGenerativeAI({
            model: model,
        });
    }
    if (provider === "azure") {
        return new openai_1.AzureChatOpenAI({
            temperature: 0,
            maxRetries: 2,
            azureOpenAIApiKey: process.env.AZURE_OPENAI_API_KEY,
            azureOpenAIApiInstanceName: process.env.AZURE_OPENAI_ENDPOINT,
            azureOpenAIApiDeploymentName: process.env.AZURE_OPENAI_API_DEPLOYMENT_NAME,
            azureOpenAIApiVersion: process.env.AZURE_OPENAI_API_VERSION,
        });
    }
    if (provider === "anthropic") {
        return new anthropic_1.ChatAnthropic({
            model: model,
            temperature: 0,
            anthropicApiKey: process.env.ANTHROPIC_API_KEY,
        });
    }
    if (provider === "vertex") {
        return new google_vertexai_1.ChatVertexAI({
            model: model,
            temperature: 0,
        });
    }
    if (provider === "ollama") {
        return new ollama_1.ChatOllama({
            model: model,
            temperature: 0,
            baseUrl: process.env.BASE_URL || "http://127.0.0.1:11434",
        });
    }
    return new openai_1.ChatOpenAI({
        modelName: model,
        temperature: 0,
    });
}