kreuzberg 4.0.0.rc2 → 4.0.1

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 (446) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +14 -14
  3. data/.rspec +3 -3
  4. data/.rubocop.yaml +1 -1
  5. data/.rubocop.yml +543 -538
  6. data/Gemfile +8 -8
  7. data/Gemfile.lock +194 -6
  8. data/README.md +391 -426
  9. data/Rakefile +34 -25
  10. data/Steepfile +51 -47
  11. data/examples/async_patterns.rb +283 -341
  12. data/ext/kreuzberg_rb/extconf.rb +65 -45
  13. data/ext/kreuzberg_rb/native/.cargo/config.toml +23 -0
  14. data/ext/kreuzberg_rb/native/Cargo.lock +7619 -6535
  15. data/ext/kreuzberg_rb/native/Cargo.toml +75 -44
  16. data/ext/kreuzberg_rb/native/README.md +425 -425
  17. data/ext/kreuzberg_rb/native/build.rs +15 -15
  18. data/ext/kreuzberg_rb/native/include/ieeefp.h +11 -11
  19. data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +14 -14
  20. data/ext/kreuzberg_rb/native/include/strings.h +20 -20
  21. data/ext/kreuzberg_rb/native/include/unistd.h +47 -47
  22. data/ext/kreuzberg_rb/native/src/lib.rs +3802 -2998
  23. data/extconf.rb +60 -28
  24. data/kreuzberg.gemspec +199 -148
  25. data/lib/kreuzberg/api_proxy.rb +126 -142
  26. data/lib/kreuzberg/cache_api.rb +67 -46
  27. data/lib/kreuzberg/cli.rb +47 -55
  28. data/lib/kreuzberg/cli_proxy.rb +117 -127
  29. data/lib/kreuzberg/config.rb +936 -691
  30. data/lib/kreuzberg/error_context.rb +136 -32
  31. data/lib/kreuzberg/errors.rb +116 -118
  32. data/lib/kreuzberg/extraction_api.rb +313 -85
  33. data/lib/kreuzberg/mcp_proxy.rb +177 -186
  34. data/lib/kreuzberg/ocr_backend_protocol.rb +40 -113
  35. data/lib/kreuzberg/post_processor_protocol.rb +15 -86
  36. data/lib/kreuzberg/result.rb +334 -216
  37. data/lib/kreuzberg/setup_lib_path.rb +99 -80
  38. data/lib/kreuzberg/types.rb +170 -0
  39. data/lib/kreuzberg/validator_protocol.rb +16 -89
  40. data/lib/kreuzberg/version.rb +5 -5
  41. data/lib/kreuzberg.rb +96 -103
  42. data/lib/libpdfium.so +0 -0
  43. data/sig/kreuzberg/internal.rbs +184 -184
  44. data/sig/kreuzberg.rbs +561 -520
  45. data/spec/binding/async_operations_spec.rb +473 -0
  46. data/spec/binding/batch_operations_spec.rb +595 -0
  47. data/spec/binding/batch_spec.rb +359 -0
  48. data/spec/binding/cache_spec.rb +227 -227
  49. data/spec/binding/cli_proxy_spec.rb +85 -85
  50. data/spec/binding/cli_spec.rb +55 -55
  51. data/spec/binding/config_result_spec.rb +377 -0
  52. data/spec/binding/config_spec.rb +419 -345
  53. data/spec/binding/config_validation_spec.rb +377 -283
  54. data/spec/binding/embeddings_spec.rb +816 -0
  55. data/spec/binding/error_handling_spec.rb +399 -213
  56. data/spec/binding/error_recovery_spec.rb +488 -0
  57. data/spec/binding/errors_spec.rb +66 -66
  58. data/spec/binding/font_config_spec.rb +220 -0
  59. data/spec/binding/images_spec.rb +738 -0
  60. data/spec/binding/keywords_extraction_spec.rb +600 -0
  61. data/spec/binding/metadata_types_spec.rb +1228 -0
  62. data/spec/binding/pages_extraction_spec.rb +471 -0
  63. data/spec/binding/plugins/ocr_backend_spec.rb +307 -307
  64. data/spec/binding/plugins/postprocessor_spec.rb +269 -269
  65. data/spec/binding/plugins/validator_spec.rb +273 -274
  66. data/spec/binding/tables_spec.rb +641 -0
  67. data/spec/fixtures/config.toml +38 -39
  68. data/spec/fixtures/config.yaml +41 -41
  69. data/spec/fixtures/invalid_config.toml +3 -4
  70. data/spec/smoke/package_spec.rb +177 -178
  71. data/spec/spec_helper.rb +40 -42
  72. data/spec/unit/config/chunking_config_spec.rb +213 -0
  73. data/spec/unit/config/embedding_config_spec.rb +343 -0
  74. data/spec/unit/config/extraction_config_spec.rb +438 -0
  75. data/spec/unit/config/font_config_spec.rb +285 -0
  76. data/spec/unit/config/hierarchy_config_spec.rb +314 -0
  77. data/spec/unit/config/image_extraction_config_spec.rb +209 -0
  78. data/spec/unit/config/image_preprocessing_config_spec.rb +249 -0
  79. data/spec/unit/config/keyword_config_spec.rb +229 -0
  80. data/spec/unit/config/language_detection_config_spec.rb +258 -0
  81. data/spec/unit/config/ocr_config_spec.rb +171 -0
  82. data/spec/unit/config/page_config_spec.rb +221 -0
  83. data/spec/unit/config/pdf_config_spec.rb +267 -0
  84. data/spec/unit/config/postprocessor_config_spec.rb +290 -0
  85. data/spec/unit/config/tesseract_config_spec.rb +181 -0
  86. data/spec/unit/config/token_reduction_config_spec.rb +251 -0
  87. data/test/metadata_types_test.rb +959 -0
  88. data/vendor/Cargo.toml +61 -0
  89. data/vendor/kreuzberg/Cargo.toml +259 -204
  90. data/vendor/kreuzberg/README.md +263 -175
  91. data/vendor/kreuzberg/build.rs +782 -474
  92. data/vendor/kreuzberg/examples/bench_fixes.rs +71 -0
  93. data/vendor/kreuzberg/examples/test_pdfium_fork.rs +62 -0
  94. data/vendor/kreuzberg/src/api/error.rs +81 -81
  95. data/vendor/kreuzberg/src/api/handlers.rs +320 -199
  96. data/vendor/kreuzberg/src/api/mod.rs +94 -79
  97. data/vendor/kreuzberg/src/api/server.rs +518 -353
  98. data/vendor/kreuzberg/src/api/types.rs +206 -170
  99. data/vendor/kreuzberg/src/cache/mod.rs +1167 -1167
  100. data/vendor/kreuzberg/src/chunking/mod.rs +2303 -677
  101. data/vendor/kreuzberg/src/chunking/processor.rs +219 -0
  102. data/vendor/kreuzberg/src/core/batch_mode.rs +95 -95
  103. data/vendor/kreuzberg/src/core/batch_optimizations.rs +385 -0
  104. data/vendor/kreuzberg/src/core/config.rs +1914 -1032
  105. data/vendor/kreuzberg/src/core/config_validation.rs +949 -0
  106. data/vendor/kreuzberg/src/core/extractor.rs +1200 -1024
  107. data/vendor/kreuzberg/src/core/formats.rs +235 -0
  108. data/vendor/kreuzberg/src/core/io.rs +329 -329
  109. data/vendor/kreuzberg/src/core/mime.rs +605 -605
  110. data/vendor/kreuzberg/src/core/mod.rs +61 -45
  111. data/vendor/kreuzberg/src/core/pipeline.rs +1223 -984
  112. data/vendor/kreuzberg/src/core/server_config.rs +1220 -0
  113. data/vendor/kreuzberg/src/embeddings.rs +471 -432
  114. data/vendor/kreuzberg/src/error.rs +431 -431
  115. data/vendor/kreuzberg/src/extraction/archive.rs +959 -954
  116. data/vendor/kreuzberg/src/extraction/capacity.rs +263 -0
  117. data/vendor/kreuzberg/src/extraction/docx.rs +404 -40
  118. data/vendor/kreuzberg/src/extraction/email.rs +855 -854
  119. data/vendor/kreuzberg/src/extraction/excel.rs +697 -688
  120. data/vendor/kreuzberg/src/extraction/html.rs +1830 -553
  121. data/vendor/kreuzberg/src/extraction/image.rs +492 -368
  122. data/vendor/kreuzberg/src/extraction/libreoffice.rs +574 -563
  123. data/vendor/kreuzberg/src/extraction/markdown.rs +216 -213
  124. data/vendor/kreuzberg/src/extraction/mod.rs +93 -81
  125. data/vendor/kreuzberg/src/extraction/office_metadata/app_properties.rs +398 -398
  126. data/vendor/kreuzberg/src/extraction/office_metadata/core_properties.rs +247 -247
  127. data/vendor/kreuzberg/src/extraction/office_metadata/custom_properties.rs +240 -240
  128. data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +130 -130
  129. data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +284 -287
  130. data/vendor/kreuzberg/src/extraction/pptx.rs +3102 -3000
  131. data/vendor/kreuzberg/src/extraction/structured.rs +491 -490
  132. data/vendor/kreuzberg/src/extraction/table.rs +329 -328
  133. data/vendor/kreuzberg/src/extraction/text.rs +277 -269
  134. data/vendor/kreuzberg/src/extraction/xml.rs +333 -333
  135. data/vendor/kreuzberg/src/extractors/archive.rs +447 -446
  136. data/vendor/kreuzberg/src/extractors/bibtex.rs +470 -469
  137. data/vendor/kreuzberg/src/extractors/docbook.rs +504 -502
  138. data/vendor/kreuzberg/src/extractors/docx.rs +400 -367
  139. data/vendor/kreuzberg/src/extractors/email.rs +157 -143
  140. data/vendor/kreuzberg/src/extractors/epub.rs +696 -707
  141. data/vendor/kreuzberg/src/extractors/excel.rs +385 -343
  142. data/vendor/kreuzberg/src/extractors/fictionbook.rs +492 -491
  143. data/vendor/kreuzberg/src/extractors/html.rs +419 -393
  144. data/vendor/kreuzberg/src/extractors/image.rs +219 -198
  145. data/vendor/kreuzberg/src/extractors/jats.rs +1054 -1051
  146. data/vendor/kreuzberg/src/extractors/jupyter.rs +368 -367
  147. data/vendor/kreuzberg/src/extractors/latex.rs +653 -652
  148. data/vendor/kreuzberg/src/extractors/markdown.rs +701 -700
  149. data/vendor/kreuzberg/src/extractors/mod.rs +429 -365
  150. data/vendor/kreuzberg/src/extractors/odt.rs +628 -628
  151. data/vendor/kreuzberg/src/extractors/opml.rs +635 -634
  152. data/vendor/kreuzberg/src/extractors/orgmode.rs +529 -528
  153. data/vendor/kreuzberg/src/extractors/pdf.rs +761 -493
  154. data/vendor/kreuzberg/src/extractors/pptx.rs +279 -248
  155. data/vendor/kreuzberg/src/extractors/rst.rs +577 -576
  156. data/vendor/kreuzberg/src/extractors/rtf.rs +809 -810
  157. data/vendor/kreuzberg/src/extractors/security.rs +484 -484
  158. data/vendor/kreuzberg/src/extractors/security_tests.rs +367 -367
  159. data/vendor/kreuzberg/src/extractors/structured.rs +142 -140
  160. data/vendor/kreuzberg/src/extractors/text.rs +265 -260
  161. data/vendor/kreuzberg/src/extractors/typst.rs +651 -650
  162. data/vendor/kreuzberg/src/extractors/xml.rs +147 -135
  163. data/vendor/kreuzberg/src/image/dpi.rs +164 -164
  164. data/vendor/kreuzberg/src/image/mod.rs +6 -6
  165. data/vendor/kreuzberg/src/image/preprocessing.rs +417 -417
  166. data/vendor/kreuzberg/src/image/resize.rs +89 -89
  167. data/vendor/kreuzberg/src/keywords/config.rs +154 -154
  168. data/vendor/kreuzberg/src/keywords/mod.rs +237 -237
  169. data/vendor/kreuzberg/src/keywords/processor.rs +275 -267
  170. data/vendor/kreuzberg/src/keywords/rake.rs +293 -293
  171. data/vendor/kreuzberg/src/keywords/types.rs +68 -68
  172. data/vendor/kreuzberg/src/keywords/yake.rs +163 -163
  173. data/vendor/kreuzberg/src/language_detection/mod.rs +985 -942
  174. data/vendor/kreuzberg/src/language_detection/processor.rs +218 -0
  175. data/vendor/kreuzberg/src/lib.rs +114 -105
  176. data/vendor/kreuzberg/src/mcp/mod.rs +35 -32
  177. data/vendor/kreuzberg/src/mcp/server.rs +2090 -1968
  178. data/vendor/kreuzberg/src/ocr/cache.rs +469 -469
  179. data/vendor/kreuzberg/src/ocr/error.rs +37 -37
  180. data/vendor/kreuzberg/src/ocr/hocr.rs +216 -216
  181. data/vendor/kreuzberg/src/ocr/language_registry.rs +520 -0
  182. data/vendor/kreuzberg/src/ocr/mod.rs +60 -58
  183. data/vendor/kreuzberg/src/ocr/processor.rs +858 -863
  184. data/vendor/kreuzberg/src/ocr/table/mod.rs +4 -4
  185. data/vendor/kreuzberg/src/ocr/table/tsv_parser.rs +144 -144
  186. data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +456 -450
  187. data/vendor/kreuzberg/src/ocr/types.rs +393 -393
  188. data/vendor/kreuzberg/src/ocr/utils.rs +47 -47
  189. data/vendor/kreuzberg/src/ocr/validation.rs +206 -206
  190. data/vendor/kreuzberg/src/panic_context.rs +154 -154
  191. data/vendor/kreuzberg/src/pdf/bindings.rs +306 -0
  192. data/vendor/kreuzberg/src/pdf/bundled.rs +408 -0
  193. data/vendor/kreuzberg/src/pdf/error.rs +214 -122
  194. data/vendor/kreuzberg/src/pdf/fonts.rs +358 -0
  195. data/vendor/kreuzberg/src/pdf/hierarchy.rs +903 -0
  196. data/vendor/kreuzberg/src/pdf/images.rs +139 -139
  197. data/vendor/kreuzberg/src/pdf/metadata.rs +509 -346
  198. data/vendor/kreuzberg/src/pdf/mod.rs +81 -50
  199. data/vendor/kreuzberg/src/pdf/rendering.rs +369 -369
  200. data/vendor/kreuzberg/src/pdf/table.rs +417 -393
  201. data/vendor/kreuzberg/src/pdf/text.rs +553 -158
  202. data/vendor/kreuzberg/src/plugins/extractor.rs +1042 -1013
  203. data/vendor/kreuzberg/src/plugins/mod.rs +212 -209
  204. data/vendor/kreuzberg/src/plugins/ocr.rs +637 -620
  205. data/vendor/kreuzberg/src/plugins/processor.rs +650 -642
  206. data/vendor/kreuzberg/src/plugins/registry.rs +1339 -1337
  207. data/vendor/kreuzberg/src/plugins/traits.rs +258 -258
  208. data/vendor/kreuzberg/src/plugins/validator.rs +967 -956
  209. data/vendor/kreuzberg/src/stopwords/mod.rs +1470 -1470
  210. data/vendor/kreuzberg/src/text/mod.rs +27 -19
  211. data/vendor/kreuzberg/src/text/quality.rs +710 -697
  212. data/vendor/kreuzberg/src/text/quality_processor.rs +231 -0
  213. data/vendor/kreuzberg/src/text/string_utils.rs +229 -217
  214. data/vendor/kreuzberg/src/text/token_reduction/cjk_utils.rs +164 -164
  215. data/vendor/kreuzberg/src/text/token_reduction/config.rs +100 -100
  216. data/vendor/kreuzberg/src/text/token_reduction/core.rs +832 -796
  217. data/vendor/kreuzberg/src/text/token_reduction/filters.rs +923 -902
  218. data/vendor/kreuzberg/src/text/token_reduction/mod.rs +160 -160
  219. data/vendor/kreuzberg/src/text/token_reduction/semantic.rs +619 -619
  220. data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +148 -147
  221. data/vendor/kreuzberg/src/text/utf8_validation.rs +193 -0
  222. data/vendor/kreuzberg/src/types.rs +1713 -903
  223. data/vendor/kreuzberg/src/utils/mod.rs +31 -17
  224. data/vendor/kreuzberg/src/utils/pool.rs +503 -0
  225. data/vendor/kreuzberg/src/utils/pool_sizing.rs +364 -0
  226. data/vendor/kreuzberg/src/utils/quality.rs +968 -959
  227. data/vendor/kreuzberg/src/utils/string_pool.rs +761 -0
  228. data/vendor/kreuzberg/src/utils/string_utils.rs +381 -381
  229. data/vendor/kreuzberg/stopwords/af_stopwords.json +53 -53
  230. data/vendor/kreuzberg/stopwords/ar_stopwords.json +482 -482
  231. data/vendor/kreuzberg/stopwords/bg_stopwords.json +261 -261
  232. data/vendor/kreuzberg/stopwords/bn_stopwords.json +400 -400
  233. data/vendor/kreuzberg/stopwords/br_stopwords.json +1205 -1205
  234. data/vendor/kreuzberg/stopwords/ca_stopwords.json +280 -280
  235. data/vendor/kreuzberg/stopwords/cs_stopwords.json +425 -425
  236. data/vendor/kreuzberg/stopwords/da_stopwords.json +172 -172
  237. data/vendor/kreuzberg/stopwords/de_stopwords.json +622 -622
  238. data/vendor/kreuzberg/stopwords/el_stopwords.json +849 -849
  239. data/vendor/kreuzberg/stopwords/en_stopwords.json +1300 -1300
  240. data/vendor/kreuzberg/stopwords/eo_stopwords.json +175 -175
  241. data/vendor/kreuzberg/stopwords/es_stopwords.json +734 -734
  242. data/vendor/kreuzberg/stopwords/et_stopwords.json +37 -37
  243. data/vendor/kreuzberg/stopwords/eu_stopwords.json +100 -100
  244. data/vendor/kreuzberg/stopwords/fa_stopwords.json +801 -801
  245. data/vendor/kreuzberg/stopwords/fi_stopwords.json +849 -849
  246. data/vendor/kreuzberg/stopwords/fr_stopwords.json +693 -693
  247. data/vendor/kreuzberg/stopwords/ga_stopwords.json +111 -111
  248. data/vendor/kreuzberg/stopwords/gl_stopwords.json +162 -162
  249. data/vendor/kreuzberg/stopwords/gu_stopwords.json +226 -226
  250. data/vendor/kreuzberg/stopwords/ha_stopwords.json +41 -41
  251. data/vendor/kreuzberg/stopwords/he_stopwords.json +196 -196
  252. data/vendor/kreuzberg/stopwords/hi_stopwords.json +227 -227
  253. data/vendor/kreuzberg/stopwords/hr_stopwords.json +181 -181
  254. data/vendor/kreuzberg/stopwords/hu_stopwords.json +791 -791
  255. data/vendor/kreuzberg/stopwords/hy_stopwords.json +47 -47
  256. data/vendor/kreuzberg/stopwords/id_stopwords.json +760 -760
  257. data/vendor/kreuzberg/stopwords/it_stopwords.json +634 -634
  258. data/vendor/kreuzberg/stopwords/ja_stopwords.json +136 -136
  259. data/vendor/kreuzberg/stopwords/kn_stopwords.json +84 -84
  260. data/vendor/kreuzberg/stopwords/ko_stopwords.json +681 -681
  261. data/vendor/kreuzberg/stopwords/ku_stopwords.json +64 -64
  262. data/vendor/kreuzberg/stopwords/la_stopwords.json +51 -51
  263. data/vendor/kreuzberg/stopwords/lt_stopwords.json +476 -476
  264. data/vendor/kreuzberg/stopwords/lv_stopwords.json +163 -163
  265. data/vendor/kreuzberg/stopwords/ml_stopwords.json +1 -1
  266. data/vendor/kreuzberg/stopwords/mr_stopwords.json +101 -101
  267. data/vendor/kreuzberg/stopwords/ms_stopwords.json +477 -477
  268. data/vendor/kreuzberg/stopwords/ne_stopwords.json +490 -490
  269. data/vendor/kreuzberg/stopwords/nl_stopwords.json +415 -415
  270. data/vendor/kreuzberg/stopwords/no_stopwords.json +223 -223
  271. data/vendor/kreuzberg/stopwords/pl_stopwords.json +331 -331
  272. data/vendor/kreuzberg/stopwords/pt_stopwords.json +562 -562
  273. data/vendor/kreuzberg/stopwords/ro_stopwords.json +436 -436
  274. data/vendor/kreuzberg/stopwords/ru_stopwords.json +561 -561
  275. data/vendor/kreuzberg/stopwords/si_stopwords.json +193 -193
  276. data/vendor/kreuzberg/stopwords/sk_stopwords.json +420 -420
  277. data/vendor/kreuzberg/stopwords/sl_stopwords.json +448 -448
  278. data/vendor/kreuzberg/stopwords/so_stopwords.json +32 -32
  279. data/vendor/kreuzberg/stopwords/st_stopwords.json +33 -33
  280. data/vendor/kreuzberg/stopwords/sv_stopwords.json +420 -420
  281. data/vendor/kreuzberg/stopwords/sw_stopwords.json +76 -76
  282. data/vendor/kreuzberg/stopwords/ta_stopwords.json +129 -129
  283. data/vendor/kreuzberg/stopwords/te_stopwords.json +54 -54
  284. data/vendor/kreuzberg/stopwords/th_stopwords.json +118 -118
  285. data/vendor/kreuzberg/stopwords/tl_stopwords.json +149 -149
  286. data/vendor/kreuzberg/stopwords/tr_stopwords.json +506 -506
  287. data/vendor/kreuzberg/stopwords/uk_stopwords.json +75 -75
  288. data/vendor/kreuzberg/stopwords/ur_stopwords.json +519 -519
  289. data/vendor/kreuzberg/stopwords/vi_stopwords.json +647 -647
  290. data/vendor/kreuzberg/stopwords/yo_stopwords.json +62 -62
  291. data/vendor/kreuzberg/stopwords/zh_stopwords.json +796 -796
  292. data/vendor/kreuzberg/stopwords/zu_stopwords.json +31 -31
  293. data/vendor/kreuzberg/tests/api_embed.rs +360 -0
  294. data/vendor/kreuzberg/tests/api_extract_multipart.rs +52 -52
  295. data/vendor/kreuzberg/tests/api_large_pdf_extraction.rs +471 -0
  296. data/vendor/kreuzberg/tests/api_large_pdf_extraction_diagnostics.rs +289 -0
  297. data/vendor/kreuzberg/tests/api_tests.rs +1472 -966
  298. data/vendor/kreuzberg/tests/archive_integration.rs +545 -543
  299. data/vendor/kreuzberg/tests/batch_orchestration.rs +587 -556
  300. data/vendor/kreuzberg/tests/batch_pooling_benchmark.rs +154 -0
  301. data/vendor/kreuzberg/tests/batch_processing.rs +328 -316
  302. data/vendor/kreuzberg/tests/bibtex_parity_test.rs +421 -421
  303. data/vendor/kreuzberg/tests/concurrency_stress.rs +541 -525
  304. data/vendor/kreuzberg/tests/config_features.rs +612 -598
  305. data/vendor/kreuzberg/tests/config_integration_test.rs +753 -0
  306. data/vendor/kreuzberg/tests/config_loading_tests.rs +416 -415
  307. data/vendor/kreuzberg/tests/core_integration.rs +519 -510
  308. data/vendor/kreuzberg/tests/csv_integration.rs +414 -414
  309. data/vendor/kreuzberg/tests/data/hierarchy_ground_truth.json +294 -0
  310. data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +500 -498
  311. data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +122 -122
  312. data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +370 -370
  313. data/vendor/kreuzberg/tests/email_integration.rs +327 -325
  314. data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +275 -275
  315. data/vendor/kreuzberg/tests/error_handling.rs +402 -393
  316. data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +228 -228
  317. data/vendor/kreuzberg/tests/format_integration.rs +165 -159
  318. data/vendor/kreuzberg/tests/helpers/mod.rs +202 -142
  319. data/vendor/kreuzberg/tests/html_table_test.rs +551 -551
  320. data/vendor/kreuzberg/tests/image_integration.rs +255 -253
  321. data/vendor/kreuzberg/tests/instrumentation_test.rs +139 -139
  322. data/vendor/kreuzberg/tests/jats_extractor_tests.rs +639 -639
  323. data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +704 -704
  324. data/vendor/kreuzberg/tests/keywords_integration.rs +479 -479
  325. data/vendor/kreuzberg/tests/keywords_quality.rs +509 -509
  326. data/vendor/kreuzberg/tests/latex_extractor_tests.rs +496 -496
  327. data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +490 -490
  328. data/vendor/kreuzberg/tests/mime_detection.rs +429 -428
  329. data/vendor/kreuzberg/tests/ocr_configuration.rs +514 -510
  330. data/vendor/kreuzberg/tests/ocr_errors.rs +698 -676
  331. data/vendor/kreuzberg/tests/ocr_language_registry.rs +191 -0
  332. data/vendor/kreuzberg/tests/ocr_quality.rs +629 -627
  333. data/vendor/kreuzberg/tests/ocr_stress.rs +469 -469
  334. data/vendor/kreuzberg/tests/odt_extractor_tests.rs +674 -695
  335. data/vendor/kreuzberg/tests/opml_extractor_tests.rs +616 -616
  336. data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +822 -822
  337. data/vendor/kreuzberg/tests/page_markers.rs +297 -0
  338. data/vendor/kreuzberg/tests/pdf_hierarchy_detection.rs +301 -0
  339. data/vendor/kreuzberg/tests/pdf_hierarchy_quality.rs +589 -0
  340. data/vendor/kreuzberg/tests/pdf_integration.rs +45 -43
  341. data/vendor/kreuzberg/tests/pdf_ocr_triggering.rs +301 -0
  342. data/vendor/kreuzberg/tests/pdf_text_merging.rs +475 -0
  343. data/vendor/kreuzberg/tests/pdfium_linking.rs +340 -0
  344. data/vendor/kreuzberg/tests/pipeline_integration.rs +1446 -1411
  345. data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +776 -771
  346. data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +577 -560
  347. data/vendor/kreuzberg/tests/plugin_system.rs +927 -921
  348. data/vendor/kreuzberg/tests/plugin_validator_test.rs +783 -783
  349. data/vendor/kreuzberg/tests/registry_integration_tests.rs +587 -586
  350. data/vendor/kreuzberg/tests/rst_extractor_tests.rs +694 -692
  351. data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +775 -776
  352. data/vendor/kreuzberg/tests/security_validation.rs +416 -415
  353. data/vendor/kreuzberg/tests/stopwords_integration_test.rs +888 -888
  354. data/vendor/kreuzberg/tests/test_fastembed.rs +631 -609
  355. data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +1260 -1259
  356. data/vendor/kreuzberg/tests/typst_extractor_tests.rs +648 -647
  357. data/vendor/kreuzberg/tests/xlsx_metadata_extraction_test.rs +87 -87
  358. data/vendor/kreuzberg-ffi/Cargo.toml +67 -0
  359. data/vendor/kreuzberg-ffi/README.md +851 -0
  360. data/vendor/kreuzberg-ffi/benches/result_view_benchmark.rs +227 -0
  361. data/vendor/kreuzberg-ffi/build.rs +168 -0
  362. data/vendor/kreuzberg-ffi/cbindgen.toml +37 -0
  363. data/vendor/kreuzberg-ffi/kreuzberg-ffi.pc.in +12 -0
  364. data/vendor/kreuzberg-ffi/kreuzberg.h +3012 -0
  365. data/vendor/kreuzberg-ffi/src/batch_streaming.rs +588 -0
  366. data/vendor/kreuzberg-ffi/src/config.rs +1341 -0
  367. data/vendor/kreuzberg-ffi/src/error.rs +901 -0
  368. data/vendor/kreuzberg-ffi/src/extraction.rs +555 -0
  369. data/vendor/kreuzberg-ffi/src/helpers.rs +879 -0
  370. data/vendor/kreuzberg-ffi/src/lib.rs +977 -0
  371. data/vendor/kreuzberg-ffi/src/memory.rs +493 -0
  372. data/vendor/kreuzberg-ffi/src/mime.rs +329 -0
  373. data/vendor/kreuzberg-ffi/src/panic_shield.rs +265 -0
  374. data/vendor/kreuzberg-ffi/src/plugins/document_extractor.rs +442 -0
  375. data/vendor/kreuzberg-ffi/src/plugins/mod.rs +14 -0
  376. data/vendor/kreuzberg-ffi/src/plugins/ocr_backend.rs +628 -0
  377. data/vendor/kreuzberg-ffi/src/plugins/post_processor.rs +438 -0
  378. data/vendor/kreuzberg-ffi/src/plugins/validator.rs +329 -0
  379. data/vendor/kreuzberg-ffi/src/result.rs +510 -0
  380. data/vendor/kreuzberg-ffi/src/result_pool.rs +639 -0
  381. data/vendor/kreuzberg-ffi/src/result_view.rs +773 -0
  382. data/vendor/kreuzberg-ffi/src/string_intern.rs +568 -0
  383. data/vendor/kreuzberg-ffi/src/types.rs +363 -0
  384. data/vendor/kreuzberg-ffi/src/util.rs +210 -0
  385. data/vendor/kreuzberg-ffi/src/validation.rs +848 -0
  386. data/vendor/kreuzberg-ffi/tests.disabled/README.md +48 -0
  387. data/vendor/kreuzberg-ffi/tests.disabled/config_loading_tests.rs +299 -0
  388. data/vendor/kreuzberg-ffi/tests.disabled/config_tests.rs +346 -0
  389. data/vendor/kreuzberg-ffi/tests.disabled/extractor_tests.rs +232 -0
  390. data/vendor/kreuzberg-ffi/tests.disabled/plugin_registration_tests.rs +470 -0
  391. data/vendor/kreuzberg-tesseract/.commitlintrc.json +13 -0
  392. data/vendor/kreuzberg-tesseract/.crate-ignore +2 -0
  393. data/vendor/kreuzberg-tesseract/Cargo.lock +2933 -0
  394. data/vendor/kreuzberg-tesseract/Cargo.toml +57 -0
  395. data/vendor/{rb-sys/LICENSE-MIT → kreuzberg-tesseract/LICENSE} +22 -21
  396. data/vendor/kreuzberg-tesseract/README.md +399 -0
  397. data/vendor/kreuzberg-tesseract/build.rs +1127 -0
  398. data/vendor/kreuzberg-tesseract/patches/README.md +71 -0
  399. data/vendor/kreuzberg-tesseract/patches/tesseract.diff +199 -0
  400. data/vendor/kreuzberg-tesseract/src/api.rs +1371 -0
  401. data/vendor/kreuzberg-tesseract/src/choice_iterator.rs +77 -0
  402. data/vendor/kreuzberg-tesseract/src/enums.rs +297 -0
  403. data/vendor/kreuzberg-tesseract/src/error.rs +81 -0
  404. data/vendor/kreuzberg-tesseract/src/lib.rs +145 -0
  405. data/vendor/kreuzberg-tesseract/src/monitor.rs +57 -0
  406. data/vendor/kreuzberg-tesseract/src/mutable_iterator.rs +197 -0
  407. data/vendor/kreuzberg-tesseract/src/page_iterator.rs +253 -0
  408. data/vendor/kreuzberg-tesseract/src/result_iterator.rs +286 -0
  409. data/vendor/kreuzberg-tesseract/src/result_renderer.rs +183 -0
  410. data/vendor/kreuzberg-tesseract/tests/integration_test.rs +211 -0
  411. metadata +196 -45
  412. data/vendor/kreuzberg/benches/otel_overhead.rs +0 -48
  413. data/vendor/kreuzberg/src/extractors/fictionbook.rs.backup2 +0 -738
  414. data/vendor/rb-sys/.cargo-ok +0 -1
  415. data/vendor/rb-sys/.cargo_vcs_info.json +0 -6
  416. data/vendor/rb-sys/Cargo.lock +0 -393
  417. data/vendor/rb-sys/Cargo.toml +0 -70
  418. data/vendor/rb-sys/Cargo.toml.orig +0 -57
  419. data/vendor/rb-sys/LICENSE-APACHE +0 -190
  420. data/vendor/rb-sys/bin/release.sh +0 -21
  421. data/vendor/rb-sys/build/features.rs +0 -108
  422. data/vendor/rb-sys/build/main.rs +0 -246
  423. data/vendor/rb-sys/build/stable_api_config.rs +0 -153
  424. data/vendor/rb-sys/build/version.rs +0 -48
  425. data/vendor/rb-sys/readme.md +0 -36
  426. data/vendor/rb-sys/src/bindings.rs +0 -21
  427. data/vendor/rb-sys/src/hidden.rs +0 -11
  428. data/vendor/rb-sys/src/lib.rs +0 -34
  429. data/vendor/rb-sys/src/macros.rs +0 -371
  430. data/vendor/rb-sys/src/memory.rs +0 -53
  431. data/vendor/rb-sys/src/ruby_abi_version.rs +0 -38
  432. data/vendor/rb-sys/src/special_consts.rs +0 -31
  433. data/vendor/rb-sys/src/stable_api/compiled.c +0 -179
  434. data/vendor/rb-sys/src/stable_api/compiled.rs +0 -257
  435. data/vendor/rb-sys/src/stable_api/ruby_2_6.rs +0 -316
  436. data/vendor/rb-sys/src/stable_api/ruby_2_7.rs +0 -316
  437. data/vendor/rb-sys/src/stable_api/ruby_3_0.rs +0 -324
  438. data/vendor/rb-sys/src/stable_api/ruby_3_1.rs +0 -317
  439. data/vendor/rb-sys/src/stable_api/ruby_3_2.rs +0 -315
  440. data/vendor/rb-sys/src/stable_api/ruby_3_3.rs +0 -326
  441. data/vendor/rb-sys/src/stable_api/ruby_3_4.rs +0 -327
  442. data/vendor/rb-sys/src/stable_api.rs +0 -261
  443. data/vendor/rb-sys/src/symbol.rs +0 -31
  444. data/vendor/rb-sys/src/tracking_allocator.rs +0 -332
  445. data/vendor/rb-sys/src/utils.rs +0 -89
  446. data/vendor/rb-sys/src/value_type.rs +0 -7
