docling-serve-client 0.4.0 → 0.5.0

package/dist/index.js

@@ -250,6 +250,70 @@ function fail(error) {
   return { ok: false, error };
 }
 
+// src/capabilities.ts
+function readCapabilities(body) {
+  if (!isPlainObject(body)) {
+    return fail(ERR.INVALID_RESPONSE("Capabilities response is not a JSON object"));
+  }
+  const targets = objectOf(body.targets);
+  const features = objectOf(body.features);
+  return ok({
+    versions: Object.fromEntries(Object.entries(objectOf(body.versions)).filter((entry) => typeof entry[1] === "string")),
+    stages: Object.fromEntries(Object.entries(objectOf(body.stages)).map(([name, stage]) => [name, readStage(objectOf(stage))])),
+    sources: stringsOf(body.sources),
+    targets: { allowed: stringsOf(targets.allowed), default: stringOf(targets.default) },
+    outputFormats: stringsOf(body.output_formats),
+    imageExportModes: stringsOf(body.image_export_modes),
+    limits: readLimits(objectOf(body.limits)),
+    features: {
+      apiKeyRequired: features.api_key_required === true,
+      artifactStorage: features.artifact_storage === true,
+      websocket: features.websocket === true
+    }
+  });
+}
+function readStage(stage) {
+  const presets = [];
+  for (const item of Array.isArray(stage.presets) ? stage.presets : []) {
+    const preset = objectOf(item);
+    if (typeof preset.id === "string") {
+      presets.push({
+        id: preset.id,
+        name: stringOf(preset.name),
+        description: typeof preset.description === "string" ? preset.description : null,
+        source: stringOf(preset.source)
+      });
+    }
+  }
+  return {
+    option: stringOf(stage.option),
+    customConfigOption: typeof stage.custom_config_option === "string" ? stage.custom_config_option : null,
+    default: stringOf(stage.default),
+    presets
+  };
+}
+function readLimits(limits) {
+  return {
+    maxDocumentTimeout: numberOf(limits.max_document_timeout),
+    maxImagesScale: numberOf(limits.max_images_scale),
+    maxSourcesPerRequest: numberOf(limits.max_sources_per_request),
+    maxNumPages: numberOf(limits.max_num_pages),
+    maxFileSize: numberOf(limits.max_file_size)
+  };
+}
+function objectOf(value) {
+  return isPlainObject(value) ? value : {};
+}
+function stringsOf(value) {
+  return Array.isArray(value) ? value.filter((item) => typeof item === "string") : [];
+}
+function stringOf(value) {
+  return typeof value === "string" ? value : "";
+}
+function numberOf(value) {
+  return typeof value === "number" && Number.isFinite(value) ? value : null;
+}
+
 // src/core/http.ts
 var STRING_TOO_LONG = "ERR_STRING_TOO_LONG";
 var MATCH_OUT_OF_MEMORY = /out of memory/i;
