@roadiehq/rag-ai-backend-retrieval-augmenter/dist/defaultInitializer.cjs.js

'use strict';

var DefaultRetrievalPipeline = require('./retrieval/DefaultRetrievalPipeline.cjs.js');
var VectorEmbeddingsRetriever = require('./retrieval/retrievers/VectorEmbeddingsRetriever.cjs.js');
var SearchRetriever = require('./retrieval/retrievers/SearchRetriever.cjs.js');
var SourceBasedRetrievalRouter = require('./retrieval/routers/SourceBasedRetrievalRouter.cjs.js');

const createDefaultRetrievalPipeline = ({
  vectorStore,
  discovery,
  logger,
  auth,
  tokenManager
}) => {
  const vectorEmbeddingsRetriever = new VectorEmbeddingsRetriever.VectorEmbeddingsRetriever({
    vectorStore,
    logger
  });
  const searchRetriever = new SearchRetriever.SearchRetriever({
    discovery,
    logger,
    auth,
    tokenManager
  });
  const sourceBasedRetrieverConfig = /* @__PURE__ */ new Map();
  ["catalog", "tech-docs", "all"].forEach((source) => {
    sourceBasedRetrieverConfig.set(source, [
      vectorEmbeddingsRetriever,
      searchRetriever
    ]);
  });
  const retrievalRouters = [
    new SourceBasedRetrievalRouter.SourceBasedRetrievalRouter({
      logger,
      retrievers: sourceBasedRetrieverConfig
    })
  ];
  return new DefaultRetrievalPipeline.DefaultRetrievalPipeline({
    routers: retrievalRouters
  });
};

exports.createDefaultRetrievalPipeline = createDefaultRetrievalPipeline;
//# sourceMappingURL=defaultInitializer.cjs.js.map