mongodb-rag-core/build/findContent/DefaultFindContent.js

Common elements used by MongoDB Chatbot Framework components.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.makeDefaultFindContent = void 0;
/**
  Basic implementation of FindContentFunc with search boosters.
 */
const makeDefaultFindContent = ({ embedder, store, findNearestNeighborsOptions, }) => {
    return async ({ query, filters = {}, limit }) => {
        const { embedding } = await embedder.embed({
            text: query,
        });
        const content = await store.findNearestNeighbors(embedding, {
            ...findNearestNeighborsOptions,
            filter: filters,
            ...(limit ? { k: limit } : {}),
        });
        return {
            queryEmbedding: embedding,
            content,
            embeddingModelName: embedder.modelName,
        };
    };
};
exports.makeDefaultFindContent = makeDefaultFindContent;
//# sourceMappingURL=DefaultFindContent.js.map