mindee/src/v2/product/split/splitRange.js

Mindee Client Library for Node.js

import { expandRange, extractMultipleSplits } from "../../../v2/fileOperations/split.js";
import { ExtractionResponse } from "../../../v2/product/index.js";
/**
 * Split inference result.
 */
export class SplitRange {
    constructor(serverResponse) {
        this.pageRange = serverResponse["page_range"];
        this.documentType = serverResponse["document_type"];
        this.extractionResponse = serverResponse["extraction_response"]
            ? new ExtractionResponse(serverResponse["extraction_response"])
            : undefined;
    }
    toString() {
        const pageRange = this.pageRange.join(",");
        return `* :Page Range: ${pageRange}\n  :Document Type: ${this.documentType}`;
    }
    /**
     * Extracts a single split from the input file.
     * @param inputSource The input file to extract from.
     */
    async extractFromFile(inputSource) {
        const pageRange = [expandRange(this.pageRange)];
        return (await extractMultipleSplits(inputSource, pageRange))[0];
    }
}