seraph-agent/dist/llm/embedding-provider.js

An extremely lightweight, SRE autonomous AI agent for seamless integration with common observability tasks.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.createEmbeddingProvider = createEmbeddingProvider;
const openai_1 = require("@langchain/openai");
const hf_embeddings_1 = require("./hf-embeddings");
function createEmbeddingProvider(config) {
    const provider = config.knowledge?.embeddings.provider || 'huggingface';
    const model = config.knowledge?.embeddings.model;
    switch (provider) {
        case 'openai':
            return new openai_1.OpenAIEmbeddings({ openAIApiKey: config.apiKey, modelName: model });
        case 'huggingface':
            return new hf_embeddings_1.HuggingFaceEmbeddings(model);
        default:
            return new hf_embeddings_1.HuggingFaceEmbeddings(model);
    }
}