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 +4 -4
- data/README.md +1 -1
- data/ext/kreuzberg_rb/native/Cargo.lock +83 -14
- data/ext/kreuzberg_rb/native/Cargo.toml +2 -1
- data/ext/kreuzberg_rb/native/src/extraction.rs +30 -0
- data/ext/kreuzberg_rb/native/src/lib.rs +5 -1
- data/lib/kreuzberg/extraction_api.rb +37 -0
- data/lib/kreuzberg/version.rb +1 -1
- data/spec/binding/render_spec.rb +91 -0
- data/vendor/Cargo.toml +3 -3
- data/vendor/kreuzberg/Cargo.toml +2 -2
- data/vendor/kreuzberg/README.md +1 -1
- data/vendor/kreuzberg/src/extraction/pst.rs +1 -0
- data/vendor/kreuzberg/src/extractors/pdf/mod.rs +1 -1
- data/vendor/kreuzberg/src/extractors/pdf/ocr.rs +33 -7
- data/vendor/kreuzberg/src/pdf/error.rs +18 -0
- data/vendor/kreuzberg/src/pdf/mod.rs +1 -1
- data/vendor/kreuzberg/src/pdf/rendering.rs +358 -32
- data/vendor/kreuzberg-ffi/Cargo.toml +3 -3
- data/vendor/kreuzberg-ffi/kreuzberg.h +127 -2
- data/vendor/kreuzberg-ffi/src/lib.rs +6 -0
- data/vendor/kreuzberg-ffi/src/rendering.rs +325 -0
- data/vendor/kreuzberg-paddle-ocr/Cargo.toml +1 -1
- data/vendor/kreuzberg-paddle-ocr/src/base_net.rs +15 -4
- data/vendor/kreuzberg-pdfium-render/Cargo.toml +1 -1
- data/vendor/kreuzberg-tesseract/Cargo.toml +1 -1
- metadata +4 -2
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA256:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: f087c152499deb621223ba54a8ca450bb0510da3b43c880f3969e33ec4b4d5e5
|
|
4
|
+
data.tar.gz: dba38092babe378ec93e0dd4f307ab44c10519623302fdf14023a52eff6549b2
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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)
|
data/lib/kreuzberg/version.rb
CHANGED
|
@@ -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.
|
|
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.
|
|
34
|
-
kreuzberg-ffi = { path = "./crates/kreuzberg-ffi", version = "4.6.
|
|
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"
|
data/vendor/kreuzberg/Cargo.toml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
[package]
|
|
2
2
|
name = "kreuzberg"
|
|
3
|
-
version = "4.6.
|
|
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.
|
|
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 }
|
data/vendor/kreuzberg/README.md
CHANGED
|
@@ -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.
|
|
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>)> {
|
|
@@ -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
|
-
|
|
745
|
-
|
|
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,
|
|
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(
|
|
805
|
-
let hints = detection_to_layout_hints(
|
|
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")]
|