@roadiehq/rag-ai-backend-retrieval-augmenter
Version:
28 lines (25 loc) • 705 B
JavaScript
;
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