@n8n/n8n-nodes-langchain/dist/nodes/embeddings/EmbeddingsNvidia/methods/searchModels.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.DEFAULT_NVIDIA_EMBEDDING_MODEL = exports.NVIDIA_EMBEDDING_MODELS = void 0;
exports.searchModels = searchModels;
exports.NVIDIA_EMBEDDING_MODELS = [
    'nvidia/llama-3.2-nv-embedqa-1b-v2',
    'nvidia/nv-embedqa-e5-v5',
    'nvidia/nv-embed-v1',
    'nvidia/llama-nemotron-embed-1b-v2',
    'nvidia/llama-nemotron-embed-300m-v2',
];
exports.DEFAULT_NVIDIA_EMBEDDING_MODEL = 'nvidia/llama-3.2-nv-embedqa-1b-v2';
async function searchModels(filter) {
    const credentials = await this.getCredentials('nvidiaApi');
    const response = (await this.helpers.httpRequestWithAuthentication.call(this, 'nvidiaApi', {
        url: `${credentials.url}/models`,
    }));
    const supported = new Set(exports.NVIDIA_EMBEDDING_MODELS);
    const search = filter?.toLowerCase();
    const results = (response.data ?? [])
        .map((model) => model.id)
        .filter((id) => supported.has(id))
        .filter((id) => !search || id.toLowerCase().includes(search))
        .sort((a, b) => a.localeCompare(b))
        .map((id) => ({ name: id, value: id }));
    return { results };
}
//# sourceMappingURL=searchModels.js.map