mongodb-rag-core
Version:
Common elements used by MongoDB Chatbot Framework components.
19 lines • 797 B
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.makeDefaultFindVerifiedAnswer = void 0;
/**
Basic implementation of FindVerifiedAnswerFunc.
*/
const makeDefaultFindVerifiedAnswer = ({ embedder, store, findNearestNeighborsOptions, }) => {
return async ({ query }) => {
const { embedding: queryEmbedding } = await embedder.embed({
text: query,
});
const answers = await store.findNearestNeighbors(queryEmbedding, findNearestNeighborsOptions);
// Sort by score descending
answers.sort((a, b) => b.score - a.score);
return { queryEmbedding, answer: answers[0] };
};
};
exports.makeDefaultFindVerifiedAnswer = makeDefaultFindVerifiedAnswer;
//# sourceMappingURL=DefaultFindVerifiedAnswer.js.map