kreuzberg 4.6.1 → 4.6.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: 7f2441d44e9083d36f9f0e60b2c155624ad251f97c6f2ca629bd3ef33aeeeb46
4
- data.tar.gz: b443884a5e4bfffa1bc916928ae97997e1ca21cdfb80e76b845c454c0aa67256
3
+ metadata.gz: f087c152499deb621223ba54a8ca450bb0510da3b43c880f3969e33ec4b4d5e5
4
+ data.tar.gz: dba38092babe378ec93e0dd4f307ab44c10519623302fdf14023a52eff6549b2
5
5
  SHA512:
6
- metadata.gz: e238336b3ceae6d2bed4bd530d393f32473a7a88f68f7e28a6300ffe2f7cadc5aee0ac834fa5712223810f9cf1f7a68a363b7e96079060565e1c9e10af9f8114
7
- data.tar.gz: 48eae2f77c78ef1b9794ac3dde62c785b32ab99253b6ab7ce8da9d8bba8a57f209fb5fc6353a0bc8f5771bd825fc3b3b29652149133bd0ba1dc29e58101185bb
6
+ metadata.gz: 2e2cb3a3636555ac5f7d54bf473e36872efef5cf1796c667f727fa8622792a5cec009799d35c5ccb776997b7c39a0e5d0745266c7c5e5604d22b4e3e3e2f5e1f
7
+ data.tar.gz: b032174ab9b1366d3d05082ef6cfe3b1b700ff66d265fb28353d69b581aee6989688dd8e01b8c516f7046a8968ca86edd8a80cb72cb48477607eb8104d7facb3
data/README.md CHANGED
@@ -22,7 +22,7 @@
22
22
  <img src="https://img.shields.io/maven-central/v/dev.kreuzberg/kreuzberg?label=Java&color=007ec6" alt="Java">
23
23
  </a>
24
24
  <a href="https://github.com/kreuzberg-dev/kreuzberg/releases">
25
- <img src="https://img.shields.io/github/v/tag/kreuzberg-dev/kreuzberg?label=Go&color=007ec6&filter=v4.6.1" alt="Go">
25
+ <img src="https://img.shields.io/github/v/tag/kreuzberg-dev/kreuzberg?label=Go&color=007ec6&filter=v4.6.2" alt="Go">
26
26
  </a>
27
27
  <a href="https://www.nuget.org/packages/Kreuzberg/">
28
28
  <img src="https://img.shields.io/nuget/v/Kreuzberg?label=C%23&color=007ec6" alt="C#">
@@ -499,6 +499,15 @@ dependencies = [
499
499
  "generic-array",
500
500
  ]
501
501
 
502
+ [[package]]
503
+ name = "block-buffer"
504
+ version = "0.12.0"
505
+ source = "registry+https://github.com/rust-lang/crates.io-index"
506
+ checksum = "cdd35008169921d80bc60d3d0ab416eecb028c4cd653352907921d95084790be"
507
+ dependencies = [
508
+ "hybrid-array",
509
+ ]
510
+
502
511
  [[package]]
503
512
  name = "block-padding"
504
513
  version = "0.3.3"
@@ -726,7 +735,7 @@ version = "0.4.4"
726
735
  source = "registry+https://github.com/rust-lang/crates.io-index"
727
736
  checksum = "773f3b9af64447d2ce9850330c473515014aa235e6a783b02db81ff39e4a3dad"
728
737
  dependencies = [
729
- "crypto-common",
738
+ "crypto-common 0.1.7",
730
739
  "inout",
731
740
  ]
732
741
 
@@ -877,6 +886,12 @@ dependencies = [
877
886
  "web-sys",
878
887
  ]
879
888
 
889
+ [[package]]
890
+ name = "const-oid"
891
+ version = "0.10.2"
892
+ source = "registry+https://github.com/rust-lang/crates.io-index"
893
+ checksum = "a6ef517f0926dd24a1582492c791b6a4818a4d94e789a334894aa15b0d12f55c"
894
+
880
895
  [[package]]
881
896
  name = "const-random"
882
897
  version = "0.1.18"
@@ -1034,6 +1049,15 @@ dependencies = [
1034
1049
  "typenum",
1035
1050
  ]
