UNPKG

@langchain/community

Version:
97 lines (96 loc) 4.24 kB
import { __exportAll } from "../../_virtual/_rolldown/runtime.js"; import { Document } from "@langchain/core/documents"; import { BufferLoader } from "@langchain/classic/document_loaders/fs/buffer"; //#region src/document_loaders/fs/docx.ts var docx_exports = /* @__PURE__ */ __exportAll({ DocxLoader: () => DocxLoader }); /** * A class that extends the `BufferLoader` class. It represents a document * loader that loads documents from DOCX files. * It has a constructor that takes a `filePathOrBlob` parameter representing the path to the word * file or a Blob object, and an optional `options` parameter of type * `DocxLoaderOptions` */ var DocxLoader = class extends BufferLoader { options = { type: "docx" }; constructor(filePathOrBlob, options) { super(filePathOrBlob); if (options) this.options = { ...options }; } /** * A method that takes a `raw` buffer and `metadata` as parameters and * returns a promise that resolves to an array of `Document` instances. It * uses the `extractRawText` function from the `mammoth` module or * `extract` method from the `word-extractor` module to extract * the raw text content from the buffer. If the extracted text content is * empty, it returns an empty array. Otherwise, it creates a new * `Document` instance with the extracted text content and the provided * metadata, and returns it as an array. * @param raw The raw buffer from which to extract text content. * @param metadata The metadata to be associated with the created `Document` instance. * @returns A promise that resolves to an array of `Document` instances. */ async parse(raw, metadata) { if (this.options.type === "doc") return this.parseDoc(raw, metadata); return this.parseDocx(raw, metadata); } /** * A private method that takes a `raw` buffer and `metadata` as parameters and * returns a promise that resolves to an array of `Document` instances. It * uses the `extractRawText` function from the `mammoth` module to extract * the raw text content from the buffer. If the extracted text content is * empty, it returns an empty array. Otherwise, it creates a new * `Document` instance with the extracted text content and the provided * metadata, and returns it as an array. * @param raw The raw buffer from which to extract text content. * @param metadata The metadata to be associated with the created `Document` instance. * @returns A promise that resolves to an array of `Document` instances. */ async parseDocx(raw, metadata) { if (this.options.type === "doc") return this.parseDoc(raw, metadata); const { extractRawText } = await DocxLoaderImports(); const docx = await extractRawText({ buffer: raw }); if (!docx.value) return []; return [new Document({ pageContent: docx.value, metadata })]; } /** * A private method that takes a `raw` buffer and `metadata` as parameters and * returns a promise that resolves to an array of `Document` instances. It * uses the `extract` method from the `word-extractor` module to extract * the raw text content from the buffer. If the extracted text content is * empty, it returns an empty array. Otherwise, it creates a new * `Document` instance with the extracted text content and the provided * metadata, and returns it as an array. * @param raw The raw buffer from which to extract text content. * @param metadata The metadata to be associated with the created `Document` instance. * @returns A promise that resolves to an array of `Document` instances. */ async parseDoc(raw, metadata) { return [new Document({ pageContent: (await new (await (DocLoaderImports()))().extract(raw)).getBody(), metadata })]; } }; async function DocxLoaderImports() { try { const { extractRawText } = await import("mammoth"); return { extractRawText }; } catch (e) { console.error(e); throw new Error("Failed to load mammoth. Please install it with eg. `npm install mammoth`."); } } async function DocLoaderImports() { try { return (await import("word-extractor")).default; } catch (e) { console.error(e); throw new Error("Failed to load word-extractor. Please install it with eg. `npm install word-extractor`."); } } //#endregion export { DocxLoader, docx_exports }; //# sourceMappingURL=docx.js.map