{"version":3,"file":"vectara_summary.cjs","names":["BaseRetriever","DEFAULT_FILTER","Document"],"sources":["../../src/retrievers/vectara_summary.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport {\n  BaseRetriever,\n  type BaseRetrieverInput,\n} from \"@langchain/core/retrievers\";\nimport { CallbackManagerForRetrieverRun } from \"@langchain/core/callbacks/manager\";\nimport {\n  VectaraStore,\n  type VectaraSummary,\n  type VectaraFilter,\n  DEFAULT_FILTER,\n} from \"../vectorstores/vectara.js\";\n\nexport interface VectaraRetrieverInput extends BaseRetrieverInput {\n  vectara: VectaraStore;\n  filter?: VectaraFilter;\n  topK?: number;\n  summaryConfig?: VectaraSummary;\n}\n\nexport class VectaraSummaryRetriever extends BaseRetriever {\n  static lc_name() {\n    return \"VectaraSummaryRetriever\";\n  }\n\n  lc_namespace = [\"langchain\", \"retrievers\"];\n\n  private filter = DEFAULT_FILTER;\n\n  private vectara: VectaraStore;\n\n  private topK: number;\n\n  private summaryConfig: VectaraSummary;\n\n  constructor(fields: VectaraRetrieverInput) {\n    super(fields);\n    this.vectara = fields.vectara;\n    this.topK = fields.topK ?? 10;\n    this.filter = fields.filter ?? DEFAULT_FILTER;\n    this.summaryConfig = fields.summaryConfig ?? {\n      enabled: false,\n      maxSummarizedResults: 0,\n      responseLang: \"eng\",\n    };\n  }\n\n  async _getRelevantDocuments(\n    query: string,\n    _callbacks?: CallbackManagerForRetrieverRun\n  ): Promise<Document[]> {\n    const summaryResult = await this.vectara.vectaraQuery(\n      query,\n      this.topK,\n      this.filter,\n      this.summaryConfig ? this.summaryConfig : undefined\n    );\n    const docs = summaryResult.documents;\n    if (this.summaryConfig.enabled) {\n      docs.push(\n        new Document({\n          pageContent: summaryResult.summary,\n          metadata: { summary: true },\n        })\n      );\n    }\n    return docs;\n  }\n}\n"],"mappings":";;;;;;;AAoBA,IAAa,0BAAb,cAA6CA,2BAAAA,cAAc;CACzD,OAAO,UAAU;AACf,SAAO;;CAGT,eAAe,CAAC,aAAa,aAAa;CAE1C,SAAiBC,6BAAAA;CAEjB;CAEA;CAEA;CAEA,YAAY,QAA+B;AACzC,QAAM,OAAO;AACb,OAAK,UAAU,OAAO;AACtB,OAAK,OAAO,OAAO,QAAQ;AAC3B,OAAK,SAAS,OAAO,UAAUA,6BAAAA;AAC/B,OAAK,gBAAgB,OAAO,iBAAiB;GAC3C,SAAS;GACT,sBAAsB;GACtB,cAAc;GACf;;CAGH,MAAM,sBACJ,OACA,YACqB;EACrB,MAAM,gBAAgB,MAAM,KAAK,QAAQ,aACvC,OACA,KAAK,MACL,KAAK,QACL,KAAK,gBAAgB,KAAK,gBAAgB,KAAA,EAC3C;EACD,MAAM,OAAO,cAAc;AAC3B,MAAI,KAAK,cAAc,QACrB,MAAK,KACH,IAAIC,0BAAAA,SAAS;GACX,aAAa,cAAc;GAC3B,UAAU,EAAE,SAAS,MAAM;GAC5B,CAAC,CACH;AAEH,SAAO"}