kreuzberg 4.0.8 → 4.1.0
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/Gemfile.lock +2 -2
- data/README.md +1 -1
- data/ext/kreuzberg_rb/native/Cargo.lock +94 -98
- data/ext/kreuzberg_rb/native/Cargo.toml +4 -2
- data/ext/kreuzberg_rb/native/src/batch.rs +139 -0
- data/ext/kreuzberg_rb/native/src/config/mod.rs +10 -0
- data/ext/kreuzberg_rb/native/src/config/types.rs +1058 -0
- data/ext/kreuzberg_rb/native/src/error_handling.rs +125 -0
- data/ext/kreuzberg_rb/native/src/extraction.rs +79 -0
- data/ext/kreuzberg_rb/native/src/gc_guarded_value.rs +35 -0
- data/ext/kreuzberg_rb/native/src/helpers.rs +176 -0
- data/ext/kreuzberg_rb/native/src/lib.rs +342 -3622
- data/ext/kreuzberg_rb/native/src/metadata.rs +34 -0
- data/ext/kreuzberg_rb/native/src/plugins/mod.rs +92 -0
- data/ext/kreuzberg_rb/native/src/plugins/ocr_backend.rs +159 -0
- data/ext/kreuzberg_rb/native/src/plugins/post_processor.rs +126 -0
- data/ext/kreuzberg_rb/native/src/plugins/validator.rs +99 -0
- data/ext/kreuzberg_rb/native/src/result.rs +326 -0
- data/ext/kreuzberg_rb/native/src/validation.rs +4 -0
- data/lib/kreuzberg/config.rb +66 -0
- data/lib/kreuzberg/result.rb +107 -2
- data/lib/kreuzberg/types.rb +104 -0
- data/lib/kreuzberg/version.rb +1 -1
- data/lib/kreuzberg.rb +0 -4
- data/sig/kreuzberg.rbs +105 -1
- data/vendor/Cargo.toml +3 -3
- data/vendor/kreuzberg/Cargo.toml +4 -3
- data/vendor/kreuzberg/README.md +1 -1
- data/vendor/kreuzberg/src/api/config.rs +69 -0
- data/vendor/kreuzberg/src/api/handlers.rs +99 -2
- data/vendor/kreuzberg/src/api/mod.rs +14 -7
- data/vendor/kreuzberg/src/api/router.rs +214 -0
- data/vendor/kreuzberg/src/api/startup.rs +243 -0
- data/vendor/kreuzberg/src/api/types.rs +78 -0
- data/vendor/kreuzberg/src/cache/cleanup.rs +277 -0
- data/vendor/kreuzberg/src/cache/core.rs +428 -0
- data/vendor/kreuzberg/src/cache/mod.rs +21 -843
- data/vendor/kreuzberg/src/cache/utilities.rs +156 -0
- data/vendor/kreuzberg/src/chunking/boundaries.rs +301 -0
- data/vendor/kreuzberg/src/chunking/builder.rs +294 -0
- data/vendor/kreuzberg/src/chunking/config.rs +52 -0
- data/vendor/kreuzberg/src/chunking/core.rs +1017 -0
- data/vendor/kreuzberg/src/chunking/mod.rs +14 -2211
- data/vendor/kreuzberg/src/chunking/processor.rs +10 -0
- data/vendor/kreuzberg/src/chunking/validation.rs +686 -0
- data/vendor/kreuzberg/src/core/config/extraction/core.rs +169 -0
- data/vendor/kreuzberg/src/core/config/extraction/env.rs +179 -0
- data/vendor/kreuzberg/src/core/config/extraction/loaders.rs +204 -0
- data/vendor/kreuzberg/src/core/config/extraction/mod.rs +42 -0
- data/vendor/kreuzberg/src/core/config/extraction/types.rs +93 -0
- data/vendor/kreuzberg/src/core/config/formats.rs +135 -0
- data/vendor/kreuzberg/src/core/config/mod.rs +20 -0
- data/vendor/kreuzberg/src/core/config/ocr.rs +73 -0
- data/vendor/kreuzberg/src/core/config/page.rs +57 -0
- data/vendor/kreuzberg/src/core/config/pdf.rs +111 -0
- data/vendor/kreuzberg/src/core/config/processing.rs +312 -0
- data/vendor/kreuzberg/src/core/config_validation/dependencies.rs +187 -0
- data/vendor/kreuzberg/src/core/config_validation/mod.rs +386 -0
- data/vendor/kreuzberg/src/core/config_validation/sections.rs +401 -0
- data/vendor/kreuzberg/src/core/extractor/batch.rs +246 -0
- data/vendor/kreuzberg/src/core/extractor/bytes.rs +116 -0
- data/vendor/kreuzberg/src/core/extractor/file.rs +240 -0
- data/vendor/kreuzberg/src/core/extractor/helpers.rs +71 -0
- data/vendor/kreuzberg/src/core/extractor/legacy.rs +62 -0
- data/vendor/kreuzberg/src/core/extractor/mod.rs +490 -0
- data/vendor/kreuzberg/src/core/extractor/sync.rs +208 -0
- data/vendor/kreuzberg/src/core/mod.rs +4 -1
- data/vendor/kreuzberg/src/core/pipeline/cache.rs +60 -0
- data/vendor/kreuzberg/src/core/pipeline/execution.rs +89 -0
- data/vendor/kreuzberg/src/core/pipeline/features.rs +108 -0
- data/vendor/kreuzberg/src/core/pipeline/format.rs +392 -0
- data/vendor/kreuzberg/src/core/pipeline/initialization.rs +67 -0
- data/vendor/kreuzberg/src/core/pipeline/mod.rs +135 -0
- data/vendor/kreuzberg/src/core/pipeline/tests.rs +975 -0
- data/vendor/kreuzberg/src/core/server_config/env.rs +90 -0
- data/vendor/kreuzberg/src/core/server_config/loader.rs +202 -0
- data/vendor/kreuzberg/src/core/server_config/mod.rs +380 -0
- data/vendor/kreuzberg/src/core/server_config/tests/basic_tests.rs +124 -0
- data/vendor/kreuzberg/src/core/server_config/tests/env_tests.rs +216 -0
- data/vendor/kreuzberg/src/core/server_config/tests/file_loading_tests.rs +341 -0
- data/vendor/kreuzberg/src/core/server_config/tests/mod.rs +5 -0
- data/vendor/kreuzberg/src/core/server_config/validation.rs +17 -0
- data/vendor/kreuzberg/src/embeddings.rs +136 -13
- data/vendor/kreuzberg/src/extraction/{archive.rs → archive/mod.rs} +45 -239
- data/vendor/kreuzberg/src/extraction/archive/sevenz.rs +98 -0
- data/vendor/kreuzberg/src/extraction/archive/tar.rs +118 -0
- data/vendor/kreuzberg/src/extraction/archive/zip.rs +101 -0
- data/vendor/kreuzberg/src/extraction/html/converter.rs +592 -0
- data/vendor/kreuzberg/src/extraction/html/image_handling.rs +95 -0
- data/vendor/kreuzberg/src/extraction/html/mod.rs +53 -0
- data/vendor/kreuzberg/src/extraction/html/processor.rs +659 -0
- data/vendor/kreuzberg/src/extraction/html/stack_management.rs +103 -0
- data/vendor/kreuzberg/src/extraction/html/types.rs +28 -0
- data/vendor/kreuzberg/src/extraction/mod.rs +6 -2
- data/vendor/kreuzberg/src/extraction/pptx/container.rs +159 -0
- data/vendor/kreuzberg/src/extraction/pptx/content_builder.rs +168 -0
- data/vendor/kreuzberg/src/extraction/pptx/elements.rs +132 -0
- data/vendor/kreuzberg/src/extraction/pptx/image_handling.rs +57 -0
- data/vendor/kreuzberg/src/extraction/pptx/metadata.rs +160 -0
- data/vendor/kreuzberg/src/extraction/pptx/mod.rs +558 -0
- data/vendor/kreuzberg/src/extraction/pptx/parser.rs +379 -0
- data/vendor/kreuzberg/src/extraction/transform/content.rs +205 -0
- data/vendor/kreuzberg/src/extraction/transform/elements.rs +211 -0
- data/vendor/kreuzberg/src/extraction/transform/mod.rs +480 -0
- data/vendor/kreuzberg/src/extraction/transform/types.rs +27 -0
- data/vendor/kreuzberg/src/extractors/archive.rs +2 -0
- data/vendor/kreuzberg/src/extractors/bibtex.rs +2 -0
- data/vendor/kreuzberg/src/extractors/djot_format/attributes.rs +134 -0
- data/vendor/kreuzberg/src/extractors/djot_format/conversion.rs +223 -0
- data/vendor/kreuzberg/src/extractors/djot_format/extractor.rs +172 -0
- data/vendor/kreuzberg/src/extractors/djot_format/mod.rs +24 -0
- data/vendor/kreuzberg/src/extractors/djot_format/parsing/block_handlers.rs +271 -0
- data/vendor/kreuzberg/src/extractors/djot_format/parsing/content_extraction.rs +257 -0
- data/vendor/kreuzberg/src/extractors/djot_format/parsing/event_handlers.rs +101 -0
- data/vendor/kreuzberg/src/extractors/djot_format/parsing/inline_handlers.rs +201 -0
- data/vendor/kreuzberg/src/extractors/djot_format/parsing/mod.rs +16 -0
- data/vendor/kreuzberg/src/extractors/djot_format/parsing/state.rs +78 -0
- data/vendor/kreuzberg/src/extractors/djot_format/parsing/table_extraction.rs +68 -0
- data/vendor/kreuzberg/src/extractors/djot_format/parsing/text_extraction.rs +61 -0
- data/vendor/kreuzberg/src/extractors/djot_format/rendering.rs +452 -0
- data/vendor/kreuzberg/src/extractors/docbook.rs +2 -0
- data/vendor/kreuzberg/src/extractors/docx.rs +12 -1
- data/vendor/kreuzberg/src/extractors/email.rs +2 -0
- data/vendor/kreuzberg/src/extractors/epub/content.rs +333 -0
- data/vendor/kreuzberg/src/extractors/epub/metadata.rs +137 -0
- data/vendor/kreuzberg/src/extractors/epub/mod.rs +186 -0
- data/vendor/kreuzberg/src/extractors/epub/parsing.rs +86 -0
- data/vendor/kreuzberg/src/extractors/excel.rs +4 -0
- data/vendor/kreuzberg/src/extractors/fictionbook.rs +2 -0
- data/vendor/kreuzberg/src/extractors/frontmatter_utils.rs +466 -0
- data/vendor/kreuzberg/src/extractors/html.rs +80 -8
- data/vendor/kreuzberg/src/extractors/image.rs +8 -1
- data/vendor/kreuzberg/src/extractors/jats/elements.rs +350 -0
- data/vendor/kreuzberg/src/extractors/jats/metadata.rs +21 -0
- data/vendor/kreuzberg/src/extractors/{jats.rs → jats/mod.rs} +10 -412
- data/vendor/kreuzberg/src/extractors/jats/parser.rs +52 -0
- data/vendor/kreuzberg/src/extractors/jupyter.rs +2 -0
- data/vendor/kreuzberg/src/extractors/latex/commands.rs +93 -0
- data/vendor/kreuzberg/src/extractors/latex/environments.rs +157 -0
- data/vendor/kreuzberg/src/extractors/latex/metadata.rs +27 -0
- data/vendor/kreuzberg/src/extractors/latex/mod.rs +146 -0
- data/vendor/kreuzberg/src/extractors/latex/parser.rs +231 -0
- data/vendor/kreuzberg/src/extractors/latex/utilities.rs +126 -0
- data/vendor/kreuzberg/src/extractors/markdown.rs +39 -162
- data/vendor/kreuzberg/src/extractors/mod.rs +9 -1
- data/vendor/kreuzberg/src/extractors/odt.rs +2 -0
- data/vendor/kreuzberg/src/extractors/opml/core.rs +165 -0
- data/vendor/kreuzberg/src/extractors/opml/mod.rs +31 -0
- data/vendor/kreuzberg/src/extractors/opml/parser.rs +479 -0
- data/vendor/kreuzberg/src/extractors/orgmode.rs +2 -0
- data/vendor/kreuzberg/src/extractors/pdf/extraction.rs +106 -0
- data/vendor/kreuzberg/src/extractors/{pdf.rs → pdf/mod.rs} +25 -324
- data/vendor/kreuzberg/src/extractors/pdf/ocr.rs +214 -0
- data/vendor/kreuzberg/src/extractors/pdf/pages.rs +51 -0
- data/vendor/kreuzberg/src/extractors/pptx.rs +9 -2
- data/vendor/kreuzberg/src/extractors/rst.rs +2 -0
- data/vendor/kreuzberg/src/extractors/rtf/encoding.rs +116 -0
- data/vendor/kreuzberg/src/extractors/rtf/formatting.rs +24 -0
- data/vendor/kreuzberg/src/extractors/rtf/images.rs +72 -0
- data/vendor/kreuzberg/src/extractors/rtf/metadata.rs +216 -0
- data/vendor/kreuzberg/src/extractors/rtf/mod.rs +142 -0
- data/vendor/kreuzberg/src/extractors/rtf/parser.rs +259 -0
- data/vendor/kreuzberg/src/extractors/rtf/tables.rs +83 -0
- data/vendor/kreuzberg/src/extractors/structured.rs +2 -0
- data/vendor/kreuzberg/src/extractors/text.rs +4 -0
- data/vendor/kreuzberg/src/extractors/typst.rs +2 -0
- data/vendor/kreuzberg/src/extractors/xml.rs +2 -0
- data/vendor/kreuzberg/src/keywords/processor.rs +14 -0
- data/vendor/kreuzberg/src/language_detection/processor.rs +10 -0
- data/vendor/kreuzberg/src/lib.rs +2 -2
- data/vendor/kreuzberg/src/mcp/errors.rs +312 -0
- data/vendor/kreuzberg/src/mcp/format.rs +211 -0
- data/vendor/kreuzberg/src/mcp/mod.rs +9 -3
- data/vendor/kreuzberg/src/mcp/params.rs +196 -0
- data/vendor/kreuzberg/src/mcp/server.rs +39 -1438
- data/vendor/kreuzberg/src/mcp/tools/cache.rs +179 -0
- data/vendor/kreuzberg/src/mcp/tools/extraction.rs +403 -0
- data/vendor/kreuzberg/src/mcp/tools/mime.rs +150 -0
- data/vendor/kreuzberg/src/mcp/tools/mod.rs +11 -0
- data/vendor/kreuzberg/src/ocr/backends/easyocr.rs +96 -0
- data/vendor/kreuzberg/src/ocr/backends/mod.rs +7 -0
- data/vendor/kreuzberg/src/ocr/backends/paddleocr.rs +27 -0
- data/vendor/kreuzberg/src/ocr/backends/tesseract.rs +134 -0
- data/vendor/kreuzberg/src/ocr/hocr.rs +60 -16
- data/vendor/kreuzberg/src/ocr/language_registry.rs +11 -235
- data/vendor/kreuzberg/src/ocr/mod.rs +1 -0
- data/vendor/kreuzberg/src/ocr/processor/config.rs +203 -0
- data/vendor/kreuzberg/src/ocr/processor/execution.rs +494 -0
- data/vendor/kreuzberg/src/ocr/processor/mod.rs +265 -0
- data/vendor/kreuzberg/src/ocr/processor/validation.rs +145 -0
- data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +41 -24
- data/vendor/kreuzberg/src/pdf/bindings.rs +21 -8
- data/vendor/kreuzberg/src/pdf/hierarchy/bounding_box.rs +289 -0
- data/vendor/kreuzberg/src/pdf/hierarchy/clustering.rs +199 -0
- data/vendor/kreuzberg/src/pdf/{hierarchy.rs → hierarchy/extraction.rs} +6 -346
- data/vendor/kreuzberg/src/pdf/hierarchy/mod.rs +18 -0
- data/vendor/kreuzberg/src/plugins/extractor/mod.rs +319 -0
- data/vendor/kreuzberg/src/plugins/extractor/registry.rs +434 -0
- data/vendor/kreuzberg/src/plugins/extractor/trait.rs +391 -0
- data/vendor/kreuzberg/src/plugins/mod.rs +13 -0
- data/vendor/kreuzberg/src/plugins/ocr.rs +11 -0
- data/vendor/kreuzberg/src/plugins/processor/mod.rs +365 -0
- data/vendor/kreuzberg/src/plugins/processor/registry.rs +37 -0
- data/vendor/kreuzberg/src/plugins/processor/trait.rs +284 -0
- data/vendor/kreuzberg/src/plugins/registry/extractor.rs +416 -0
- data/vendor/kreuzberg/src/plugins/registry/mod.rs +116 -0
- data/vendor/kreuzberg/src/plugins/registry/ocr.rs +293 -0
- data/vendor/kreuzberg/src/plugins/registry/processor.rs +304 -0
- data/vendor/kreuzberg/src/plugins/registry/validator.rs +238 -0
- data/vendor/kreuzberg/src/plugins/validator/mod.rs +424 -0
- data/vendor/kreuzberg/src/plugins/validator/registry.rs +355 -0
- data/vendor/kreuzberg/src/plugins/validator/trait.rs +276 -0
- data/vendor/kreuzberg/src/stopwords/languages/asian.rs +40 -0
- data/vendor/kreuzberg/src/stopwords/languages/germanic.rs +36 -0
- data/vendor/kreuzberg/src/stopwords/languages/mod.rs +10 -0
- data/vendor/kreuzberg/src/stopwords/languages/other.rs +44 -0
- data/vendor/kreuzberg/src/stopwords/languages/romance.rs +36 -0
- data/vendor/kreuzberg/src/stopwords/languages/slavic.rs +36 -0
- data/vendor/kreuzberg/src/stopwords/mod.rs +7 -33
- data/vendor/kreuzberg/src/text/quality.rs +1 -1
- data/vendor/kreuzberg/src/text/quality_processor.rs +10 -0
- data/vendor/kreuzberg/src/text/token_reduction/core/analysis.rs +238 -0
- data/vendor/kreuzberg/src/text/token_reduction/core/mod.rs +8 -0
- data/vendor/kreuzberg/src/text/token_reduction/core/punctuation.rs +54 -0
- data/vendor/kreuzberg/src/text/token_reduction/core/reducer.rs +384 -0
- data/vendor/kreuzberg/src/text/token_reduction/core/sentence_selection.rs +68 -0
- data/vendor/kreuzberg/src/text/token_reduction/core/word_filtering.rs +156 -0
- data/vendor/kreuzberg/src/text/token_reduction/filters/general.rs +377 -0
- data/vendor/kreuzberg/src/text/token_reduction/filters/html.rs +51 -0
- data/vendor/kreuzberg/src/text/token_reduction/filters/markdown.rs +285 -0
- data/vendor/kreuzberg/src/text/token_reduction/filters.rs +131 -246
- data/vendor/kreuzberg/src/types/djot.rs +209 -0
- data/vendor/kreuzberg/src/types/extraction.rs +301 -0
- data/vendor/kreuzberg/src/types/formats.rs +443 -0
- data/vendor/kreuzberg/src/types/metadata.rs +560 -0
- data/vendor/kreuzberg/src/types/mod.rs +281 -0
- data/vendor/kreuzberg/src/types/page.rs +182 -0
- data/vendor/kreuzberg/src/types/serde_helpers.rs +132 -0
- data/vendor/kreuzberg/src/types/tables.rs +39 -0
- data/vendor/kreuzberg/src/utils/quality/heuristics.rs +58 -0
- data/vendor/kreuzberg/src/utils/{quality.rs → quality/mod.rs} +168 -489
- data/vendor/kreuzberg/src/utils/quality/patterns.rs +117 -0
- data/vendor/kreuzberg/src/utils/quality/scoring.rs +178 -0
- data/vendor/kreuzberg/src/utils/string_pool/buffer_pool.rs +325 -0
- data/vendor/kreuzberg/src/utils/string_pool/interned.rs +102 -0
- data/vendor/kreuzberg/src/utils/string_pool/language_pool.rs +119 -0
- data/vendor/kreuzberg/src/utils/string_pool/mime_pool.rs +235 -0
- data/vendor/kreuzberg/src/utils/string_pool/mod.rs +41 -0
- data/vendor/kreuzberg/tests/api_chunk.rs +313 -0
- data/vendor/kreuzberg/tests/api_embed.rs +6 -9
- data/vendor/kreuzberg/tests/batch_orchestration.rs +1 -0
- data/vendor/kreuzberg/tests/concurrency_stress.rs +7 -0
- data/vendor/kreuzberg/tests/core_integration.rs +1 -0
- data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +130 -0
- data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +5 -14
- data/vendor/kreuzberg/tests/format_integration.rs +2 -0
- data/vendor/kreuzberg/tests/helpers/mod.rs +1 -0
- data/vendor/kreuzberg/tests/html_table_test.rs +11 -11
- data/vendor/kreuzberg/tests/ocr_configuration.rs +16 -0
- data/vendor/kreuzberg/tests/ocr_errors.rs +18 -0
- data/vendor/kreuzberg/tests/ocr_quality.rs +9 -0
- data/vendor/kreuzberg/tests/ocr_stress.rs +1 -0
- data/vendor/kreuzberg/tests/pipeline_integration.rs +50 -0
- data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +13 -0
- data/vendor/kreuzberg/tests/plugin_system.rs +12 -0
- data/vendor/kreuzberg/tests/registry_integration_tests.rs +2 -0
- data/vendor/kreuzberg-ffi/Cargo.toml +2 -1
- data/vendor/kreuzberg-ffi/benches/result_view_benchmark.rs +2 -0
- data/vendor/kreuzberg-ffi/kreuzberg.h +347 -178
- data/vendor/kreuzberg-ffi/src/config/html.rs +318 -0
- data/vendor/kreuzberg-ffi/src/config/loader.rs +154 -0
- data/vendor/kreuzberg-ffi/src/config/merge.rs +104 -0
- data/vendor/kreuzberg-ffi/src/config/mod.rs +385 -0
- data/vendor/kreuzberg-ffi/src/config/parse.rs +91 -0
- data/vendor/kreuzberg-ffi/src/config/serialize.rs +118 -0
- data/vendor/kreuzberg-ffi/src/config_builder.rs +598 -0
- data/vendor/kreuzberg-ffi/src/error.rs +46 -14
- data/vendor/kreuzberg-ffi/src/helpers.rs +10 -0
- data/vendor/kreuzberg-ffi/src/html_options.rs +421 -0
- data/vendor/kreuzberg-ffi/src/lib.rs +16 -0
- data/vendor/kreuzberg-ffi/src/panic_shield.rs +11 -0
- data/vendor/kreuzberg-ffi/src/plugins/ocr_backend.rs +2 -0
- data/vendor/kreuzberg-ffi/src/result.rs +148 -122
- data/vendor/kreuzberg-ffi/src/result_view.rs +4 -0
- data/vendor/kreuzberg-tesseract/Cargo.toml +2 -2
- metadata +200 -28
- data/vendor/kreuzberg/src/api/server.rs +0 -518
- data/vendor/kreuzberg/src/core/config.rs +0 -1914
- data/vendor/kreuzberg/src/core/config_validation.rs +0 -949
- data/vendor/kreuzberg/src/core/extractor.rs +0 -1200
- data/vendor/kreuzberg/src/core/pipeline.rs +0 -1223
- data/vendor/kreuzberg/src/core/server_config.rs +0 -1220
- data/vendor/kreuzberg/src/extraction/html.rs +0 -1830
- data/vendor/kreuzberg/src/extraction/pptx.rs +0 -3102
- data/vendor/kreuzberg/src/extractors/epub.rs +0 -696
- data/vendor/kreuzberg/src/extractors/latex.rs +0 -653
- data/vendor/kreuzberg/src/extractors/opml.rs +0 -635
- data/vendor/kreuzberg/src/extractors/rtf.rs +0 -809
- data/vendor/kreuzberg/src/ocr/processor.rs +0 -858
- data/vendor/kreuzberg/src/plugins/extractor.rs +0 -1042
- data/vendor/kreuzberg/src/plugins/processor.rs +0 -650
- data/vendor/kreuzberg/src/plugins/registry.rs +0 -1339
- data/vendor/kreuzberg/src/plugins/validator.rs +0 -967
- data/vendor/kreuzberg/src/text/token_reduction/core.rs +0 -832
- data/vendor/kreuzberg/src/types.rs +0 -1713
- data/vendor/kreuzberg/src/utils/string_pool.rs +0 -762
- data/vendor/kreuzberg-ffi/src/config.rs +0 -1341
|
@@ -1,1914 +0,0 @@
|
|
|
1
|
-
//! Configuration loading and management.
|
|
2
|
-
//!
|
|
3
|
-
//! This module provides utilities for loading extraction configuration from various
|
|
4
|
-
//! sources (TOML, YAML, JSON) and discovering configuration files in the project hierarchy.
|
|
5
|
-
|
|
6
|
-
use crate::{KreuzbergError, Result};
|
|
7
|
-
use dashmap::DashMap;
|
|
8
|
-
use serde::{Deserialize, Serialize};
|
|
9
|
-
use std::collections::HashSet;
|
|
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
|
-
}
|
|
49
|
-
|
|
50
|
-
/// Main extraction configuration.
|
|
51
|
-
///
|
|
52
|
-
/// This struct contains all configuration options for the extraction process.
|
|
53
|
-
/// It can be loaded from TOML, YAML, or JSON files, or created programmatically.
|
|
54
|
-
///
|
|
55
|
-
/// # Example
|
|
56
|
-
///
|
|
57
|
-
/// ```rust
|
|
58
|
-
/// use kreuzberg::core::config::ExtractionConfig;
|
|
59
|
-
///
|
|
60
|
-
/// // Create with defaults
|
|
61
|
-
/// let config = ExtractionConfig::default();
|
|
62
|
-
///
|
|
63
|
-
/// // Load from TOML file
|
|
64
|
-
/// // let config = ExtractionConfig::from_toml_file("kreuzberg.toml")?;
|
|
65
|
-
/// ```
|
|
66
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
67
|
-
pub struct ExtractionConfig {
|
|
68
|
-
/// Enable caching of extraction results
|
|
69
|
-
#[serde(default = "default_true")]
|
|
70
|
-
pub use_cache: bool,
|
|
71
|
-
|
|
72
|
-
/// Enable quality post-processing
|
|
73
|
-
#[serde(default = "default_true")]
|
|
74
|
-
pub enable_quality_processing: bool,
|
|
75
|
-
|
|
76
|
-
/// OCR configuration (None = OCR disabled)
|
|
77
|
-
#[serde(default)]
|
|
78
|
-
pub ocr: Option<OcrConfig>,
|
|
79
|
-
|
|
80
|
-
/// Force OCR even for searchable PDFs
|
|
81
|
-
#[serde(default)]
|
|
82
|
-
pub force_ocr: bool,
|
|
83
|
-
|
|
84
|
-
/// Text chunking configuration (None = chunking disabled)
|
|
85
|
-
#[serde(default)]
|
|
86
|
-
pub chunking: Option<ChunkingConfig>,
|
|
87
|
-
|
|
88
|
-
/// Image extraction configuration (None = no image extraction)
|
|
89
|
-
#[serde(default)]
|
|
90
|
-
pub images: Option<ImageExtractionConfig>,
|
|
91
|
-
|
|
92
|
-
/// PDF-specific options (None = use defaults)
|
|
93
|
-
#[cfg(feature = "pdf")]
|
|
94
|
-
#[serde(default)]
|
|
95
|
-
pub pdf_options: Option<PdfConfig>,
|
|
96
|
-
|
|
97
|
-
/// Token reduction configuration (None = no token reduction)
|
|
98
|
-
#[serde(default)]
|
|
99
|
-
pub token_reduction: Option<TokenReductionConfig>,
|
|
100
|
-
|
|
101
|
-
/// Language detection configuration (None = no language detection)
|
|
102
|
-
#[serde(default)]
|
|
103
|
-
pub language_detection: Option<LanguageDetectionConfig>,
|
|
104
|
-
|
|
105
|
-
/// Page extraction configuration (None = no page tracking)
|
|
106
|
-
#[serde(default)]
|
|
107
|
-
pub pages: Option<PageConfig>,
|
|
108
|
-
|
|
109
|
-
/// Keyword extraction configuration (None = no keyword extraction)
|
|
110
|
-
#[cfg(any(feature = "keywords-yake", feature = "keywords-rake"))]
|
|
111
|
-
#[serde(default)]
|
|
112
|
-
pub keywords: Option<crate::keywords::KeywordConfig>,
|
|
113
|
-
|
|
114
|
-
/// Post-processor configuration (None = use defaults)
|
|
115
|
-
#[serde(default)]
|
|
116
|
-
pub postprocessor: Option<PostProcessorConfig>,
|
|
117
|
-
|
|
118
|
-
/// HTML to Markdown conversion options (None = use defaults)
|
|
119
|
-
///
|
|
120
|
-
/// Configure how HTML documents are converted to Markdown, including heading styles,
|
|
121
|
-
/// list formatting, code block styles, and preprocessing options.
|
|
122
|
-
#[cfg(feature = "html")]
|
|
123
|
-
#[serde(default)]
|
|
124
|
-
pub html_options: Option<html_to_markdown_rs::ConversionOptions>,
|
|
125
|
-
|
|
126
|
-
/// Maximum concurrent extractions in batch operations (None = num_cpus * 2).
|
|
127
|
-
///
|
|
128
|
-
/// Limits parallelism to prevent resource exhaustion when processing
|
|
129
|
-
/// large batches. Defaults to twice the number of CPU cores.
|
|
130
|
-
#[serde(default)]
|
|
131
|
-
pub max_concurrent_extractions: Option<usize>,
|
|
132
|
-
}
|
|
133
|
-
|
|
134
|
-
/// Post-processor configuration.
|
|
135
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
136
|
-
pub struct PostProcessorConfig {
|
|
137
|
-
/// Enable post-processors
|
|
138
|
-
#[serde(default = "default_true")]
|
|
139
|
-
pub enabled: bool,
|
|
140
|
-
|
|
141
|
-
/// Whitelist of processor names to run (None = all enabled)
|
|
142
|
-
#[serde(default)]
|
|
143
|
-
pub enabled_processors: Option<Vec<String>>,
|
|
144
|
-
|
|
145
|
-
/// Blacklist of processor names to skip (None = none disabled)
|
|
146
|
-
#[serde(default)]
|
|
147
|
-
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
|
-
}
|
|
172
|
-
|
|
173
|
-
/// OCR configuration.
|
|
174
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
175
|
-
pub struct OcrConfig {
|
|
176
|
-
/// OCR backend: tesseract, easyocr, paddleocr
|
|
177
|
-
#[serde(default = "default_tesseract_backend")]
|
|
178
|
-
pub backend: String,
|
|
179
|
-
|
|
180
|
-
/// Language code (e.g., "eng", "deu")
|
|
181
|
-
#[serde(default = "default_eng")]
|
|
182
|
-
pub language: String,
|
|
183
|
-
|
|
184
|
-
/// Tesseract-specific configuration (optional)
|
|
185
|
-
#[serde(default)]
|
|
186
|
-
pub tesseract_config: Option<crate::types::TesseractConfig>,
|
|
187
|
-
}
|
|
188
|
-
|
|
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
|
-
/// Chunking configuration.
|
|
200
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
201
|
-
pub struct ChunkingConfig {
|
|
202
|
-
/// Maximum characters per chunk
|
|
203
|
-
#[serde(default = "default_chunk_size")]
|
|
204
|
-
pub max_chars: usize,
|
|
205
|
-
|
|
206
|
-
/// Overlap between chunks in characters
|
|
207
|
-
#[serde(default = "default_chunk_overlap")]
|
|
208
|
-
pub max_overlap: usize,
|
|
209
|
-
|
|
210
|
-
/// Optional embedding configuration for chunk embeddings
|
|
211
|
-
#[serde(skip_serializing_if = "Option::is_none")]
|
|
212
|
-
pub embedding: Option<EmbeddingConfig>,
|
|
213
|
-
|
|
214
|
-
/// Use a preset configuration (overrides individual settings if provided)
|
|
215
|
-
#[serde(skip_serializing_if = "Option::is_none")]
|
|
216
|
-
pub preset: Option<String>,
|
|
217
|
-
}
|
|
218
|
-
|
|
219
|
-
/// Embedding configuration for text chunks.
|
|
220
|
-
///
|
|
221
|
-
/// Configures embedding generation using ONNX models via fastembed-rs.
|
|
222
|
-
/// Requires the `embeddings` feature to be enabled.
|
|
223
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
224
|
-
pub struct EmbeddingConfig {
|
|
225
|
-
/// The embedding model to use
|
|
226
|
-
pub model: EmbeddingModelType,
|
|
227
|
-
|
|
228
|
-
/// Whether to normalize embedding vectors (recommended for cosine similarity)
|
|
229
|
-
#[serde(default = "default_normalize")]
|
|
230
|
-
pub normalize: bool,
|
|
231
|
-
|
|
232
|
-
/// Batch size for embedding generation
|
|
233
|
-
#[serde(default = "default_batch_size")]
|
|
234
|
-
pub batch_size: usize,
|
|
235
|
-
|
|
236
|
-
/// Show model download progress
|
|
237
|
-
#[serde(default)]
|
|
238
|
-
pub show_download_progress: bool,
|
|
239
|
-
|
|
240
|
-
/// Custom cache directory for model files
|
|
241
|
-
///
|
|
242
|
-
/// Defaults to `~/.cache/kreuzberg/embeddings/` if not specified.
|
|
243
|
-
/// Allows full customization of model download location.
|
|
244
|
-
#[serde(skip_serializing_if = "Option::is_none")]
|
|
245
|
-
pub cache_dir: Option<std::path::PathBuf>,
|
|
246
|
-
}
|
|
247
|
-
|
|
248
|
-
impl Default for EmbeddingConfig {
|
|
249
|
-
fn default() -> Self {
|
|
250
|
-
Self {
|
|
251
|
-
model: EmbeddingModelType::Preset {
|
|
252
|
-
name: "balanced".to_string(),
|
|
253
|
-
},
|
|
254
|
-
normalize: true,
|
|
255
|
-
batch_size: 32,
|
|
256
|
-
show_download_progress: false,
|
|
257
|
-
cache_dir: None,
|
|
258
|
-
}
|
|
259
|
-
}
|
|
260
|
-
}
|
|
261
|
-
|
|
262
|
-
/// Embedding model types supported by Kreuzberg.
|
|
263
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
264
|
-
#[serde(tag = "type", rename_all = "snake_case")]
|
|
265
|
-
pub enum EmbeddingModelType {
|
|
266
|
-
/// Use a preset model configuration (recommended)
|
|
267
|
-
Preset { name: String },
|
|
268
|
-
|
|
269
|
-
/// Use a specific fastembed model by name
|
|
270
|
-
#[cfg(feature = "embeddings")]
|
|
271
|
-
FastEmbed { model: String, dimensions: usize },
|
|
272
|
-
|
|
273
|
-
/// Use a custom ONNX model from HuggingFace
|
|
274
|
-
Custom { model_id: String, dimensions: usize },
|
|
275
|
-
}
|
|
276
|
-
|
|
277
|
-
/// Image extraction configuration.
|
|
278
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
279
|
-
pub struct ImageExtractionConfig {
|
|
280
|
-
/// Extract images from documents
|
|
281
|
-
#[serde(default = "default_true")]
|
|
282
|
-
pub extract_images: bool,
|
|
283
|
-
|
|
284
|
-
/// Target DPI for image normalization
|
|
285
|
-
#[serde(default = "default_target_dpi")]
|
|
286
|
-
pub target_dpi: i32,
|
|
287
|
-
|
|
288
|
-
/// Maximum dimension for images (width or height)
|
|
289
|
-
#[serde(default = "default_max_dimension")]
|
|
290
|
-
pub max_image_dimension: i32,
|
|
291
|
-
|
|
292
|
-
/// Automatically adjust DPI based on image content
|
|
293
|
-
#[serde(default = "default_true")]
|
|
294
|
-
pub auto_adjust_dpi: bool,
|
|
295
|
-
|
|
296
|
-
/// Minimum DPI threshold
|
|
297
|
-
#[serde(default = "default_min_dpi")]
|
|
298
|
-
pub min_dpi: i32,
|
|
299
|
-
|
|
300
|
-
/// Maximum DPI threshold
|
|
301
|
-
#[serde(default = "default_max_dpi")]
|
|
302
|
-
pub max_dpi: i32,
|
|
303
|
-
}
|
|
304
|
-
|
|
305
|
-
/// PDF-specific configuration.
|
|
306
|
-
#[cfg(feature = "pdf")]
|
|
307
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
308
|
-
pub struct PdfConfig {
|
|
309
|
-
/// Extract images from PDF
|
|
310
|
-
#[serde(default)]
|
|
311
|
-
pub extract_images: bool,
|
|
312
|
-
|
|
313
|
-
/// List of passwords to try when opening encrypted PDFs
|
|
314
|
-
#[serde(default)]
|
|
315
|
-
pub passwords: Option<Vec<String>>,
|
|
316
|
-
|
|
317
|
-
/// Extract PDF metadata
|
|
318
|
-
#[serde(default = "default_true")]
|
|
319
|
-
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
|
-
}
|
|
375
|
-
|
|
376
|
-
/// Token reduction configuration.
|
|
377
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
378
|
-
pub struct TokenReductionConfig {
|
|
379
|
-
/// Reduction mode: "off", "light", "moderate", "aggressive", "maximum"
|
|
380
|
-
#[serde(default = "default_reduction_mode")]
|
|
381
|
-
pub mode: String,
|
|
382
|
-
|
|
383
|
-
/// Preserve important words (capitalized, technical terms)
|
|
384
|
-
#[serde(default = "default_true")]
|
|
385
|
-
pub preserve_important_words: bool,
|
|
386
|
-
}
|
|
387
|
-
|
|
388
|
-
/// Language detection configuration.
|
|
389
|
-
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
390
|
-
pub struct LanguageDetectionConfig {
|
|
391
|
-
/// Enable language detection
|
|
392
|
-
#[serde(default = "default_true")]
|
|
393
|
-
pub enabled: bool,
|
|
394
|
-
|
|
395
|
-
/// Minimum confidence threshold (0.0-1.0)
|
|
396
|
-
#[serde(default = "default_confidence")]
|
|
397
|
-
pub min_confidence: f64,
|
|
398
|
-
|
|
399
|
-
/// Detect multiple languages in the document
|
|
400
|
-
#[serde(default)]
|
|
401
|
-
pub detect_multiple: bool,
|
|
402
|
-
}
|
|
403
|
-
|
|
404
|
-
fn default_true() -> bool {
|
|
405
|
-
true
|
|
406
|
-
}
|
|
407
|
-
fn default_eng() -> String {
|
|
408
|
-
"eng".to_string()
|
|
409
|
-
}
|
|
410
|
-
fn default_tesseract_backend() -> String {
|
|
411
|
-
"tesseract".to_string()
|
|
412
|
-
}
|
|
413
|
-
fn default_page_marker_format() -> String {
|
|
414
|
-
"\n\n<!-- PAGE {page_num} -->\n\n".to_string()
|
|
415
|
-
}
|
|
416
|
-
fn default_chunk_size() -> usize {
|
|
417
|
-
1000
|
|
418
|
-
}
|
|
419
|
-
fn default_chunk_overlap() -> usize {
|
|
420
|
-
200
|
|
421
|
-
}
|
|
422
|
-
fn default_normalize() -> bool {
|
|
423
|
-
true
|
|
424
|
-
}
|
|
425
|
-
fn default_batch_size() -> usize {
|
|
426
|
-
32
|
|
427
|
-
}
|
|
428
|
-
fn default_target_dpi() -> i32 {
|
|
429
|
-
300
|
|
430
|
-
}
|
|
431
|
-
fn default_max_dimension() -> i32 {
|
|
432
|
-
4096
|
|
433
|
-
}
|
|
434
|
-
fn default_min_dpi() -> i32 {
|
|
435
|
-
72
|
|
436
|
-
}
|
|
437
|
-
fn default_max_dpi() -> i32 {
|
|
438
|
-
600
|
|
439
|
-
}
|
|
440
|
-
fn default_reduction_mode() -> String {
|
|
441
|
-
"off".to_string()
|
|
442
|
-
}
|
|
443
|
-
fn default_confidence() -> f64 {
|
|
444
|
-
0.8
|
|
445
|
-
}
|
|
446
|
-
|
|
447
|
-
impl Default for ExtractionConfig {
|
|
448
|
-
fn default() -> Self {
|
|
449
|
-
Self {
|
|
450
|
-
use_cache: true,
|
|
451
|
-
enable_quality_processing: true,
|
|
452
|
-
ocr: None,
|
|
453
|
-
force_ocr: false,
|
|
454
|
-
chunking: None,
|
|
455
|
-
images: None,
|
|
456
|
-
#[cfg(feature = "pdf")]
|
|
457
|
-
pdf_options: None,
|
|
458
|
-
token_reduction: None,
|
|
459
|
-
language_detection: None,
|
|
460
|
-
pages: None,
|
|
461
|
-
#[cfg(any(feature = "keywords-yake", feature = "keywords-rake"))]
|
|
462
|
-
keywords: None,
|
|
463
|
-
postprocessor: None,
|
|
464
|
-
#[cfg(feature = "html")]
|
|
465
|
-
html_options: None,
|
|
466
|
-
max_concurrent_extractions: None,
|
|
467
|
-
}
|
|
468
|
-
}
|
|
469
|
-
}
|
|
470
|
-
|
|
471
|
-
impl Default for PostProcessorConfig {
|
|
472
|
-
fn default() -> Self {
|
|
473
|
-
Self {
|
|
474
|
-
enabled: true,
|
|
475
|
-
enabled_processors: None,
|
|
476
|
-
disabled_processors: None,
|
|
477
|
-
enabled_set: None,
|
|
478
|
-
disabled_set: None,
|
|
479
|
-
}
|
|
480
|
-
}
|
|
481
|
-
}
|
|
482
|
-
|
|
483
|
-
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
|
-
/// Load configuration from a TOML file.
|
|
662
|
-
///
|
|
663
|
-
/// # Arguments
|
|
664
|
-
///
|
|
665
|
-
/// * `path` - Path to the TOML file
|
|
666
|
-
///
|
|
667
|
-
/// # Errors
|
|
668
|
-
///
|
|
669
|
-
/// Returns `KreuzbergError::Validation` if file doesn't exist or is invalid TOML.
|
|
670
|
-
pub fn from_toml_file(path: impl AsRef<Path>) -> Result<Self> {
|
|
671
|
-
let path = path.as_ref();
|
|
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))
|
|
677
|
-
})?;
|
|
678
|
-
|
|
679
|
-
if let Some(entry) = CONFIG_CACHE.get(path)
|
|
680
|
-
&& entry.0 == mtime
|
|
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)
|
|
695
|
-
}
|
|
696
|
-
|
|
697
|
-
/// Load configuration from a YAML file.
|
|
698
|
-
pub fn from_yaml_file(path: impl AsRef<Path>) -> Result<Self> {
|
|
699
|
-
let path = path.as_ref();
|
|
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))
|
|
705
|
-
})?;
|
|
706
|
-
|
|
707
|
-
if let Some(entry) = CONFIG_CACHE.get(path)
|
|
708
|
-
&& entry.0 == mtime
|
|
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)
|
|
723
|
-
}
|
|
724
|
-
|
|
725
|
-
/// Load configuration from a JSON file.
|
|
726
|
-
pub fn from_json_file(path: impl AsRef<Path>) -> Result<Self> {
|
|
727
|
-
let path = path.as_ref();
|
|
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))
|
|
733
|
-
})?;
|
|
734
|
-
|
|
735
|
-
if let Some(entry) = CONFIG_CACHE.get(path)
|
|
736
|
-
&& entry.0 == mtime
|
|
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)
|
|
751
|
-
}
|
|
752
|
-
|
|
753
|
-
/// Load configuration from a file, auto-detecting format by extension.
|
|
754
|
-
///
|
|
755
|
-
/// Supported formats:
|
|
756
|
-
/// - `.toml` - TOML format
|
|
757
|
-
/// - `.yaml` - YAML format
|
|
758
|
-
/// - `.json` - JSON format
|
|
759
|
-
///
|
|
760
|
-
/// # Arguments
|
|
761
|
-
///
|
|
762
|
-
/// * `path` - Path to the configuration file
|
|
763
|
-
///
|
|
764
|
-
/// # Errors
|
|
765
|
-
///
|
|
766
|
-
/// Returns `KreuzbergError::Validation` if:
|
|
767
|
-
/// - File doesn't exist
|
|
768
|
-
/// - File extension is not supported
|
|
769
|
-
/// - File content is invalid for the detected format
|
|
770
|
-
///
|
|
771
|
-
/// # Example
|
|
772
|
-
///
|
|
773
|
-
/// ```rust
|
|
774
|
-
/// use kreuzberg::core::config::ExtractionConfig;
|
|
775
|
-
///
|
|
776
|
-
/// // Auto-detects TOML format
|
|
777
|
-
/// // let config = ExtractionConfig::from_file("kreuzberg.toml")?;
|
|
778
|
-
///
|
|
779
|
-
/// // Auto-detects YAML format
|
|
780
|
-
/// // let config = ExtractionConfig::from_file("kreuzberg.yaml")?;
|
|
781
|
-
/// ```
|
|
782
|
-
pub fn from_file(path: impl AsRef<Path>) -> Result<Self> {
|
|
783
|
-
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
|
-
let extension = path.extension().and_then(|ext| ext.to_str()).ok_or_else(|| {
|
|
798
|
-
KreuzbergError::validation(format!(
|
|
799
|
-
"Cannot determine file format: no extension found in {}",
|
|
800
|
-
path.display()
|
|
801
|
-
))
|
|
802
|
-
})?;
|
|
803
|
-
|
|
804
|
-
let config = match extension.to_lowercase().as_str() {
|
|
805
|
-
"toml" => Self::from_toml_file(path)?,
|
|
806
|
-
"yaml" | "yml" => Self::from_yaml_file(path)?,
|
|
807
|
-
"json" => Self::from_json_file(path)?,
|
|
808
|
-
_ => {
|
|
809
|
-
return Err(KreuzbergError::validation(format!(
|
|
810
|
-
"Unsupported config file format: .{}. Supported formats: .toml, .yaml, .json",
|
|
811
|
-
extension
|
|
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)
|
|
820
|
-
}
|
|
821
|
-
|
|
822
|
-
/// Discover configuration file in parent directories.
|
|
823
|
-
///
|
|
824
|
-
/// Searches for `kreuzberg.toml` in current directory and parent directories.
|
|
825
|
-
///
|
|
826
|
-
/// # Returns
|
|
827
|
-
///
|
|
828
|
-
/// - `Some(config)` if found
|
|
829
|
-
/// - `None` if no config file found
|
|
830
|
-
pub fn discover() -> Result<Option<Self>> {
|
|
831
|
-
let mut current = std::env::current_dir().map_err(KreuzbergError::Io)?;
|
|
832
|
-
|
|
833
|
-
loop {
|
|
834
|
-
let kreuzberg_toml = current.join("kreuzberg.toml");
|
|
835
|
-
if kreuzberg_toml.exists() {
|
|
836
|
-
return Ok(Some(Self::from_toml_file(kreuzberg_toml)?));
|
|
837
|
-
}
|
|
838
|
-
|
|
839
|
-
if let Some(parent) = current.parent() {
|
|
840
|
-
current = parent.to_path_buf();
|
|
841
|
-
} else {
|
|
842
|
-
break;
|
|
843
|
-
}
|
|
844
|
-
}
|
|
845
|
-
|
|
846
|
-
Ok(None)
|
|
847
|
-
}
|
|
848
|
-
}
|
|
849
|
-
|
|
850
|
-
#[cfg(test)]
|
|
851
|
-
#[allow(unsafe_code)]
|
|
852
|
-
mod tests {
|
|
853
|
-
use super::*;
|
|
854
|
-
use std::fs;
|
|
855
|
-
use tempfile::tempdir;
|
|
856
|
-
|
|
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
|
-
#[test]
|
|
882
|
-
fn test_default_config() {
|
|
883
|
-
let config = ExtractionConfig::default();
|
|
884
|
-
assert!(config.use_cache);
|
|
885
|
-
assert!(config.enable_quality_processing);
|
|
886
|
-
assert!(config.ocr.is_none());
|
|
887
|
-
}
|
|
888
|
-
|
|
889
|
-
#[test]
|
|
890
|
-
fn test_from_toml_file() {
|
|
891
|
-
let dir = tempdir().unwrap();
|
|
892
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
893
|
-
|
|
894
|
-
fs::write(
|
|
895
|
-
&config_path,
|
|
896
|
-
r#"
|
|
897
|
-
use_cache = false
|
|
898
|
-
enable_quality_processing = true
|
|
899
|
-
"#,
|
|
900
|
-
)
|
|
901
|
-
.unwrap();
|
|
902
|
-
|
|
903
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
904
|
-
assert!(!config.use_cache);
|
|
905
|
-
assert!(config.enable_quality_processing);
|
|
906
|
-
}
|
|
907
|
-
|
|
908
|
-
#[test]
|
|
909
|
-
fn test_from_file_auto_detects_toml() {
|
|
910
|
-
let dir = tempdir().unwrap();
|
|
911
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
912
|
-
|
|
913
|
-
fs::write(
|
|
914
|
-
&config_path,
|
|
915
|
-
r#"
|
|
916
|
-
use_cache = false
|
|
917
|
-
enable_quality_processing = true
|
|
918
|
-
"#,
|
|
919
|
-
)
|
|
920
|
-
.unwrap();
|
|
921
|
-
|
|
922
|
-
let config = ExtractionConfig::from_file(&config_path).unwrap();
|
|
923
|
-
assert!(!config.use_cache);
|
|
924
|
-
assert!(config.enable_quality_processing);
|
|
925
|
-
}
|
|
926
|
-
|
|
927
|
-
#[test]
|
|
928
|
-
fn test_from_file_auto_detects_yaml() {
|
|
929
|
-
let dir = tempdir().unwrap();
|
|
930
|
-
let config_path = dir.path().join("kreuzberg.yaml");
|
|
931
|
-
|
|
932
|
-
fs::write(
|
|
933
|
-
&config_path,
|
|
934
|
-
r#"
|
|
935
|
-
use_cache: true
|
|
936
|
-
enable_quality_processing: false
|
|
937
|
-
"#,
|
|
938
|
-
)
|
|
939
|
-
.unwrap();
|
|
940
|
-
|
|
941
|
-
let config = ExtractionConfig::from_file(&config_path).unwrap();
|
|
942
|
-
assert!(config.use_cache);
|
|
943
|
-
assert!(!config.enable_quality_processing);
|
|
944
|
-
}
|
|
945
|
-
|
|
946
|
-
#[test]
|
|
947
|
-
fn test_from_file_unsupported_extension() {
|
|
948
|
-
let dir = tempdir().unwrap();
|
|
949
|
-
let config_path = dir.path().join("kreuzberg.txt");
|
|
950
|
-
|
|
951
|
-
fs::write(&config_path, "use_cache = false").unwrap();
|
|
952
|
-
|
|
953
|
-
let result = ExtractionConfig::from_file(&config_path);
|
|
954
|
-
assert!(result.is_err());
|
|
955
|
-
let err = result.unwrap_err();
|
|
956
|
-
assert!(err.to_string().contains("Unsupported config file format"));
|
|
957
|
-
}
|
|
958
|
-
|
|
959
|
-
#[test]
|
|
960
|
-
fn test_from_file_no_extension() {
|
|
961
|
-
let dir = tempdir().unwrap();
|
|
962
|
-
let config_path = dir.path().join("kreuzberg");
|
|
963
|
-
|
|
964
|
-
fs::write(&config_path, "use_cache = false").unwrap();
|
|
965
|
-
|
|
966
|
-
let result = ExtractionConfig::from_file(&config_path);
|
|
967
|
-
assert!(result.is_err());
|
|
968
|
-
let err = result.unwrap_err();
|
|
969
|
-
assert!(err.to_string().contains("no extension found"));
|
|
970
|
-
}
|
|
971
|
-
|
|
972
|
-
#[test]
|
|
973
|
-
fn test_discover_kreuzberg_toml() {
|
|
974
|
-
let dir = tempdir().unwrap();
|
|
975
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
976
|
-
|
|
977
|
-
fs::write(
|
|
978
|
-
&config_path,
|
|
979
|
-
r#"
|
|
980
|
-
use_cache = false
|
|
981
|
-
enable_quality_processing = true
|
|
982
|
-
"#,
|
|
983
|
-
)
|
|
984
|
-
.unwrap();
|
|
985
|
-
|
|
986
|
-
let original_dir = std::env::current_dir().unwrap();
|
|
987
|
-
std::env::set_current_dir(&dir).unwrap();
|
|
988
|
-
|
|
989
|
-
let result = std::panic::catch_unwind(|| {
|
|
990
|
-
let config = ExtractionConfig::discover().unwrap();
|
|
991
|
-
assert!(config.is_some());
|
|
992
|
-
assert!(!config.unwrap().use_cache);
|
|
993
|
-
});
|
|
994
|
-
|
|
995
|
-
std::env::set_current_dir(&original_dir).unwrap();
|
|
996
|
-
|
|
997
|
-
if let Err(e) = result {
|
|
998
|
-
std::panic::resume_unwind(e);
|
|
999
|
-
}
|
|
1000
|
-
}
|
|
1001
|
-
|
|
1002
|
-
#[test]
|
|
1003
|
-
fn test_v4_config_with_ocr_and_chunking() {
|
|
1004
|
-
let dir = tempdir().unwrap();
|
|
1005
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1006
|
-
|
|
1007
|
-
fs::write(
|
|
1008
|
-
&config_path,
|
|
1009
|
-
r#"
|
|
1010
|
-
use_cache = true
|
|
1011
|
-
enable_quality_processing = false
|
|
1012
|
-
|
|
1013
|
-
[ocr]
|
|
1014
|
-
backend = "tesseract"
|
|
1015
|
-
language = "eng"
|
|
1016
|
-
|
|
1017
|
-
[chunking]
|
|
1018
|
-
max_chars = 2000
|
|
1019
|
-
max_overlap = 300
|
|
1020
|
-
"#,
|
|
1021
|
-
)
|
|
1022
|
-
.unwrap();
|
|
1023
|
-
|
|
1024
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1025
|
-
assert!(config.use_cache);
|
|
1026
|
-
assert!(!config.enable_quality_processing);
|
|
1027
|
-
assert!(config.ocr.is_some());
|
|
1028
|
-
assert_eq!(config.ocr.unwrap().backend, "tesseract");
|
|
1029
|
-
assert!(config.chunking.is_some());
|
|
1030
|
-
assert_eq!(config.chunking.unwrap().max_chars, 2000);
|
|
1031
|
-
}
|
|
1032
|
-
|
|
1033
|
-
#[test]
|
|
1034
|
-
fn test_config_with_image_extraction() {
|
|
1035
|
-
let dir = tempdir().unwrap();
|
|
1036
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1037
|
-
|
|
1038
|
-
fs::write(
|
|
1039
|
-
&config_path,
|
|
1040
|
-
r#"
|
|
1041
|
-
use_cache = true
|
|
1042
|
-
|
|
1043
|
-
[images]
|
|
1044
|
-
extract_images = true
|
|
1045
|
-
target_dpi = 300
|
|
1046
|
-
max_image_dimension = 4096
|
|
1047
|
-
auto_adjust_dpi = true
|
|
1048
|
-
min_dpi = 72
|
|
1049
|
-
max_dpi = 600
|
|
1050
|
-
"#,
|
|
1051
|
-
)
|
|
1052
|
-
.unwrap();
|
|
1053
|
-
|
|
1054
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1055
|
-
assert!(config.images.is_some());
|
|
1056
|
-
let images = config.images.unwrap();
|
|
1057
|
-
assert!(images.extract_images);
|
|
1058
|
-
assert_eq!(images.target_dpi, 300);
|
|
1059
|
-
assert_eq!(images.max_image_dimension, 4096);
|
|
1060
|
-
assert!(images.auto_adjust_dpi);
|
|
1061
|
-
assert_eq!(images.min_dpi, 72);
|
|
1062
|
-
assert_eq!(images.max_dpi, 600);
|
|
1063
|
-
}
|
|
1064
|
-
|
|
1065
|
-
#[test]
|
|
1066
|
-
#[cfg(feature = "pdf")]
|
|
1067
|
-
fn test_config_with_pdf_options() {
|
|
1068
|
-
let dir = tempdir().unwrap();
|
|
1069
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1070
|
-
|
|
1071
|
-
fs::write(
|
|
1072
|
-
&config_path,
|
|
1073
|
-
r#"
|
|
1074
|
-
use_cache = true
|
|
1075
|
-
|
|
1076
|
-
[pdf_options]
|
|
1077
|
-
extract_images = true
|
|
1078
|
-
passwords = ["password1", "password2"]
|
|
1079
|
-
extract_metadata = true
|
|
1080
|
-
"#,
|
|
1081
|
-
)
|
|
1082
|
-
.unwrap();
|
|
1083
|
-
|
|
1084
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1085
|
-
assert!(config.pdf_options.is_some());
|
|
1086
|
-
let pdf = config.pdf_options.unwrap();
|
|
1087
|
-
assert!(pdf.extract_images);
|
|
1088
|
-
assert!(pdf.extract_metadata);
|
|
1089
|
-
assert!(pdf.passwords.is_some());
|
|
1090
|
-
let passwords = pdf.passwords.unwrap();
|
|
1091
|
-
assert_eq!(passwords.len(), 2);
|
|
1092
|
-
assert_eq!(passwords[0], "password1");
|
|
1093
|
-
assert_eq!(passwords[1], "password2");
|
|
1094
|
-
}
|
|
1095
|
-
|
|
1096
|
-
#[test]
|
|
1097
|
-
fn test_config_with_token_reduction() {
|
|
1098
|
-
let dir = tempdir().unwrap();
|
|
1099
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1100
|
-
|
|
1101
|
-
fs::write(
|
|
1102
|
-
&config_path,
|
|
1103
|
-
r#"
|
|
1104
|
-
use_cache = true
|
|
1105
|
-
|
|
1106
|
-
[token_reduction]
|
|
1107
|
-
mode = "aggressive"
|
|
1108
|
-
preserve_important_words = true
|
|
1109
|
-
"#,
|
|
1110
|
-
)
|
|
1111
|
-
.unwrap();
|
|
1112
|
-
|
|
1113
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1114
|
-
assert!(config.token_reduction.is_some());
|
|
1115
|
-
let token = config.token_reduction.unwrap();
|
|
1116
|
-
assert_eq!(token.mode, "aggressive");
|
|
1117
|
-
assert!(token.preserve_important_words);
|
|
1118
|
-
}
|
|
1119
|
-
|
|
1120
|
-
#[test]
|
|
1121
|
-
fn test_config_with_language_detection() {
|
|
1122
|
-
let dir = tempdir().unwrap();
|
|
1123
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1124
|
-
|
|
1125
|
-
fs::write(
|
|
1126
|
-
&config_path,
|
|
1127
|
-
r#"
|
|
1128
|
-
use_cache = true
|
|
1129
|
-
|
|
1130
|
-
[language_detection]
|
|
1131
|
-
enabled = true
|
|
1132
|
-
min_confidence = 0.9
|
|
1133
|
-
detect_multiple = true
|
|
1134
|
-
"#,
|
|
1135
|
-
)
|
|
1136
|
-
.unwrap();
|
|
1137
|
-
|
|
1138
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1139
|
-
assert!(config.language_detection.is_some());
|
|
1140
|
-
let lang = config.language_detection.unwrap();
|
|
1141
|
-
assert!(lang.enabled);
|
|
1142
|
-
assert_eq!(lang.min_confidence, 0.9);
|
|
1143
|
-
assert!(lang.detect_multiple);
|
|
1144
|
-
}
|
|
1145
|
-
|
|
1146
|
-
#[test]
|
|
1147
|
-
fn test_config_with_all_optional_fields() {
|
|
1148
|
-
let dir = tempdir().unwrap();
|
|
1149
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1150
|
-
|
|
1151
|
-
fs::write(
|
|
1152
|
-
&config_path,
|
|
1153
|
-
r#"
|
|
1154
|
-
use_cache = true
|
|
1155
|
-
enable_quality_processing = true
|
|
1156
|
-
force_ocr = false
|
|
1157
|
-
|
|
1158
|
-
[ocr]
|
|
1159
|
-
backend = "tesseract"
|
|
1160
|
-
language = "eng"
|
|
1161
|
-
|
|
1162
|
-
[chunking]
|
|
1163
|
-
max_chars = 1500
|
|
1164
|
-
max_overlap = 250
|
|
1165
|
-
|
|
1166
|
-
[images]
|
|
1167
|
-
extract_images = true
|
|
1168
|
-
target_dpi = 300
|
|
1169
|
-
|
|
1170
|
-
[pdf_options]
|
|
1171
|
-
extract_images = false
|
|
1172
|
-
extract_metadata = true
|
|
1173
|
-
|
|
1174
|
-
[token_reduction]
|
|
1175
|
-
mode = "moderate"
|
|
1176
|
-
|
|
1177
|
-
[language_detection]
|
|
1178
|
-
enabled = true
|
|
1179
|
-
"#,
|
|
1180
|
-
)
|
|
1181
|
-
.unwrap();
|
|
1182
|
-
|
|
1183
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1184
|
-
assert!(config.use_cache);
|
|
1185
|
-
assert!(config.enable_quality_processing);
|
|
1186
|
-
assert!(!config.force_ocr);
|
|
1187
|
-
assert!(config.ocr.is_some());
|
|
1188
|
-
assert!(config.chunking.is_some());
|
|
1189
|
-
assert!(config.images.is_some());
|
|
1190
|
-
assert!(config.token_reduction.is_some());
|
|
1191
|
-
assert!(config.language_detection.is_some());
|
|
1192
|
-
#[cfg(feature = "pdf")]
|
|
1193
|
-
assert!(config.pdf_options.is_some());
|
|
1194
|
-
}
|
|
1195
|
-
|
|
1196
|
-
#[test]
|
|
1197
|
-
fn test_image_config_defaults() {
|
|
1198
|
-
let dir = tempdir().unwrap();
|
|
1199
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1200
|
-
|
|
1201
|
-
fs::write(
|
|
1202
|
-
&config_path,
|
|
1203
|
-
r#"
|
|
1204
|
-
[images]
|
|
1205
|
-
"#,
|
|
1206
|
-
)
|
|
1207
|
-
.unwrap();
|
|
1208
|
-
|
|
1209
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1210
|
-
let images = config.images.unwrap();
|
|
1211
|
-
assert!(images.extract_images);
|
|
1212
|
-
assert_eq!(images.target_dpi, 300);
|
|
1213
|
-
assert_eq!(images.max_image_dimension, 4096);
|
|
1214
|
-
assert!(images.auto_adjust_dpi);
|
|
1215
|
-
assert_eq!(images.min_dpi, 72);
|
|
1216
|
-
assert_eq!(images.max_dpi, 600);
|
|
1217
|
-
}
|
|
1218
|
-
|
|
1219
|
-
#[test]
|
|
1220
|
-
fn test_token_reduction_defaults() {
|
|
1221
|
-
let dir = tempdir().unwrap();
|
|
1222
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1223
|
-
|
|
1224
|
-
fs::write(
|
|
1225
|
-
&config_path,
|
|
1226
|
-
r#"
|
|
1227
|
-
[token_reduction]
|
|
1228
|
-
"#,
|
|
1229
|
-
)
|
|
1230
|
-
.unwrap();
|
|
1231
|
-
|
|
1232
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1233
|
-
let token = config.token_reduction.unwrap();
|
|
1234
|
-
assert_eq!(token.mode, "off");
|
|
1235
|
-
assert!(token.preserve_important_words);
|
|
1236
|
-
}
|
|
1237
|
-
|
|
1238
|
-
#[test]
|
|
1239
|
-
fn test_language_detection_defaults() {
|
|
1240
|
-
let dir = tempdir().unwrap();
|
|
1241
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1242
|
-
|
|
1243
|
-
fs::write(
|
|
1244
|
-
&config_path,
|
|
1245
|
-
r#"
|
|
1246
|
-
[language_detection]
|
|
1247
|
-
"#,
|
|
1248
|
-
)
|
|
1249
|
-
.unwrap();
|
|
1250
|
-
|
|
1251
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1252
|
-
let lang = config.language_detection.unwrap();
|
|
1253
|
-
assert!(lang.enabled);
|
|
1254
|
-
assert_eq!(lang.min_confidence, 0.8);
|
|
1255
|
-
assert!(!lang.detect_multiple);
|
|
1256
|
-
}
|
|
1257
|
-
|
|
1258
|
-
#[test]
|
|
1259
|
-
#[cfg(feature = "pdf")]
|
|
1260
|
-
fn test_pdf_config_defaults() {
|
|
1261
|
-
let dir = tempdir().unwrap();
|
|
1262
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1263
|
-
|
|
1264
|
-
fs::write(
|
|
1265
|
-
&config_path,
|
|
1266
|
-
r#"
|
|
1267
|
-
[pdf_options]
|
|
1268
|
-
"#,
|
|
1269
|
-
)
|
|
1270
|
-
.unwrap();
|
|
1271
|
-
|
|
1272
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1273
|
-
let pdf = config.pdf_options.unwrap();
|
|
1274
|
-
assert!(!pdf.extract_images);
|
|
1275
|
-
assert!(pdf.extract_metadata);
|
|
1276
|
-
assert!(pdf.passwords.is_none());
|
|
1277
|
-
}
|
|
1278
|
-
|
|
1279
|
-
#[test]
|
|
1280
|
-
fn test_config_with_postprocessor() {
|
|
1281
|
-
let dir = tempdir().unwrap();
|
|
1282
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1283
|
-
|
|
1284
|
-
fs::write(
|
|
1285
|
-
&config_path,
|
|
1286
|
-
r#"
|
|
1287
|
-
use_cache = true
|
|
1288
|
-
|
|
1289
|
-
[postprocessor]
|
|
1290
|
-
enabled = true
|
|
1291
|
-
enabled_processors = ["entity_extraction", "keyword_extraction"]
|
|
1292
|
-
"#,
|
|
1293
|
-
)
|
|
1294
|
-
.unwrap();
|
|
1295
|
-
|
|
1296
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1297
|
-
assert!(config.postprocessor.is_some());
|
|
1298
|
-
let pp = config.postprocessor.unwrap();
|
|
1299
|
-
assert!(pp.enabled);
|
|
1300
|
-
assert!(pp.enabled_processors.is_some());
|
|
1301
|
-
let enabled = pp.enabled_processors.unwrap();
|
|
1302
|
-
assert_eq!(enabled.len(), 2);
|
|
1303
|
-
assert_eq!(enabled[0], "entity_extraction");
|
|
1304
|
-
assert_eq!(enabled[1], "keyword_extraction");
|
|
1305
|
-
assert!(pp.disabled_processors.is_none());
|
|
1306
|
-
}
|
|
1307
|
-
|
|
1308
|
-
#[test]
|
|
1309
|
-
fn test_postprocessor_config_defaults() {
|
|
1310
|
-
let dir = tempdir().unwrap();
|
|
1311
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1312
|
-
|
|
1313
|
-
fs::write(
|
|
1314
|
-
&config_path,
|
|
1315
|
-
r#"
|
|
1316
|
-
[postprocessor]
|
|
1317
|
-
"#,
|
|
1318
|
-
)
|
|
1319
|
-
.unwrap();
|
|
1320
|
-
|
|
1321
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1322
|
-
let pp = config.postprocessor.unwrap();
|
|
1323
|
-
assert!(pp.enabled);
|
|
1324
|
-
assert!(pp.enabled_processors.is_none());
|
|
1325
|
-
assert!(pp.disabled_processors.is_none());
|
|
1326
|
-
}
|
|
1327
|
-
|
|
1328
|
-
#[test]
|
|
1329
|
-
fn test_postprocessor_config_disabled() {
|
|
1330
|
-
let dir = tempdir().unwrap();
|
|
1331
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1332
|
-
|
|
1333
|
-
fs::write(
|
|
1334
|
-
&config_path,
|
|
1335
|
-
r#"
|
|
1336
|
-
[postprocessor]
|
|
1337
|
-
enabled = false
|
|
1338
|
-
"#,
|
|
1339
|
-
)
|
|
1340
|
-
.unwrap();
|
|
1341
|
-
|
|
1342
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1343
|
-
let pp = config.postprocessor.unwrap();
|
|
1344
|
-
assert!(!pp.enabled);
|
|
1345
|
-
}
|
|
1346
|
-
|
|
1347
|
-
#[test]
|
|
1348
|
-
fn test_postprocessor_config_blacklist() {
|
|
1349
|
-
let dir = tempdir().unwrap();
|
|
1350
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1351
|
-
|
|
1352
|
-
fs::write(
|
|
1353
|
-
&config_path,
|
|
1354
|
-
r#"
|
|
1355
|
-
[postprocessor]
|
|
1356
|
-
disabled_processors = ["category_extraction"]
|
|
1357
|
-
"#,
|
|
1358
|
-
)
|
|
1359
|
-
.unwrap();
|
|
1360
|
-
|
|
1361
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1362
|
-
let pp = config.postprocessor.unwrap();
|
|
1363
|
-
assert!(pp.enabled);
|
|
1364
|
-
assert!(pp.disabled_processors.is_some());
|
|
1365
|
-
let disabled = pp.disabled_processors.unwrap();
|
|
1366
|
-
assert_eq!(disabled.len(), 1);
|
|
1367
|
-
assert_eq!(disabled[0], "category_extraction");
|
|
1368
|
-
}
|
|
1369
|
-
|
|
1370
|
-
#[test]
|
|
1371
|
-
fn test_config_with_tesseract_config() {
|
|
1372
|
-
let dir = tempdir().unwrap();
|
|
1373
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1374
|
-
|
|
1375
|
-
fs::write(
|
|
1376
|
-
&config_path,
|
|
1377
|
-
r#"
|
|
1378
|
-
use_cache = true
|
|
1379
|
-
|
|
1380
|
-
[ocr]
|
|
1381
|
-
backend = "tesseract"
|
|
1382
|
-
language = "eng"
|
|
1383
|
-
|
|
1384
|
-
[ocr.tesseract_config]
|
|
1385
|
-
psm = 6
|
|
1386
|
-
output_format = "text"
|
|
1387
|
-
enable_table_detection = false
|
|
1388
|
-
tessedit_char_whitelist = "0123456789"
|
|
1389
|
-
"#,
|
|
1390
|
-
)
|
|
1391
|
-
.unwrap();
|
|
1392
|
-
|
|
1393
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1394
|
-
assert!(config.ocr.is_some());
|
|
1395
|
-
let ocr = config.ocr.unwrap();
|
|
1396
|
-
assert_eq!(ocr.backend, "tesseract");
|
|
1397
|
-
assert_eq!(ocr.language, "eng");
|
|
1398
|
-
assert!(ocr.tesseract_config.is_some());
|
|
1399
|
-
|
|
1400
|
-
let tess = ocr.tesseract_config.unwrap();
|
|
1401
|
-
assert_eq!(tess.psm, 6);
|
|
1402
|
-
assert_eq!(tess.output_format, "text");
|
|
1403
|
-
assert!(!tess.enable_table_detection);
|
|
1404
|
-
assert_eq!(tess.tessedit_char_whitelist, "0123456789");
|
|
1405
|
-
}
|
|
1406
|
-
|
|
1407
|
-
#[test]
|
|
1408
|
-
fn test_ocr_config_without_tesseract_config() {
|
|
1409
|
-
let dir = tempdir().unwrap();
|
|
1410
|
-
let config_path = dir.path().join("kreuzberg.toml");
|
|
1411
|
-
|
|
1412
|
-
fs::write(
|
|
1413
|
-
&config_path,
|
|
1414
|
-
r#"
|
|
1415
|
-
[ocr]
|
|
1416
|
-
backend = "easyocr"
|
|
1417
|
-
language = "eng"
|
|
1418
|
-
"#,
|
|
1419
|
-
)
|
|
1420
|
-
.unwrap();
|
|
1421
|
-
|
|
1422
|
-
let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
|
|
1423
|
-
assert!(config.ocr.is_some());
|
|
1424
|
-
let ocr = config.ocr.unwrap();
|
|
1425
|
-
assert_eq!(ocr.backend, "easyocr");
|
|
1426
|
-
assert_eq!(ocr.language, "eng");
|
|
1427
|
-
assert!(ocr.tesseract_config.is_none());
|
|
1428
|
-
}
|
|
1429
|
-
|
|
1430
|
-
#[test]
|
|
1431
|
-
fn test_tesseract_config_defaults() {
|
|
1432
|
-
let tess = crate::types::TesseractConfig::default();
|
|
1433
|
-
assert_eq!(tess.language, "eng");
|
|
1434
|
-
assert_eq!(tess.psm, 3);
|
|
1435
|
-
assert_eq!(tess.output_format, "markdown");
|
|
1436
|
-
assert!(tess.enable_table_detection);
|
|
1437
|
-
assert_eq!(tess.table_min_confidence, 0.0);
|
|
1438
|
-
assert_eq!(tess.table_column_threshold, 50);
|
|
1439
|
-
assert_eq!(tess.table_row_threshold_ratio, 0.5);
|
|
1440
|
-
assert!(tess.use_cache);
|
|
1441
|
-
assert!(tess.classify_use_pre_adapted_templates);
|
|
1442
|
-
assert!(!tess.language_model_ngram_on);
|
|
1443
|
-
assert!(tess.tessedit_dont_blkrej_good_wds);
|
|
1444
|
-
assert!(tess.tessedit_dont_rowrej_good_wds);
|
|
1445
|
-
assert!(tess.tessedit_enable_dict_correction);
|
|
1446
|
-
assert_eq!(tess.tessedit_char_whitelist, "");
|
|
1447
|
-
assert!(tess.tessedit_use_primary_params_model);
|
|
1448
|
-
assert!(tess.textord_space_size_is_variable);
|
|
1449
|
-
assert!(!tess.thresholding_method);
|
|
1450
|
-
}
|
|
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
|
-
}
|