@nomyx/assistant
Version:
A powerful assistant library and cli for your AI projects. works with Vertex AI (Claude and Gemini)
35 lines (34 loc) • 1.09 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.DEFAULT_OPENROUTER_MODEL = exports.OPENROUTER_MODEL_CONFIG = void 0;
exports.getModelConfig = getModelConfig;
exports.getMaxTokens = getMaxTokens;
exports.supportsTools = supportsTools;
exports.OPENROUTER_MODEL_CONFIG = {
'nousresearch/hermes-3-llama-3.1-405b:free': {
maxTokens: 13072,
supportsTools: false,
},
'google/gemini-pro-1.5-exp': {
maxTokens: 1000000,
supportsTools: true,
},
'meta-llama/llama-3.1-405b-instruct:free': {
maxTokens: 13072,
supportsTools: false,
},
'liquid/lfm-40b': {
maxTokens: 32768,
supportsTools: false,
},
};
exports.DEFAULT_OPENROUTER_MODEL = 'openai/gpt-3.5-turbo';
function getModelConfig(model) {
return exports.OPENROUTER_MODEL_CONFIG[model] || exports.OPENROUTER_MODEL_CONFIG[exports.DEFAULT_OPENROUTER_MODEL];
}
function getMaxTokens(model) {
return getModelConfig(model).maxTokens;
}
function supportsTools(model) {
return getModelConfig(model).supportsTools;
}