1036
1051
 
1052
+ [[package]]
1053
+ name = "crypto-common"
1054
+ version = "0.2.1"
1055
+ source = "registry+https://github.com/rust-lang/crates.io-index"
1056
+ checksum = "77727bb15fa921304124b128af125e7e3b968275d1b108b379190264f4423710"
1057
+ dependencies = [
1058
+ "hybrid-array",
1059
+ ]
1060
+
1037
1061
  [[package]]
1038
1062
  name = "ctor"
1039
1063
  version = "0.6.3"
@@ -1232,8 +1256,19 @@ version = "0.10.7"
1232
1256
  source = "registry+https://github.com/rust-lang/crates.io-index"
1233
1257
  checksum = "9ed9a281f7bc9b7576e61468ba615a66a5c8cfdff42420a70aa82701a3b1e292"
1234
1258
  dependencies = [
1235
- "block-buffer",
1236
- "crypto-common",
1259
+ "block-buffer 0.10.4",
1260
+ "crypto-common 0.1.7",
1261
+ ]
1262
+
1263
+ [[package]]
1264
+ name = "digest"
1265
+ version = "0.11.2"
1266
+ source = "registry+https://github.com/rust-lang/crates.io-index"
1267
+ checksum = "4850db49bf08e663084f7fb5c87d202ef91a3907271aff24a94eb97ff039153c"
1268
+ dependencies = [
1269
+ "block-buffer 0.12.0",
1270
+ "const-oid",
1271
+ "crypto-common 0.2.1",
1237
1272
  ]
1238
1273
 
1239
1274
  [[package]]
@@ -2061,6 +2096,15 @@ version = "1.0.3"
2061
2096
  source = "registry+https://github.com/rust-lang/crates.io-index"
2062
2097
  checksum = "df3b46402a9d5adb4c86a0cf463f42e19994e3ee891101b1841f30a545cb49a9"
2063
2098
 
2099
+ [[package]]
2100
+ name = "hybrid-array"
2101
+ version = "0.4.8"
2102
+ source = "registry+https://github.com/rust-lang/crates.io-index"
2103
+ checksum = "8655f91cd07f2b9d0c24137bd650fe69617773435ee5ec83022377777ce65ef1"
2104
+ dependencies = [
2105
+ "typenum",
2106
+ ]
2107
+
2064
2108
  [[package]]
2065
2109
  name = "hyper"
2066
2110
  version = "1.8.1"