@@ -368,7 +432,8 @@ var DEFAULT_OPTIONS = Object.freeze({
   pipeline: "standard",
   pdfBackend: "docling_parse",
   ocr: true,
-  forceOcr: false
+  forceOcr: false,
+  ocrEngine: "auto"
 });
 var CALL_OPTION_RULES = {
   headingHierarchy: isBoolean,
@@ -378,7 +443,8 @@ var CALL_OPTION_RULES = {
   pipeline: (value) => isOneOf(value, PIPELINES),
   pdfBackend: (value) => isOneOf(value, PDF_BACKENDS),
   ocr: isBoolean,
-  forceOcr: isBoolean
+  forceOcr: isBoolean,
+  ocrEngine: (value) => typeof value === "string" && value !== ""
 };
 var CLIENT_OPTION_RULES = {
   ...CALL_OPTION_RULES,
@@ -635,10 +701,11 @@ function isPageNumber(value) {
 }
 
 // src/client.ts
-var FIXED_FORM_FIELDS = [
-  ["do_table_structure", "true"],
-  ["table_mode", "accurate"]
-];
+var TABLE_STRUCTURE_PRESET = "tableformer_v1_accurate";
+var VLM_PIPELINE_PRESET = "granite_docling";
+var CHART_EXTRACTION_PRESET = "granite_vision_v4";
+var CODE_FORMULA_PRESET = "codeformulav2";
+var MATCH_UPPER_CASE = /[A-Z]/g;
 var INVALID_INPUT = "Expected a non-empty PDF, PNG, JPEG, TIFF or WebP as Uint8Array";
 var SERVER_ANSWERED = new Set([
   "http",
@@ -657,13 +724,18 @@ function createDoclingClient(config) {
     convertDocument,
     submitMarkdown,
     submitDocument,
-    check
+    check,
+    capabilities
   };
   async function convertMarkdown(input, options) {
     const call = resolveMarkdownCall(defaults, options);
     const fields = { format: "md", embedImages: call.output.images === "extract", pageImages: false };
     const res = await post("/v1/convert/file", input, call.options, fields, call.signal);
-    return res.ok ? withVersions(readMarkdown(res.data, call.output)) : res;
+    if (!res.ok) {
+      return res;
+    }
+    const converted = await withVersions(readMarkdown(res.data.body, call.output));
+    return converted.ok ? ok({ ...converted.data, settings: res.data.settings }) : converted;
   }
   async function convertDocument(input, options) {
     const call = resolveDocumentCall(defaults, options);
@@ -673,13 +745,17 @@ function createDoclingClient(config) {
       pageImages: call.pageImages
     };
     const res = await post("/v1/convert/file", input, call.options, fields, call.signal);
-    return res.ok ? withVersions(readDocument(res.data)) : res;
+    if (!res.ok) {
+      return res;
+    }
+    const converted = await withVersions(readDocument(res.data.body));
+    return converted.ok ? ok({ ...converted.data, settings: res.data.settings }) : converted;
   }
   async function submitMarkdown(input, options) {
     const call = resolveMarkdownCall(defaults, options);
     const fields = { format: "md", embedImages: call.output.images === "extract", pageImages: false };
-    const id = await submit(input, call.options, fields, call.signal);
-    return id.ok ? ok({ kind: "markdown", id: id.data, output: call.output }) : id;
+    const task = await submit(input, call.options, fields, call.signal);
+    return task.ok ? ok({ kind: "markdown", id: task.data.id, output: call.output, settings: task.data.settings }) : task;
   }
   async function submitDocument(input, options) {
     const call = resolveDocumentCall(defaults, options);
@@ -688,8 +764,8 @@ function createDoclingClient(config) {
       embedImages: call.images === "embedded",
       pageImages: call.pageImages
     };
-    const id = await submit(input, call.options, fields, call.signal);
-    return id.ok ? ok({ kind: "document", id: id.data }) : id;
+    const task = await submit(input, call.options, fields, call.signal);
+    return task.ok ? ok({ kind: "document", id: task.data.id, settings: task.data.settings }) : task;
   }
   async function check(job, options) {
     const { signal } = resolveCheckOptions(options);
@@ -733,12 +809,21 @@ function createDoclingClient(config) {
         return failWithVersions(ERR.INVALID_RESPONSE(`Unexpected task status: ${String(task.task_status)}`));
     }
   }
+  async function capabilities(options) {
+    const { signal } = resolveCheckOptions(options);
+    const res = await get("/v1/capabilities", signal);
+    if (!res.ok) {
+      return failWithVersions(res.error);
+    }
+    const read = readCapabilities(res.data);
+    return read.ok ? read : failWithVersions(read.error);
+  }
   async function submit(input, options, fields, signal) {
     const res = await post("/v1/convert/file/async", input, options, fields, signal);
     if (!res.ok) {
       return res;
     }
-    const id = res.data?.task_id;
+    const id = res.data.body?.task_id;
     if (readText(id) === undefined) {
       return failWithVersions(ERR.INVALID_RESPONSE("Response has no task_id"));
     }
@@ -746,16 +831,17 @@ function createDoclingClient(config) {
       const shown = JSON.stringify(id).slice(0, MAX_SHOWN_ID_LENGTH);
       return failWithVersions(ERR.INVALID_RESPONSE(`Response has an unusable task_id: ${shown}`));
     }
-    return ok(id);
+    return ok({ id, settings: res.data.settings });
   }
   async function post(route, input, options, fields, signal) {
     const format = readInputFormat(input);
     if (!format.ok) {
       return format;
     }
+    const settings = buildSettings(format.data, options, fields);
     let body;
     try {
-      body = buildFormData(input, format.data, options, fields);
+      body = buildFormData(input, format.data, settings);
     } catch (error) {
       return fail(ERR.INVALID_INPUT(`Input could not be prepared for upload: ${errorMessage(error)}`));
     }
@@ -766,7 +852,7 @@ function createDoclingClient(config) {
       body,
       ...limited === undefined ? {} : { signal: limited }
     });
-    return res.ok ? res : failWithVersions(res.error);
+    return res.ok ? ok({ body: res.data, settings }) : failWithVersions(res.error);
   }
   function get(route, signal, limit = true) {
     const limited = limit ? withTimeout(signal, timeoutMs) : signal;
@@ -822,43 +908,51 @@ function readInputFormat(input) {
   const offset = format.fromFormat === "pdf" ? findTruncatedXref(bytes) : undefined;
   return offset === undefined ? ok(format) : fail(ERR.INVALID_INPUT(`PDF is truncated: cross-reference offset ${offset} lies past the end of the file (${bytes.length} bytes)`));
 }
-function buildFormData(input, { fromFormat, mimeType, fileName }, options, { format, embedImages, pageImages }) {
+function buildSettings({ fromFormat }, options, { format, embedImages, pageImages }) {
+  const isImage = fromFormat === "image";
+  const isMarkdown = format === "md";
+  return {
+    fromFormats: fromFormat,
+    toFormats: format,
+    imageExportMode: embedImages || pageImages ? "embedded" : "placeholder",
+    includeImages: embedImages,
+    includePageImages: pageImages,
+    imagesScale: PAGE_IMAGE_SCALE,
+    mdCompactTables: isMarkdown ? true : null,
+    mdPageBreakPlaceholder: isMarkdown ? "" : null,
+    doOcr: isImage || options.ocr,
+    forceOcr: !isImage && options.forceOcr,
+    ocrPreset: options.ocrEngine,
+    ocrLang: [...options.ocrLang ?? []],
+    pdfBackend: options.pdfBackend,
+    pipeline: options.pipeline,
+    vlmPipelinePreset: options.pipeline === "vlm" ? VLM_PIPELINE_PRESET : null,
+    doTableStructure: true,
+    tableMode: "accurate",
+    tableStructurePreset: TABLE_STRUCTURE_PRESET,
+    tableCellMatching: true,
+    doPdfHeadingHierarchy: options.headingHierarchy,
+    doFormulaEnrichment: options.formulaEnrichment,
+    doCodeEnrichment: options.codeEnrichment,
+    codeFormulaPreset: options.formulaEnrichment || options.codeEnrichment ? CODE_FORMULA_PRESET : null,
+    doChartExtraction: options.chartExtraction,
+    chartExtractionPreset: options.chartExtraction ? CHART_EXTRACTION_PRESET : null,
+    doPictureClassification: false,
+    doPictureDescription: false,
+    pageRange: options.pageRange === undefined ? null : [options.pageRange[0], options.pageRange[1]],
+    documentTimeout: options.documentTimeout ?? null,
+    abortOnError: false
+  };
+}
+function buildFormData(input, { mimeType, fileName }, settings) {
   const form = new FormData;
   const bytes = input.buffer instanceof ArrayBuffer ? input : new Uint8Array(input);
   form.append("files", new Blob([bytes], { type: mimeType }), fileName);
-  form.append("from_formats", fromFormat);
-  for (const [name, value] of FIXED_FORM_FIELDS) {
-    form.append(name, value);
-  }
-  form.append("to_formats", format);
-  if (format === "md") {
-    form.append("md_compact_tables", "true");
-  }
-  form.append("include_images", String(embedImages));
-  form.append("include_page_images", String(pageImages));
-  form.append("image_export_mode", embedImages || pageImages ? "embedded" : "placeholder");
-  form.append("images_scale", String(PAGE_IMAGE_SCALE));
-  form.append("do_pdf_heading_hierarchy", String(options.headingHierarchy));
-  form.append("do_formula_enrichment", String(options.formulaEnrichment));
-  form.append("do_code_enrichment", String(options.codeEnrichment));
-  if (options.formulaEnrichment || options.codeEnrichment) {
-    form.append("code_formula_preset", "codeformulav2");
-  }
-  form.append("do_chart_extraction", String(options.chartExtraction));
-  form.append("pipeline", options.pipeline);
-  form.append("pdf_backend", options.pdfBackend);
-  const isImage = fromFormat === "image";
-  form.append("do_ocr", String(isImage || options.ocr));
-  form.append("force_ocr", String(!isImage && options.forceOcr));
-  for (const lang of options.ocrLang ?? []) {
-    form.append("ocr_lang", lang);
-  }
-  if (options.pageRange !== undefined) {
-    form.append("page_range", String(options.pageRange[0]));
-    form.append("page_range", String(options.pageRange[1]));
-  }
-  if (options.documentTimeout !== undefined) {
-    form.append("document_timeout", String(options.documentTimeout));
+  for (const [key, value] of Object.entries(settings)) {
+    const name = key.replace(MATCH_UPPER_CASE, (letter) => `_${letter.toLowerCase()}`);
+    for (const item of value === null ? [] : Array.isArray(value) ? value : [value]) {
+      form.append(name, String(item));
+    }
   }
   return form;
 }
@@ -870,5 +964,5 @@ export {
   toPageImageBox
 };
 
-//# debugId=B34739AC554F7F9C64756E2164756E21
+//# debugId=A74469BA02A838B164756E2164756E21
 //# sourceMappingURL=index.js.map