UNPKG

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

Version:

28 lines (25 loc) 705 B
'use strict'; class SourceBasedRetrievalRouter { logger; retrievers; constructor({ logger, retrievers }) { this.retrievers = retrievers; this.logger = logger; } async determineRetriever(_query, source) { if (this.retrievers.has(source)) { return this.retrievers.get(source); } this.logger.warn( `Attempted to determine augmentation retriever for a source not implemented yet: ${source}` ); throw new Error( `Attempting to determine augmentation retriever for a source not implemented yet: ${source}` ); } } exports.SourceBasedRetrievalRouter = SourceBasedRetrievalRouter; //# sourceMappingURL=SourceBasedRetrievalRouter.cjs.js.map