@@ -1,1032 +1,1914 @@
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 serde::{Deserialize, Serialize};
8
- use std::path::Path;
9
-
10
- /// Main extraction configuration.
11
- ///
12
- /// This struct contains all configuration options for the extraction process.
13
- /// It can be loaded from TOML, YAML, or JSON files, or created programmatically.
14
- ///
15
- /// # Example
16
- ///
17
- /// ```rust
18
- /// use kreuzberg::core::config::ExtractionConfig;
19
- ///
20
- /// // Create with defaults
21
- /// let config = ExtractionConfig::default();
22
- ///
23
- /// // Load from TOML file
24
- /// // let config = ExtractionConfig::from_toml_file("kreuzberg.toml")?;
25
- /// ```
26
- #[derive(Debug, Clone, Serialize, Deserialize)]
27
- pub struct ExtractionConfig {
28
- /// Enable caching of extraction results
29
- #[serde(default = "default_true")]
30
- pub use_cache: bool,
31
-
32
- /// Enable quality post-processing
33
- #[serde(default = "default_true")]
34
- pub enable_quality_processing: bool,
35
-
36
- /// OCR configuration (None = OCR disabled)
37
- #[serde(default)]
38
- pub ocr: Option<OcrConfig>,
39
-
40
- /// Force OCR even for searchable PDFs
41
- #[serde(default)]
42
- pub force_ocr: bool,
43
-
44
- /// Text chunking configuration (None = chunking disabled)
45
- #[serde(default)]
46
- pub chunking: Option<ChunkingConfig>,
47
-
48
- /// Image extraction configuration (None = no image extraction)
49
- #[serde(default)]
50
- pub images: Option<ImageExtractionConfig>,
51
-
52
- /// PDF-specific options (None = use defaults)
53
- #[serde(default)]
54
- pub pdf_options: Option<PdfConfig>,
55
-
56
- /// Token reduction configuration (None = no token reduction)
57
- #[serde(default)]
58
- pub token_reduction: Option<TokenReductionConfig>,
59
-
60
- /// Language detection configuration (None = no language detection)
61
- #[serde(default)]
62
- pub language_detection: Option<LanguageDetectionConfig>,
63
-
64
- /// Keyword extraction configuration (None = no keyword extraction)
65
- #[cfg(any(feature = "keywords-yake", feature = "keywords-rake"))]
66
- #[serde(default)]
67
- pub keywords: Option<crate::keywords::KeywordConfig>,
68
-
69
- /// Post-processor configuration (None = use defaults)
70
- #[serde(default)]
71
- pub postprocessor: Option<PostProcessorConfig>,
72
-
73
- /// HTML conversion options (None = use defaults)
74
- ///
75
- /// Note: This field cannot be deserialized from TOML/YAML/JSON files.
76
- /// Set it programmatically after loading config.
77
- #[cfg(feature = "html")]
78
- #[serde(skip)]
79
- pub html_options: Option<html_to_markdown_rs::ConversionOptions>,
80
-
81
- /// Maximum concurrent extractions in batch operations (None = num_cpus * 2).
82
- ///
83
- /// Limits parallelism to prevent resource exhaustion when processing
84
- /// large batches. Defaults to twice the number of CPU cores.
85
- #[serde(default)]
86
- pub max_concurrent_extractions: Option<usize>,
87
- }
88
-
89
- /// Post-processor configuration.
90
- #[derive(Debug, Clone, Serialize, Deserialize)]
91
- pub struct PostProcessorConfig {
92
- /// Enable post-processors
93
- #[serde(default = "default_true")]
94
- pub enabled: bool,
95
-
96
- /// Whitelist of processor names to run (None = all enabled)
97
- #[serde(default)]
98
- pub enabled_processors: Option<Vec<String>>,
99
-
100
- /// Blacklist of processor names to skip (None = none disabled)
101
- #[serde(default)]
102
- pub disabled_processors: Option<Vec<String>>,
103
- }
104
-
105
- /// OCR configuration.
106
- #[derive(Debug, Clone, Serialize, Deserialize)]
107
- pub struct OcrConfig {
108
- /// OCR backend: tesseract, easyocr, paddleocr
109
- #[serde(default = "default_tesseract_backend")]
110
- pub backend: String,
111
-
112
- /// Language code (e.g., "eng", "deu")
113
- #[serde(default = "default_eng")]
114
- pub language: String,
115
-
116
- /// Tesseract-specific configuration (optional)
117
- #[serde(default)]
118
- pub tesseract_config: Option<crate::types::TesseractConfig>,
119
- }
120
-
121
- /// Chunking configuration.
122
- #[derive(Debug, Clone, Serialize, Deserialize)]
123
- pub struct ChunkingConfig {
124
- /// Maximum characters per chunk
125
- #[serde(default = "default_chunk_size")]
126
- pub max_chars: usize,
127
-
128
- /// Overlap between chunks in characters
129
- #[serde(default = "default_chunk_overlap")]
130
- pub max_overlap: usize,
131
-
132
- /// Optional embedding configuration for chunk embeddings
133
- #[serde(skip_serializing_if = "Option::is_none")]
134
- pub embedding: Option<EmbeddingConfig>,
135
-
136
- /// Use a preset configuration (overrides individual settings if provided)
137
- #[serde(skip_serializing_if = "Option::is_none")]
138
- pub preset: Option<String>,
139
- }
140
-
141
- /// Embedding configuration for text chunks.
142
- ///
143
- /// Configures embedding generation using ONNX models via fastembed-rs.
144
- /// Requires the `embeddings` feature to be enabled.
145
- #[derive(Debug, Clone, Serialize, Deserialize)]
146
- pub struct EmbeddingConfig {
147
- /// The embedding model to use
148
- pub model: EmbeddingModelType,
149
-
150
- /// Whether to normalize embedding vectors (recommended for cosine similarity)
151
- #[serde(default = "default_normalize")]
152
- pub normalize: bool,
153
-
154
- /// Batch size for embedding generation
155
- #[serde(default = "default_batch_size")]
156
- pub batch_size: usize,
157
-
158
- /// Show model download progress
159
- #[serde(default)]
160
- pub show_download_progress: bool,
161
-
162
- /// Custom cache directory for model files
163
- ///
164
- /// Defaults to `~/.cache/kreuzberg/embeddings/` if not specified.
165
- /// Allows full customization of model download location.
166
- #[serde(skip_serializing_if = "Option::is_none")]
167
- pub cache_dir: Option<std::path::PathBuf>,
168
- }
169
-
170
- impl Default for EmbeddingConfig {
171
- fn default() -> Self {
172
- Self {
173
- model: EmbeddingModelType::Preset {
174
- name: "balanced".to_string(),
175
- },
176
- normalize: true,
177
- batch_size: 32,
178
- show_download_progress: false,
179
- cache_dir: None,
180
- }
181
- }
182
- }
183
-
184
- /// Embedding model types supported by Kreuzberg.
185
- #[derive(Debug, Clone, Serialize, Deserialize)]
186
- #[serde(tag = "type", rename_all = "snake_case")]
187
- pub enum EmbeddingModelType {
188
- /// Use a preset model configuration (recommended)
189
- Preset { name: String },
190
-
191
- /// Use a specific fastembed model by name
192
- #[cfg(feature = "embeddings")]
193
- FastEmbed { model: String, dimensions: usize },
194
-
195
- /// Use a custom ONNX model from HuggingFace
196
- Custom { model_id: String, dimensions: usize },
197
- }
198
-
199
- /// Image extraction configuration.
200
- #[derive(Debug, Clone, Serialize, Deserialize)]
201
- pub struct ImageExtractionConfig {
202
- /// Extract images from documents
203
- #[serde(default = "default_true")]
204
- pub extract_images: bool,
205
-
206
- /// Target DPI for image normalization
207
- #[serde(default = "default_target_dpi")]
208
- pub target_dpi: i32,
209
-
210
- /// Maximum dimension for images (width or height)
211
- #[serde(default = "default_max_dimension")]
212
- pub max_image_dimension: i32,
213
-
214
- /// Automatically adjust DPI based on image content
215
- #[serde(default = "default_true")]
216
- pub auto_adjust_dpi: bool,
217
-
218
- /// Minimum DPI threshold
219
- #[serde(default = "default_min_dpi")]
220
- pub min_dpi: i32,
221
-
222
- /// Maximum DPI threshold
223
- #[serde(default = "default_max_dpi")]
224
- pub max_dpi: i32,
225
- }
226
-
227
- /// PDF-specific configuration.
228
- #[derive(Debug, Clone, Serialize, Deserialize)]
229
- pub struct PdfConfig {
230
- /// Extract images from PDF
231
- #[serde(default)]
232
- pub extract_images: bool,
233
-
234
- /// List of passwords to try when opening encrypted PDFs
235
- #[serde(default)]
236
- pub passwords: Option<Vec<String>>,
237
-
238
- /// Extract PDF metadata
239
- #[serde(default = "default_true")]
240
- pub extract_metadata: bool,
241
- }
242
-
243
- /// Token reduction configuration.
244
- #[derive(Debug, Clone, Serialize, Deserialize)]
245
- pub struct TokenReductionConfig {
246
- /// Reduction mode: "off", "light", "moderate", "aggressive", "maximum"
247
- #[serde(default = "default_reduction_mode")]
248
- pub mode: String,
249
-
250
- /// Preserve important words (capitalized, technical terms)
251
- #[serde(default = "default_true")]
252
- pub preserve_important_words: bool,
253
- }
254
-
255
- /// Language detection configuration.
256
- #[derive(Debug, Clone, Serialize, Deserialize)]
257
- pub struct LanguageDetectionConfig {
258
- /// Enable language detection
259
- #[serde(default = "default_true")]
260
- pub enabled: bool,
261
-
262
- /// Minimum confidence threshold (0.0-1.0)
263
- #[serde(default = "default_confidence")]
264
- pub min_confidence: f64,
265
-
266
- /// Detect multiple languages in the document
267
- #[serde(default)]
268
- pub detect_multiple: bool,
269
- }
270
-
271
- fn default_true() -> bool {
272
- true
273
- }
274
- fn default_eng() -> String {
275
- "eng".to_string()
276
- }
277
- fn default_tesseract_backend() -> String {
278
- "tesseract".to_string()
279
- }
280
- fn default_chunk_size() -> usize {
281
- 1000
282
- }
283
- fn default_chunk_overlap() -> usize {
284
- 200
285
- }
286
- fn default_normalize() -> bool {
287
- true
288
- }
289
- fn default_batch_size() -> usize {
290
- 32
291
- }
292
- fn default_target_dpi() -> i32 {
293
- 300
294
- }
295
- fn default_max_dimension() -> i32 {
296
- 4096
297
- }
298
- fn default_min_dpi() -> i32 {
299
- 72
300
- }
301
- fn default_max_dpi() -> i32 {
302
- 600
303
- }
304
- fn default_reduction_mode() -> String {
305
- "off".to_string()
306
- }
307
- fn default_confidence() -> f64 {
308
- 0.8
309
- }
310
-
311
- impl Default for ExtractionConfig {
312
- fn default() -> Self {
313
- Self {
314
- use_cache: true,
315
- enable_quality_processing: true,
316
- ocr: None,
317
- force_ocr: false,
318
- chunking: None,
319
- images: None,
320
- pdf_options: None,
321
- token_reduction: None,
322
- language_detection: None,
323
- #[cfg(any(feature = "keywords-yake", feature = "keywords-rake"))]
324
- keywords: None,
325
- postprocessor: None,
326
- #[cfg(feature = "html")]
327
- html_options: None,
328
- max_concurrent_extractions: None,
329
- }
330
- }
331
- }
332
-
333
- impl Default for PostProcessorConfig {
334
- fn default() -> Self {
335
- Self {
336
- enabled: true,
337
- enabled_processors: None,
338
- disabled_processors: None,
339
- }
340
- }
341
- }
342
-
343
- impl ExtractionConfig {
344
- /// Load configuration from a TOML file.
345
- ///
346
- /// # Arguments
347
- ///
348
- /// * `path` - Path to the TOML file
349
- ///
350
- /// # Errors
351
- ///
352
- /// Returns `KreuzbergError::Validation` if file doesn't exist or is invalid TOML.
353
- pub fn from_toml_file(path: impl AsRef<Path>) -> Result<Self> {
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))
356
- })?;
357
-
358
- toml::from_str(&content)
359
- .map_err(|e| KreuzbergError::validation(format!("Invalid TOML in {}: {}", path.as_ref().display(), e)))
360
- }
361
-
362
- /// Load configuration from a YAML file.
363
- pub fn from_yaml_file(path: impl AsRef<Path>) -> Result<Self> {
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))
366
- })?;
367
-
368
- serde_yaml_ng::from_str(&content)
369
- .map_err(|e| KreuzbergError::validation(format!("Invalid YAML in {}: {}", path.as_ref().display(), e)))
370
- }
371
-
372
- /// Load configuration from a JSON file.
373
- pub fn from_json_file(path: impl AsRef<Path>) -> Result<Self> {
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))
376
- })?;
377
-
378
- serde_json::from_str(&content)
379
- .map_err(|e| KreuzbergError::validation(format!("Invalid JSON in {}: {}", path.as_ref().display(), e)))
380
- }
381
-
382
- /// Load configuration from a file, auto-detecting format by extension.
383
- ///
384
- /// Supported formats:
385
- /// - `.toml` - TOML format
386
- /// - `.yaml` - YAML format
387
- /// - `.json` - JSON format
388
- ///
389
- /// # Arguments
390
- ///
391
- /// * `path` - Path to the configuration file
392
- ///
393
- /// # Errors
394
- ///
395
- /// Returns `KreuzbergError::Validation` if:
396
- /// - File doesn't exist
397
- /// - File extension is not supported
398
- /// - File content is invalid for the detected format
399
- ///
400
- /// # Example
401
- ///
402
- /// ```rust
403
- /// use kreuzberg::core::config::ExtractionConfig;
404
- ///
405
- /// // Auto-detects TOML format
406
- /// // let config = ExtractionConfig::from_file("kreuzberg.toml")?;
407
- ///
408
- /// // Auto-detects YAML format
409
- /// // let config = ExtractionConfig::from_file("kreuzberg.yaml")?;
410
- /// ```
411
- pub fn from_file(path: impl AsRef<Path>) -> Result<Self> {
412
- let path = path.as_ref();
413
- let extension = path.extension().and_then(|ext| ext.to_str()).ok_or_else(|| {
414
- KreuzbergError::validation(format!(
415
- "Cannot determine file format: no extension found in {}",
416
- path.display()
417
- ))
418
- })?;
419
-
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
- }
429
- }
430
-
431
- /// Discover configuration file in parent directories.
432
- ///
433
- /// Searches for `kreuzberg.toml` in current directory and parent directories.
434
- ///
435
- /// # Returns
436
- ///
437
- /// - `Some(config)` if found
438
- /// - `None` if no config file found
439
- pub fn discover() -> Result<Option<Self>> {
440
- let mut current = std::env::current_dir().map_err(KreuzbergError::Io)?;
441
-
442
- loop {
443
- let kreuzberg_toml = current.join("kreuzberg.toml");
444
- if kreuzberg_toml.exists() {
445
- return Ok(Some(Self::from_toml_file(kreuzberg_toml)?));
446
- }
447
-
448
- if let Some(parent) = current.parent() {
449
- current = parent.to_path_buf();
450
- } else {
451
- break;
452
- }
453
- }
454
-
455
- Ok(None)
456
- }
457
- }
458
-
459
- #[cfg(test)]
460
- mod tests {
461
- use super::*;
462
- use std::fs;
463
- use tempfile::tempdir;
464
-
465
- #[test]
466
- fn test_default_config() {
467
- let config = ExtractionConfig::default();
468
- assert!(config.use_cache);
469
- assert!(config.enable_quality_processing);
470
- assert!(config.ocr.is_none());
471
- }
472
-
473
- #[test]
474
- fn test_from_toml_file() {
475
- let dir = tempdir().unwrap();
476
- let config_path = dir.path().join("kreuzberg.toml");
477
-
478
- fs::write(
479
- &config_path,
480
- r#"
481
- use_cache = false
482
- enable_quality_processing = true
483
- "#,
484
- )
485
- .unwrap();
486
-
487
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
488
- assert!(!config.use_cache);
489
- assert!(config.enable_quality_processing);
490
- }
491
-
492
- #[test]
493
- fn test_from_file_auto_detects_toml() {
494
- let dir = tempdir().unwrap();
495
- let config_path = dir.path().join("kreuzberg.toml");
496
-
497
- fs::write(
498
- &config_path,
499
- r#"
500
- use_cache = false
501
- enable_quality_processing = true
502
- "#,
503
- )
504
- .unwrap();
505
-
506
- let config = ExtractionConfig::from_file(&config_path).unwrap();
507
- assert!(!config.use_cache);
508
- assert!(config.enable_quality_processing);
509
- }
510
-
511
- #[test]
512
- fn test_from_file_auto_detects_yaml() {
513
- let dir = tempdir().unwrap();
514
- let config_path = dir.path().join("kreuzberg.yaml");
515
-
516
- fs::write(
517
- &config_path,
518
- r#"
519
- use_cache: true
520
- enable_quality_processing: false
521
- "#,
522
- )
523
- .unwrap();
524
-
525
- let config = ExtractionConfig::from_file(&config_path).unwrap();
526
- assert!(config.use_cache);
527
- assert!(!config.enable_quality_processing);
528
- }
529
-
530
- #[test]
531
- fn test_from_file_unsupported_extension() {
532
- let dir = tempdir().unwrap();
533
- let config_path = dir.path().join("kreuzberg.txt");
534
-
535
- fs::write(&config_path, "use_cache = false").unwrap();
536
-
537
- let result = ExtractionConfig::from_file(&config_path);
538
- assert!(result.is_err());
539
- let err = result.unwrap_err();
540
- assert!(err.to_string().contains("Unsupported config file format"));
541
- }
542
-
543
- #[test]
544
- fn test_from_file_no_extension() {
545
- let dir = tempdir().unwrap();
546
- let config_path = dir.path().join("kreuzberg");
547
-
548
- fs::write(&config_path, "use_cache = false").unwrap();
549
-
550
- let result = ExtractionConfig::from_file(&config_path);
551
- assert!(result.is_err());
552
- let err = result.unwrap_err();
553
- assert!(err.to_string().contains("no extension found"));
554
- }
555
-
556
- #[test]
557
- fn test_discover_kreuzberg_toml() {
558
- let dir = tempdir().unwrap();
559
- let config_path = dir.path().join("kreuzberg.toml");
560
-
561
- fs::write(
562
- &config_path,
563
- r#"
564
- use_cache = false
565
- enable_quality_processing = true
566
- "#,
567
- )
568
- .unwrap();
569
-
570
- let original_dir = std::env::current_dir().unwrap();
571
- std::env::set_current_dir(&dir).unwrap();
572
-
573
- let result = std::panic::catch_unwind(|| {
574
- let config = ExtractionConfig::discover().unwrap();
575
- assert!(config.is_some());
576
- assert!(!config.unwrap().use_cache);
577
- });
578
-
579
- std::env::set_current_dir(&original_dir).unwrap();
580
-
581
- if let Err(e) = result {
582
- std::panic::resume_unwind(e);
583
- }
584
- }
585
-
586
- #[test]
587
- fn test_v4_config_with_ocr_and_chunking() {
588
- let dir = tempdir().unwrap();
589
- let config_path = dir.path().join("kreuzberg.toml");
590
-
591
- fs::write(
592
- &config_path,
593
- r#"
594
- use_cache = true
595
- enable_quality_processing = false
596
-
597
- [ocr]
598
- backend = "tesseract"
599
- language = "eng"
600
-
601
- [chunking]
602
- max_chars = 2000
603
- max_overlap = 300
604
- "#,
605
- )
606
- .unwrap();
607
-
608
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
609
- assert!(config.use_cache);
610
- assert!(!config.enable_quality_processing);
611
- assert!(config.ocr.is_some());
612
- assert_eq!(config.ocr.unwrap().backend, "tesseract");
613
- assert!(config.chunking.is_some());
614
- assert_eq!(config.chunking.unwrap().max_chars, 2000);
615
- }
616
-
617
- #[test]
618
- fn test_config_with_image_extraction() {
619
- let dir = tempdir().unwrap();
620
- let config_path = dir.path().join("kreuzberg.toml");
621
-
622
- fs::write(
623
- &config_path,
624
- r#"
625
- use_cache = true
626
-
627
- [images]
628
- extract_images = true
629
- target_dpi = 300
630
- max_image_dimension = 4096
631
- auto_adjust_dpi = true
632
- min_dpi = 72
633
- max_dpi = 600
634
- "#,
635
- )
636
- .unwrap();
637
-
638
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
639
- assert!(config.images.is_some());
640
- let images = config.images.unwrap();
641
- assert!(images.extract_images);
642
- assert_eq!(images.target_dpi, 300);
643
- assert_eq!(images.max_image_dimension, 4096);
644
- assert!(images.auto_adjust_dpi);
645
- assert_eq!(images.min_dpi, 72);
646
- assert_eq!(images.max_dpi, 600);
647
- }
648
-
649
- #[test]
650
- fn test_config_with_pdf_options() {
651
- let dir = tempdir().unwrap();
652
- let config_path = dir.path().join("kreuzberg.toml");
653
-
654
- fs::write(
655
- &config_path,
656
- r#"
657
- use_cache = true
658
-
659
- [pdf_options]
660
- extract_images = true
661
- passwords = ["password1", "password2"]
662
- extract_metadata = true
663
- "#,
664
- )
665
- .unwrap();
666
-
667
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
668
- assert!(config.pdf_options.is_some());
669
- let pdf = config.pdf_options.unwrap();
670
- assert!(pdf.extract_images);
671
- assert!(pdf.extract_metadata);
672
- assert!(pdf.passwords.is_some());
673
- let passwords = pdf.passwords.unwrap();
674
- assert_eq!(passwords.len(), 2);
675
- assert_eq!(passwords[0], "password1");
676
- assert_eq!(passwords[1], "password2");
677
- }
678
-
679
- #[test]
680
- fn test_config_with_token_reduction() {
681
- let dir = tempdir().unwrap();
682
- let config_path = dir.path().join("kreuzberg.toml");
683
-
684
- fs::write(
685
- &config_path,
686
- r#"
687
- use_cache = true
688
-
689
- [token_reduction]
690
- mode = "aggressive"
691
- preserve_important_words = true
692
- "#,
693
- )
694
- .unwrap();
695
-
696
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
697
- assert!(config.token_reduction.is_some());
698
- let token = config.token_reduction.unwrap();
699
- assert_eq!(token.mode, "aggressive");
700
- assert!(token.preserve_important_words);
701
- }
702
-
703
- #[test]
704
- fn test_config_with_language_detection() {
705
- let dir = tempdir().unwrap();
706
- let config_path = dir.path().join("kreuzberg.toml");
707
-
708
- fs::write(
709
- &config_path,
710
- r#"
711
- use_cache = true
712
-
713
- [language_detection]
714
- enabled = true
715
- min_confidence = 0.9
716
- detect_multiple = true
717
- "#,
718
- )
719
- .unwrap();
720
-
721
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
722
- assert!(config.language_detection.is_some());
723
- let lang = config.language_detection.unwrap();
724
- assert!(lang.enabled);
725
- assert_eq!(lang.min_confidence, 0.9);
726
- assert!(lang.detect_multiple);
727
- }
728
-
729
- #[test]
730
- fn test_config_with_all_optional_fields() {
731
- let dir = tempdir().unwrap();
732
- let config_path = dir.path().join("kreuzberg.toml");
733
-
734
- fs::write(
735
- &config_path,
736
- r#"
737
- use_cache = true
738
- enable_quality_processing = true
739
- force_ocr = false
740
-
741
- [ocr]
742
- backend = "tesseract"
743
- language = "eng"
744
-
745
- [chunking]
746
- max_chars = 1500
747
- max_overlap = 250
748
-
749
- [images]
750
- extract_images = true
751
- target_dpi = 300
752
-
753
- [pdf_options]
754
- extract_images = false
755
- extract_metadata = true
756
-
757
- [token_reduction]
758
- mode = "moderate"
759
-
760
- [language_detection]
761
- enabled = true
762
- "#,
763
- )
764
- .unwrap();
765
-
766
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
767
- assert!(config.use_cache);
768
- assert!(config.enable_quality_processing);
769
- assert!(!config.force_ocr);
770
- assert!(config.ocr.is_some());
771
- assert!(config.chunking.is_some());
772
- assert!(config.images.is_some());
773
- assert!(config.pdf_options.is_some());
774
- assert!(config.token_reduction.is_some());
775
- assert!(config.language_detection.is_some());
776
- }
777
-
778
- #[test]
779
- fn test_image_config_defaults() {
780
- let dir = tempdir().unwrap();
781
- let config_path = dir.path().join("kreuzberg.toml");
782
-
783
- fs::write(
784
- &config_path,
785
- r#"
786
- [images]
787
- "#,
788
- )
789
- .unwrap();
790
-
791
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
792
- let images = config.images.unwrap();
793
- assert!(images.extract_images);
794
- assert_eq!(images.target_dpi, 300);
795
- assert_eq!(images.max_image_dimension, 4096);
796
- assert!(images.auto_adjust_dpi);
797
- assert_eq!(images.min_dpi, 72);
798
- assert_eq!(images.max_dpi, 600);
799
- }
800
-
801
- #[test]
802
- fn test_token_reduction_defaults() {
803
- let dir = tempdir().unwrap();
804
- let config_path = dir.path().join("kreuzberg.toml");
805
-
806
- fs::write(
807
- &config_path,
808
- r#"
809
- [token_reduction]
810
- "#,
811
- )
812
- .unwrap();
813
-
814
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
815
- let token = config.token_reduction.unwrap();
816
- assert_eq!(token.mode, "off");
817
- assert!(token.preserve_important_words);
818
- }
819
-
820
- #[test]
821
- fn test_language_detection_defaults() {
822
- let dir = tempdir().unwrap();
823
- let config_path = dir.path().join("kreuzberg.toml");
824
-
825
- fs::write(
826
- &config_path,
827
- r#"
828
- [language_detection]
829
- "#,
830
- )
831
- .unwrap();
832
-
833
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
834
- let lang = config.language_detection.unwrap();
835
- assert!(lang.enabled);
836
- assert_eq!(lang.min_confidence, 0.8);
837
- assert!(!lang.detect_multiple);
838
- }
839
-
840
- #[test]
841
- fn test_pdf_config_defaults() {
842
- let dir = tempdir().unwrap();
843
- let config_path = dir.path().join("kreuzberg.toml");
844
-
845
- fs::write(
846
- &config_path,
847
- r#"
848
- [pdf_options]
849
- "#,
850
- )
851
- .unwrap();
852
-
853
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
854
- let pdf = config.pdf_options.unwrap();
855
- assert!(!pdf.extract_images);
856
- assert!(pdf.extract_metadata);
857
- assert!(pdf.passwords.is_none());
858
- }
859
-
860
- #[test]
861
- fn test_config_with_postprocessor() {
862
- let dir = tempdir().unwrap();
863
- let config_path = dir.path().join("kreuzberg.toml");
864
-
865
- fs::write(
866
- &config_path,
867
- r#"
868
- use_cache = true
869
-
870
- [postprocessor]
871
- enabled = true
872
- enabled_processors = ["entity_extraction", "keyword_extraction"]
873
- "#,
874
- )
875
- .unwrap();
876
-
877
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
878
- assert!(config.postprocessor.is_some());
879
- let pp = config.postprocessor.unwrap();
880
- assert!(pp.enabled);
881
- assert!(pp.enabled_processors.is_some());
882
- let enabled = pp.enabled_processors.unwrap();
883
- assert_eq!(enabled.len(), 2);
884
- assert_eq!(enabled[0], "entity_extraction");
885
- assert_eq!(enabled[1], "keyword_extraction");
886
- assert!(pp.disabled_processors.is_none());
887
- }
888
-
889
- #[test]
890
- fn test_postprocessor_config_defaults() {
891
- let dir = tempdir().unwrap();
892
- let config_path = dir.path().join("kreuzberg.toml");
893
-
894
- fs::write(
895
- &config_path,
896
- r#"
897
- [postprocessor]
898
- "#,
899
- )
900
- .unwrap();
901
-
902
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
903
- let pp = config.postprocessor.unwrap();
904
- assert!(pp.enabled);
905
- assert!(pp.enabled_processors.is_none());
906
- assert!(pp.disabled_processors.is_none());
907
- }
908
-
909
- #[test]
910
- fn test_postprocessor_config_disabled() {
911
- let dir = tempdir().unwrap();
912
- let config_path = dir.path().join("kreuzberg.toml");
913
-
914
- fs::write(
915
- &config_path,
916
- r#"
917
- [postprocessor]
918
- enabled = false
919
- "#,
920
- )
921
- .unwrap();
922
-
923
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
924
- let pp = config.postprocessor.unwrap();
925
- assert!(!pp.enabled);
926
- }
927
-
928
- #[test]
929
- fn test_postprocessor_config_blacklist() {
930
- let dir = tempdir().unwrap();
931
- let config_path = dir.path().join("kreuzberg.toml");
932
-
933
- fs::write(
934
- &config_path,
935
- r#"
936
- [postprocessor]
937
- disabled_processors = ["category_extraction"]
938
- "#,
939
- )
940
- .unwrap();
941
-
942
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
943
- let pp = config.postprocessor.unwrap();
944
- assert!(pp.enabled);
945
- assert!(pp.disabled_processors.is_some());
946
- let disabled = pp.disabled_processors.unwrap();
947
- assert_eq!(disabled.len(), 1);
948
- assert_eq!(disabled[0], "category_extraction");
949
- }
950
-
951
- #[test]
952
- fn test_config_with_tesseract_config() {
953
- let dir = tempdir().unwrap();
954
- let config_path = dir.path().join("kreuzberg.toml");
955
-
956
- fs::write(
957
- &config_path,
958
- r#"
959
- use_cache = true
960
-
961
- [ocr]
962
- backend = "tesseract"
963
- language = "eng"
964
-
965
- [ocr.tesseract_config]
966
- psm = 6
967
- output_format = "text"
968
- enable_table_detection = false
969
- tessedit_char_whitelist = "0123456789"
970
- "#,
971
- )
972
- .unwrap();
973
-
974
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
975
- assert!(config.ocr.is_some());
976
- let ocr = config.ocr.unwrap();
977
- assert_eq!(ocr.backend, "tesseract");
978
- assert_eq!(ocr.language, "eng");
979
- assert!(ocr.tesseract_config.is_some());
980
-
981
- let tess = ocr.tesseract_config.unwrap();
982
- assert_eq!(tess.psm, 6);
983
- assert_eq!(tess.output_format, "text");
984
- assert!(!tess.enable_table_detection);
985
- assert_eq!(tess.tessedit_char_whitelist, "0123456789");
986
- }
987
-
988
- #[test]
989
- fn test_ocr_config_without_tesseract_config() {
990
- let dir = tempdir().unwrap();
991
- let config_path = dir.path().join("kreuzberg.toml");
992
-
993
- fs::write(
994
- &config_path,
995
- r#"
996
- [ocr]
997
- backend = "easyocr"
998
- language = "eng"
999
- "#,
1000
- )
1001
- .unwrap();
1002
-
1003
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1004
- assert!(config.ocr.is_some());
1005
- let ocr = config.ocr.unwrap();
1006
- assert_eq!(ocr.backend, "easyocr");
1007
- assert_eq!(ocr.language, "eng");
1008
- assert!(ocr.tesseract_config.is_none());
1009
- }
1010
-
1011
- #[test]
1012
- fn test_tesseract_config_defaults() {
1013
- let tess = crate::types::TesseractConfig::default();
1014
- assert_eq!(tess.language, "eng");
1015
- assert_eq!(tess.psm, 3);
1016
- assert_eq!(tess.output_format, "markdown");
1017
- assert!(tess.enable_table_detection);
1018
- assert_eq!(tess.table_min_confidence, 0.0);
1019
- assert_eq!(tess.table_column_threshold, 50);
1020
- assert_eq!(tess.table_row_threshold_ratio, 0.5);
1021
- assert!(tess.use_cache);
1022
- assert!(tess.classify_use_pre_adapted_templates);
1023
- assert!(!tess.language_model_ngram_on);
1024
- assert!(tess.tessedit_dont_blkrej_good_wds);
1025
- assert!(tess.tessedit_dont_rowrej_good_wds);
1026
- assert!(tess.tessedit_enable_dict_correction);
1027
- assert_eq!(tess.tessedit_char_whitelist, "");
1028
- assert!(tess.tessedit_use_primary_params_model);
1029
- assert!(tess.textord_space_size_is_variable);
1030
- assert!(!tess.thresholding_method);
1031
- }
1032
- }
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 conversion options (None = use defaults)
119
+ ///
120
+ /// Note: This field cannot be deserialized from TOML/YAML/JSON files.
121
+ /// Set it programmatically after loading config.
122
+ #[cfg(feature = "html")]
123
+ #[serde(skip)]
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
+ }