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

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

import { trimModelName } from "./index.js";
import { GeminiModelDefinition } from "./models/gemini.js";
import { ClaudeModelDefinition } from "./models/claude.js";
import { LLamaModelDefinition } from "./models/llama.js";
export function getModelDefinition(model) {
    const splits = model.split("/");
    const publisher = splits[1];
    const modelName = trimModelName(splits[splits.length - 1]);
    if (publisher?.includes("anthropic")) {
        return new ClaudeModelDefinition(modelName);
    }
    else if (publisher?.includes("google")) {
        return new GeminiModelDefinition(modelName);
    }
    else if (publisher?.includes("meta")) {
        return new LLamaModelDefinition(modelName);
    }
    //Fallback, assume it is Gemini.
    return new GeminiModelDefinition(modelName);
}
//# sourceMappingURL=models.js.map