kreuzberg 4.0.0.pre.rc.29 → 4.0.0.rc1
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/.gitignore +0 -6
- data/.rubocop.yaml +534 -1
- data/Gemfile +2 -1
- data/Gemfile.lock +28 -116
- data/README.md +269 -629
- data/Rakefile +0 -9
- data/Steepfile +4 -8
- data/examples/async_patterns.rb +58 -1
- data/ext/kreuzberg_rb/extconf.rb +5 -35
- data/ext/kreuzberg_rb/native/Cargo.toml +16 -55
- data/ext/kreuzberg_rb/native/build.rs +14 -12
- data/ext/kreuzberg_rb/native/include/ieeefp.h +1 -1
- data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +1 -1
- data/ext/kreuzberg_rb/native/include/strings.h +2 -2
- data/ext/kreuzberg_rb/native/include/unistd.h +1 -1
- data/ext/kreuzberg_rb/native/src/lib.rs +34 -897
- data/extconf.rb +6 -38
- data/kreuzberg.gemspec +20 -114
- data/lib/kreuzberg/api_proxy.rb +18 -2
- data/lib/kreuzberg/cache_api.rb +0 -22
- data/lib/kreuzberg/cli.rb +10 -2
- data/lib/kreuzberg/cli_proxy.rb +10 -0
- data/lib/kreuzberg/config.rb +22 -274
- data/lib/kreuzberg/errors.rb +7 -73
- data/lib/kreuzberg/extraction_api.rb +8 -237
- data/lib/kreuzberg/mcp_proxy.rb +11 -2
- data/lib/kreuzberg/ocr_backend_protocol.rb +73 -0
- data/lib/kreuzberg/post_processor_protocol.rb +71 -0
- data/lib/kreuzberg/result.rb +33 -151
- data/lib/kreuzberg/setup_lib_path.rb +2 -22
- data/lib/kreuzberg/validator_protocol.rb +73 -0
- data/lib/kreuzberg/version.rb +1 -1
- data/lib/kreuzberg.rb +13 -27
- data/pkg/kreuzberg-4.0.0.rc1.gem +0 -0
- data/sig/kreuzberg.rbs +12 -105
- data/spec/binding/cache_spec.rb +22 -22
- data/spec/binding/cli_proxy_spec.rb +4 -2
- data/spec/binding/cli_spec.rb +11 -12
- data/spec/binding/config_spec.rb +0 -74
- data/spec/binding/config_validation_spec.rb +6 -100
- data/spec/binding/error_handling_spec.rb +97 -283
- data/spec/binding/plugins/ocr_backend_spec.rb +8 -8
- data/spec/binding/plugins/postprocessor_spec.rb +11 -11
- data/spec/binding/plugins/validator_spec.rb +13 -12
- data/spec/examples.txt +104 -0
- data/spec/fixtures/config.toml +1 -0
- data/spec/fixtures/config.yaml +1 -0
- data/spec/fixtures/invalid_config.toml +1 -0
- data/spec/smoke/package_spec.rb +3 -2
- data/spec/spec_helper.rb +3 -1
- data/vendor/kreuzberg/Cargo.toml +67 -192
- data/vendor/kreuzberg/README.md +9 -97
- data/vendor/kreuzberg/build.rs +194 -516
- data/vendor/kreuzberg/src/api/handlers.rs +9 -130
- data/vendor/kreuzberg/src/api/mod.rs +3 -18
- data/vendor/kreuzberg/src/api/server.rs +71 -236
- data/vendor/kreuzberg/src/api/types.rs +7 -43
- data/vendor/kreuzberg/src/bin/profile_extract.rs +455 -0
- data/vendor/kreuzberg/src/cache/mod.rs +3 -27
- data/vendor/kreuzberg/src/chunking/mod.rs +79 -1705
- data/vendor/kreuzberg/src/core/batch_mode.rs +0 -60
- data/vendor/kreuzberg/src/core/config.rs +23 -905
- data/vendor/kreuzberg/src/core/extractor.rs +106 -403
- data/vendor/kreuzberg/src/core/io.rs +2 -4
- data/vendor/kreuzberg/src/core/mime.rs +12 -2
- data/vendor/kreuzberg/src/core/mod.rs +3 -22
- data/vendor/kreuzberg/src/core/pipeline.rs +78 -395
- data/vendor/kreuzberg/src/embeddings.rs +21 -169
- data/vendor/kreuzberg/src/error.rs +2 -2
- data/vendor/kreuzberg/src/extraction/archive.rs +31 -36
- data/vendor/kreuzberg/src/extraction/docx.rs +1 -365
- data/vendor/kreuzberg/src/extraction/email.rs +11 -12
- data/vendor/kreuzberg/src/extraction/excel.rs +129 -138
- data/vendor/kreuzberg/src/extraction/html.rs +170 -1447
- data/vendor/kreuzberg/src/extraction/image.rs +14 -138
- data/vendor/kreuzberg/src/extraction/libreoffice.rs +3 -13
- data/vendor/kreuzberg/src/extraction/mod.rs +5 -21
- data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +0 -2
- data/vendor/kreuzberg/src/extraction/pandoc/batch.rs +275 -0
- data/vendor/kreuzberg/src/extraction/pandoc/mime_types.rs +178 -0
- data/vendor/kreuzberg/src/extraction/pandoc/mod.rs +491 -0
- data/vendor/kreuzberg/src/extraction/pandoc/server.rs +496 -0
- data/vendor/kreuzberg/src/extraction/pandoc/subprocess.rs +1188 -0
- data/vendor/kreuzberg/src/extraction/pandoc/version.rs +162 -0
- data/vendor/kreuzberg/src/extraction/pptx.rs +94 -196
- data/vendor/kreuzberg/src/extraction/structured.rs +4 -5
- data/vendor/kreuzberg/src/extraction/table.rs +1 -2
- data/vendor/kreuzberg/src/extraction/text.rs +10 -18
- data/vendor/kreuzberg/src/extractors/archive.rs +0 -22
- data/vendor/kreuzberg/src/extractors/docx.rs +148 -69
- data/vendor/kreuzberg/src/extractors/email.rs +9 -37
- data/vendor/kreuzberg/src/extractors/excel.rs +40 -81
- data/vendor/kreuzberg/src/extractors/html.rs +173 -182
- data/vendor/kreuzberg/src/extractors/image.rs +8 -32
- data/vendor/kreuzberg/src/extractors/mod.rs +10 -171
- data/vendor/kreuzberg/src/extractors/pandoc.rs +201 -0
- data/vendor/kreuzberg/src/extractors/pdf.rs +64 -329
- data/vendor/kreuzberg/src/extractors/pptx.rs +34 -79
- data/vendor/kreuzberg/src/extractors/structured.rs +0 -16
- data/vendor/kreuzberg/src/extractors/text.rs +7 -30
- data/vendor/kreuzberg/src/extractors/xml.rs +8 -27
- data/vendor/kreuzberg/src/keywords/processor.rs +1 -9
- data/vendor/kreuzberg/src/keywords/rake.rs +1 -0
- data/vendor/kreuzberg/src/language_detection/mod.rs +51 -94
- data/vendor/kreuzberg/src/lib.rs +5 -17
- data/vendor/kreuzberg/src/mcp/mod.rs +1 -4
- data/vendor/kreuzberg/src/mcp/server.rs +21 -145
- data/vendor/kreuzberg/src/ocr/mod.rs +0 -2
- data/vendor/kreuzberg/src/ocr/processor.rs +8 -19
- data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +0 -2
- data/vendor/kreuzberg/src/pdf/error.rs +1 -93
- data/vendor/kreuzberg/src/pdf/metadata.rs +100 -263
- data/vendor/kreuzberg/src/pdf/mod.rs +2 -33
- data/vendor/kreuzberg/src/pdf/rendering.rs +12 -12
- data/vendor/kreuzberg/src/pdf/table.rs +64 -61
- data/vendor/kreuzberg/src/pdf/text.rs +24 -416
- data/vendor/kreuzberg/src/plugins/extractor.rs +8 -40
- data/vendor/kreuzberg/src/plugins/mod.rs +0 -3
- data/vendor/kreuzberg/src/plugins/ocr.rs +14 -22
- data/vendor/kreuzberg/src/plugins/processor.rs +1 -10
- data/vendor/kreuzberg/src/plugins/registry.rs +0 -15
- data/vendor/kreuzberg/src/plugins/validator.rs +8 -20
- data/vendor/kreuzberg/src/stopwords/mod.rs +2 -2
- data/vendor/kreuzberg/src/text/mod.rs +0 -8
- data/vendor/kreuzberg/src/text/quality.rs +15 -28
- data/vendor/kreuzberg/src/text/string_utils.rs +10 -22
- data/vendor/kreuzberg/src/text/token_reduction/core.rs +50 -86
- data/vendor/kreuzberg/src/text/token_reduction/filters.rs +16 -37
- data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +1 -2
- data/vendor/kreuzberg/src/types.rs +67 -907
- data/vendor/kreuzberg/src/utils/mod.rs +0 -14
- data/vendor/kreuzberg/src/utils/quality.rs +3 -12
- data/vendor/kreuzberg/tests/api_tests.rs +0 -506
- data/vendor/kreuzberg/tests/archive_integration.rs +0 -2
- data/vendor/kreuzberg/tests/batch_orchestration.rs +12 -57
- data/vendor/kreuzberg/tests/batch_processing.rs +8 -32
- data/vendor/kreuzberg/tests/chunking_offset_demo.rs +92 -0
- data/vendor/kreuzberg/tests/concurrency_stress.rs +8 -40
- data/vendor/kreuzberg/tests/config_features.rs +1 -33
- data/vendor/kreuzberg/tests/config_loading_tests.rs +39 -16
- data/vendor/kreuzberg/tests/core_integration.rs +9 -35
- data/vendor/kreuzberg/tests/csv_integration.rs +81 -71
- data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +25 -23
- data/vendor/kreuzberg/tests/email_integration.rs +1 -3
- data/vendor/kreuzberg/tests/error_handling.rs +34 -43
- data/vendor/kreuzberg/tests/format_integration.rs +1 -7
- data/vendor/kreuzberg/tests/helpers/mod.rs +0 -60
- data/vendor/kreuzberg/tests/image_integration.rs +0 -2
- data/vendor/kreuzberg/tests/mime_detection.rs +16 -17
- data/vendor/kreuzberg/tests/ocr_configuration.rs +0 -4
- data/vendor/kreuzberg/tests/ocr_errors.rs +0 -22
- data/vendor/kreuzberg/tests/ocr_quality.rs +0 -2
- data/vendor/kreuzberg/tests/pandoc_integration.rs +503 -0
- data/vendor/kreuzberg/tests/pdf_integration.rs +0 -2
- data/vendor/kreuzberg/tests/pipeline_integration.rs +2 -36
- data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +0 -5
- data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +1 -17
- data/vendor/kreuzberg/tests/plugin_system.rs +0 -6
- data/vendor/kreuzberg/tests/registry_integration_tests.rs +22 -2
- data/vendor/kreuzberg/tests/security_validation.rs +1 -13
- data/vendor/kreuzberg/tests/test_fastembed.rs +23 -45
- metadata +25 -171
- data/.rubocop.yml +0 -543
- data/ext/kreuzberg_rb/native/.cargo/config.toml +0 -23
- data/ext/kreuzberg_rb/native/Cargo.lock +0 -7619
- data/lib/kreuzberg/error_context.rb +0 -136
- data/lib/kreuzberg/types.rb +0 -170
- data/lib/libpdfium.so +0 -0
- data/spec/binding/async_operations_spec.rb +0 -473
- data/spec/binding/batch_operations_spec.rb +0 -595
- data/spec/binding/batch_spec.rb +0 -359
- data/spec/binding/config_result_spec.rb +0 -377
- data/spec/binding/embeddings_spec.rb +0 -816
- data/spec/binding/error_recovery_spec.rb +0 -488
- data/spec/binding/font_config_spec.rb +0 -220
- data/spec/binding/images_spec.rb +0 -738
- data/spec/binding/keywords_extraction_spec.rb +0 -600
- data/spec/binding/metadata_types_spec.rb +0 -1228
- data/spec/binding/pages_extraction_spec.rb +0 -471
- data/spec/binding/tables_spec.rb +0 -641
- data/spec/unit/config/chunking_config_spec.rb +0 -213
- data/spec/unit/config/embedding_config_spec.rb +0 -343
- data/spec/unit/config/extraction_config_spec.rb +0 -438
- data/spec/unit/config/font_config_spec.rb +0 -285
- data/spec/unit/config/hierarchy_config_spec.rb +0 -314
- data/spec/unit/config/image_extraction_config_spec.rb +0 -209
- data/spec/unit/config/image_preprocessing_config_spec.rb +0 -249
- data/spec/unit/config/keyword_config_spec.rb +0 -229
- data/spec/unit/config/language_detection_config_spec.rb +0 -258
- data/spec/unit/config/ocr_config_spec.rb +0 -171
- data/spec/unit/config/page_config_spec.rb +0 -221
- data/spec/unit/config/pdf_config_spec.rb +0 -267
- data/spec/unit/config/postprocessor_config_spec.rb +0 -290
- data/spec/unit/config/tesseract_config_spec.rb +0 -181
- data/spec/unit/config/token_reduction_config_spec.rb +0 -251
- data/test/metadata_types_test.rb +0 -959
- data/vendor/Cargo.toml +0 -61
- data/vendor/kreuzberg/examples/bench_fixes.rs +0 -71
- data/vendor/kreuzberg/examples/test_pdfium_fork.rs +0 -62
- data/vendor/kreuzberg/src/chunking/processor.rs +0 -219
- data/vendor/kreuzberg/src/core/batch_optimizations.rs +0 -385
- data/vendor/kreuzberg/src/core/config_validation.rs +0 -949
- data/vendor/kreuzberg/src/core/formats.rs +0 -235
- data/vendor/kreuzberg/src/core/server_config.rs +0 -1220
- data/vendor/kreuzberg/src/extraction/capacity.rs +0 -263
- data/vendor/kreuzberg/src/extraction/markdown.rs +0 -216
- data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +0 -284
- data/vendor/kreuzberg/src/extractors/bibtex.rs +0 -470
- data/vendor/kreuzberg/src/extractors/docbook.rs +0 -504
- data/vendor/kreuzberg/src/extractors/epub.rs +0 -696
- data/vendor/kreuzberg/src/extractors/fictionbook.rs +0 -492
- data/vendor/kreuzberg/src/extractors/jats.rs +0 -1054
- data/vendor/kreuzberg/src/extractors/jupyter.rs +0 -368
- data/vendor/kreuzberg/src/extractors/latex.rs +0 -653
- data/vendor/kreuzberg/src/extractors/markdown.rs +0 -701
- data/vendor/kreuzberg/src/extractors/odt.rs +0 -628
- data/vendor/kreuzberg/src/extractors/opml.rs +0 -635
- data/vendor/kreuzberg/src/extractors/orgmode.rs +0 -529
- data/vendor/kreuzberg/src/extractors/rst.rs +0 -577
- data/vendor/kreuzberg/src/extractors/rtf.rs +0 -809
- data/vendor/kreuzberg/src/extractors/security.rs +0 -484
- data/vendor/kreuzberg/src/extractors/security_tests.rs +0 -367
- data/vendor/kreuzberg/src/extractors/typst.rs +0 -651
- data/vendor/kreuzberg/src/language_detection/processor.rs +0 -218
- data/vendor/kreuzberg/src/ocr/language_registry.rs +0 -520
- data/vendor/kreuzberg/src/panic_context.rs +0 -154
- data/vendor/kreuzberg/src/pdf/bindings.rs +0 -306
- data/vendor/kreuzberg/src/pdf/bundled.rs +0 -408
- data/vendor/kreuzberg/src/pdf/fonts.rs +0 -358
- data/vendor/kreuzberg/src/pdf/hierarchy.rs +0 -903
- data/vendor/kreuzberg/src/text/quality_processor.rs +0 -231
- data/vendor/kreuzberg/src/text/utf8_validation.rs +0 -193
- data/vendor/kreuzberg/src/utils/pool.rs +0 -503
- data/vendor/kreuzberg/src/utils/pool_sizing.rs +0 -364
- data/vendor/kreuzberg/src/utils/string_pool.rs +0 -761
- data/vendor/kreuzberg/tests/api_embed.rs +0 -360
- data/vendor/kreuzberg/tests/api_extract_multipart.rs +0 -52
- data/vendor/kreuzberg/tests/api_large_pdf_extraction.rs +0 -471
- data/vendor/kreuzberg/tests/api_large_pdf_extraction_diagnostics.rs +0 -289
- data/vendor/kreuzberg/tests/batch_pooling_benchmark.rs +0 -154
- data/vendor/kreuzberg/tests/bibtex_parity_test.rs +0 -421
- data/vendor/kreuzberg/tests/config_integration_test.rs +0 -753
- data/vendor/kreuzberg/tests/data/hierarchy_ground_truth.json +0 -294
- data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +0 -500
- data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +0 -370
- data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +0 -275
- data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +0 -228
- data/vendor/kreuzberg/tests/html_table_test.rs +0 -551
- data/vendor/kreuzberg/tests/instrumentation_test.rs +0 -139
- data/vendor/kreuzberg/tests/jats_extractor_tests.rs +0 -639
- data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +0 -704
- data/vendor/kreuzberg/tests/latex_extractor_tests.rs +0 -496
- data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +0 -490
- data/vendor/kreuzberg/tests/ocr_language_registry.rs +0 -191
- data/vendor/kreuzberg/tests/odt_extractor_tests.rs +0 -674
- data/vendor/kreuzberg/tests/opml_extractor_tests.rs +0 -616
- data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +0 -822
- data/vendor/kreuzberg/tests/page_markers.rs +0 -297
- data/vendor/kreuzberg/tests/pdf_hierarchy_detection.rs +0 -301
- data/vendor/kreuzberg/tests/pdf_hierarchy_quality.rs +0 -589
- data/vendor/kreuzberg/tests/pdf_ocr_triggering.rs +0 -301
- data/vendor/kreuzberg/tests/pdf_text_merging.rs +0 -475
- data/vendor/kreuzberg/tests/pdfium_linking.rs +0 -340
- data/vendor/kreuzberg/tests/rst_extractor_tests.rs +0 -694
- data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +0 -775
- data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +0 -1260
- data/vendor/kreuzberg/tests/typst_extractor_tests.rs +0 -648
- data/vendor/kreuzberg-ffi/Cargo.toml +0 -67
- data/vendor/kreuzberg-ffi/README.md +0 -851
- data/vendor/kreuzberg-ffi/benches/result_view_benchmark.rs +0 -227
- data/vendor/kreuzberg-ffi/build.rs +0 -168
- data/vendor/kreuzberg-ffi/cbindgen.toml +0 -37
- data/vendor/kreuzberg-ffi/kreuzberg-ffi.pc.in +0 -12
- data/vendor/kreuzberg-ffi/kreuzberg.h +0 -3012
- data/vendor/kreuzberg-ffi/src/batch_streaming.rs +0 -588
- data/vendor/kreuzberg-ffi/src/config.rs +0 -1341
- data/vendor/kreuzberg-ffi/src/error.rs +0 -901
- data/vendor/kreuzberg-ffi/src/extraction.rs +0 -555
- data/vendor/kreuzberg-ffi/src/helpers.rs +0 -879
- data/vendor/kreuzberg-ffi/src/lib.rs +0 -977
- data/vendor/kreuzberg-ffi/src/memory.rs +0 -493
- data/vendor/kreuzberg-ffi/src/mime.rs +0 -329
- data/vendor/kreuzberg-ffi/src/panic_shield.rs +0 -265
- data/vendor/kreuzberg-ffi/src/plugins/document_extractor.rs +0 -442
- data/vendor/kreuzberg-ffi/src/plugins/mod.rs +0 -14
- data/vendor/kreuzberg-ffi/src/plugins/ocr_backend.rs +0 -628
- data/vendor/kreuzberg-ffi/src/plugins/post_processor.rs +0 -438
- data/vendor/kreuzberg-ffi/src/plugins/validator.rs +0 -329
- data/vendor/kreuzberg-ffi/src/result.rs +0 -510
- data/vendor/kreuzberg-ffi/src/result_pool.rs +0 -639
- data/vendor/kreuzberg-ffi/src/result_view.rs +0 -773
- data/vendor/kreuzberg-ffi/src/string_intern.rs +0 -568
- data/vendor/kreuzberg-ffi/src/types.rs +0 -363
- data/vendor/kreuzberg-ffi/src/util.rs +0 -210
- data/vendor/kreuzberg-ffi/src/validation.rs +0 -848
- data/vendor/kreuzberg-ffi/tests.disabled/README.md +0 -48
- data/vendor/kreuzberg-ffi/tests.disabled/config_loading_tests.rs +0 -299
- data/vendor/kreuzberg-ffi/tests.disabled/config_tests.rs +0 -346
- data/vendor/kreuzberg-ffi/tests.disabled/extractor_tests.rs +0 -232
- data/vendor/kreuzberg-ffi/tests.disabled/plugin_registration_tests.rs +0 -470
- data/vendor/kreuzberg-tesseract/.commitlintrc.json +0 -13
- data/vendor/kreuzberg-tesseract/.crate-ignore +0 -2
- data/vendor/kreuzberg-tesseract/Cargo.lock +0 -2933
- data/vendor/kreuzberg-tesseract/Cargo.toml +0 -57
- data/vendor/kreuzberg-tesseract/LICENSE +0 -22
- data/vendor/kreuzberg-tesseract/README.md +0 -399
- data/vendor/kreuzberg-tesseract/build.rs +0 -1127
- data/vendor/kreuzberg-tesseract/patches/README.md +0 -71
- data/vendor/kreuzberg-tesseract/patches/tesseract.diff +0 -199
- data/vendor/kreuzberg-tesseract/src/api.rs +0 -1371
- data/vendor/kreuzberg-tesseract/src/choice_iterator.rs +0 -77
- data/vendor/kreuzberg-tesseract/src/enums.rs +0 -297
- data/vendor/kreuzberg-tesseract/src/error.rs +0 -81
- data/vendor/kreuzberg-tesseract/src/lib.rs +0 -145
- data/vendor/kreuzberg-tesseract/src/monitor.rs +0 -57
- data/vendor/kreuzberg-tesseract/src/mutable_iterator.rs +0 -197
- data/vendor/kreuzberg-tesseract/src/page_iterator.rs +0 -253
- data/vendor/kreuzberg-tesseract/src/result_iterator.rs +0 -286
- data/vendor/kreuzberg-tesseract/src/result_renderer.rs +0 -183
- data/vendor/kreuzberg-tesseract/tests/integration_test.rs +0 -211
|
@@ -1,227 +0,0 @@
|
|
|
1
|
-
use criterion::{BenchmarkId, Criterion, Throughput, criterion_group, criterion_main};
|
|
2
|
-
use kreuzberg::types::{Chunk, ChunkMetadata, ExtractionResult, Metadata, PageStructure, PageUnitType};
|
|
3
|
-
use kreuzberg_ffi::{CExtractionResultView, kreuzberg_get_result_view};
|
|
4
|
-
use std::ffi::CString;
|
|
5
|
-
use std::hint;
|
|
6
|
-
use std::mem;
|
|
7
|
-
|
|
8
|
-
/// Create a test extraction result with configurable complexity
|
|
9
|
-
fn create_test_result(content_size: usize, chunk_count: usize) -> ExtractionResult {
|
|
10
|
-
let mut metadata = Metadata {
|
|
11
|
-
title: Some("Benchmark Test Document".to_string()),
|
|
12
|
-
language: Some("en".to_string()),
|
|
13
|
-
created_at: Some("2025-01-01T00:00:00Z".to_string()),
|
|
14
|
-
subject: Some("Performance Testing".to_string()),
|
|
15
|
-
..Default::default()
|
|
16
|
-
};
|
|
17
|
-
|
|
18
|
-
let page_structure = PageStructure {
|
|
19
|
-
total_count: 100,
|
|
20
|
-
unit_type: PageUnitType::Page,
|
|
21
|
-
boundaries: None,
|
|
22
|
-
pages: None,
|
|
23
|
-
};
|
|
24
|
-
metadata.pages = Some(page_structure);
|
|
25
|
-
|
|
26
|
-
let content = format!(
|
|
27
|
-
"{}{}{}",
|
|
28
|
-
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. ",
|
|
29
|
-
"Sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. ".repeat(content_size / 120),
|
|
30
|
-
"Final sentence."
|
|
31
|
-
);
|
|
32
|
-
|
|
33
|
-
let chunks = if chunk_count > 0 {
|
|
34
|
-
let chunk_size = content.len() / chunk_count;
|
|
35
|
-
Some(
|
|
36
|
-
(0..chunk_count)
|
|
37
|
-
.map(|i| {
|
|
38
|
-
let start = i * chunk_size;
|
|
39
|
-
let end = if i == chunk_count - 1 {
|
|
40
|
-
content.len()
|
|
41
|
-
} else {
|
|
42
|
-
(i + 1) * chunk_size
|
|
43
|
-
};
|
|
44
|
-
Chunk {
|
|
45
|
-
content: content[start..end].to_string(),
|
|
46
|
-
embedding: None,
|
|
47
|
-
metadata: ChunkMetadata {
|
|
48
|
-
byte_start: start,
|
|
49
|
-
byte_end: end,
|
|
50
|
-
token_count: Some((end - start) / 4),
|
|
51
|
-
chunk_index: i,
|
|
52
|
-
total_chunks: chunk_count,
|
|
53
|
-
first_page: Some(1 + (i / 10)),
|
|
54
|
-
last_page: Some(1 + (i / 10)),
|
|
55
|
-
},
|
|
56
|
-
}
|
|
57
|
-
})
|
|
58
|
-
.collect(),
|
|
59
|
-
)
|
|
60
|
-
} else {
|
|
61
|
-
None
|
|
62
|
-
};
|
|
63
|
-
|
|
64
|
-
ExtractionResult {
|
|
65
|
-
content,
|
|
66
|
-
mime_type: "application/pdf".to_string(),
|
|
67
|
-
metadata,
|
|
68
|
-
tables: vec![],
|
|
69
|
-
detected_languages: Some(vec!["en".to_string(), "de".to_string()]),
|
|
70
|
-
chunks,
|
|
71
|
-
images: None,
|
|
72
|
-
pages: None,
|
|
73
|
-
}
|
|
74
|
-
}
|
|
75
|
-
|
|
76
|
-
/// Benchmark: Zero-copy result view creation
|
|
77
|
-
fn bench_zero_copy_view_creation(c: &mut Criterion) {
|
|
78
|
-
let mut group = c.benchmark_group("result_view_creation");
|
|
79
|
-
|
|
80
|
-
for size in [1_000, 10_000, 100_000, 1_000_000].iter() {
|
|
81
|
-
group.throughput(Throughput::Bytes(*size as u64));
|
|
82
|
-
group.bench_with_input(BenchmarkId::new("zero_copy", size), size, |b, &size| {
|
|
83
|
-
let result = create_test_result(size, 10);
|
|
84
|
-
let result_ptr = &result as *const ExtractionResult;
|
|
85
|
-
|
|
86
|
-
b.iter(|| {
|
|
87
|
-
let mut view: CExtractionResultView = unsafe { mem::zeroed() };
|
|
88
|
-
unsafe {
|
|
89
|
-
kreuzberg_get_result_view(hint::black_box(result_ptr), &mut view);
|
|
90
|
-
}
|
|
91
|
-
hint::black_box(view);
|
|
92
|
-
});
|
|
93
|
-
});
|
|
94
|
-
}
|
|
95
|
-
|
|
96
|
-
group.finish();
|
|
97
|
-
}
|
|
98
|
-
|
|
99
|
-
/// Benchmark: Copy-based approach (simulated via CString allocation)
|
|
100
|
-
fn bench_copy_based_approach(c: &mut Criterion) {
|
|
101
|
-
let mut group = c.benchmark_group("result_copy_based");
|
|
102
|
-
|
|
103
|
-
for size in [1_000, 10_000, 100_000, 1_000_000].iter() {
|
|
104
|
-
group.throughput(Throughput::Bytes(*size as u64));
|
|
105
|
-
group.bench_with_input(BenchmarkId::new("copy_with_cstring", size), size, |b, &size| {
|
|
106
|
-
let result = create_test_result(size, 10);
|
|
107
|
-
|
|
108
|
-
b.iter(|| {
|
|
109
|
-
let content_cstr = CString::new(result.content.as_str()).unwrap();
|
|
110
|
-
let mime_cstr = CString::new(result.mime_type.as_str()).unwrap();
|
|
111
|
-
let language_cstr = result
|
|
112
|
-
.metadata
|
|
113
|
-
.language
|
|
114
|
-
.as_ref()
|
|
115
|
-
.map(|s| CString::new(s.as_str()).unwrap());
|
|
116
|
-
let title_cstr = result
|
|
117
|
-
.metadata
|
|
118
|
-
.title
|
|
119
|
-
.as_ref()
|
|
120
|
-
.map(|s| CString::new(s.as_str()).unwrap());
|
|
121
|
-
|
|
122
|
-
hint::black_box(content_cstr);
|
|
123
|
-
hint::black_box(mime_cstr);
|
|
124
|
-
hint::black_box(language_cstr);
|
|
125
|
-
hint::black_box(title_cstr);
|
|
126
|
-
});
|
|
127
|
-
});
|
|
128
|
-
}
|
|
129
|
-
|
|
130
|
-
group.finish();
|
|
131
|
-
}
|
|
132
|
-
|
|
133
|
-
/// Benchmark: Field access from zero-copy view
|
|
134
|
-
fn bench_zero_copy_field_access(c: &mut Criterion) {
|
|
135
|
-
let mut group = c.benchmark_group("result_view_field_access");
|
|
136
|
-
|
|
137
|
-
let result = create_test_result(100_000, 50);
|
|
138
|
-
let result_ptr = &result as *const ExtractionResult;
|
|
139
|
-
let mut view: CExtractionResultView = unsafe { mem::zeroed() };
|
|
140
|
-
unsafe {
|
|
141
|
-
kreuzberg_get_result_view(result_ptr, &mut view);
|
|
142
|
-
}
|
|
143
|
-
|
|
144
|
-
group.bench_function("access_content_length", |b| {
|
|
145
|
-
b.iter(|| {
|
|
146
|
-
hint::black_box(view.content_len);
|
|
147
|
-
});
|
|
148
|
-
});
|
|
149
|
-
|
|
150
|
-
group.bench_function("access_all_counts", |b| {
|
|
151
|
-
b.iter(|| {
|
|
152
|
-
hint::black_box(view.table_count);
|
|
153
|
-
hint::black_box(view.chunk_count);
|
|
154
|
-
hint::black_box(view.detected_language_count);
|
|
155
|
-
hint::black_box(view.image_count);
|
|
156
|
-
hint::black_box(view.page_count);
|
|
157
|
-
});
|
|
158
|
-
});
|
|
159
|
-
|
|
160
|
-
group.bench_function("read_content_slice", |b| {
|
|
161
|
-
b.iter(|| {
|
|
162
|
-
let content_slice = unsafe { std::slice::from_raw_parts(view.content_ptr, view.content_len) };
|
|
163
|
-
hint::black_box(content_slice);
|
|
164
|
-
});
|
|
165
|
-
});
|
|
166
|
-
|
|
167
|
-
group.finish();
|
|
168
|
-
}
|
|
169
|
-
|
|
170
|
-
/// Benchmark: Multiple views from same result (zero overhead)
|
|
171
|
-
fn bench_multiple_views(c: &mut Criterion) {
|
|
172
|
-
let result = create_test_result(50_000, 20);
|
|
173
|
-
let result_ptr = &result as *const ExtractionResult;
|
|
174
|
-
|
|
175
|
-
c.bench_function("create_multiple_views", |b| {
|
|
176
|
-
b.iter(|| {
|
|
177
|
-
for _ in 0..10 {
|
|
178
|
-
let mut view: CExtractionResultView = unsafe { mem::zeroed() };
|
|
179
|
-
unsafe {
|
|
180
|
-
kreuzberg_get_result_view(hint::black_box(result_ptr), &mut view);
|
|
181
|
-
}
|
|
182
|
-
hint::black_box(view);
|
|
183
|
-
}
|
|
184
|
-
});
|
|
185
|
-
});
|
|
186
|
-
}
|
|
187
|
-
|
|
188
|
-
/// Benchmark: Zero-copy vs JSON serialization overhead
|
|
189
|
-
fn bench_vs_json_serialization(c: &mut Criterion) {
|
|
190
|
-
let mut group = c.benchmark_group("result_vs_json");
|
|
191
|
-
|
|
192
|
-
for size in [10_000, 100_000].iter() {
|
|
193
|
-
group.throughput(Throughput::Bytes(*size as u64));
|
|
194
|
-
|
|
195
|
-
let result = create_test_result(*size, 10);
|
|
196
|
-
let result_ptr = &result as *const ExtractionResult;
|
|
197
|
-
|
|
198
|
-
group.bench_with_input(BenchmarkId::new("zero_copy", size), size, |b, _| {
|
|
199
|
-
b.iter(|| {
|
|
200
|
-
let mut view: CExtractionResultView = unsafe { mem::zeroed() };
|
|
201
|
-
unsafe {
|
|
202
|
-
kreuzberg_get_result_view(hint::black_box(result_ptr), &mut view);
|
|
203
|
-
}
|
|
204
|
-
hint::black_box(view);
|
|
205
|
-
});
|
|
206
|
-
});
|
|
207
|
-
|
|
208
|
-
group.bench_with_input(BenchmarkId::new("json_serialize", size), size, |b, _| {
|
|
209
|
-
b.iter(|| {
|
|
210
|
-
let json = serde_json::to_string(&result.metadata).unwrap();
|
|
211
|
-
hint::black_box(json);
|
|
212
|
-
});
|
|
213
|
-
});
|
|
214
|
-
}
|
|
215
|
-
|
|
216
|
-
group.finish();
|
|
217
|
-
}
|
|
218
|
-
|
|
219
|
-
criterion_group!(
|
|
220
|
-
benches,
|
|
221
|
-
bench_zero_copy_view_creation,
|
|
222
|
-
bench_copy_based_approach,
|
|
223
|
-
bench_zero_copy_field_access,
|
|
224
|
-
bench_multiple_views,
|
|
225
|
-
bench_vs_json_serialization
|
|
226
|
-
);
|
|
227
|
-
criterion_main!(benches);
|
|
@@ -1,168 +0,0 @@
|
|
|
1
|
-
use std::env;
|
|
2
|
-
use std::fs;
|
|
3
|
-
use std::path::{Path, PathBuf};
|
|
4
|
-
|
|
5
|
-
fn main() {
|
|
6
|
-
if let Err(e) = run() {
|
|
7
|
-
eprintln!("Build script error: {}", e);
|
|
8
|
-
std::process::exit(1);
|
|
9
|
-
}
|
|
10
|
-
}
|
|
11
|
-
|
|
12
|
-
fn run() -> Result<(), String> {
|
|
13
|
-
let crate_dir = env::var("CARGO_MANIFEST_DIR").map_err(|_| "CARGO_MANIFEST_DIR not set".to_string())?;
|
|
14
|
-
|
|
15
|
-
let config =
|
|
16
|
-
cbindgen::Config::from_file("cbindgen.toml").map_err(|e| format!("Failed to load cbindgen config: {}", e))?;
|
|
17
|
-
|
|
18
|
-
cbindgen::generate_with_config(&crate_dir, config)
|
|
19
|
-
.map_err(|e| format!("Failed to generate C bindings: {}", e))?
|
|
20
|
-
.write_to_file("kreuzberg.h");
|
|
21
|
-
|
|
22
|
-
let pc_template = std::fs::read_to_string("kreuzberg-ffi.pc.in")
|
|
23
|
-
.map_err(|e| format!("Failed to read pkg-config template: {}", e))?;
|
|
24
|
-
|
|
25
|
-
let version = env::var("CARGO_PKG_VERSION").map_err(|_| "CARGO_PKG_VERSION not set".to_string())?;
|
|
26
|
-
|
|
27
|
-
let repo_root = Path::new(&crate_dir).parent().and_then(|p| p.parent()).ok_or_else(|| {
|
|
28
|
-
"CARGO_MANIFEST_DIR did not have expected depth (expected crates/kreuzberg-ffi/...)".to_string()
|
|
29
|
-
})?;
|
|
30
|
-
|
|
31
|
-
let dev_prefix = repo_root.to_string_lossy().replace('\\', "/");
|
|
32
|
-
|
|
33
|
-
let target_os = env::var("CARGO_CFG_TARGET_OS").unwrap_or_else(|_| "unknown".to_string());
|
|
34
|
-
let target_env = env::var("CARGO_CFG_TARGET_ENV").unwrap_or_else(|_| "gnu".to_string());
|
|
35
|
-
|
|
36
|
-
let libs_private = match target_os.as_str() {
|
|
37
|
-
"linux" => "-lpthread -ldl -lm",
|
|
38
|
-
"macos" => "-framework CoreFoundation -framework Security -lpthread",
|
|
39
|
-
"windows" => match target_env.as_str() {
|
|
40
|
-
"msvc" => "-lws2_32 -luserenv -lbcrypt",
|
|
41
|
-
"gnu" => "-lpthread -lws2_32 -luserenv -lbcrypt -static-libgcc -static-libstdc++",
|
|
42
|
-
_ => "-lws2_32 -luserenv -lbcrypt",
|
|
43
|
-
},
|
|
44
|
-
_ => "",
|
|
45
|
-
};
|
|
46
|
-
|
|
47
|
-
let out_dir = PathBuf::from(env::var("OUT_DIR").map_err(|_| "OUT_DIR not set".to_string())?);
|
|
48
|
-
let profile_dir = out_dir
|
|
49
|
-
.ancestors()
|
|
50
|
-
.nth(3)
|
|
51
|
-
.ok_or_else(|| "OUT_DIR did not have expected depth (expected target/{debug,release}/build/...)".to_string())?;
|
|
52
|
-
|
|
53
|
-
copy_pdfium_to_profile_dir(profile_dir)?;
|
|
54
|
-
|
|
55
|
-
let dev_libdir = profile_dir.to_string_lossy().replace('\\', "/");
|
|
56
|
-
let dev_includedir = format!("{}/crates/kreuzberg-ffi", dev_prefix);
|
|
57
|
-
let dev_pc = format!(
|
|
58
|
-
r#"prefix={}
|
|
59
|
-
exec_prefix=${{prefix}}
|
|
60
|
-
libdir={}
|
|
61
|
-
includedir={}
|
|
62
|
-
|
|
63
|
-
Name: kreuzberg-ffi
|
|
64
|
-
Description: C FFI bindings for Kreuzberg document intelligence library
|
|
65
|
-
Version: {}
|
|
66
|
-
URL: https://kreuzberg.dev
|
|
67
|
-
Libs: -L${{libdir}} -lkreuzberg_ffi
|
|
68
|
-
Libs.private: {}
|
|
69
|
-
Cflags: -I${{includedir}}
|
|
70
|
-
"#,
|
|
71
|
-
dev_prefix, dev_libdir, dev_includedir, version, libs_private
|
|
72
|
-
);
|
|
73
|
-
std::fs::write("kreuzberg-ffi.pc", dev_pc).map_err(|e| format!("Failed to write development pkg-config: {}", e))?;
|
|
74
|
-
|
|
75
|
-
let install_pc = pc_template
|
|
76
|
-
.replace("@PREFIX@", "/usr/local")
|
|
77
|
-
.replace("@VERSION@", &version)
|
|
78
|
-
.replace("@LIBS_PRIVATE@", libs_private);
|
|
79
|
-
std::fs::write("kreuzberg-ffi-install.pc", install_pc)
|
|
80
|
-
.map_err(|e| format!("Failed to write installation pkg-config: {}", e))?;
|
|
81
|
-
|
|
82
|
-
#[cfg(target_os = "macos")]
|
|
83
|
-
{
|
|
84
|
-
println!("cargo:rustc-link-arg=-rpath");
|
|
85
|
-
println!("cargo:rustc-link-arg=@loader_path");
|
|
86
|
-
|
|
87
|
-
println!("cargo:rustc-link-arg=-rpath");
|
|
88
|
-
println!("cargo:rustc-link-arg=@executable_path/../target/release");
|
|
89
|
-
}
|
|
90
|
-
|
|
91
|
-
println!("cargo:rerun-if-changed=cbindgen.toml");
|
|
92
|
-
println!("cargo:rerun-if-changed=src/lib.rs");
|
|
93
|
-
println!("cargo:rerun-if-changed=kreuzberg-ffi.pc.in");
|
|
94
|
-
|
|
95
|
-
Ok(())
|
|
96
|
-
}
|
|
97
|
-
|
|
98
|
-
/// Copy bundled PDFium library from kreuzberg build output to profile directory.
|
|
99
|
-
/// This enables Java/other language bindings to find libpdfium.dylib/so/dll at runtime.
|
|
100
|
-
fn copy_pdfium_to_profile_dir(profile_dir: &Path) -> Result<(), String> {
|
|
101
|
-
let build_dir = profile_dir.join("build");
|
|
102
|
-
|
|
103
|
-
if let Ok(entries) = fs::read_dir(&build_dir) {
|
|
104
|
-
for entry in entries.flatten() {
|
|
105
|
-
let path = entry.path();
|
|
106
|
-
if path.is_dir()
|
|
107
|
-
&& path
|
|
108
|
-
.file_name()
|
|
109
|
-
.is_some_and(|n| n.to_string_lossy().starts_with("kreuzberg-"))
|
|
110
|
-
{
|
|
111
|
-
let out_dir = path.join("out");
|
|
112
|
-
if out_dir.exists() {
|
|
113
|
-
if copy_pdfium_from_dir(&out_dir, profile_dir).is_err() {
|
|
114
|
-
continue;
|
|
115
|
-
} else {
|
|
116
|
-
return Ok(());
|
|
117
|
-
}
|
|
118
|
-
}
|
|
119
|
-
}
|
|
120
|
-
}
|
|
121
|
-
}
|
|
122
|
-
|
|
123
|
-
eprintln!("Warning: bundled PDFium library not found in build output. Some features may not work.");
|
|
124
|
-
eprintln!("If PDFium is needed, ensure the 'bundled-pdfium' feature is enabled.");
|
|
125
|
-
|
|
126
|
-
Ok(())
|
|
127
|
-
}
|
|
128
|
-
|
|
129
|
-
/// Copy PDFium library files from source directory to destination.
|
|
130
|
-
fn copy_pdfium_from_dir(src_dir: &Path, dest_dir: &Path) -> Result<(), String> {
|
|
131
|
-
let entries = fs::read_dir(src_dir).map_err(|e| format!("Failed to read {}: {}", src_dir.display(), e))?;
|
|
132
|
-
|
|
133
|
-
for entry in entries.flatten() {
|
|
134
|
-
let path = entry.path();
|
|
135
|
-
let file_name = path.file_name().ok_or("No file name")?;
|
|
136
|
-
let file_name_str = file_name.to_string_lossy();
|
|
137
|
-
|
|
138
|
-
if file_name_str.starts_with("libpdfium") || file_name_str.starts_with("pdfium") {
|
|
139
|
-
let dest_file = dest_dir.join(file_name);
|
|
140
|
-
|
|
141
|
-
if dest_file.exists() {
|
|
142
|
-
eprintln!(
|
|
143
|
-
"PDFium library already exists at {}, skipping copy",
|
|
144
|
-
dest_file.display()
|
|
145
|
-
);
|
|
146
|
-
return Ok(());
|
|
147
|
-
}
|
|
148
|
-
|
|
149
|
-
match fs::copy(&path, &dest_file) {
|
|
150
|
-
Ok(bytes_copied) => {
|
|
151
|
-
eprintln!(
|
|
152
|
-
"Copied {} ({} bytes) to {}",
|
|
153
|
-
path.display(),
|
|
154
|
-
bytes_copied,
|
|
155
|
-
dest_file.display()
|
|
156
|
-
);
|
|
157
|
-
return Ok(());
|
|
158
|
-
}
|
|
159
|
-
Err(e) => {
|
|
160
|
-
eprintln!("Failed to copy {} to {}: {}", path.display(), dest_file.display(), e);
|
|
161
|
-
return Err(format!("Failed to copy PDFium: {}", e));
|
|
162
|
-
}
|
|
163
|
-
}
|
|
164
|
-
}
|
|
165
|
-
}
|
|
166
|
-
|
|
167
|
-
Err(format!("PDFium library not found in {}", src_dir.display()))
|
|
168
|
-
}
|
|
@@ -1,37 +0,0 @@
|
|
|
1
|
-
language = "C"
|
|
2
|
-
include_guard = "KREUZBERG_FFI_H"
|
|
3
|
-
pragma_once = true
|
|
4
|
-
header = "/* Auto-generated C bindings for Kreuzberg */"
|
|
5
|
-
autogen_warning = "/* Warning, this file is autogenerated by cbindgen. Don't modify this manually. */"
|
|
6
|
-
documentation = true
|
|
7
|
-
line_length = 100
|
|
8
|
-
after_includes = """
|
|
9
|
-
/**
|
|
10
|
-
* Opaque type for extraction configuration.
|
|
11
|
-
* This is an opaque pointer type - callers should not access its internals.
|
|
12
|
-
*/
|
|
13
|
-
typedef struct ExtractionConfig ExtractionConfig;
|
|
14
|
-
|
|
15
|
-
/**
|
|
16
|
-
* Opaque type for extraction result.
|
|
17
|
-
* This is an opaque pointer type - callers should not access its internals.
|
|
18
|
-
* Use the kreuzberg_result_* accessor functions to extract data.
|
|
19
|
-
*/
|
|
20
|
-
typedef struct ExtractionResult ExtractionResult;
|
|
21
|
-
"""
|
|
22
|
-
|
|
23
|
-
[export.rename]
|
|
24
|
-
ExtractionResult = "ExtractionResult"
|
|
25
|
-
|
|
26
|
-
[export]
|
|
27
|
-
include = [
|
|
28
|
-
"kreuzberg_extract_file_sync",
|
|
29
|
-
"kreuzberg_free_string",
|
|
30
|
-
"kreuzberg_free_result",
|
|
31
|
-
"kreuzberg_last_error",
|
|
32
|
-
"kreuzberg_version",
|
|
33
|
-
]
|
|
34
|
-
|
|
35
|
-
[parse]
|
|
36
|
-
parse_deps = false
|
|
37
|
-
include = []
|
|
@@ -1,12 +0,0 @@
|
|
|
1
|
-
prefix=@PREFIX@
|
|
2
|
-
exec_prefix=${prefix}
|
|
3
|
-
libdir=${exec_prefix}/lib
|
|
4
|
-
includedir=${prefix}/include
|
|
5
|
-
|
|
6
|
-
Name: kreuzberg-ffi
|
|
7
|
-
Description: C FFI bindings for Kreuzberg document intelligence library
|
|
8
|
-
Version: @VERSION@
|
|
9
|
-
URL: https://kreuzberg.dev
|
|
10
|
-
Libs: -L${libdir} -lkreuzberg_ffi
|
|
11
|
-
Libs.private: @LIBS_PRIVATE@
|
|
12
|
-
Cflags: -I${includedir}
|