@llumiverse/drivers/lib/cjs/vertexai/models.js

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

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.getModelDefinition = getModelDefinition;
const index_js_1 = require("./index.js");
const gemini_js_1 = require("./models/gemini.js");
const claude_js_1 = require("./models/claude.js");
const llama_js_1 = require("./models/llama.js");
function getModelDefinition(model) {
    const splits = model.split("/");
    const publisher = splits[1];
    const modelName = (0, index_js_1.trimModelName)(splits[splits.length - 1]);
    if (publisher?.includes("anthropic")) {
        return new claude_js_1.ClaudeModelDefinition(modelName);
    }
    else if (publisher?.includes("google")) {
        return new gemini_js_1.GeminiModelDefinition(modelName);
    }
    else if (publisher?.includes("meta")) {
        return new llama_js_1.LLamaModelDefinition(modelName);
    }
    //Fallback, assume it is Gemini.
    return new gemini_js_1.GeminiModelDefinition(modelName);
}
//# sourceMappingURL=models.js.map