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

'use strict';

var SearchClient = require('./SearchClient.cjs.js');

class SearchRetriever {
  searchClient;
  logger;
  constructor({
    discovery,
    logger,
    searchClient,
    auth,
    tokenManager
  }) {
    this.searchClient = searchClient ?? new SearchClient.SearchClient({
      discoveryApi: discovery,
      logger: logger.child({ label: "rag-ai-searchclient" }),
      auth,
      tokenManager
    });
    this.logger = logger;
  }
  get id() {
    return "SearchRetriever";
  }
  async retrieve(query, source) {
    const queryResults = await this.searchClient.query({
      term: query,
      source
    });
    this.logger.info(
      `Received ${queryResults.length} results when querying augmentations from search.`
    );
    return queryResults;
  }
}

exports.SearchRetriever = SearchRetriever;
//# sourceMappingURL=SearchRetriever.cjs.js.map