@langchain/community
Version:
Third-party integrations for LangChain.js
51 lines (50 loc) • 1.7 kB
JavaScript
import { Readability } from "@mozilla/readability";
import { JSDOM } from "jsdom";
import { MappingDocumentTransformer, Document, } from "@langchain/core/documents";
/**
* A transformer that uses the Mozilla Readability library to extract the
* main content from a web page.
* @example
* ```typescript
* const loader = new CheerioWebBaseLoader("https://example.com/article");
* const docs = await loader.load();
*
* const splitter = new RecursiveCharacterTextSplitter({
* maxCharacterCount: 5000,
* });
* const transformer = new MozillaReadabilityTransformer();
*
* // The sequence processes the loaded documents through the splitter and then the transformer.
* const sequence = splitter.pipe(transformer);
*
* // Invoke the sequence to transform the documents into a more readable format.
* const newDocuments = await sequence.invoke(docs);
*
* console.log(newDocuments);
* ```
*/
export class MozillaReadabilityTransformer extends MappingDocumentTransformer {
static lc_name() {
return "MozillaReadabilityTransformer";
}
constructor(options = {}) {
super(options);
Object.defineProperty(this, "options", {
enumerable: true,
configurable: true,
writable: true,
value: options
});
}
async _transformDocument(document) {
const doc = new JSDOM(document.pageContent);
const readability = new Readability(doc.window.document, this.options);
const result = readability.parse();
return new Document({
pageContent: result?.textContent ?? "",
metadata: {
...document.metadata,
},
});
}
}