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
|
@@ -4,48 +4,8 @@
|
|
|
4
4
|
//! sources (TOML, YAML, JSON) and discovering configuration files in the project hierarchy.
|
|
5
5
|
|
|
6
6
|
use crate::{KreuzbergError, Result};
|
|
7
|
-
use dashmap::DashMap;
|
|
8
7
|
use serde::{Deserialize, Serialize};
|
|
9
|
-
use std::
|
|
10
|
-
use std::path::{Path, PathBuf};
|
|
11
|
-
use std::sync::{Arc, LazyLock};
|
|
12
|
-
use std::time::SystemTime;
|
|
13
|
-
|
|
14
|
-
static CONFIG_CACHE: LazyLock<DashMap<PathBuf, (SystemTime, Arc<ExtractionConfig>)>> = LazyLock::new(DashMap::new);
|
|
15
|
-
|
|
16
|
-
/// Page extraction and tracking configuration.
|
|
17
|
-
///
|
|
18
|
-
/// Controls how pages are extracted, tracked, and represented in the extraction results.
|
|
19
|
-
/// When `None`, page tracking is disabled.
|
|
20
|
-
///
|
|
21
|
-
/// Page range tracking in chunk metadata (first_page/last_page) is automatically enabled
|
|
22
|
-
/// when page boundaries are available and chunking is configured.
|
|
23
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
24
|
-
#[serde(default)]
|
|
25
|
-
pub struct PageConfig {
|
|
26
|
-
/// Extract pages as separate array (ExtractionResult.pages)
|
|
27
|
-
#[serde(default)]
|
|
28
|
-
pub extract_pages: bool,
|
|
29
|
-
|
|
30
|
-
/// Insert page markers in main content string
|
|
31
|
-
#[serde(default)]
|
|
32
|
-
pub insert_page_markers: bool,
|
|
33
|
-
|
|
34
|
-
/// Page marker format (use {page_num} placeholder)
|
|
35
|
-
/// Default: "\n\n<!-- PAGE {page_num} -->\n\n"
|
|
36
|
-
#[serde(default = "default_page_marker_format")]
|
|
37
|
-
pub marker_format: String,
|
|
38
|
-
}
|
|
39
|
-
|
|
40
|
-
impl Default for PageConfig {
|
|
41
|
-
fn default() -> Self {
|
|
42
|
-
Self {
|
|
43
|
-
extract_pages: false,
|
|
44
|
-
insert_page_markers: false,
|
|
45
|
-
marker_format: "\n\n<!-- PAGE {page_num} -->\n\n".to_string(),
|
|
46
|
-
}
|
|
47
|
-
}
|
|
48
|
-
}
|
|
8
|
+
use std::path::Path;
|
|
49
9
|
|
|
50
10
|
/// Main extraction configuration.
|
|
51
11
|
///
|
|
@@ -90,7 +50,6 @@ pub struct ExtractionConfig {
|
|
|
90
50
|
pub images: Option<ImageExtractionConfig>,
|
|
91
51
|
|
|
92
52
|
/// PDF-specific options (None = use defaults)
|
|
93
|
-
#[cfg(feature = "pdf")]
|
|
94
53
|
#[serde(default)]
|
|
95
54
|
pub pdf_options: Option<PdfConfig>,
|
|
96
55
|
|
|
@@ -102,10 +61,6 @@ pub struct ExtractionConfig {
|
|
|
102
61
|
#[serde(default)]
|
|
103
62
|
pub language_detection: Option<LanguageDetectionConfig>,
|
|
104
63
|
|
|
105
|
-
/// Page extraction configuration (None = no page tracking)
|
|
106
|
-
#[serde(default)]
|
|
107
|
-
pub pages: Option<PageConfig>,
|
|
108
|
-
|
|
109
64
|
/// Keyword extraction configuration (None = no keyword extraction)
|
|
110
65
|
#[cfg(any(feature = "keywords-yake", feature = "keywords-rake"))]
|
|
111
66
|
#[serde(default)]
|
|
@@ -145,29 +100,6 @@ pub struct PostProcessorConfig {
|
|
|
145
100
|
/// Blacklist of processor names to skip (None = none disabled)
|
|
146
101
|
#[serde(default)]
|
|
147
102
|
pub disabled_processors: Option<Vec<String>>,
|
|
148
|
-
|
|
149
|
-
/// Pre-computed HashSet for O(1) enabled processor lookup
|
|
150
|
-
#[serde(skip)]
|
|
151
|
-
pub enabled_set: Option<HashSet<String>>,
|
|
152
|
-
|
|
153
|
-
/// Pre-computed HashSet for O(1) disabled processor lookup
|
|
154
|
-
#[serde(skip)]
|
|
155
|
-
pub disabled_set: Option<HashSet<String>>,
|
|
156
|
-
}
|
|
157
|
-
|
|
158
|
-
impl PostProcessorConfig {
|
|
159
|
-
/// Pre-compute HashSets for O(1) processor name lookups.
|
|
160
|
-
///
|
|
161
|
-
/// This method converts the enabled/disabled processor Vec to HashSet
|
|
162
|
-
/// for constant-time lookups in the pipeline.
|
|
163
|
-
pub fn build_lookup_sets(&mut self) {
|
|
164
|
-
if let Some(ref enabled) = self.enabled_processors {
|
|
165
|
-
self.enabled_set = Some(enabled.iter().cloned().collect());
|
|
166
|
-
}
|
|
167
|
-
if let Some(ref disabled) = self.disabled_processors {
|
|
168
|
-
self.disabled_set = Some(disabled.iter().cloned().collect());
|
|
169
|
-
}
|
|
170
|
-
}
|
|
171
103
|
}
|
|
172
104
|
|
|
173
105
|
/// OCR configuration.
|
|
@@ -186,16 +118,6 @@ pub struct OcrConfig {
|
|
|
186
118
|
pub tesseract_config: Option<crate::types::TesseractConfig>,
|
|
187
119
|
}
|
|
188
120
|
|
|
189
|
-
impl Default for OcrConfig {
|
|
190
|
-
fn default() -> Self {
|
|
191
|
-
Self {
|
|
192
|
-
backend: default_tesseract_backend(),
|
|
193
|
-
language: default_eng(),
|
|
194
|
-
tesseract_config: None,
|
|
195
|
-
}
|
|
196
|
-
}
|
|
197
|
-
}
|
|
198
|
-
|
|
199
121
|
/// Chunking configuration.
|
|
200
122
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
201
123
|
pub struct ChunkingConfig {
|
|
@@ -303,7 +225,6 @@ pub struct ImageExtractionConfig {
|
|
|
303
225
|
}
|
|
304
226
|
|
|
305
227
|
/// PDF-specific configuration.
|
|
306
|
-
#[cfg(feature = "pdf")]
|
|
307
228
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
308
229
|
pub struct PdfConfig {
|
|
309
230
|
/// Extract images from PDF
|
|
@@ -317,60 +238,6 @@ pub struct PdfConfig {
|
|
|
317
238
|
/// Extract PDF metadata
|
|
318
239
|
#[serde(default = "default_true")]
|
|
319
240
|
pub extract_metadata: bool,
|
|
320
|
-
|
|
321
|
-
/// Hierarchy extraction configuration (None = hierarchy extraction disabled)
|
|
322
|
-
#[serde(default)]
|
|
323
|
-
pub hierarchy: Option<HierarchyConfig>,
|
|
324
|
-
}
|
|
325
|
-
|
|
326
|
-
/// Hierarchy extraction configuration for PDF text structure analysis.
|
|
327
|
-
///
|
|
328
|
-
/// Enables extraction of document hierarchy levels (H1-H6) based on font size
|
|
329
|
-
/// clustering and semantic analysis. When enabled, hierarchical blocks are
|
|
330
|
-
/// included in page content.
|
|
331
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
332
|
-
pub struct HierarchyConfig {
|
|
333
|
-
/// Enable hierarchy extraction
|
|
334
|
-
#[serde(default = "default_true")]
|
|
335
|
-
pub enabled: bool,
|
|
336
|
-
|
|
337
|
-
/// Number of font size clusters to use for hierarchy levels (1-7)
|
|
338
|
-
///
|
|
339
|
-
/// Default: 6, which provides H1-H6 heading levels with body text.
|
|
340
|
-
/// Larger values create more fine-grained hierarchy levels.
|
|
341
|
-
#[serde(default = "default_k_clusters")]
|
|
342
|
-
pub k_clusters: usize,
|
|
343
|
-
|
|
344
|
-
/// Include bounding box information in hierarchy blocks
|
|
345
|
-
#[serde(default = "default_true")]
|
|
346
|
-
pub include_bbox: bool,
|
|
347
|
-
|
|
348
|
-
/// OCR coverage threshold for smart OCR triggering (0.0-1.0)
|
|
349
|
-
///
|
|
350
|
-
/// Determines when OCR should be triggered based on text block coverage.
|
|
351
|
-
/// OCR is triggered when text blocks cover less than this fraction of the page.
|
|
352
|
-
/// Default: 0.5 (trigger OCR if less than 50% of page has text)
|
|
353
|
-
#[serde(default = "default_ocr_coverage_threshold")]
|
|
354
|
-
pub ocr_coverage_threshold: Option<f32>,
|
|
355
|
-
}
|
|
356
|
-
|
|
357
|
-
impl Default for HierarchyConfig {
|
|
358
|
-
fn default() -> Self {
|
|
359
|
-
Self {
|
|
360
|
-
enabled: true,
|
|
361
|
-
k_clusters: 6,
|
|
362
|
-
include_bbox: true,
|
|
363
|
-
ocr_coverage_threshold: None,
|
|
364
|
-
}
|
|
365
|
-
}
|
|
366
|
-
}
|
|
367
|
-
|
|
368
|
-
fn default_k_clusters() -> usize {
|
|
369
|
-
6
|
|
370
|
-
}
|
|
371
|
-
|
|
372
|
-
fn default_ocr_coverage_threshold() -> Option<f32> {
|
|
373
|
-
None
|
|
374
241
|
}
|
|
375
242
|
|
|
376
243
|
/// Token reduction configuration.
|
|
@@ -410,9 +277,6 @@ fn default_eng() -> String {
|
|
|
410
277
|
fn default_tesseract_backend() -> String {
|
|
411
278
|
"tesseract".to_string()
|
|
412
279
|
}
|
|
413
|
-
fn default_page_marker_format() -> String {
|
|
414
|
-
"\n\n<!-- PAGE {page_num} -->\n\n".to_string()
|
|
415
|
-
}
|
|
416
280
|
fn default_chunk_size() -> usize {
|
|
417
281
|
1000
|
|
418
282
|
}
|
|
@@ -453,11 +317,9 @@ impl Default for ExtractionConfig {
|
|
|
453
317
|
force_ocr: false,
|
|
454
318
|
chunking: None,
|
|
455
319
|
images: None,
|
|
456
|
-
#[cfg(feature = "pdf")]
|
|
457
320
|
pdf_options: None,
|
|
458
321
|
token_reduction: None,
|
|
459
322
|
language_detection: None,
|
|
460
|
-
pages: None,
|
|
461
323
|
#[cfg(any(feature = "keywords-yake", feature = "keywords-rake"))]
|
|
462
324
|
keywords: None,
|
|
463
325
|
postprocessor: None,
|
|
@@ -474,190 +336,11 @@ impl Default for PostProcessorConfig {
|
|
|
474
336
|
enabled: true,
|
|
475
337
|
enabled_processors: None,
|
|
476
338
|
disabled_processors: None,
|
|
477
|
-
enabled_set: None,
|
|
478
|
-
disabled_set: None,
|
|
479
339
|
}
|
|
480
340
|
}
|
|
481
341
|
}
|
|
482
342
|
|
|
483
343
|
impl ExtractionConfig {
|
|
484
|
-
/// Check if image processing is needed by examining OCR and image extraction settings.
|
|
485
|
-
///
|
|
486
|
-
/// Returns `true` if either OCR is enabled or image extraction is configured,
|
|
487
|
-
/// indicating that image decompression and processing should occur.
|
|
488
|
-
/// Returns `false` if both are disabled, allowing optimization to skip unnecessary
|
|
489
|
-
/// image decompression for text-only extraction workflows.
|
|
490
|
-
///
|
|
491
|
-
/// # Optimization Impact
|
|
492
|
-
/// For text-only extractions (no OCR, no image extraction), skipping image
|
|
493
|
-
/// decompression can improve CPU utilization by 5-10% by avoiding wasteful
|
|
494
|
-
/// image I/O and processing when results won't be used.
|
|
495
|
-
pub fn needs_image_processing(&self) -> bool {
|
|
496
|
-
let ocr_enabled = self.ocr.is_some();
|
|
497
|
-
|
|
498
|
-
let image_extraction_enabled = self.images.as_ref().map(|i| i.extract_images).unwrap_or(false);
|
|
499
|
-
|
|
500
|
-
ocr_enabled || image_extraction_enabled
|
|
501
|
-
}
|
|
502
|
-
|
|
503
|
-
/// Apply environment variable overrides to configuration.
|
|
504
|
-
///
|
|
505
|
-
/// Environment variables have the highest precedence and will override any values
|
|
506
|
-
/// loaded from configuration files. This method supports the following environment variables:
|
|
507
|
-
///
|
|
508
|
-
/// - `KREUZBERG_OCR_LANGUAGE`: OCR language (ISO 639-1 or 639-3 code, e.g., "eng", "fra", "deu")
|
|
509
|
-
/// - `KREUZBERG_OCR_BACKEND`: OCR backend ("tesseract", "easyocr", or "paddleocr")
|
|
510
|
-
/// - `KREUZBERG_CHUNKING_MAX_CHARS`: Maximum characters per chunk (positive integer)
|
|
511
|
-
/// - `KREUZBERG_CHUNKING_MAX_OVERLAP`: Maximum overlap between chunks (non-negative integer)
|
|
512
|
-
/// - `KREUZBERG_CACHE_ENABLED`: Cache enabled flag ("true" or "false")
|
|
513
|
-
/// - `KREUZBERG_TOKEN_REDUCTION_MODE`: Token reduction mode ("off", "light", "moderate", "aggressive", or "maximum")
|
|
514
|
-
///
|
|
515
|
-
/// # Behavior
|
|
516
|
-
///
|
|
517
|
-
/// - If an environment variable is set and valid, it overrides the current configuration value
|
|
518
|
-
/// - If a required parent config is `None` (e.g., `self.ocr` is None), it's created with defaults before applying the override
|
|
519
|
-
/// - Invalid values return a `KreuzbergError::Validation` with helpful error messages
|
|
520
|
-
/// - Missing or unset environment variables are silently ignored
|
|
521
|
-
///
|
|
522
|
-
/// # Example
|
|
523
|
-
///
|
|
524
|
-
/// ```rust
|
|
525
|
-
/// # use kreuzberg::core::config::ExtractionConfig;
|
|
526
|
-
/// # fn example() -> kreuzberg::Result<()> {
|
|
527
|
-
/// let mut config = ExtractionConfig::from_file("config.toml")?;
|
|
528
|
-
/// // Set KREUZBERG_OCR_LANGUAGE=fra before calling
|
|
529
|
-
/// config.apply_env_overrides()?; // OCR language is now "fra"
|
|
530
|
-
/// # Ok(())
|
|
531
|
-
/// # }
|
|
532
|
-
/// ```
|
|
533
|
-
///
|
|
534
|
-
/// # Errors
|
|
535
|
-
///
|
|
536
|
-
/// Returns `KreuzbergError::Validation` if:
|
|
537
|
-
/// - An environment variable contains an invalid value
|
|
538
|
-
/// - A number cannot be parsed as the expected type
|
|
539
|
-
/// - A boolean is not "true" or "false"
|
|
540
|
-
pub fn apply_env_overrides(&mut self) -> Result<()> {
|
|
541
|
-
use crate::core::config_validation::{
|
|
542
|
-
validate_chunking_params, validate_language_code, validate_ocr_backend, validate_token_reduction_level,
|
|
543
|
-
};
|
|
544
|
-
|
|
545
|
-
// KREUZBERG_OCR_LANGUAGE override
|
|
546
|
-
if let Ok(lang) = std::env::var("KREUZBERG_OCR_LANGUAGE") {
|
|
547
|
-
validate_language_code(&lang)?;
|
|
548
|
-
if self.ocr.is_none() {
|
|
549
|
-
self.ocr = Some(OcrConfig::default());
|
|
550
|
-
}
|
|
551
|
-
if let Some(ref mut ocr) = self.ocr {
|
|
552
|
-
ocr.language = lang;
|
|
553
|
-
}
|
|
554
|
-
}
|
|
555
|
-
|
|
556
|
-
// KREUZBERG_OCR_BACKEND override
|
|
557
|
-
if let Ok(backend) = std::env::var("KREUZBERG_OCR_BACKEND") {
|
|
558
|
-
validate_ocr_backend(&backend)?;
|
|
559
|
-
if self.ocr.is_none() {
|
|
560
|
-
self.ocr = Some(OcrConfig::default());
|
|
561
|
-
}
|
|
562
|
-
if let Some(ref mut ocr) = self.ocr {
|
|
563
|
-
ocr.backend = backend;
|
|
564
|
-
}
|
|
565
|
-
}
|
|
566
|
-
|
|
567
|
-
// KREUZBERG_CHUNKING_MAX_CHARS override
|
|
568
|
-
if let Ok(max_chars_str) = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS") {
|
|
569
|
-
let max_chars: usize = max_chars_str.parse().map_err(|_| KreuzbergError::Validation {
|
|
570
|
-
message: format!(
|
|
571
|
-
"Invalid value for KREUZBERG_CHUNKING_MAX_CHARS: '{}'. Must be a positive integer.",
|
|
572
|
-
max_chars_str
|
|
573
|
-
),
|
|
574
|
-
source: None,
|
|
575
|
-
})?;
|
|
576
|
-
|
|
577
|
-
if max_chars == 0 {
|
|
578
|
-
return Err(KreuzbergError::Validation {
|
|
579
|
-
message: "KREUZBERG_CHUNKING_MAX_CHARS must be greater than 0".to_string(),
|
|
580
|
-
source: None,
|
|
581
|
-
});
|
|
582
|
-
}
|
|
583
|
-
|
|
584
|
-
if self.chunking.is_none() {
|
|
585
|
-
self.chunking = Some(ChunkingConfig {
|
|
586
|
-
max_chars: default_chunk_size(),
|
|
587
|
-
max_overlap: default_chunk_overlap(),
|
|
588
|
-
embedding: None,
|
|
589
|
-
preset: None,
|
|
590
|
-
});
|
|
591
|
-
}
|
|
592
|
-
|
|
593
|
-
if let Some(ref mut chunking) = self.chunking {
|
|
594
|
-
// Validate against current overlap before updating
|
|
595
|
-
validate_chunking_params(max_chars, chunking.max_overlap)?;
|
|
596
|
-
chunking.max_chars = max_chars;
|
|
597
|
-
}
|
|
598
|
-
}
|
|
599
|
-
|
|
600
|
-
// KREUZBERG_CHUNKING_MAX_OVERLAP override
|
|
601
|
-
if let Ok(max_overlap_str) = std::env::var("KREUZBERG_CHUNKING_MAX_OVERLAP") {
|
|
602
|
-
let max_overlap: usize = max_overlap_str.parse().map_err(|_| KreuzbergError::Validation {
|
|
603
|
-
message: format!(
|
|
604
|
-
"Invalid value for KREUZBERG_CHUNKING_MAX_OVERLAP: '{}'. Must be a non-negative integer.",
|
|
605
|
-
max_overlap_str
|
|
606
|
-
),
|
|
607
|
-
source: None,
|
|
608
|
-
})?;
|
|
609
|
-
|
|
610
|
-
if self.chunking.is_none() {
|
|
611
|
-
self.chunking = Some(ChunkingConfig {
|
|
612
|
-
max_chars: default_chunk_size(),
|
|
613
|
-
max_overlap: default_chunk_overlap(),
|
|
614
|
-
embedding: None,
|
|
615
|
-
preset: None,
|
|
616
|
-
});
|
|
617
|
-
}
|
|
618
|
-
|
|
619
|
-
if let Some(ref mut chunking) = self.chunking {
|
|
620
|
-
// Validate against current max_chars before updating
|
|
621
|
-
validate_chunking_params(chunking.max_chars, max_overlap)?;
|
|
622
|
-
chunking.max_overlap = max_overlap;
|
|
623
|
-
}
|
|
624
|
-
}
|
|
625
|
-
|
|
626
|
-
// KREUZBERG_CACHE_ENABLED override
|
|
627
|
-
if let Ok(cache_str) = std::env::var("KREUZBERG_CACHE_ENABLED") {
|
|
628
|
-
let cache_enabled = match cache_str.to_lowercase().as_str() {
|
|
629
|
-
"true" => true,
|
|
630
|
-
"false" => false,
|
|
631
|
-
_ => {
|
|
632
|
-
return Err(KreuzbergError::Validation {
|
|
633
|
-
message: format!(
|
|
634
|
-
"Invalid value for KREUZBERG_CACHE_ENABLED: '{}'. Must be 'true' or 'false'.",
|
|
635
|
-
cache_str
|
|
636
|
-
),
|
|
637
|
-
source: None,
|
|
638
|
-
});
|
|
639
|
-
}
|
|
640
|
-
};
|
|
641
|
-
self.use_cache = cache_enabled;
|
|
642
|
-
}
|
|
643
|
-
|
|
644
|
-
// KREUZBERG_TOKEN_REDUCTION_MODE override
|
|
645
|
-
if let Ok(mode) = std::env::var("KREUZBERG_TOKEN_REDUCTION_MODE") {
|
|
646
|
-
validate_token_reduction_level(&mode)?;
|
|
647
|
-
if self.token_reduction.is_none() {
|
|
648
|
-
self.token_reduction = Some(TokenReductionConfig {
|
|
649
|
-
mode: default_reduction_mode(),
|
|
650
|
-
preserve_important_words: default_true(),
|
|
651
|
-
});
|
|
652
|
-
}
|
|
653
|
-
if let Some(ref mut token_reduction) = self.token_reduction {
|
|
654
|
-
token_reduction.mode = mode;
|
|
655
|
-
}
|
|
656
|
-
}
|
|
657
|
-
|
|
658
|
-
Ok(())
|
|
659
|
-
}
|
|
660
|
-
|
|
661
344
|
/// Load configuration from a TOML file.
|
|
662
345
|
///
|
|
663
346
|
/// # Arguments
|
|
@@ -668,86 +351,32 @@ impl ExtractionConfig {
|
|
|
668
351
|
///
|
|
669
352
|
/// Returns `KreuzbergError::Validation` if file doesn't exist or is invalid TOML.
|
|
670
353
|
pub fn from_toml_file(path: impl AsRef<Path>) -> Result<Self> {
|
|
671
|
-
let
|
|
672
|
-
|
|
673
|
-
let metadata = std::fs::metadata(path)
|
|
674
|
-
.map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
|
|
675
|
-
let mtime = metadata.modified().map_err(|e| {
|
|
676
|
-
KreuzbergError::validation(format!("Failed to get modification time for {}: {}", path.display(), e))
|
|
354
|
+
let content = std::fs::read_to_string(path.as_ref()).map_err(|e| {
|
|
355
|
+
KreuzbergError::validation(format!("Failed to read config file {}: {}", path.as_ref().display(), e))
|
|
677
356
|
})?;
|
|
678
357
|
|
|
679
|
-
|
|
680
|
-
|
|
681
|
-
{
|
|
682
|
-
return Ok((*entry.1).clone());
|
|
683
|
-
}
|
|
684
|
-
|
|
685
|
-
let content = std::fs::read_to_string(path)
|
|
686
|
-
.map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
|
|
687
|
-
|
|
688
|
-
let config: Self = toml::from_str(&content)
|
|
689
|
-
.map_err(|e| KreuzbergError::validation(format!("Invalid TOML in {}: {}", path.display(), e)))?;
|
|
690
|
-
|
|
691
|
-
let config_arc = Arc::new(config.clone());
|
|
692
|
-
CONFIG_CACHE.insert(path.to_path_buf(), (mtime, config_arc));
|
|
693
|
-
|
|
694
|
-
Ok(config)
|
|
358
|
+
toml::from_str(&content)
|
|
359
|
+
.map_err(|e| KreuzbergError::validation(format!("Invalid TOML in {}: {}", path.as_ref().display(), e)))
|
|
695
360
|
}
|
|
696
361
|
|
|
697
362
|
/// Load configuration from a YAML file.
|
|
698
363
|
pub fn from_yaml_file(path: impl AsRef<Path>) -> Result<Self> {
|
|
699
|
-
let
|
|
700
|
-
|
|
701
|
-
let metadata = std::fs::metadata(path)
|
|
702
|
-
.map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
|
|
703
|
-
let mtime = metadata.modified().map_err(|e| {
|
|
704
|
-
KreuzbergError::validation(format!("Failed to get modification time for {}: {}", path.display(), e))
|
|
364
|
+
let content = std::fs::read_to_string(path.as_ref()).map_err(|e| {
|
|
365
|
+
KreuzbergError::validation(format!("Failed to read config file {}: {}", path.as_ref().display(), e))
|
|
705
366
|
})?;
|
|
706
367
|
|
|
707
|
-
|
|
708
|
-
|
|
709
|
-
{
|
|
710
|
-
return Ok((*entry.1).clone());
|
|
711
|
-
}
|
|
712
|
-
|
|
713
|
-
let content = std::fs::read_to_string(path)
|
|
714
|
-
.map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
|
|
715
|
-
|
|
716
|
-
let config: Self = serde_yaml_ng::from_str(&content)
|
|
717
|
-
.map_err(|e| KreuzbergError::validation(format!("Invalid YAML in {}: {}", path.display(), e)))?;
|
|
718
|
-
|
|
719
|
-
let config_arc = Arc::new(config.clone());
|
|
720
|
-
CONFIG_CACHE.insert(path.to_path_buf(), (mtime, config_arc));
|
|
721
|
-
|
|
722
|
-
Ok(config)
|
|
368
|
+
serde_yaml_ng::from_str(&content)
|
|
369
|
+
.map_err(|e| KreuzbergError::validation(format!("Invalid YAML in {}: {}", path.as_ref().display(), e)))
|
|
723
370
|
}
|
|
724
371
|
|
|
725
372
|
/// Load configuration from a JSON file.
|
|
726
373
|
pub fn from_json_file(path: impl AsRef<Path>) -> Result<Self> {
|
|
727
|
-
let
|
|
728
|
-
|
|
729
|
-
let metadata = std::fs::metadata(path)
|
|
730
|
-
.map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
|
|
731
|
-
let mtime = metadata.modified().map_err(|e| {
|
|
732
|
-
KreuzbergError::validation(format!("Failed to get modification time for {}: {}", path.display(), e))
|
|
374
|
+
let content = std::fs::read_to_string(path.as_ref()).map_err(|e| {
|
|
375
|
+
KreuzbergError::validation(format!("Failed to read config file {}: {}", path.as_ref().display(), e))
|
|
733
376
|
})?;
|
|
734
377
|
|
|
735
|
-
|
|
736
|
-
|
|
737
|
-
{
|
|
738
|
-
return Ok((*entry.1).clone());
|
|
739
|
-
}
|
|
740
|
-
|
|
741
|
-
let content = std::fs::read_to_string(path)
|
|
742
|
-
.map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
|
|
743
|
-
|
|
744
|
-
let config: Self = serde_json::from_str(&content)
|
|
745
|
-
.map_err(|e| KreuzbergError::validation(format!("Invalid JSON in {}: {}", path.display(), e)))?;
|
|
746
|
-
|
|
747
|
-
let config_arc = Arc::new(config.clone());
|
|
748
|
-
CONFIG_CACHE.insert(path.to_path_buf(), (mtime, config_arc));
|
|
749
|
-
|
|
750
|
-
Ok(config)
|
|
378
|
+
serde_json::from_str(&content)
|
|
379
|
+
.map_err(|e| KreuzbergError::validation(format!("Invalid JSON in {}: {}", path.as_ref().display(), e)))
|
|
751
380
|
}
|
|
752
381
|
|
|
753
382
|
/// Load configuration from a file, auto-detecting format by extension.
|
|
@@ -781,19 +410,6 @@ impl ExtractionConfig {
|
|
|
781
410
|
/// ```
|
|
782
411
|
pub fn from_file(path: impl AsRef<Path>) -> Result<Self> {
|
|
783
412
|
let path = path.as_ref();
|
|
784
|
-
|
|
785
|
-
let metadata = std::fs::metadata(path)
|
|
786
|
-
.map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
|
|
787
|
-
let mtime = metadata.modified().map_err(|e| {
|
|
788
|
-
KreuzbergError::validation(format!("Failed to get modification time for {}: {}", path.display(), e))
|
|
789
|
-
})?;
|
|
790
|
-
|
|
791
|
-
if let Some(entry) = CONFIG_CACHE.get(path)
|
|
792
|
-
&& entry.0 == mtime
|
|
793
|
-
{
|
|
794
|
-
return Ok((*entry.1).clone());
|
|
795
|
-
}
|
|
796
|
-
|
|
797
413
|
let extension = path.extension().and_then(|ext| ext.to_str()).ok_or_else(|| {
|
|
798
414
|
KreuzbergError::validation(format!(
|
|
799
415
|
"Cannot determine file format: no extension found in {}",
|
|
@@ -801,22 +417,15 @@ impl ExtractionConfig {
|
|
|
801
417
|
))
|
|
802
418
|
})?;
|
|
803
419
|
|
|
804
|
-
|
|
805
|
-
"toml" => Self::from_toml_file(path)
|
|
806
|
-
"yaml" | "yml" => Self::from_yaml_file(path)
|
|
807
|
-
"json" => Self::from_json_file(path)
|
|
808
|
-
_ =>
|
|
809
|
-
|
|
810
|
-
|
|
811
|
-
|
|
812
|
-
|
|
813
|
-
}
|
|
814
|
-
};
|
|
815
|
-
|
|
816
|
-
let config_arc = Arc::new(config.clone());
|
|
817
|
-
CONFIG_CACHE.insert(path.to_path_buf(), (mtime, config_arc));
|
|
818
|
-
|
|
819
|
-
Ok(config)
|
|
420
|
+
match extension.to_lowercase().as_str() {
|
|
421
|
+
"toml" => Self::from_toml_file(path),
|
|
422
|
+
"yaml" | "yml" => Self::from_yaml_file(path),
|
|
423
|
+
"json" => Self::from_json_file(path),
|
|
424
|
+
_ => Err(KreuzbergError::validation(format!(
|
|
425
|
+
"Unsupported config file format: .{}. Supported formats: .toml, .yaml, .json",
|
|
426
|
+
extension
|
|
427
|
+
))),
|
|
428
|
+
}
|
|
820
429
|
}
|
|
821
430
|
|
|
822
431
|
/// Discover configuration file in parent directories.
|
|
@@ -848,36 +457,11 @@ impl ExtractionConfig {
|
|
|
848
457
|
}
|
|
849
458
|
|
|
850
459
|
#[cfg(test)]
|
|
851
|
-
#[allow(unsafe_code)]
|
|
852
460
|
mod tests {
|
|
853
461
|
use super::*;
|
|
854
462
|
use std::fs;
|
|
855
463
|
use tempfile::tempdir;
|
|
856
464
|
|
|
857
|
-
fn set_env(key: &str, value: &str) {
|
|
858
|
-
unsafe {
|
|
859
|
-
std::env::set_var(key, value);
|
|
860
|
-
}
|
|
861
|
-
}
|
|
862
|
-
|
|
863
|
-
#[allow(unsafe_code)]
|
|
864
|
-
fn remove_env(key: &str) {
|
|
865
|
-
unsafe {
|
|
866
|
-
std::env::remove_var(key);
|
|
867
|
-
}
|
|
868
|
-
}
|
|
869
|
-
|
|
870
|
-
#[allow(unsafe_code)]
|
|
871
|
-
fn restore_env(key: &str, value: Option<String>) {
|
|
872
|
-
unsafe {
|
|
873
|
-
if let Some(v) = value {
|
|
874
|
-
std::env::set_var(key, v);
|
|
875
|
-
} else {
|
|
876
|
-
std::env::remove_var(key);
|
|
877
|
-
}
|
|
878
|
-
}
|
|
879
|
-
}
|
|
880
|
-
|
|
881
465
|
#[test]
|
|
882
466
|
fn test_default_config() {
|
|
883
467
|
let config = ExtractionConfig::default();
|
|
@@ -1063,7 +647,6 @@ max_dpi = 600
|
|
|
1063
647
|
}
|
|
1064
648
|
|
|
1065
649
|
#[test]
|
|
1066
|
-
#[cfg(feature = "pdf")]
|
|
1067
650
|
fn test_config_with_pdf_options() {
|
|
1068
651
|
let dir = tempdir().unwrap();
|
|
1069
652
|
let config_path = dir.path().join("kreuzberg.toml");
|
|
@@ -1187,10 +770,9 @@ enabled = true
|
|
|
1187
770
|
assert!(config.ocr.is_some());
|
|
1188
771
|
assert!(config.chunking.is_some());
|
|
1189
772
|
assert!(config.images.is_some());
|
|
773
|
+
assert!(config.pdf_options.is_some());
|
|
1190
774
|
assert!(config.token_reduction.is_some());
|
|
1191
775
|
assert!(config.language_detection.is_some());
|
|
1192
|
-
#[cfg(feature = "pdf")]
|
|
1193
|
-
assert!(config.pdf_options.is_some());
|
|
1194
776
|
}
|
|
1195
777
|
|
|
1196
778
|
#[test]
|
|
@@ -1256,7 +838,6 @@ enabled = true
|
|
|
1256
838
|
}
|
|
1257
839
|
|
|
1258
840
|
#[test]
|
|
1259
|
-
#[cfg(feature = "pdf")]
|
|
1260
841
|
fn test_pdf_config_defaults() {
|
|
1261
842
|
let dir = tempdir().unwrap();
|
|
1262
843
|
let config_path = dir.path().join("kreuzberg.toml");
|
|
@@ -1448,467 +1029,4 @@ language = "eng"
|
|
|
1448
1029
|
assert!(tess.textord_space_size_is_variable);
|
|
1449
1030
|
assert!(!tess.thresholding_method);
|
|
1450
1031
|
}
|
|
1451
|
-
|
|
1452
|
-
#[test]
|
|
1453
|
-
fn test_config_cache_hit_same_mtime() {
|
|
1454
|
-
let dir = tempdir().unwrap();
|
|
1455
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1456
|
-
|
|
1457
|
-
fs::write(
|
|
1458
|
-
&config_path,
|
|
1459
|
-
r#"
|
|
1460
|
-
use_cache = false
|
|
1461
|
-
enable_quality_processing = true
|
|
1462
|
-
"#,
|
|
1463
|
-
)
|
|
1464
|
-
.unwrap();
|
|
1465
|
-
|
|
1466
|
-
let config1 = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1467
|
-
let config2 = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1468
|
-
|
|
1469
|
-
assert_eq!(config1.use_cache, config2.use_cache);
|
|
1470
|
-
assert_eq!(config1.enable_quality_processing, config2.enable_quality_processing);
|
|
1471
|
-
}
|
|
1472
|
-
|
|
1473
|
-
#[test]
|
|
1474
|
-
fn test_config_cache_invalidation_on_mtime_change() {
|
|
1475
|
-
let dir = tempdir().unwrap();
|
|
1476
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1477
|
-
|
|
1478
|
-
fs::write(
|
|
1479
|
-
&config_path,
|
|
1480
|
-
r#"
|
|
1481
|
-
use_cache = false
|
|
1482
|
-
enable_quality_processing = true
|
|
1483
|
-
"#,
|
|
1484
|
-
)
|
|
1485
|
-
.unwrap();
|
|
1486
|
-
|
|
1487
|
-
let config1 = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1488
|
-
assert!(!config1.use_cache);
|
|
1489
|
-
|
|
1490
|
-
std::thread::sleep(std::time::Duration::from_secs(1));
|
|
1491
|
-
|
|
1492
|
-
fs::write(
|
|
1493
|
-
&config_path,
|
|
1494
|
-
r#"
|
|
1495
|
-
use_cache = true
|
|
1496
|
-
enable_quality_processing = false
|
|
1497
|
-
"#,
|
|
1498
|
-
)
|
|
1499
|
-
.unwrap();
|
|
1500
|
-
|
|
1501
|
-
let config2 = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1502
|
-
assert!(config2.use_cache);
|
|
1503
|
-
assert!(!config2.enable_quality_processing);
|
|
1504
|
-
}
|
|
1505
|
-
|
|
1506
|
-
#[test]
|
|
1507
|
-
fn test_config_cache_works_with_json() {
|
|
1508
|
-
let dir = tempdir().unwrap();
|
|
1509
|
-
let config_path = dir.path().join("kreuzberg.json");
|
|
1510
|
-
|
|
1511
|
-
fs::write(
|
|
1512
|
-
&config_path,
|
|
1513
|
-
r#"{"use_cache": true, "enable_quality_processing": false}"#,
|
|
1514
|
-
)
|
|
1515
|
-
.unwrap();
|
|
1516
|
-
|
|
1517
|
-
let config1 = ExtractionConfig::from_json_file(&config_path).unwrap();
|
|
1518
|
-
let config2 = ExtractionConfig::from_json_file(&config_path).unwrap();
|
|
1519
|
-
|
|
1520
|
-
assert!(config1.use_cache);
|
|
1521
|
-
assert!(!config1.enable_quality_processing);
|
|
1522
|
-
assert_eq!(config1.use_cache, config2.use_cache);
|
|
1523
|
-
}
|
|
1524
|
-
|
|
1525
|
-
#[test]
|
|
1526
|
-
fn test_config_cache_works_with_yaml() {
|
|
1527
|
-
let dir = tempdir().unwrap();
|
|
1528
|
-
let config_path = dir.path().join("kreuzberg.yaml");
|
|
1529
|
-
|
|
1530
|
-
fs::write(
|
|
1531
|
-
&config_path,
|
|
1532
|
-
r#"
|
|
1533
|
-
use_cache: true
|
|
1534
|
-
enable_quality_processing: false
|
|
1535
|
-
"#,
|
|
1536
|
-
)
|
|
1537
|
-
.unwrap();
|
|
1538
|
-
|
|
1539
|
-
let config1 = ExtractionConfig::from_yaml_file(&config_path).unwrap();
|
|
1540
|
-
let config2 = ExtractionConfig::from_yaml_file(&config_path).unwrap();
|
|
1541
|
-
|
|
1542
|
-
assert!(config1.use_cache);
|
|
1543
|
-
assert!(!config1.enable_quality_processing);
|
|
1544
|
-
assert_eq!(config1.use_cache, config2.use_cache);
|
|
1545
|
-
}
|
|
1546
|
-
|
|
1547
|
-
#[test]
|
|
1548
|
-
#[serial_test::serial]
|
|
1549
|
-
fn test_apply_env_overrides_ocr_language() {
|
|
1550
|
-
let original_lang = std::env::var("KREUZBERG_OCR_LANGUAGE").ok();
|
|
1551
|
-
|
|
1552
|
-
set_env("KREUZBERG_OCR_LANGUAGE", "fra");
|
|
1553
|
-
let mut config = ExtractionConfig::default();
|
|
1554
|
-
assert!(config.ocr.is_none());
|
|
1555
|
-
|
|
1556
|
-
config.apply_env_overrides().unwrap();
|
|
1557
|
-
|
|
1558
|
-
assert!(config.ocr.is_some());
|
|
1559
|
-
assert_eq!(config.ocr.unwrap().language, "fra");
|
|
1560
|
-
|
|
1561
|
-
restore_env("KREUZBERG_OCR_LANGUAGE", original_lang);
|
|
1562
|
-
}
|
|
1563
|
-
|
|
1564
|
-
#[test]
|
|
1565
|
-
#[serial_test::serial]
|
|
1566
|
-
fn test_apply_env_overrides_ocr_backend() {
|
|
1567
|
-
let original_backend = std::env::var("KREUZBERG_OCR_BACKEND").ok();
|
|
1568
|
-
|
|
1569
|
-
set_env("KREUZBERG_OCR_BACKEND", "easyocr");
|
|
1570
|
-
let mut config = ExtractionConfig::default();
|
|
1571
|
-
assert!(config.ocr.is_none());
|
|
1572
|
-
|
|
1573
|
-
config.apply_env_overrides().unwrap();
|
|
1574
|
-
|
|
1575
|
-
assert!(config.ocr.is_some());
|
|
1576
|
-
assert_eq!(config.ocr.unwrap().backend, "easyocr");
|
|
1577
|
-
|
|
1578
|
-
restore_env("KREUZBERG_OCR_BACKEND", original_backend);
|
|
1579
|
-
}
|
|
1580
|
-
|
|
1581
|
-
#[test]
|
|
1582
|
-
#[serial_test::serial]
|
|
1583
|
-
fn test_apply_env_overrides_invalid_ocr_language() {
|
|
1584
|
-
let original_lang = std::env::var("KREUZBERG_OCR_LANGUAGE").ok();
|
|
1585
|
-
|
|
1586
|
-
set_env("KREUZBERG_OCR_LANGUAGE", "invalid_lang");
|
|
1587
|
-
let mut config = ExtractionConfig::default();
|
|
1588
|
-
|
|
1589
|
-
let result = config.apply_env_overrides();
|
|
1590
|
-
assert!(result.is_err());
|
|
1591
|
-
let err_msg = result.unwrap_err().to_string();
|
|
1592
|
-
assert!(err_msg.contains("Invalid language code"));
|
|
1593
|
-
|
|
1594
|
-
restore_env("KREUZBERG_OCR_LANGUAGE", original_lang);
|
|
1595
|
-
}
|
|
1596
|
-
|
|
1597
|
-
#[test]
|
|
1598
|
-
#[serial_test::serial]
|
|
1599
|
-
fn test_apply_env_overrides_invalid_ocr_backend() {
|
|
1600
|
-
let original_backend = std::env::var("KREUZBERG_OCR_BACKEND").ok();
|
|
1601
|
-
|
|
1602
|
-
set_env("KREUZBERG_OCR_BACKEND", "invalid_backend");
|
|
1603
|
-
let mut config = ExtractionConfig::default();
|
|
1604
|
-
|
|
1605
|
-
let result = config.apply_env_overrides();
|
|
1606
|
-
assert!(result.is_err());
|
|
1607
|
-
let err_msg = result.unwrap_err().to_string();
|
|
1608
|
-
assert!(err_msg.contains("Invalid OCR backend"));
|
|
1609
|
-
|
|
1610
|
-
restore_env("KREUZBERG_OCR_BACKEND", original_backend);
|
|
1611
|
-
}
|
|
1612
|
-
|
|
1613
|
-
#[test]
|
|
1614
|
-
#[serial_test::serial]
|
|
1615
|
-
fn test_apply_env_overrides_chunking_max_chars() {
|
|
1616
|
-
let original_max_chars = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS").ok();
|
|
1617
|
-
|
|
1618
|
-
set_env("KREUZBERG_CHUNKING_MAX_CHARS", "2000");
|
|
1619
|
-
let mut config = ExtractionConfig::default();
|
|
1620
|
-
assert!(config.chunking.is_none());
|
|
1621
|
-
|
|
1622
|
-
config.apply_env_overrides().unwrap();
|
|
1623
|
-
|
|
1624
|
-
assert!(config.chunking.is_some());
|
|
1625
|
-
assert_eq!(config.chunking.unwrap().max_chars, 2000);
|
|
1626
|
-
|
|
1627
|
-
restore_env("KREUZBERG_CHUNKING_MAX_CHARS", original_max_chars);
|
|
1628
|
-
}
|
|
1629
|
-
|
|
1630
|
-
#[test]
|
|
1631
|
-
#[serial_test::serial]
|
|
1632
|
-
fn test_apply_env_overrides_chunking_max_overlap() {
|
|
1633
|
-
let original_overlap = std::env::var("KREUZBERG_CHUNKING_MAX_OVERLAP").ok();
|
|
1634
|
-
|
|
1635
|
-
set_env("KREUZBERG_CHUNKING_MAX_OVERLAP", "300");
|
|
1636
|
-
let mut config = ExtractionConfig::default();
|
|
1637
|
-
assert!(config.chunking.is_none());
|
|
1638
|
-
|
|
1639
|
-
config.apply_env_overrides().unwrap();
|
|
1640
|
-
|
|
1641
|
-
assert!(config.chunking.is_some());
|
|
1642
|
-
assert_eq!(config.chunking.unwrap().max_overlap, 300);
|
|
1643
|
-
|
|
1644
|
-
restore_env("KREUZBERG_CHUNKING_MAX_OVERLAP", original_overlap);
|
|
1645
|
-
}
|
|
1646
|
-
|
|
1647
|
-
#[test]
|
|
1648
|
-
#[serial_test::serial]
|
|
1649
|
-
fn test_apply_env_overrides_invalid_chunking_max_chars() {
|
|
1650
|
-
let original_max_chars = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS").ok();
|
|
1651
|
-
|
|
1652
|
-
set_env("KREUZBERG_CHUNKING_MAX_CHARS", "not_a_number");
|
|
1653
|
-
let mut config = ExtractionConfig::default();
|
|
1654
|
-
|
|
1655
|
-
let result = config.apply_env_overrides();
|
|
1656
|
-
assert!(result.is_err());
|
|
1657
|
-
let err_msg = result.unwrap_err().to_string();
|
|
1658
|
-
assert!(err_msg.contains("KREUZBERG_CHUNKING_MAX_CHARS"));
|
|
1659
|
-
|
|
1660
|
-
restore_env("KREUZBERG_CHUNKING_MAX_CHARS", original_max_chars);
|
|
1661
|
-
}
|
|
1662
|
-
|
|
1663
|
-
#[test]
|
|
1664
|
-
#[serial_test::serial]
|
|
1665
|
-
fn test_apply_env_overrides_chunking_max_chars_zero() {
|
|
1666
|
-
let original_max_chars = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS").ok();
|
|
1667
|
-
|
|
1668
|
-
set_env("KREUZBERG_CHUNKING_MAX_CHARS", "0");
|
|
1669
|
-
let mut config = ExtractionConfig::default();
|
|
1670
|
-
|
|
1671
|
-
let result = config.apply_env_overrides();
|
|
1672
|
-
assert!(result.is_err());
|
|
1673
|
-
let err_msg = result.unwrap_err().to_string();
|
|
1674
|
-
assert!(err_msg.contains("greater than 0"));
|
|
1675
|
-
|
|
1676
|
-
restore_env("KREUZBERG_CHUNKING_MAX_CHARS", original_max_chars);
|
|
1677
|
-
}
|
|
1678
|
-
|
|
1679
|
-
#[test]
|
|
1680
|
-
#[serial_test::serial]
|
|
1681
|
-
fn test_apply_env_overrides_chunking_overlap_validation() {
|
|
1682
|
-
let original_overlap = std::env::var("KREUZBERG_CHUNKING_MAX_OVERLAP").ok();
|
|
1683
|
-
|
|
1684
|
-
set_env("KREUZBERG_CHUNKING_MAX_OVERLAP", "1500");
|
|
1685
|
-
// Set chunking with max_chars = 1000
|
|
1686
|
-
let mut config = ExtractionConfig {
|
|
1687
|
-
chunking: Some(ChunkingConfig {
|
|
1688
|
-
max_chars: 1000,
|
|
1689
|
-
max_overlap: 200,
|
|
1690
|
-
embedding: None,
|
|
1691
|
-
preset: None,
|
|
1692
|
-
}),
|
|
1693
|
-
..Default::default()
|
|
1694
|
-
};
|
|
1695
|
-
|
|
1696
|
-
let result = config.apply_env_overrides();
|
|
1697
|
-
assert!(result.is_err());
|
|
1698
|
-
let err_msg = result.unwrap_err().to_string();
|
|
1699
|
-
assert!(err_msg.contains("overlap"));
|
|
1700
|
-
|
|
1701
|
-
restore_env("KREUZBERG_CHUNKING_MAX_OVERLAP", original_overlap);
|
|
1702
|
-
}
|
|
1703
|
-
|
|
1704
|
-
#[test]
|
|
1705
|
-
#[serial_test::serial]
|
|
1706
|
-
fn test_apply_env_overrides_cache_enabled_true() {
|
|
1707
|
-
let original_cache = std::env::var("KREUZBERG_CACHE_ENABLED").ok();
|
|
1708
|
-
|
|
1709
|
-
set_env("KREUZBERG_CACHE_ENABLED", "true");
|
|
1710
|
-
let mut config = ExtractionConfig {
|
|
1711
|
-
use_cache: false,
|
|
1712
|
-
..Default::default()
|
|
1713
|
-
};
|
|
1714
|
-
|
|
1715
|
-
config.apply_env_overrides().unwrap();
|
|
1716
|
-
assert!(config.use_cache);
|
|
1717
|
-
|
|
1718
|
-
restore_env("KREUZBERG_CACHE_ENABLED", original_cache);
|
|
1719
|
-
}
|
|
1720
|
-
|
|
1721
|
-
#[test]
|
|
1722
|
-
#[serial_test::serial]
|
|
1723
|
-
fn test_apply_env_overrides_cache_enabled_false() {
|
|
1724
|
-
let original_cache = std::env::var("KREUZBERG_CACHE_ENABLED").ok();
|
|
1725
|
-
|
|
1726
|
-
set_env("KREUZBERG_CACHE_ENABLED", "false");
|
|
1727
|
-
let mut config = ExtractionConfig {
|
|
1728
|
-
use_cache: true,
|
|
1729
|
-
..Default::default()
|
|
1730
|
-
};
|
|
1731
|
-
|
|
1732
|
-
config.apply_env_overrides().unwrap();
|
|
1733
|
-
assert!(!config.use_cache);
|
|
1734
|
-
|
|
1735
|
-
restore_env("KREUZBERG_CACHE_ENABLED", original_cache);
|
|
1736
|
-
}
|
|
1737
|
-
|
|
1738
|
-
#[test]
|
|
1739
|
-
#[serial_test::serial]
|
|
1740
|
-
fn test_apply_env_overrides_cache_enabled_case_insensitive() {
|
|
1741
|
-
let original_cache = std::env::var("KREUZBERG_CACHE_ENABLED").ok();
|
|
1742
|
-
|
|
1743
|
-
set_env("KREUZBERG_CACHE_ENABLED", "TRUE");
|
|
1744
|
-
let mut config = ExtractionConfig {
|
|
1745
|
-
use_cache: false,
|
|
1746
|
-
..Default::default()
|
|
1747
|
-
};
|
|
1748
|
-
|
|
1749
|
-
config.apply_env_overrides().unwrap();
|
|
1750
|
-
assert!(config.use_cache);
|
|
1751
|
-
|
|
1752
|
-
set_env("KREUZBERG_CACHE_ENABLED", "FaLsE");
|
|
1753
|
-
config.use_cache = true;
|
|
1754
|
-
config.apply_env_overrides().unwrap();
|
|
1755
|
-
assert!(!config.use_cache);
|
|
1756
|
-
|
|
1757
|
-
restore_env("KREUZBERG_CACHE_ENABLED", original_cache);
|
|
1758
|
-
}
|
|
1759
|
-
|
|
1760
|
-
#[test]
|
|
1761
|
-
#[serial_test::serial]
|
|
1762
|
-
fn test_apply_env_overrides_invalid_cache_enabled() {
|
|
1763
|
-
let original_cache = std::env::var("KREUZBERG_CACHE_ENABLED").ok();
|
|
1764
|
-
|
|
1765
|
-
set_env("KREUZBERG_CACHE_ENABLED", "maybe");
|
|
1766
|
-
let mut config = ExtractionConfig::default();
|
|
1767
|
-
|
|
1768
|
-
let result = config.apply_env_overrides();
|
|
1769
|
-
assert!(result.is_err());
|
|
1770
|
-
let err_msg = result.unwrap_err().to_string();
|
|
1771
|
-
assert!(err_msg.contains("KREUZBERG_CACHE_ENABLED"));
|
|
1772
|
-
assert!(err_msg.contains("'true' or 'false'"));
|
|
1773
|
-
|
|
1774
|
-
restore_env("KREUZBERG_CACHE_ENABLED", original_cache);
|
|
1775
|
-
}
|
|
1776
|
-
|
|
1777
|
-
#[test]
|
|
1778
|
-
#[serial_test::serial]
|
|
1779
|
-
fn test_apply_env_overrides_token_reduction_mode() {
|
|
1780
|
-
let original_mode = std::env::var("KREUZBERG_TOKEN_REDUCTION_MODE").ok();
|
|
1781
|
-
|
|
1782
|
-
set_env("KREUZBERG_TOKEN_REDUCTION_MODE", "moderate");
|
|
1783
|
-
let mut config = ExtractionConfig::default();
|
|
1784
|
-
assert!(config.token_reduction.is_none());
|
|
1785
|
-
|
|
1786
|
-
config.apply_env_overrides().unwrap();
|
|
1787
|
-
|
|
1788
|
-
assert!(config.token_reduction.is_some());
|
|
1789
|
-
assert_eq!(config.token_reduction.unwrap().mode, "moderate");
|
|
1790
|
-
|
|
1791
|
-
restore_env("KREUZBERG_TOKEN_REDUCTION_MODE", original_mode);
|
|
1792
|
-
}
|
|
1793
|
-
|
|
1794
|
-
#[test]
|
|
1795
|
-
#[serial_test::serial]
|
|
1796
|
-
fn test_apply_env_overrides_invalid_token_reduction_mode() {
|
|
1797
|
-
let original_mode = std::env::var("KREUZBERG_TOKEN_REDUCTION_MODE").ok();
|
|
1798
|
-
|
|
1799
|
-
set_env("KREUZBERG_TOKEN_REDUCTION_MODE", "extreme");
|
|
1800
|
-
let mut config = ExtractionConfig::default();
|
|
1801
|
-
|
|
1802
|
-
let result = config.apply_env_overrides();
|
|
1803
|
-
assert!(result.is_err());
|
|
1804
|
-
let err_msg = result.unwrap_err().to_string();
|
|
1805
|
-
assert!(err_msg.contains("Invalid token reduction level"));
|
|
1806
|
-
|
|
1807
|
-
restore_env("KREUZBERG_TOKEN_REDUCTION_MODE", original_mode);
|
|
1808
|
-
}
|
|
1809
|
-
|
|
1810
|
-
#[test]
|
|
1811
|
-
#[serial_test::serial]
|
|
1812
|
-
fn test_apply_env_overrides_multiple_overrides() {
|
|
1813
|
-
let original_lang = std::env::var("KREUZBERG_OCR_LANGUAGE").ok();
|
|
1814
|
-
let original_backend = std::env::var("KREUZBERG_OCR_BACKEND").ok();
|
|
1815
|
-
let original_max_chars = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS").ok();
|
|
1816
|
-
let original_cache = std::env::var("KREUZBERG_CACHE_ENABLED").ok();
|
|
1817
|
-
|
|
1818
|
-
set_env("KREUZBERG_OCR_LANGUAGE", "deu");
|
|
1819
|
-
set_env("KREUZBERG_OCR_BACKEND", "paddleocr");
|
|
1820
|
-
set_env("KREUZBERG_CHUNKING_MAX_CHARS", "3000");
|
|
1821
|
-
set_env("KREUZBERG_CACHE_ENABLED", "false");
|
|
1822
|
-
|
|
1823
|
-
let mut config = ExtractionConfig::default();
|
|
1824
|
-
config.apply_env_overrides().unwrap();
|
|
1825
|
-
|
|
1826
|
-
let ocr = config.ocr.as_ref().unwrap();
|
|
1827
|
-
assert_eq!(ocr.language, "deu");
|
|
1828
|
-
assert_eq!(ocr.backend, "paddleocr");
|
|
1829
|
-
let chunking = config.chunking.as_ref().unwrap();
|
|
1830
|
-
assert_eq!(chunking.max_chars, 3000);
|
|
1831
|
-
assert!(!config.use_cache);
|
|
1832
|
-
|
|
1833
|
-
restore_env("KREUZBERG_OCR_LANGUAGE", original_lang);
|
|
1834
|
-
restore_env("KREUZBERG_OCR_BACKEND", original_backend);
|
|
1835
|
-
restore_env("KREUZBERG_CHUNKING_MAX_CHARS", original_max_chars);
|
|
1836
|
-
restore_env("KREUZBERG_CACHE_ENABLED", original_cache);
|
|
1837
|
-
}
|
|
1838
|
-
|
|
1839
|
-
#[test]
|
|
1840
|
-
#[serial_test::serial]
|
|
1841
|
-
fn test_apply_env_overrides_preserves_existing_ocr_config() {
|
|
1842
|
-
let original_lang = std::env::var("KREUZBERG_OCR_LANGUAGE").ok();
|
|
1843
|
-
|
|
1844
|
-
set_env("KREUZBERG_OCR_LANGUAGE", "fra");
|
|
1845
|
-
|
|
1846
|
-
let mut config = ExtractionConfig {
|
|
1847
|
-
ocr: Some(OcrConfig {
|
|
1848
|
-
backend: "tesseract".to_string(),
|
|
1849
|
-
language: "eng".to_string(),
|
|
1850
|
-
tesseract_config: None,
|
|
1851
|
-
}),
|
|
1852
|
-
..Default::default()
|
|
1853
|
-
};
|
|
1854
|
-
|
|
1855
|
-
config.apply_env_overrides().unwrap();
|
|
1856
|
-
|
|
1857
|
-
let ocr = config.ocr.unwrap();
|
|
1858
|
-
assert_eq!(ocr.language, "fra");
|
|
1859
|
-
assert_eq!(ocr.backend, "tesseract"); // Not overridden
|
|
1860
|
-
|
|
1861
|
-
restore_env("KREUZBERG_OCR_LANGUAGE", original_lang);
|
|
1862
|
-
}
|
|
1863
|
-
|
|
1864
|
-
#[test]
|
|
1865
|
-
#[serial_test::serial]
|
|
1866
|
-
fn test_apply_env_overrides_no_env_vars() {
|
|
1867
|
-
// Ensure no env vars are set
|
|
1868
|
-
remove_env("KREUZBERG_OCR_LANGUAGE");
|
|
1869
|
-
remove_env("KREUZBERG_OCR_BACKEND");
|
|
1870
|
-
remove_env("KREUZBERG_CHUNKING_MAX_CHARS");
|
|
1871
|
-
remove_env("KREUZBERG_CHUNKING_MAX_OVERLAP");
|
|
1872
|
-
remove_env("KREUZBERG_CACHE_ENABLED");
|
|
1873
|
-
remove_env("KREUZBERG_TOKEN_REDUCTION_MODE");
|
|
1874
|
-
|
|
1875
|
-
let mut config = ExtractionConfig {
|
|
1876
|
-
use_cache: false,
|
|
1877
|
-
ocr: Some(OcrConfig {
|
|
1878
|
-
backend: "tesseract".to_string(),
|
|
1879
|
-
language: "eng".to_string(),
|
|
1880
|
-
tesseract_config: None,
|
|
1881
|
-
}),
|
|
1882
|
-
..Default::default()
|
|
1883
|
-
};
|
|
1884
|
-
|
|
1885
|
-
// Should not change anything
|
|
1886
|
-
config.apply_env_overrides().unwrap();
|
|
1887
|
-
|
|
1888
|
-
assert!(!config.use_cache);
|
|
1889
|
-
assert_eq!(config.ocr.unwrap().language, "eng");
|
|
1890
|
-
}
|
|
1891
|
-
|
|
1892
|
-
#[test]
|
|
1893
|
-
#[serial_test::serial]
|
|
1894
|
-
fn test_apply_env_overrides_chunking_params_validation() {
|
|
1895
|
-
let original_max_chars = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS").ok();
|
|
1896
|
-
|
|
1897
|
-
set_env("KREUZBERG_CHUNKING_MAX_CHARS", "500");
|
|
1898
|
-
let mut config = ExtractionConfig {
|
|
1899
|
-
chunking: Some(ChunkingConfig {
|
|
1900
|
-
max_chars: 1000,
|
|
1901
|
-
max_overlap: 600, // Valid: < 1000
|
|
1902
|
-
embedding: None,
|
|
1903
|
-
preset: None,
|
|
1904
|
-
}),
|
|
1905
|
-
..Default::default()
|
|
1906
|
-
};
|
|
1907
|
-
|
|
1908
|
-
let result = config.apply_env_overrides();
|
|
1909
|
-
assert!(result.is_err());
|
|
1910
|
-
// After override, max_chars = 500, max_overlap = 600 (invalid: >= 500)
|
|
1911
|
-
|
|
1912
|
-
restore_env("KREUZBERG_CHUNKING_MAX_CHARS", original_max_chars);
|
|
1913
|
-
}
|
|
1914
1032
|
}
|