mongodb-rag-core
Version:
Common elements used by MongoDB Chatbot Framework components.
26 lines • 994 B
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.makeDefaultFindContent = void 0;
/**
Basic implementation of FindContentFunc with search boosters.
*/
const makeDefaultFindContent = ({ embedder, store, findNearestNeighborsOptions, searchBoosters, }) => {
return async ({ query }) => {
const { embedding } = await embedder.embed({
text: query,
});
let content = await store.findNearestNeighbors(embedding, findNearestNeighborsOptions);
for (const booster of searchBoosters ?? []) {
if (await booster.shouldBoost({ text: query })) {
content = await booster.boost({
existingResults: content,
embedding,
store,
});
}
}
return { queryEmbedding: embedding, content };
};
};
exports.makeDefaultFindContent = makeDefaultFindContent;
//# sourceMappingURL=DefaultFindContent.js.map