@@ -2612,7 +2656,7 @@ dependencies = [
2612
2656
 
2613
2657
  [[package]]
2614
2658
  name = "kreuzberg"
2615
- version = "4.6.0"
2659
+ version = "4.6.2"
2616
2660
  dependencies = [
2617
2661
  "ahash",
2618
2662
  "async-trait",
@@ -2626,6 +2670,7 @@ dependencies = [
2626
2670
  "calamine",
2627
2671
  "cfb 0.14.0",
2628
2672
  "chardetng",
2673
+ "chrono",
2629
2674
  "dashmap",
2630
2675
  "dbase",
2631
2676
  "encoding_rs",
@@ -2658,6 +2703,7 @@ dependencies = [
2658
2703
  "opentelemetry_sdk",
2659
2704
  "org",
2660
2705
  "ort",
2706
+ "outlook-pst",
2661
2707
  "parking_lot",
2662
2708
  "pastey 0.2.1",
2663
2709
  "pkg-config",
@@ -2673,9 +2719,10 @@ dependencies = [
2673
2719
  "serde_json",
2674
2720
  "serde_yaml_ng",
2675
2721
  "sevenz-rust2",
2676
- "sha2",
2722
+ "sha2 0.11.0",
2677
2723
  "snap",
2678
2724
  "tar",
2725
+ "tempfile",
2679
2726
  "text-splitter",
2680
2727
  "thiserror 2.0.18",
2681
2728
  "tiff",
@@ -2696,7 +2743,7 @@ dependencies = [
2696
2743
 
2697
2744
  [[package]]
2698
2745
  name = "kreuzberg-ffi"
2699
- version = "4.6.0"
2746
+ version = "4.6.2"
2700
2747
  dependencies = [
2701
2748
  "ahash",
2702
2749
  "async-trait",
@@ -2712,7 +2759,7 @@ dependencies = [
2712
2759
 
2713
2760
  [[package]]
2714
2761
  name = "kreuzberg-paddle-ocr"
2715
- version = "4.6.0"
2762
+ version = "4.6.2"
2716
2763
  dependencies = [
2717
2764
  "geo-clipper",
2718
2765
  "geo-types",
@@ -2726,7 +2773,7 @@ dependencies = [
2726
2773
 
2727
2774
  [[package]]
2728
2775
  name = "kreuzberg-pdfium-render"
2729
- version = "4.6.0"
2776
+ version = "4.6.2"
2730
2777
  dependencies = [
2731
2778
  "bitflags",
2732
2779
  "bytemuck",
@@ -2749,7 +2796,7 @@ dependencies = [
2749
2796
 
2750
2797
  [[package]]
2751
2798
  name = "kreuzberg-rb"
2752
- version = "4.6.0"
2799
+ version = "4.6.2"
2753
2800
  dependencies = [
2754
2801
  "async-trait",
2755
2802
  "html-to-markdown-rs",
@@ -2766,7 +2813,7 @@ dependencies = [
2766
2813
 
2767
2814
  [[package]]
2768
2815
  name = "kreuzberg-tesseract"
2769
- version = "4.6.0"
2816
+ version = "4.6.2"
2770
2817
  dependencies = [
2771
2818
  "cc",
2772
2819
  "cmake",
@@ -2931,7 +2978,7 @@ dependencies = [
2931
2978
  "rand 0.10.0",
2932
2979
  "rangemap",
2933
2980
  "rayon",
2934
- "sha2",
2981
+ "sha2 0.10.9",
2935
2982
  "stringprep",
2936
2983
  "thiserror 2.0.18",
2937
2984
  "time",
@@ -3068,7 +3115,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
3068
3115
  checksum = "d89e7ee0cfbedfc4da3340218492196241d89eefb6dab27de5df917a6d2e78cf"
3069
3116
  dependencies = [
3070
3117
  "cfg-if",
3071
- "digest",
3118
+ "digest 0.10.7",
3072
3119
  ]
3073
3120
 
3074
3121
  [[package]]
@@ -3540,6 +3587,17 @@ dependencies = [
3540
3587
  "ureq 3.3.0",
3541
3588
  ]
3542
3589
 
3590
+ [[package]]
3591
+ name = "outlook-pst"
3592
+ version = "1.2.0"
3593
+ source = "registry+https://github.com/rust-lang/crates.io-index"
3594
+ checksum = "299eace9d895cc31490927d7de779f8e8f3deb5fb6bd1e68eb192aa1da19bd47"
3595
+ dependencies = [
3596
+ "byteorder",
3597
+ "thiserror 2.0.18",
3598
+ "tracing",
3599
+ ]
3600
+
3543
3601
  [[package]]
3544
3602
  name = "parking_lot"
3545
3603
  version = "0.12.5"
@@ -4652,7 +4710,7 @@ dependencies = [
4652
4710
  "js-sys",
4653
4711
  "lzma-rust2 0.16.2",
4654
4712
  "ppmd-rust",
4655
- "sha2",
4713
+ "sha2 0.10.9",
4656
4714
  "wasm-bindgen",
4657
4715
  ]
4658
4716
 
@@ -4664,7 +4722,18 @@ checksum = "a7507d819769d01a365ab707794a4084392c824f54a7a6a7862f8c3d0892b283"
4664
4722
  dependencies = [
4665
4723
  "cfg-if",
4666
4724
  "cpufeatures 0.2.17",
4667
- "digest",
4725
+ "digest 0.10.7",
4726
+ ]
4727
+
4728
+ [[package]]
4729
+ name = "sha2"
4730
+ version = "0.11.0"
4731
+ source = "registry+https://github.com/rust-lang/crates.io-index"
4732
+ checksum = "446ba717509524cb3f22f17ecc096f10f4822d76ab5c0b9822c5f9c284e825f4"
4733
+ dependencies = [
4734
+ "cfg-if",
4735
+ "cpufeatures 0.3.0",
4736
+ "digest 0.11.2",
4668
4737
  ]
4669
4738
 
4670
4739
  [[package]]
@@ -1,6 +1,6 @@
1
1
  [package]
2
2
  name = "kreuzberg-rb"
3
- version = "4.6.1"
3
+ version = "4.6.2"
4
4
  edition = "2024"
5
5
  rust-version = "1.91"
6
6
  authors = ["Na'aman Hirschfeld <naaman@kreuzberg.dev>"]
@@ -36,6 +36,7 @@ kreuzberg = { path = "../../../vendor/kreuzberg", default-features = false, feat
36
36
  "chunking",
37
37
  "chunking-tokenizers",
38
38
  "embeddings",
39
+ "ort-bundled",
39
40
  "quality",
40
41
  "keywords",
41
42
  "api",
@@ -58,6 +58,36 @@ pub fn extract_file(args: &[Value]) -> Result<RHash, Error> {
58
58
  extraction_result_to_ruby(&ruby, result)
59
59
  }
60
60
 
61
+ /// Iterate over PDF pages, yielding (page_index, png_bytes) per page to a Ruby block.
62
+ pub fn render_pdf_pages_iter(path: String, dpi: i32) -> Result<(), Error> {
63
+ let ruby = Ruby::get().expect("Ruby not initialized");
64
+ let dpi_opt = if dpi <= 0 { None } else { Some(dpi) };
65
+
66
+ let iter = kreuzberg::pdf::PdfPageIterator::from_file(&path, dpi_opt, None)
67
+ .map_err(|e| kreuzberg_error(e.into()))?;
68
+
69
+ for result in iter {
70
+ let (page_index, png_bytes) = result.map_err(|e| kreuzberg_error(e.into()))?;
71
+ let rb_index = ruby.integer_from_i64(page_index as i64);
72
+ let rb_bytes = ruby.str_from_slice(&png_bytes);
73
+ let _: magnus::Value = ruby.yield_values((rb_index, rb_bytes))?;
74
+ }
75
+
76
+ Ok(())
77
+ }
78
+
79
+ /// Render a single PDF page to PNG bytes.
80
+ pub fn native_render_pdf_page(path: String, page_index: i64, dpi: i64) -> Result<Vec<u8>, Error> {
81
+ if page_index < 0 {
82
+ return Err(crate::error_handling::runtime_error("page_index must be non-negative"));
83
+ }
84
+ let pdf_bytes = std::fs::read(&path)
85
+ .map_err(|e| crate::error_handling::runtime_error(format!("Failed to read file: {}", e)))?;
86
+ let dpi_opt = if dpi <= 0 { None } else { Some(dpi as i32) };
87
+ kreuzberg::pdf::render_pdf_page_to_png(&pdf_bytes, page_index as usize, dpi_opt, None)
88
+ .map_err(|e| kreuzberg_error(e.into()))
89
+ }
90
+
61
91
  /// Extract content from bytes (asynchronous)
62
92
  pub fn extract_bytes(args: &[Value]) -> Result<RHash, Error> {
63
93
  let ruby = Ruby::get().expect("Ruby not initialized");
@@ -23,7 +23,7 @@ pub use gc_guarded_value::GcGuardedValue;
23
23
  pub use helpers::{get_kw, set_hash_entry, json_value_to_ruby, ruby_value_to_json, cache_root_dir, cache_directories};
24
24
  pub use config::parse_extraction_config;
25
25
  pub use result::extraction_result_to_ruby;
26
- pub use extraction::{extract_file_sync, extract_bytes_sync, extract_file, extract_bytes};
26
+ pub use extraction::{extract_file_sync, extract_bytes_sync, extract_file, extract_bytes, render_pdf_pages_iter, native_render_pdf_page};
27
27
  pub use batch::{
28
28
  batch_extract_files_sync, batch_extract_bytes_sync, batch_extract_files, batch_extract_bytes,
29
29
  };
@@ -442,6 +442,10 @@ fn init(ruby: &Ruby) -> Result<(), Error> {
442
442
  module.define_module_function("batch_extract_files", function!(batch_extract_files, -1))?;
443
443
  module.define_module_function("batch_extract_bytes", function!(batch_extract_bytes, -1))?;
444
444
 
445
+ // PDF page iterator
446
+ module.define_module_function("native_render_pdf_pages_iter", function!(render_pdf_pages_iter, 2))?;
447
+ module.define_module_function("native_render_pdf_page", function!(native_render_pdf_page, 3))?;
448
+
445
449
  // Cache functions
446
450
  module.define_module_function("clear_cache", function!(ruby_clear_cache, 0))?;
447
451
  module.define_module_function("cache_stats", function!(ruby_cache_stats, 0))?;
@@ -319,6 +319,43 @@ module Kreuzberg
319
319
  results
320
320
  end
321
321
 
322
+ # Render a single PDF page as a PNG image.
323
+ #
324
+ # @param path [String, Pathname] Path to the PDF file
325
+ # @param page_index [Integer] Zero-based page index
326
+ # @param dpi [Integer] Rendering resolution (default 150)
327
+ # @return [String] PNG-encoded binary string
328
+ # @raise [Errors::IOError] If the file cannot be read
329
+ # @raise [Errors::ParsingError] If rendering fails
330
+ def render_pdf_page(path, page_index, dpi: 150)
331
+ path_str = path.to_s
332
+ raise ArgumentError, 'page_index must be non-negative' if page_index.negative?
333
+ raise Errors::IOError, "File not found: #{path_str}" unless File.exist?(path_str)
334
+
335
+ native_render_pdf_page(path_str, page_index, dpi)
336
+ end
337
+
338
+ # Iterate over pages of a PDF lazily, yielding each page as it is rendered.
339
+ #
340
+ # Each page is rendered via the native FFI iterator, so only one page is in
341
+ # memory at a time.
342
+ #
343
+ # @param path [String, Pathname] Path to the PDF file
344
+ # @param dpi [Integer] Rendering resolution (default 150)
345
+ # @yieldparam page_index [Integer] Zero-based page index
346
+ # @yieldparam png_bytes [String] PNG-encoded binary string for the page
347
+ # @return [Enumerator] if no block is given
348
+ # @raise [Errors::IOError] If the file cannot be read
349
+ # @raise [Errors::ParsingError] If rendering fails
350
+ def render_pdf_pages_iter(path, dpi: 150, &block)
351
+ path_str = path.to_s
352
+ raise Errors::IOError, "File not found: #{path_str}" unless File.exist?(path_str)
353
+
354
+ return enum_for(:render_pdf_pages_iter, path, dpi: dpi) unless block
355
+
356
+ native_render_pdf_pages_iter(path_str, dpi, &block)
357
+ end
358
+
322
359
  def normalize_config(config)
323
360
  return {} if config.nil?
324
361
  return config if config.is_a?(Hash)
@@ -1,5 +1,5 @@
1
1
  # frozen_string_literal: true
2
2
 
3
3
  module Kreuzberg
4
- VERSION = '4.6.1'
4
+ VERSION = '4.6.2'
5
5
  end
@@ -0,0 +1,91 @@
1
+ # Hand-written binding-specific edge case tests for PDF rendering.
2
+ # Happy-path render tests are auto-generated from fixtures in e2e/.
3
+ # These tests cover error handling, validation, and lifecycle patterns
4
+ # that vary per language and can't be generated uniformly.
5
+
6
+ # frozen_string_literal: true
7
+
8
+ require 'spec_helper'
9
+
10
+ RSpec.describe 'PDF Rendering' do
11
+ it 'exposes rendering methods' do
12
+ expect(Kreuzberg).to respond_to(:render_pdf_page)
13
+ expect(Kreuzberg).to respond_to(:render_pdf_pages_iter)
14
+ end
15
+
16
+ describe '.render_pdf_page' do
17
+ it 'raises an error for a nonexistent file' do
18
+ expect do
19
+ Kreuzberg.render_pdf_page('/nonexistent/path/to/document.pdf', 0)
20
+ end.to raise_error(Kreuzberg::Errors::IOError)
21
+ end
22
+
23
+ it 'raises an error for an out-of-bounds page index' do
24
+ pdf_path = test_document_path('pdf/tiny.pdf')
25
+ skip 'Test PDF not available' unless File.exist?(pdf_path)
26
+
27
+ expect do
28
+ Kreuzberg.render_pdf_page(pdf_path, 9999)
29
+ end.to raise_error(StandardError)
30
+ end
31
+ end
32
+
33
+ describe '.render_pdf_page with negative index' do
34
+ it 'raises ArgumentError for a negative page index' do
35
+ pdf_path = test_document_path('pdf/tiny.pdf')
36
+ skip 'Test PDF not available' unless File.exist?(pdf_path)
37
+
38
+ expect do
39
+ Kreuzberg.render_pdf_page(pdf_path, -1)
40
+ end.to raise_error(ArgumentError)
41
+ end
42
+ end
43
+
44
+ describe '.render_pdf_pages_iter' do
45
+ it 'raises an error for a nonexistent file' do
46
+ expect do
47
+ Kreuzberg.render_pdf_pages_iter('/nonexistent/path/to/document.pdf') { |_, _| nil }
48
+ end.to raise_error(Kreuzberg::Errors::IOError)
49
+ end
50
+ end
51
+
52
+ describe '.render_pdf_page with empty path' do
53
+ it 'raises an error for an empty path' do
54
+ expect do
55
+ Kreuzberg.render_pdf_page('', 0)
56
+ end.to raise_error(StandardError)
57
+ end
58
+ end
59
+
60
+ describe '.render_pdf_pages_iter cleanup' do
61
+ it 'handles iterator cleanup without fully consuming' do
62
+ pdf_path = test_document_path('pdf/tiny.pdf')
63
+ skip 'Test PDF not available' unless File.exist?(pdf_path)
64
+
65
+ # Iterate but stop immediately — no crash
66
+ Kreuzberg.render_pdf_pages_iter(pdf_path) do |_page_index, _png_data|
67
+ break
68
+ end
69
+ end
70
+ end
71
+
72
+ describe '.render_pdf_pages_iter early termination' do
73
+ it 'returns valid PNG for the first page then stops' do
74
+ pdf_path = test_document_path('pdf/tiny.pdf')
75
+ skip 'Test PDF not available' unless File.exist?(pdf_path)
76
+
77
+ first_png = nil
78
+ Kreuzberg.render_pdf_pages_iter(pdf_path) do |page_index, png_data|
79
+ expect(page_index).to eq(0)
80
+ expect(png_data).to be_a(String)
81
+ expect(png_data.bytesize).to be > 8
82
+ # PNG magic bytes
83
+ expect(png_data.bytes[0..3]).to eq([0x89, 0x50, 0x4E, 0x47])
84
+ first_png = png_data
85
+ break
86
+ end
87
+
88
+ expect(first_png).not_to be_nil
89
+ end
90
+ end
91
+ end
data/vendor/Cargo.toml CHANGED
@@ -2,7 +2,7 @@
2
2
  members = ["kreuzberg", "kreuzberg-ffi", "kreuzberg-tesseract", "kreuzberg-paddle-ocr", "kreuzberg-pdfium-render"]
3
3
 
4
4
  [workspace.package]
5
- version = "4.6.1"
5
+ version = "4.6.2"
6
6
  edition = "2024"
7
7
  rust-version = "1.91"
8
8
  authors = ["Na'aman Hirschfeld <naaman@kreuzberg.dev>"]
@@ -30,8 +30,8 @@ html-to-markdown-rs = { version = "2.29.0", default-features = false }
30
30
  image = { version = "0.25.10", default-features = false }
31
31
  itertools = "0.14"
32
32
  js-sys = "0.3"
33
- kreuzberg = { path = "./crates/kreuzberg", version = "4.6.1", default-features = false }
34
- kreuzberg-ffi = { path = "./crates/kreuzberg-ffi", version = "4.6.1" }
33
+ kreuzberg = { path = "./crates/kreuzberg", version = "4.6.2", default-features = false }
34
+ kreuzberg-ffi = { path = "./crates/kreuzberg-ffi", version = "4.6.2" }
35
35
  lazy_static = "1.5.0"
36
36
  libc = "0.2.183"
37
37
  log = "0.4"
@@ -1,6 +1,6 @@
1
1
  [package]
2
2
  name = "kreuzberg"
3
- version = "4.6.1"
3
+ version = "4.6.2"
4
4
  edition = "2024"
5
5
  rust-version = "1.91"
6
6
  authors = ["Na'aman Hirschfeld <naaman@kreuzberg.dev>"]
@@ -313,7 +313,7 @@ serde = { version = "1.0.228", features = ["derive"] }
313
313
  serde_json = { version = "1.0.149" }
314
314
  serde_yaml_ng = "0.10.0"
315
315
  sevenz-rust2 = { version = "0.20.2", optional = true }
316
- sha2 = { version = "0.10", optional = true }
316
+ sha2 = { version = "0.11", optional = true }
317
317
  simdutf8 = { version = "0.1", optional = true }
318
318
  snap = { version = "1.1", optional = true }
319
319
  tar = { version = "0.4.45", optional = true }
@@ -18,7 +18,7 @@ High-performance document intelligence library for Rust. Extract text, metadata,
18
18
 
19
19
  This is the core Rust library that powers the Python, TypeScript, and Ruby bindings.
20
20
 
21
- > **🚀 Version 4.6.1 Release**
21
+ > **🚀 Version 4.6.2 Release**
22
22
  > This is a pre-release version. We invite you to test the library and [report any issues](https://github.com/kreuzberg-dev/kreuzberg/issues) you encounter.
23
23
  >
24
24
  > **Note**: The Rust crate is not currently published to crates.io for this RC. Use git dependencies or language bindings (Python, TypeScript, Ruby) instead.
@@ -80,6 +80,7 @@ pub fn extract_pst_messages(pst_data: &[u8]) -> Result<(Vec<EmailExtractionResul
80
80
  /// occurs when the full PST is first read into a `Vec<u8>` and then written
81
81
  /// back out to a tempfile before parsing.
82
82
  #[cfg(feature = "email")]
83
+ #[allow(dead_code)]
83
84
  pub(crate) fn extract_pst_from_path(
84
85
  path: &std::path::Path,
85
86
  ) -> Result<(Vec<EmailExtractionResult>, Vec<ProcessingWarning>)> {
@@ -294,7 +294,7 @@ impl PdfExtractor {
294
294
  let content = &*derotated;
295
295
 
296
296
  #[cfg(feature = "pdf")]
297
- #[allow(unused_variables)]
297
+ #[allow(unused_variables, unused_mut)]
298
298
  let (
299
299
  mut pdf_metadata,
300
300
  native_text,
@@ -716,7 +716,7 @@ pub(crate) async fn extract_with_ocr(
716
716
  // Sequential post-processing for this batch utilizing TATR.
717
717
  for offset in 0..batch_count {
718
718
  let page_idx = batch_start + offset;
719
- let ocr_result = batch_ocr_results[offset].take().expect("OCR result missing for page");
719
+ let mut ocr_result = batch_ocr_results[offset].take().expect("OCR result missing for page");
720
720
  #[cfg(feature = "layout-detection")]
721
721
  let height = encoded_batch[offset].3;
722
722
 
@@ -731,7 +731,10 @@ pub(crate) async fn extract_with_ocr(
731
731
  }
732
732
 
733
733
  // Accumulate OCR elements from this page.
734
- if let Some(ref elems) = ocr_result.ocr_elements {
734
+ if let Some(ref mut elems) = ocr_result.ocr_elements {
735
+ for elem in elems.iter_mut() {
736
+ elem.page_number = page_idx + 1;
737
+ }
735
738
  all_ocr_elements.extend(elems.iter().cloned());
736
739
  }
737
740
 
@@ -741,8 +744,31 @@ pub(crate) async fn extract_with_ocr(
741
744
  && !elements.is_empty()
742
745
  {
743
746
  let detection = detections.get(page_idx);
744
- let recognized_tables = match (detection, tatr_model.as_mut()) {
745
- (Some(det), Some(model)) => {
747
+
748
+ // Scale layout detection bounding boxes from layout-model resolution
749
+ // (e.g. 640×640) to OCR render resolution so that coordinates are
750
+ // consistent when passed to recognize_page_tables and detection_to_layout_hints.
751
+ // `width` and `height` come from the encoded PNG dimensions, i.e. they are
752
+ // the actual pixel dimensions of the OCR-rendered page image.
753
+ let ocr_render_width = encoded_batch[offset].2;
754
+ let ocr_render_height = encoded_batch[offset].3;
755
+ let scaled_detection: Option<crate::layout::DetectionResult> = detection.map(|det| {
756
+ let sx = ocr_render_width as f32 / det.page_width as f32;
757
+ let sy = ocr_render_height as f32 / det.page_height as f32;
758
+ let mut scaled = det.clone();
759
+ scaled.page_width = ocr_render_width;
760
+ scaled.page_height = ocr_render_height;
761
+ for region in &mut scaled.detections {
762
+ region.bbox.x1 *= sx;
763
+ region.bbox.y1 *= sy;
764
+ region.bbox.x2 *= sx;
765
+ region.bbox.y2 *= sy;
766
+ }
767
+ scaled
768
+ });
769
+
770
+ let recognized_tables = match (scaled_detection.as_ref(), tatr_model.as_mut()) {
771
+ (Some(scaled_det), Some(model)) => {
746
772
  // Decode the page image from its PNG for TATR table recognition.
747
773
  // When pre-rendered images are available, use them directly.
748
774
  // Otherwise, decode from the PNG we already encoded.
@@ -757,7 +783,7 @@ pub(crate) async fn extract_with_ocr(
757
783
  })?;
758
784
  decoded.to_rgb8()
759
785
  };
760
- crate::ocr::layout_assembly::recognize_page_tables(&rgb, det, elements, model)
786
+ crate::ocr::layout_assembly::recognize_page_tables(&rgb, scaled_det, elements, model)
761
787
  }
762
788
  _ => Vec::new(),
763
789
  };
@@ -801,8 +827,8 @@ pub(crate) async fn extract_with_ocr(
801
827
  crate::pdf::markdown::reorder_elements_reading_order(&mut page_content.elements);
802
828
  let mut paragraphs = crate::pdf::markdown::content_to_paragraphs(&page_content);
803
829
 
804
- if let Some(det) = detection {
805
- let hints = detection_to_layout_hints(det, height as f32);
830
+ if let Some(ref scaled_det) = scaled_detection {
831
+ let hints = detection_to_layout_hints(scaled_det, height as f32);
806
832
  crate::pdf::markdown::layout_classify::apply_layout_overrides(
807
833
  &mut paragraphs,
808
834
  &hints,
@@ -76,6 +76,24 @@ pub(crate) fn format_pdfium_error<E: std::fmt::Debug>(error: E) -> String {
76
76
  }
77
77
  }
78
78
 
79
+ /// Classify a PDFium document-load error into the appropriate `PdfError` variant.
80
+ ///
81
+ /// Password-related errors are distinguished by inspecting the formatted message.
82
+ /// When `password` is `Some`, a password-related failure means the password was wrong;
83
+ /// when `None`, it means the document requires one.
84
+ pub(crate) fn classify_pdfium_load_error<E: std::fmt::Debug>(error: E, password: Option<&str>) -> PdfError {
85
+ let err_msg = format_pdfium_error(error);
86
+ if err_msg.contains("password") || err_msg.contains("Password") {
87
+ if password.is_some() {
88
+ PdfError::InvalidPassword
89
+ } else {
90
+ PdfError::PasswordRequired
91
+ }
92
+ } else {
93
+ PdfError::InvalidPdf(err_msg)
94
+ }
95
+ }
96
+
79
97
  #[cfg(test)]
80
98
  mod tests {
81
99
  use super::*;
@@ -99,7 +99,7 @@ pub use images::{PdfImage, PdfImageExtractor, extract_images_from_pdf};
99
99
  #[cfg(feature = "pdf")]
100
100
  pub use metadata::extract_metadata;
101
101
  #[cfg(feature = "pdf")]
102
- pub use rendering::{PageRenderOptions, render_page_to_image};
102
+ pub use rendering::{PageRenderOptions, PdfPageIterator, render_page_to_image, render_pdf_page_to_png};
103
103
  #[cfg(feature = "pdf")]
104
104
  pub use table::extract_words_from_page;
105
105
  #[cfg(feature = "pdf")]