UNPKG

mongodb-chatbot-server

Version:

A chatbot server for retrieval augmented generation (RAG).

38 lines (37 loc) 1.24 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.makeDefaultReferenceLinks = void 0; /** The default reference format returns the following for chunks from _unique_ pages: ```js { title: chunk.metadata.pageTitle ?? chunk.url, // if title doesn't exist, just put url url: chunk.url // this always exists } ``` */ const makeDefaultReferenceLinks = (chunks) => { // Filter chunks with unique URLs const uniqueUrls = new Set(); const uniqueChunks = chunks.filter((chunk) => { if (!uniqueUrls.has(chunk.url)) { uniqueUrls.add(chunk.url); return true; // Keep the chunk as it has a unique URL } return false; // Discard the chunk as its URL is not unique }); return uniqueChunks.map((chunk) => { const url = new URL(chunk.url).href; const title = chunk.metadata?.pageTitle ?? url; return { title, url, metadata: { sourceName: chunk.sourceName, tags: chunk.metadata?.tags ?? [], }, }; }); }; exports.makeDefaultReferenceLinks = makeDefaultReferenceLinks; //# sourceMappingURL=makeDefaultReferenceLinks.js.map