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.
Files changed (321) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +0 -6
  3. data/.rubocop.yaml +534 -1
  4. data/Gemfile +2 -1
  5. data/Gemfile.lock +28 -116
  6. data/README.md +269 -629
  7. data/Rakefile +0 -9
  8. data/Steepfile +4 -8
  9. data/examples/async_patterns.rb +58 -1
  10. data/ext/kreuzberg_rb/extconf.rb +5 -35
  11. data/ext/kreuzberg_rb/native/Cargo.toml +16 -55
  12. data/ext/kreuzberg_rb/native/build.rs +14 -12
  13. data/ext/kreuzberg_rb/native/include/ieeefp.h +1 -1
  14. data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +1 -1
  15. data/ext/kreuzberg_rb/native/include/strings.h +2 -2
  16. data/ext/kreuzberg_rb/native/include/unistd.h +1 -1
  17. data/ext/kreuzberg_rb/native/src/lib.rs +34 -897
  18. data/extconf.rb +6 -38
  19. data/kreuzberg.gemspec +20 -114
  20. data/lib/kreuzberg/api_proxy.rb +18 -2
  21. data/lib/kreuzberg/cache_api.rb +0 -22
  22. data/lib/kreuzberg/cli.rb +10 -2
  23. data/lib/kreuzberg/cli_proxy.rb +10 -0
  24. data/lib/kreuzberg/config.rb +22 -274
  25. data/lib/kreuzberg/errors.rb +7 -73
  26. data/lib/kreuzberg/extraction_api.rb +8 -237
  27. data/lib/kreuzberg/mcp_proxy.rb +11 -2
  28. data/lib/kreuzberg/ocr_backend_protocol.rb +73 -0
  29. data/lib/kreuzberg/post_processor_protocol.rb +71 -0
  30. data/lib/kreuzberg/result.rb +33 -151
  31. data/lib/kreuzberg/setup_lib_path.rb +2 -22
  32. data/lib/kreuzberg/validator_protocol.rb +73 -0
  33. data/lib/kreuzberg/version.rb +1 -1
  34. data/lib/kreuzberg.rb +13 -27
  35. data/pkg/kreuzberg-4.0.0.rc1.gem +0 -0
  36. data/sig/kreuzberg.rbs +12 -105
  37. data/spec/binding/cache_spec.rb +22 -22
  38. data/spec/binding/cli_proxy_spec.rb +4 -2
  39. data/spec/binding/cli_spec.rb +11 -12
  40. data/spec/binding/config_spec.rb +0 -74
  41. data/spec/binding/config_validation_spec.rb +6 -100
  42. data/spec/binding/error_handling_spec.rb +97 -283
  43. data/spec/binding/plugins/ocr_backend_spec.rb +8 -8
  44. data/spec/binding/plugins/postprocessor_spec.rb +11 -11
  45. data/spec/binding/plugins/validator_spec.rb +13 -12
  46. data/spec/examples.txt +104 -0
  47. data/spec/fixtures/config.toml +1 -0
  48. data/spec/fixtures/config.yaml +1 -0
  49. data/spec/fixtures/invalid_config.toml +1 -0
  50. data/spec/smoke/package_spec.rb +3 -2
  51. data/spec/spec_helper.rb +3 -1
  52. data/vendor/kreuzberg/Cargo.toml +67 -192
  53. data/vendor/kreuzberg/README.md +9 -97
  54. data/vendor/kreuzberg/build.rs +194 -516
  55. data/vendor/kreuzberg/src/api/handlers.rs +9 -130
  56. data/vendor/kreuzberg/src/api/mod.rs +3 -18
  57. data/vendor/kreuzberg/src/api/server.rs +71 -236
  58. data/vendor/kreuzberg/src/api/types.rs +7 -43
  59. data/vendor/kreuzberg/src/bin/profile_extract.rs +455 -0
  60. data/vendor/kreuzberg/src/cache/mod.rs +3 -27
  61. data/vendor/kreuzberg/src/chunking/mod.rs +79 -1705
  62. data/vendor/kreuzberg/src/core/batch_mode.rs +0 -60
  63. data/vendor/kreuzberg/src/core/config.rs +23 -905
  64. data/vendor/kreuzberg/src/core/extractor.rs +106 -403
  65. data/vendor/kreuzberg/src/core/io.rs +2 -4
  66. data/vendor/kreuzberg/src/core/mime.rs +12 -2
  67. data/vendor/kreuzberg/src/core/mod.rs +3 -22
  68. data/vendor/kreuzberg/src/core/pipeline.rs +78 -395
  69. data/vendor/kreuzberg/src/embeddings.rs +21 -169
  70. data/vendor/kreuzberg/src/error.rs +2 -2
  71. data/vendor/kreuzberg/src/extraction/archive.rs +31 -36
  72. data/vendor/kreuzberg/src/extraction/docx.rs +1 -365
  73. data/vendor/kreuzberg/src/extraction/email.rs +11 -12
  74. data/vendor/kreuzberg/src/extraction/excel.rs +129 -138
  75. data/vendor/kreuzberg/src/extraction/html.rs +170 -1447
  76. data/vendor/kreuzberg/src/extraction/image.rs +14 -138
  77. data/vendor/kreuzberg/src/extraction/libreoffice.rs +3 -13
  78. data/vendor/kreuzberg/src/extraction/mod.rs +5 -21
  79. data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +0 -2
  80. data/vendor/kreuzberg/src/extraction/pandoc/batch.rs +275 -0
  81. data/vendor/kreuzberg/src/extraction/pandoc/mime_types.rs +178 -0
  82. data/vendor/kreuzberg/src/extraction/pandoc/mod.rs +491 -0
  83. data/vendor/kreuzberg/src/extraction/pandoc/server.rs +496 -0
  84. data/vendor/kreuzberg/src/extraction/pandoc/subprocess.rs +1188 -0
  85. data/vendor/kreuzberg/src/extraction/pandoc/version.rs +162 -0
  86. data/vendor/kreuzberg/src/extraction/pptx.rs +94 -196
  87. data/vendor/kreuzberg/src/extraction/structured.rs +4 -5
  88. data/vendor/kreuzberg/src/extraction/table.rs +1 -2
  89. data/vendor/kreuzberg/src/extraction/text.rs +10 -18
  90. data/vendor/kreuzberg/src/extractors/archive.rs +0 -22
  91. data/vendor/kreuzberg/src/extractors/docx.rs +148 -69
  92. data/vendor/kreuzberg/src/extractors/email.rs +9 -37
  93. data/vendor/kreuzberg/src/extractors/excel.rs +40 -81
  94. data/vendor/kreuzberg/src/extractors/html.rs +173 -182
  95. data/vendor/kreuzberg/src/extractors/image.rs +8 -32
  96. data/vendor/kreuzberg/src/extractors/mod.rs +10 -171
  97. data/vendor/kreuzberg/src/extractors/pandoc.rs +201 -0
  98. data/vendor/kreuzberg/src/extractors/pdf.rs +64 -329
  99. data/vendor/kreuzberg/src/extractors/pptx.rs +34 -79
  100. data/vendor/kreuzberg/src/extractors/structured.rs +0 -16
  101. data/vendor/kreuzberg/src/extractors/text.rs +7 -30
  102. data/vendor/kreuzberg/src/extractors/xml.rs +8 -27
  103. data/vendor/kreuzberg/src/keywords/processor.rs +1 -9
  104. data/vendor/kreuzberg/src/keywords/rake.rs +1 -0
  105. data/vendor/kreuzberg/src/language_detection/mod.rs +51 -94
  106. data/vendor/kreuzberg/src/lib.rs +5 -17
  107. data/vendor/kreuzberg/src/mcp/mod.rs +1 -4
  108. data/vendor/kreuzberg/src/mcp/server.rs +21 -145
  109. data/vendor/kreuzberg/src/ocr/mod.rs +0 -2
  110. data/vendor/kreuzberg/src/ocr/processor.rs +8 -19
  111. data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +0 -2
  112. data/vendor/kreuzberg/src/pdf/error.rs +1 -93
  113. data/vendor/kreuzberg/src/pdf/metadata.rs +100 -263
  114. data/vendor/kreuzberg/src/pdf/mod.rs +2 -33
  115. data/vendor/kreuzberg/src/pdf/rendering.rs +12 -12
  116. data/vendor/kreuzberg/src/pdf/table.rs +64 -61
  117. data/vendor/kreuzberg/src/pdf/text.rs +24 -416
  118. data/vendor/kreuzberg/src/plugins/extractor.rs +8 -40
  119. data/vendor/kreuzberg/src/plugins/mod.rs +0 -3
  120. data/vendor/kreuzberg/src/plugins/ocr.rs +14 -22
  121. data/vendor/kreuzberg/src/plugins/processor.rs +1 -10
  122. data/vendor/kreuzberg/src/plugins/registry.rs +0 -15
  123. data/vendor/kreuzberg/src/plugins/validator.rs +8 -20
  124. data/vendor/kreuzberg/src/stopwords/mod.rs +2 -2
  125. data/vendor/kreuzberg/src/text/mod.rs +0 -8
  126. data/vendor/kreuzberg/src/text/quality.rs +15 -28
  127. data/vendor/kreuzberg/src/text/string_utils.rs +10 -22
  128. data/vendor/kreuzberg/src/text/token_reduction/core.rs +50 -86
  129. data/vendor/kreuzberg/src/text/token_reduction/filters.rs +16 -37
  130. data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +1 -2
  131. data/vendor/kreuzberg/src/types.rs +67 -907
  132. data/vendor/kreuzberg/src/utils/mod.rs +0 -14
  133. data/vendor/kreuzberg/src/utils/quality.rs +3 -12
  134. data/vendor/kreuzberg/tests/api_tests.rs +0 -506
  135. data/vendor/kreuzberg/tests/archive_integration.rs +0 -2
  136. data/vendor/kreuzberg/tests/batch_orchestration.rs +12 -57
  137. data/vendor/kreuzberg/tests/batch_processing.rs +8 -32
  138. data/vendor/kreuzberg/tests/chunking_offset_demo.rs +92 -0
  139. data/vendor/kreuzberg/tests/concurrency_stress.rs +8 -40
  140. data/vendor/kreuzberg/tests/config_features.rs +1 -33
  141. data/vendor/kreuzberg/tests/config_loading_tests.rs +39 -16
  142. data/vendor/kreuzberg/tests/core_integration.rs +9 -35
  143. data/vendor/kreuzberg/tests/csv_integration.rs +81 -71
  144. data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +25 -23
  145. data/vendor/kreuzberg/tests/email_integration.rs +1 -3
  146. data/vendor/kreuzberg/tests/error_handling.rs +34 -43
  147. data/vendor/kreuzberg/tests/format_integration.rs +1 -7
  148. data/vendor/kreuzberg/tests/helpers/mod.rs +0 -60
  149. data/vendor/kreuzberg/tests/image_integration.rs +0 -2
  150. data/vendor/kreuzberg/tests/mime_detection.rs +16 -17
  151. data/vendor/kreuzberg/tests/ocr_configuration.rs +0 -4
  152. data/vendor/kreuzberg/tests/ocr_errors.rs +0 -22
  153. data/vendor/kreuzberg/tests/ocr_quality.rs +0 -2
  154. data/vendor/kreuzberg/tests/pandoc_integration.rs +503 -0
  155. data/vendor/kreuzberg/tests/pdf_integration.rs +0 -2
  156. data/vendor/kreuzberg/tests/pipeline_integration.rs +2 -36
  157. data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +0 -5
  158. data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +1 -17
  159. data/vendor/kreuzberg/tests/plugin_system.rs +0 -6
  160. data/vendor/kreuzberg/tests/registry_integration_tests.rs +22 -2
  161. data/vendor/kreuzberg/tests/security_validation.rs +1 -13
  162. data/vendor/kreuzberg/tests/test_fastembed.rs +23 -45
  163. metadata +25 -171
  164. data/.rubocop.yml +0 -543
  165. data/ext/kreuzberg_rb/native/.cargo/config.toml +0 -23
  166. data/ext/kreuzberg_rb/native/Cargo.lock +0 -7619
  167. data/lib/kreuzberg/error_context.rb +0 -136
  168. data/lib/kreuzberg/types.rb +0 -170
  169. data/lib/libpdfium.so +0 -0
  170. data/spec/binding/async_operations_spec.rb +0 -473
  171. data/spec/binding/batch_operations_spec.rb +0 -595
  172. data/spec/binding/batch_spec.rb +0 -359
  173. data/spec/binding/config_result_spec.rb +0 -377
  174. data/spec/binding/embeddings_spec.rb +0 -816
  175. data/spec/binding/error_recovery_spec.rb +0 -488
  176. data/spec/binding/font_config_spec.rb +0 -220
  177. data/spec/binding/images_spec.rb +0 -738
  178. data/spec/binding/keywords_extraction_spec.rb +0 -600
  179. data/spec/binding/metadata_types_spec.rb +0 -1228
  180. data/spec/binding/pages_extraction_spec.rb +0 -471
  181. data/spec/binding/tables_spec.rb +0 -641
  182. data/spec/unit/config/chunking_config_spec.rb +0 -213
  183. data/spec/unit/config/embedding_config_spec.rb +0 -343
  184. data/spec/unit/config/extraction_config_spec.rb +0 -438
  185. data/spec/unit/config/font_config_spec.rb +0 -285
  186. data/spec/unit/config/hierarchy_config_spec.rb +0 -314
  187. data/spec/unit/config/image_extraction_config_spec.rb +0 -209
  188. data/spec/unit/config/image_preprocessing_config_spec.rb +0 -249
  189. data/spec/unit/config/keyword_config_spec.rb +0 -229
  190. data/spec/unit/config/language_detection_config_spec.rb +0 -258
  191. data/spec/unit/config/ocr_config_spec.rb +0 -171
  192. data/spec/unit/config/page_config_spec.rb +0 -221
  193. data/spec/unit/config/pdf_config_spec.rb +0 -267
  194. data/spec/unit/config/postprocessor_config_spec.rb +0 -290
  195. data/spec/unit/config/tesseract_config_spec.rb +0 -181
  196. data/spec/unit/config/token_reduction_config_spec.rb +0 -251
  197. data/test/metadata_types_test.rb +0 -959
  198. data/vendor/Cargo.toml +0 -61
  199. data/vendor/kreuzberg/examples/bench_fixes.rs +0 -71
  200. data/vendor/kreuzberg/examples/test_pdfium_fork.rs +0 -62
  201. data/vendor/kreuzberg/src/chunking/processor.rs +0 -219
  202. data/vendor/kreuzberg/src/core/batch_optimizations.rs +0 -385
  203. data/vendor/kreuzberg/src/core/config_validation.rs +0 -949
  204. data/vendor/kreuzberg/src/core/formats.rs +0 -235
  205. data/vendor/kreuzberg/src/core/server_config.rs +0 -1220
  206. data/vendor/kreuzberg/src/extraction/capacity.rs +0 -263
  207. data/vendor/kreuzberg/src/extraction/markdown.rs +0 -216
  208. data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +0 -284
  209. data/vendor/kreuzberg/src/extractors/bibtex.rs +0 -470
  210. data/vendor/kreuzberg/src/extractors/docbook.rs +0 -504
  211. data/vendor/kreuzberg/src/extractors/epub.rs +0 -696
  212. data/vendor/kreuzberg/src/extractors/fictionbook.rs +0 -492
  213. data/vendor/kreuzberg/src/extractors/jats.rs +0 -1054
  214. data/vendor/kreuzberg/src/extractors/jupyter.rs +0 -368
  215. data/vendor/kreuzberg/src/extractors/latex.rs +0 -653
  216. data/vendor/kreuzberg/src/extractors/markdown.rs +0 -701
  217. data/vendor/kreuzberg/src/extractors/odt.rs +0 -628
  218. data/vendor/kreuzberg/src/extractors/opml.rs +0 -635
  219. data/vendor/kreuzberg/src/extractors/orgmode.rs +0 -529
  220. data/vendor/kreuzberg/src/extractors/rst.rs +0 -577
  221. data/vendor/kreuzberg/src/extractors/rtf.rs +0 -809
  222. data/vendor/kreuzberg/src/extractors/security.rs +0 -484
  223. data/vendor/kreuzberg/src/extractors/security_tests.rs +0 -367
  224. data/vendor/kreuzberg/src/extractors/typst.rs +0 -651
  225. data/vendor/kreuzberg/src/language_detection/processor.rs +0 -218
  226. data/vendor/kreuzberg/src/ocr/language_registry.rs +0 -520
  227. data/vendor/kreuzberg/src/panic_context.rs +0 -154
  228. data/vendor/kreuzberg/src/pdf/bindings.rs +0 -306
  229. data/vendor/kreuzberg/src/pdf/bundled.rs +0 -408
  230. data/vendor/kreuzberg/src/pdf/fonts.rs +0 -358
  231. data/vendor/kreuzberg/src/pdf/hierarchy.rs +0 -903
  232. data/vendor/kreuzberg/src/text/quality_processor.rs +0 -231
  233. data/vendor/kreuzberg/src/text/utf8_validation.rs +0 -193
  234. data/vendor/kreuzberg/src/utils/pool.rs +0 -503
  235. data/vendor/kreuzberg/src/utils/pool_sizing.rs +0 -364
  236. data/vendor/kreuzberg/src/utils/string_pool.rs +0 -761
  237. data/vendor/kreuzberg/tests/api_embed.rs +0 -360
  238. data/vendor/kreuzberg/tests/api_extract_multipart.rs +0 -52
  239. data/vendor/kreuzberg/tests/api_large_pdf_extraction.rs +0 -471
  240. data/vendor/kreuzberg/tests/api_large_pdf_extraction_diagnostics.rs +0 -289
  241. data/vendor/kreuzberg/tests/batch_pooling_benchmark.rs +0 -154
  242. data/vendor/kreuzberg/tests/bibtex_parity_test.rs +0 -421
  243. data/vendor/kreuzberg/tests/config_integration_test.rs +0 -753
  244. data/vendor/kreuzberg/tests/data/hierarchy_ground_truth.json +0 -294
  245. data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +0 -500
  246. data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +0 -370
  247. data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +0 -275
  248. data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +0 -228
  249. data/vendor/kreuzberg/tests/html_table_test.rs +0 -551
  250. data/vendor/kreuzberg/tests/instrumentation_test.rs +0 -139
  251. data/vendor/kreuzberg/tests/jats_extractor_tests.rs +0 -639
  252. data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +0 -704
  253. data/vendor/kreuzberg/tests/latex_extractor_tests.rs +0 -496
  254. data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +0 -490
  255. data/vendor/kreuzberg/tests/ocr_language_registry.rs +0 -191
  256. data/vendor/kreuzberg/tests/odt_extractor_tests.rs +0 -674
  257. data/vendor/kreuzberg/tests/opml_extractor_tests.rs +0 -616
  258. data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +0 -822
  259. data/vendor/kreuzberg/tests/page_markers.rs +0 -297
  260. data/vendor/kreuzberg/tests/pdf_hierarchy_detection.rs +0 -301
  261. data/vendor/kreuzberg/tests/pdf_hierarchy_quality.rs +0 -589
  262. data/vendor/kreuzberg/tests/pdf_ocr_triggering.rs +0 -301
  263. data/vendor/kreuzberg/tests/pdf_text_merging.rs +0 -475
  264. data/vendor/kreuzberg/tests/pdfium_linking.rs +0 -340
  265. data/vendor/kreuzberg/tests/rst_extractor_tests.rs +0 -694
  266. data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +0 -775
  267. data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +0 -1260
  268. data/vendor/kreuzberg/tests/typst_extractor_tests.rs +0 -648
  269. data/vendor/kreuzberg-ffi/Cargo.toml +0 -67
  270. data/vendor/kreuzberg-ffi/README.md +0 -851
  271. data/vendor/kreuzberg-ffi/benches/result_view_benchmark.rs +0 -227
  272. data/vendor/kreuzberg-ffi/build.rs +0 -168
  273. data/vendor/kreuzberg-ffi/cbindgen.toml +0 -37
  274. data/vendor/kreuzberg-ffi/kreuzberg-ffi.pc.in +0 -12
  275. data/vendor/kreuzberg-ffi/kreuzberg.h +0 -3012
  276. data/vendor/kreuzberg-ffi/src/batch_streaming.rs +0 -588
  277. data/vendor/kreuzberg-ffi/src/config.rs +0 -1341
  278. data/vendor/kreuzberg-ffi/src/error.rs +0 -901
  279. data/vendor/kreuzberg-ffi/src/extraction.rs +0 -555
  280. data/vendor/kreuzberg-ffi/src/helpers.rs +0 -879
  281. data/vendor/kreuzberg-ffi/src/lib.rs +0 -977
  282. data/vendor/kreuzberg-ffi/src/memory.rs +0 -493
  283. data/vendor/kreuzberg-ffi/src/mime.rs +0 -329
  284. data/vendor/kreuzberg-ffi/src/panic_shield.rs +0 -265
  285. data/vendor/kreuzberg-ffi/src/plugins/document_extractor.rs +0 -442
  286. data/vendor/kreuzberg-ffi/src/plugins/mod.rs +0 -14
  287. data/vendor/kreuzberg-ffi/src/plugins/ocr_backend.rs +0 -628
  288. data/vendor/kreuzberg-ffi/src/plugins/post_processor.rs +0 -438
  289. data/vendor/kreuzberg-ffi/src/plugins/validator.rs +0 -329
  290. data/vendor/kreuzberg-ffi/src/result.rs +0 -510
  291. data/vendor/kreuzberg-ffi/src/result_pool.rs +0 -639
  292. data/vendor/kreuzberg-ffi/src/result_view.rs +0 -773
  293. data/vendor/kreuzberg-ffi/src/string_intern.rs +0 -568
  294. data/vendor/kreuzberg-ffi/src/types.rs +0 -363
  295. data/vendor/kreuzberg-ffi/src/util.rs +0 -210
  296. data/vendor/kreuzberg-ffi/src/validation.rs +0 -848
  297. data/vendor/kreuzberg-ffi/tests.disabled/README.md +0 -48
  298. data/vendor/kreuzberg-ffi/tests.disabled/config_loading_tests.rs +0 -299
  299. data/vendor/kreuzberg-ffi/tests.disabled/config_tests.rs +0 -346
  300. data/vendor/kreuzberg-ffi/tests.disabled/extractor_tests.rs +0 -232
  301. data/vendor/kreuzberg-ffi/tests.disabled/plugin_registration_tests.rs +0 -470
  302. data/vendor/kreuzberg-tesseract/.commitlintrc.json +0 -13
  303. data/vendor/kreuzberg-tesseract/.crate-ignore +0 -2
  304. data/vendor/kreuzberg-tesseract/Cargo.lock +0 -2933
  305. data/vendor/kreuzberg-tesseract/Cargo.toml +0 -57
  306. data/vendor/kreuzberg-tesseract/LICENSE +0 -22
  307. data/vendor/kreuzberg-tesseract/README.md +0 -399
  308. data/vendor/kreuzberg-tesseract/build.rs +0 -1127
  309. data/vendor/kreuzberg-tesseract/patches/README.md +0 -71
  310. data/vendor/kreuzberg-tesseract/patches/tesseract.diff +0 -199
  311. data/vendor/kreuzberg-tesseract/src/api.rs +0 -1371
  312. data/vendor/kreuzberg-tesseract/src/choice_iterator.rs +0 -77
  313. data/vendor/kreuzberg-tesseract/src/enums.rs +0 -297
  314. data/vendor/kreuzberg-tesseract/src/error.rs +0 -81
  315. data/vendor/kreuzberg-tesseract/src/lib.rs +0 -145
  316. data/vendor/kreuzberg-tesseract/src/monitor.rs +0 -57
  317. data/vendor/kreuzberg-tesseract/src/mutable_iterator.rs +0 -197
  318. data/vendor/kreuzberg-tesseract/src/page_iterator.rs +0 -253
  319. data/vendor/kreuzberg-tesseract/src/result_iterator.rs +0 -286
  320. data/vendor/kreuzberg-tesseract/src/result_renderer.rs +0 -183
  321. 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::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
- }
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 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))
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
- 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)
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 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))
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
- 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)
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 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))
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
- 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)
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
- 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)
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
  }