@roadiehq/rag-ai-backend-retrieval-augmenter
Version:
29 lines (26 loc) • 667 B
JavaScript
;
class VectorEmbeddingsRetriever {
logger;
vectorStore;
constructor({
vectorStore,
logger
}) {
this.vectorStore = vectorStore;
this.logger = logger;
}
get id() {
return "VectorEmbeddingsRetriever";
}
async retrieve(query, source, _filter) {
const embeddings = await this.vectorStore.similaritySearch(query, {
source: source !== "all" ? source : void 0
});
this.logger.info(
`Received ${embeddings.length} embeddings from Vector store`
);
return embeddings;
}
}
exports.VectorEmbeddingsRetriever = VectorEmbeddingsRetriever;
//# sourceMappingURL=VectorEmbeddingsRetriever.cjs.js.map