UNPKG

@roadiehq/rag-ai-backend-retrieval-augmenter

Version:

29 lines (26 loc) 667 B
'use strict'; 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