kreuzberg 4.0.8 → 4.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (308) hide show
  1. checksums.yaml +4 -4
  2. data/Gemfile.lock +2 -2
  3. data/README.md +1 -1
  4. data/ext/kreuzberg_rb/native/Cargo.lock +94 -98
  5. data/ext/kreuzberg_rb/native/Cargo.toml +4 -2
  6. data/ext/kreuzberg_rb/native/src/batch.rs +139 -0
  7. data/ext/kreuzberg_rb/native/src/config/mod.rs +10 -0
  8. data/ext/kreuzberg_rb/native/src/config/types.rs +1058 -0
  9. data/ext/kreuzberg_rb/native/src/error_handling.rs +125 -0
  10. data/ext/kreuzberg_rb/native/src/extraction.rs +79 -0
  11. data/ext/kreuzberg_rb/native/src/gc_guarded_value.rs +35 -0
  12. data/ext/kreuzberg_rb/native/src/helpers.rs +176 -0
  13. data/ext/kreuzberg_rb/native/src/lib.rs +342 -3622
  14. data/ext/kreuzberg_rb/native/src/metadata.rs +34 -0
  15. data/ext/kreuzberg_rb/native/src/plugins/mod.rs +92 -0
  16. data/ext/kreuzberg_rb/native/src/plugins/ocr_backend.rs +159 -0
  17. data/ext/kreuzberg_rb/native/src/plugins/post_processor.rs +126 -0
  18. data/ext/kreuzberg_rb/native/src/plugins/validator.rs +99 -0
  19. data/ext/kreuzberg_rb/native/src/result.rs +326 -0
  20. data/ext/kreuzberg_rb/native/src/validation.rs +4 -0
  21. data/lib/kreuzberg/config.rb +66 -0
  22. data/lib/kreuzberg/result.rb +107 -2
  23. data/lib/kreuzberg/types.rb +104 -0
  24. data/lib/kreuzberg/version.rb +1 -1
  25. data/lib/kreuzberg.rb +0 -4
  26. data/sig/kreuzberg.rbs +105 -1
  27. data/vendor/Cargo.toml +3 -3
  28. data/vendor/kreuzberg/Cargo.toml +4 -3
  29. data/vendor/kreuzberg/README.md +1 -1
  30. data/vendor/kreuzberg/src/api/config.rs +69 -0
  31. data/vendor/kreuzberg/src/api/handlers.rs +99 -2
  32. data/vendor/kreuzberg/src/api/mod.rs +14 -7
  33. data/vendor/kreuzberg/src/api/router.rs +214 -0
  34. data/vendor/kreuzberg/src/api/startup.rs +243 -0
  35. data/vendor/kreuzberg/src/api/types.rs +78 -0
  36. data/vendor/kreuzberg/src/cache/cleanup.rs +277 -0
  37. data/vendor/kreuzberg/src/cache/core.rs +428 -0
  38. data/vendor/kreuzberg/src/cache/mod.rs +21 -843
  39. data/vendor/kreuzberg/src/cache/utilities.rs +156 -0
  40. data/vendor/kreuzberg/src/chunking/boundaries.rs +301 -0
  41. data/vendor/kreuzberg/src/chunking/builder.rs +294 -0
  42. data/vendor/kreuzberg/src/chunking/config.rs +52 -0
  43. data/vendor/kreuzberg/src/chunking/core.rs +1017 -0
  44. data/vendor/kreuzberg/src/chunking/mod.rs +14 -2211
  45. data/vendor/kreuzberg/src/chunking/processor.rs +10 -0
  46. data/vendor/kreuzberg/src/chunking/validation.rs +686 -0
  47. data/vendor/kreuzberg/src/core/config/extraction/core.rs +169 -0
  48. data/vendor/kreuzberg/src/core/config/extraction/env.rs +179 -0
  49. data/vendor/kreuzberg/src/core/config/extraction/loaders.rs +204 -0
  50. data/vendor/kreuzberg/src/core/config/extraction/mod.rs +42 -0
  51. data/vendor/kreuzberg/src/core/config/extraction/types.rs +93 -0
  52. data/vendor/kreuzberg/src/core/config/formats.rs +135 -0
  53. data/vendor/kreuzberg/src/core/config/mod.rs +20 -0
  54. data/vendor/kreuzberg/src/core/config/ocr.rs +73 -0
  55. data/vendor/kreuzberg/src/core/config/page.rs +57 -0
  56. data/vendor/kreuzberg/src/core/config/pdf.rs +111 -0
  57. data/vendor/kreuzberg/src/core/config/processing.rs +312 -0
  58. data/vendor/kreuzberg/src/core/config_validation/dependencies.rs +187 -0
  59. data/vendor/kreuzberg/src/core/config_validation/mod.rs +386 -0
  60. data/vendor/kreuzberg/src/core/config_validation/sections.rs +401 -0
  61. data/vendor/kreuzberg/src/core/extractor/batch.rs +246 -0
  62. data/vendor/kreuzberg/src/core/extractor/bytes.rs +116 -0
  63. data/vendor/kreuzberg/src/core/extractor/file.rs +240 -0
  64. data/vendor/kreuzberg/src/core/extractor/helpers.rs +71 -0
  65. data/vendor/kreuzberg/src/core/extractor/legacy.rs +62 -0
  66. data/vendor/kreuzberg/src/core/extractor/mod.rs +490 -0
  67. data/vendor/kreuzberg/src/core/extractor/sync.rs +208 -0
  68. data/vendor/kreuzberg/src/core/mod.rs +4 -1
  69. data/vendor/kreuzberg/src/core/pipeline/cache.rs +60 -0
  70. data/vendor/kreuzberg/src/core/pipeline/execution.rs +89 -0
  71. data/vendor/kreuzberg/src/core/pipeline/features.rs +108 -0
  72. data/vendor/kreuzberg/src/core/pipeline/format.rs +392 -0
  73. data/vendor/kreuzberg/src/core/pipeline/initialization.rs +67 -0
  74. data/vendor/kreuzberg/src/core/pipeline/mod.rs +135 -0
  75. data/vendor/kreuzberg/src/core/pipeline/tests.rs +975 -0
  76. data/vendor/kreuzberg/src/core/server_config/env.rs +90 -0
  77. data/vendor/kreuzberg/src/core/server_config/loader.rs +202 -0
  78. data/vendor/kreuzberg/src/core/server_config/mod.rs +380 -0
  79. data/vendor/kreuzberg/src/core/server_config/tests/basic_tests.rs +124 -0
  80. data/vendor/kreuzberg/src/core/server_config/tests/env_tests.rs +216 -0
  81. data/vendor/kreuzberg/src/core/server_config/tests/file_loading_tests.rs +341 -0
  82. data/vendor/kreuzberg/src/core/server_config/tests/mod.rs +5 -0
  83. data/vendor/kreuzberg/src/core/server_config/validation.rs +17 -0
  84. data/vendor/kreuzberg/src/embeddings.rs +136 -13
  85. data/vendor/kreuzberg/src/extraction/{archive.rs → archive/mod.rs} +45 -239
  86. data/vendor/kreuzberg/src/extraction/archive/sevenz.rs +98 -0
  87. data/vendor/kreuzberg/src/extraction/archive/tar.rs +118 -0
  88. data/vendor/kreuzberg/src/extraction/archive/zip.rs +101 -0
  89. data/vendor/kreuzberg/src/extraction/html/converter.rs +592 -0
  90. data/vendor/kreuzberg/src/extraction/html/image_handling.rs +95 -0
  91. data/vendor/kreuzberg/src/extraction/html/mod.rs +53 -0
  92. data/vendor/kreuzberg/src/extraction/html/processor.rs +659 -0
  93. data/vendor/kreuzberg/src/extraction/html/stack_management.rs +103 -0
  94. data/vendor/kreuzberg/src/extraction/html/types.rs +28 -0
  95. data/vendor/kreuzberg/src/extraction/mod.rs +6 -2
  96. data/vendor/kreuzberg/src/extraction/pptx/container.rs +159 -0
  97. data/vendor/kreuzberg/src/extraction/pptx/content_builder.rs +168 -0
  98. data/vendor/kreuzberg/src/extraction/pptx/elements.rs +132 -0
  99. data/vendor/kreuzberg/src/extraction/pptx/image_handling.rs +57 -0
  100. data/vendor/kreuzberg/src/extraction/pptx/metadata.rs +160 -0
  101. data/vendor/kreuzberg/src/extraction/pptx/mod.rs +558 -0
  102. data/vendor/kreuzberg/src/extraction/pptx/parser.rs +379 -0
  103. data/vendor/kreuzberg/src/extraction/transform/content.rs +205 -0
  104. data/vendor/kreuzberg/src/extraction/transform/elements.rs +211 -0
  105. data/vendor/kreuzberg/src/extraction/transform/mod.rs +480 -0
  106. data/vendor/kreuzberg/src/extraction/transform/types.rs +27 -0
  107. data/vendor/kreuzberg/src/extractors/archive.rs +2 -0
  108. data/vendor/kreuzberg/src/extractors/bibtex.rs +2 -0
  109. data/vendor/kreuzberg/src/extractors/djot_format/attributes.rs +134 -0
  110. data/vendor/kreuzberg/src/extractors/djot_format/conversion.rs +223 -0
  111. data/vendor/kreuzberg/src/extractors/djot_format/extractor.rs +172 -0
  112. data/vendor/kreuzberg/src/extractors/djot_format/mod.rs +24 -0
  113. data/vendor/kreuzberg/src/extractors/djot_format/parsing/block_handlers.rs +271 -0
  114. data/vendor/kreuzberg/src/extractors/djot_format/parsing/content_extraction.rs +257 -0
  115. data/vendor/kreuzberg/src/extractors/djot_format/parsing/event_handlers.rs +101 -0
  116. data/vendor/kreuzberg/src/extractors/djot_format/parsing/inline_handlers.rs +201 -0
  117. data/vendor/kreuzberg/src/extractors/djot_format/parsing/mod.rs +16 -0
  118. data/vendor/kreuzberg/src/extractors/djot_format/parsing/state.rs +78 -0
  119. data/vendor/kreuzberg/src/extractors/djot_format/parsing/table_extraction.rs +68 -0
  120. data/vendor/kreuzberg/src/extractors/djot_format/parsing/text_extraction.rs +61 -0
  121. data/vendor/kreuzberg/src/extractors/djot_format/rendering.rs +452 -0
  122. data/vendor/kreuzberg/src/extractors/docbook.rs +2 -0
  123. data/vendor/kreuzberg/src/extractors/docx.rs +12 -1
  124. data/vendor/kreuzberg/src/extractors/email.rs +2 -0
  125. data/vendor/kreuzberg/src/extractors/epub/content.rs +333 -0
  126. data/vendor/kreuzberg/src/extractors/epub/metadata.rs +137 -0
  127. data/vendor/kreuzberg/src/extractors/epub/mod.rs +186 -0
  128. data/vendor/kreuzberg/src/extractors/epub/parsing.rs +86 -0
  129. data/vendor/kreuzberg/src/extractors/excel.rs +4 -0
  130. data/vendor/kreuzberg/src/extractors/fictionbook.rs +2 -0
  131. data/vendor/kreuzberg/src/extractors/frontmatter_utils.rs +466 -0
  132. data/vendor/kreuzberg/src/extractors/html.rs +80 -8
  133. data/vendor/kreuzberg/src/extractors/image.rs +8 -1
  134. data/vendor/kreuzberg/src/extractors/jats/elements.rs +350 -0
  135. data/vendor/kreuzberg/src/extractors/jats/metadata.rs +21 -0
  136. data/vendor/kreuzberg/src/extractors/{jats.rs → jats/mod.rs} +10 -412
  137. data/vendor/kreuzberg/src/extractors/jats/parser.rs +52 -0
  138. data/vendor/kreuzberg/src/extractors/jupyter.rs +2 -0
  139. data/vendor/kreuzberg/src/extractors/latex/commands.rs +93 -0
  140. data/vendor/kreuzberg/src/extractors/latex/environments.rs +157 -0
  141. data/vendor/kreuzberg/src/extractors/latex/metadata.rs +27 -0
  142. data/vendor/kreuzberg/src/extractors/latex/mod.rs +146 -0
  143. data/vendor/kreuzberg/src/extractors/latex/parser.rs +231 -0
  144. data/vendor/kreuzberg/src/extractors/latex/utilities.rs +126 -0
  145. data/vendor/kreuzberg/src/extractors/markdown.rs +39 -162
  146. data/vendor/kreuzberg/src/extractors/mod.rs +9 -1
  147. data/vendor/kreuzberg/src/extractors/odt.rs +2 -0
  148. data/vendor/kreuzberg/src/extractors/opml/core.rs +165 -0
  149. data/vendor/kreuzberg/src/extractors/opml/mod.rs +31 -0
  150. data/vendor/kreuzberg/src/extractors/opml/parser.rs +479 -0
  151. data/vendor/kreuzberg/src/extractors/orgmode.rs +2 -0
  152. data/vendor/kreuzberg/src/extractors/pdf/extraction.rs +106 -0
  153. data/vendor/kreuzberg/src/extractors/{pdf.rs → pdf/mod.rs} +25 -324
  154. data/vendor/kreuzberg/src/extractors/pdf/ocr.rs +214 -0
  155. data/vendor/kreuzberg/src/extractors/pdf/pages.rs +51 -0
  156. data/vendor/kreuzberg/src/extractors/pptx.rs +9 -2
  157. data/vendor/kreuzberg/src/extractors/rst.rs +2 -0
  158. data/vendor/kreuzberg/src/extractors/rtf/encoding.rs +116 -0
  159. data/vendor/kreuzberg/src/extractors/rtf/formatting.rs +24 -0
  160. data/vendor/kreuzberg/src/extractors/rtf/images.rs +72 -0
  161. data/vendor/kreuzberg/src/extractors/rtf/metadata.rs +216 -0
  162. data/vendor/kreuzberg/src/extractors/rtf/mod.rs +142 -0
  163. data/vendor/kreuzberg/src/extractors/rtf/parser.rs +259 -0
  164. data/vendor/kreuzberg/src/extractors/rtf/tables.rs +83 -0
  165. data/vendor/kreuzberg/src/extractors/structured.rs +2 -0
  166. data/vendor/kreuzberg/src/extractors/text.rs +4 -0
  167. data/vendor/kreuzberg/src/extractors/typst.rs +2 -0
  168. data/vendor/kreuzberg/src/extractors/xml.rs +2 -0
  169. data/vendor/kreuzberg/src/keywords/processor.rs +14 -0
  170. data/vendor/kreuzberg/src/language_detection/processor.rs +10 -0
  171. data/vendor/kreuzberg/src/lib.rs +2 -2
  172. data/vendor/kreuzberg/src/mcp/errors.rs +312 -0
  173. data/vendor/kreuzberg/src/mcp/format.rs +211 -0
  174. data/vendor/kreuzberg/src/mcp/mod.rs +9 -3
  175. data/vendor/kreuzberg/src/mcp/params.rs +196 -0
  176. data/vendor/kreuzberg/src/mcp/server.rs +39 -1438
  177. data/vendor/kreuzberg/src/mcp/tools/cache.rs +179 -0
  178. data/vendor/kreuzberg/src/mcp/tools/extraction.rs +403 -0
  179. data/vendor/kreuzberg/src/mcp/tools/mime.rs +150 -0
  180. data/vendor/kreuzberg/src/mcp/tools/mod.rs +11 -0
  181. data/vendor/kreuzberg/src/ocr/backends/easyocr.rs +96 -0
  182. data/vendor/kreuzberg/src/ocr/backends/mod.rs +7 -0
  183. data/vendor/kreuzberg/src/ocr/backends/paddleocr.rs +27 -0
  184. data/vendor/kreuzberg/src/ocr/backends/tesseract.rs +134 -0
  185. data/vendor/kreuzberg/src/ocr/hocr.rs +60 -16
  186. data/vendor/kreuzberg/src/ocr/language_registry.rs +11 -235
  187. data/vendor/kreuzberg/src/ocr/mod.rs +1 -0
  188. data/vendor/kreuzberg/src/ocr/processor/config.rs +203 -0
  189. data/vendor/kreuzberg/src/ocr/processor/execution.rs +494 -0
  190. data/vendor/kreuzberg/src/ocr/processor/mod.rs +265 -0
  191. data/vendor/kreuzberg/src/ocr/processor/validation.rs +145 -0
  192. data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +41 -24
  193. data/vendor/kreuzberg/src/pdf/bindings.rs +21 -8
  194. data/vendor/kreuzberg/src/pdf/hierarchy/bounding_box.rs +289 -0
  195. data/vendor/kreuzberg/src/pdf/hierarchy/clustering.rs +199 -0
  196. data/vendor/kreuzberg/src/pdf/{hierarchy.rs → hierarchy/extraction.rs} +6 -346
  197. data/vendor/kreuzberg/src/pdf/hierarchy/mod.rs +18 -0
  198. data/vendor/kreuzberg/src/plugins/extractor/mod.rs +319 -0
  199. data/vendor/kreuzberg/src/plugins/extractor/registry.rs +434 -0
  200. data/vendor/kreuzberg/src/plugins/extractor/trait.rs +391 -0
  201. data/vendor/kreuzberg/src/plugins/mod.rs +13 -0
  202. data/vendor/kreuzberg/src/plugins/ocr.rs +11 -0
  203. data/vendor/kreuzberg/src/plugins/processor/mod.rs +365 -0
  204. data/vendor/kreuzberg/src/plugins/processor/registry.rs +37 -0
  205. data/vendor/kreuzberg/src/plugins/processor/trait.rs +284 -0
  206. data/vendor/kreuzberg/src/plugins/registry/extractor.rs +416 -0
  207. data/vendor/kreuzberg/src/plugins/registry/mod.rs +116 -0
  208. data/vendor/kreuzberg/src/plugins/registry/ocr.rs +293 -0
  209. data/vendor/kreuzberg/src/plugins/registry/processor.rs +304 -0
  210. data/vendor/kreuzberg/src/plugins/registry/validator.rs +238 -0
  211. data/vendor/kreuzberg/src/plugins/validator/mod.rs +424 -0
  212. data/vendor/kreuzberg/src/plugins/validator/registry.rs +355 -0
  213. data/vendor/kreuzberg/src/plugins/validator/trait.rs +276 -0
  214. data/vendor/kreuzberg/src/stopwords/languages/asian.rs +40 -0
  215. data/vendor/kreuzberg/src/stopwords/languages/germanic.rs +36 -0
  216. data/vendor/kreuzberg/src/stopwords/languages/mod.rs +10 -0
  217. data/vendor/kreuzberg/src/stopwords/languages/other.rs +44 -0
  218. data/vendor/kreuzberg/src/stopwords/languages/romance.rs +36 -0
  219. data/vendor/kreuzberg/src/stopwords/languages/slavic.rs +36 -0
  220. data/vendor/kreuzberg/src/stopwords/mod.rs +7 -33
  221. data/vendor/kreuzberg/src/text/quality.rs +1 -1
  222. data/vendor/kreuzberg/src/text/quality_processor.rs +10 -0
  223. data/vendor/kreuzberg/src/text/token_reduction/core/analysis.rs +238 -0
  224. data/vendor/kreuzberg/src/text/token_reduction/core/mod.rs +8 -0
  225. data/vendor/kreuzberg/src/text/token_reduction/core/punctuation.rs +54 -0
  226. data/vendor/kreuzberg/src/text/token_reduction/core/reducer.rs +384 -0
  227. data/vendor/kreuzberg/src/text/token_reduction/core/sentence_selection.rs +68 -0
  228. data/vendor/kreuzberg/src/text/token_reduction/core/word_filtering.rs +156 -0
  229. data/vendor/kreuzberg/src/text/token_reduction/filters/general.rs +377 -0
  230. data/vendor/kreuzberg/src/text/token_reduction/filters/html.rs +51 -0
  231. data/vendor/kreuzberg/src/text/token_reduction/filters/markdown.rs +285 -0
  232. data/vendor/kreuzberg/src/text/token_reduction/filters.rs +131 -246
  233. data/vendor/kreuzberg/src/types/djot.rs +209 -0
  234. data/vendor/kreuzberg/src/types/extraction.rs +301 -0
  235. data/vendor/kreuzberg/src/types/formats.rs +443 -0
  236. data/vendor/kreuzberg/src/types/metadata.rs +560 -0
  237. data/vendor/kreuzberg/src/types/mod.rs +281 -0
  238. data/vendor/kreuzberg/src/types/page.rs +182 -0
  239. data/vendor/kreuzberg/src/types/serde_helpers.rs +132 -0
  240. data/vendor/kreuzberg/src/types/tables.rs +39 -0
  241. data/vendor/kreuzberg/src/utils/quality/heuristics.rs +58 -0
  242. data/vendor/kreuzberg/src/utils/{quality.rs → quality/mod.rs} +168 -489
  243. data/vendor/kreuzberg/src/utils/quality/patterns.rs +117 -0
  244. data/vendor/kreuzberg/src/utils/quality/scoring.rs +178 -0
  245. data/vendor/kreuzberg/src/utils/string_pool/buffer_pool.rs +325 -0
  246. data/vendor/kreuzberg/src/utils/string_pool/interned.rs +102 -0
  247. data/vendor/kreuzberg/src/utils/string_pool/language_pool.rs +119 -0
  248. data/vendor/kreuzberg/src/utils/string_pool/mime_pool.rs +235 -0
  249. data/vendor/kreuzberg/src/utils/string_pool/mod.rs +41 -0
  250. data/vendor/kreuzberg/tests/api_chunk.rs +313 -0
  251. data/vendor/kreuzberg/tests/api_embed.rs +6 -9
  252. data/vendor/kreuzberg/tests/batch_orchestration.rs +1 -0
  253. data/vendor/kreuzberg/tests/concurrency_stress.rs +7 -0
  254. data/vendor/kreuzberg/tests/core_integration.rs +1 -0
  255. data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +130 -0
  256. data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +5 -14
  257. data/vendor/kreuzberg/tests/format_integration.rs +2 -0
  258. data/vendor/kreuzberg/tests/helpers/mod.rs +1 -0
  259. data/vendor/kreuzberg/tests/html_table_test.rs +11 -11
  260. data/vendor/kreuzberg/tests/ocr_configuration.rs +16 -0
  261. data/vendor/kreuzberg/tests/ocr_errors.rs +18 -0
  262. data/vendor/kreuzberg/tests/ocr_quality.rs +9 -0
  263. data/vendor/kreuzberg/tests/ocr_stress.rs +1 -0
  264. data/vendor/kreuzberg/tests/pipeline_integration.rs +50 -0
  265. data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +13 -0
  266. data/vendor/kreuzberg/tests/plugin_system.rs +12 -0
  267. data/vendor/kreuzberg/tests/registry_integration_tests.rs +2 -0
  268. data/vendor/kreuzberg-ffi/Cargo.toml +2 -1
  269. data/vendor/kreuzberg-ffi/benches/result_view_benchmark.rs +2 -0
  270. data/vendor/kreuzberg-ffi/kreuzberg.h +347 -178
  271. data/vendor/kreuzberg-ffi/src/config/html.rs +318 -0
  272. data/vendor/kreuzberg-ffi/src/config/loader.rs +154 -0
  273. data/vendor/kreuzberg-ffi/src/config/merge.rs +104 -0
  274. data/vendor/kreuzberg-ffi/src/config/mod.rs +385 -0
  275. data/vendor/kreuzberg-ffi/src/config/parse.rs +91 -0
  276. data/vendor/kreuzberg-ffi/src/config/serialize.rs +118 -0
  277. data/vendor/kreuzberg-ffi/src/config_builder.rs +598 -0
  278. data/vendor/kreuzberg-ffi/src/error.rs +46 -14
  279. data/vendor/kreuzberg-ffi/src/helpers.rs +10 -0
  280. data/vendor/kreuzberg-ffi/src/html_options.rs +421 -0
  281. data/vendor/kreuzberg-ffi/src/lib.rs +16 -0
  282. data/vendor/kreuzberg-ffi/src/panic_shield.rs +11 -0
  283. data/vendor/kreuzberg-ffi/src/plugins/ocr_backend.rs +2 -0
  284. data/vendor/kreuzberg-ffi/src/result.rs +148 -122
  285. data/vendor/kreuzberg-ffi/src/result_view.rs +4 -0
  286. data/vendor/kreuzberg-tesseract/Cargo.toml +2 -2
  287. metadata +200 -28
  288. data/vendor/kreuzberg/src/api/server.rs +0 -518
  289. data/vendor/kreuzberg/src/core/config.rs +0 -1914
  290. data/vendor/kreuzberg/src/core/config_validation.rs +0 -949
  291. data/vendor/kreuzberg/src/core/extractor.rs +0 -1200
  292. data/vendor/kreuzberg/src/core/pipeline.rs +0 -1223
  293. data/vendor/kreuzberg/src/core/server_config.rs +0 -1220
  294. data/vendor/kreuzberg/src/extraction/html.rs +0 -1830
  295. data/vendor/kreuzberg/src/extraction/pptx.rs +0 -3102
  296. data/vendor/kreuzberg/src/extractors/epub.rs +0 -696
  297. data/vendor/kreuzberg/src/extractors/latex.rs +0 -653
  298. data/vendor/kreuzberg/src/extractors/opml.rs +0 -635
  299. data/vendor/kreuzberg/src/extractors/rtf.rs +0 -809
  300. data/vendor/kreuzberg/src/ocr/processor.rs +0 -858
  301. data/vendor/kreuzberg/src/plugins/extractor.rs +0 -1042
  302. data/vendor/kreuzberg/src/plugins/processor.rs +0 -650
  303. data/vendor/kreuzberg/src/plugins/registry.rs +0 -1339
  304. data/vendor/kreuzberg/src/plugins/validator.rs +0 -967
  305. data/vendor/kreuzberg/src/text/token_reduction/core.rs +0 -832
  306. data/vendor/kreuzberg/src/types.rs +0 -1713
  307. data/vendor/kreuzberg/src/utils/string_pool.rs +0 -762
  308. data/vendor/kreuzberg-ffi/src/config.rs +0 -1341
@@ -1,1914 +0,0 @@
1
- //! Configuration loading and management.
2
- //!
3
- //! This module provides utilities for loading extraction configuration from various
4
- //! sources (TOML, YAML, JSON) and discovering configuration files in the project hierarchy.
5
-
6
- use crate::{KreuzbergError, Result};
7
- use dashmap::DashMap;
8
- use serde::{Deserialize, Serialize};
9
- use std::collections::HashSet;
10
- use std::path::{Path, PathBuf};
11
- use std::sync::{Arc, LazyLock};
12
- use std::time::SystemTime;
13
-
14
- static CONFIG_CACHE: LazyLock<DashMap<PathBuf, (SystemTime, Arc<ExtractionConfig>)>> = LazyLock::new(DashMap::new);
15
-
16
- /// Page extraction and tracking configuration.
17
- ///
18
- /// Controls how pages are extracted, tracked, and represented in the extraction results.
19
- /// When `None`, page tracking is disabled.
20
- ///
21
- /// Page range tracking in chunk metadata (first_page/last_page) is automatically enabled
22
- /// when page boundaries are available and chunking is configured.
23
- #[derive(Debug, Clone, Serialize, Deserialize)]
24
- #[serde(default)]
25
- pub struct PageConfig {
26
- /// Extract pages as separate array (ExtractionResult.pages)
27
- #[serde(default)]
28
- pub extract_pages: bool,
29
-
30
- /// Insert page markers in main content string
31
- #[serde(default)]
32
- pub insert_page_markers: bool,
33
-
34
- /// Page marker format (use {page_num} placeholder)
35
- /// Default: "\n\n<!-- PAGE {page_num} -->\n\n"
36
- #[serde(default = "default_page_marker_format")]
37
- pub marker_format: String,
38
- }
39
-
40
- impl Default for PageConfig {
41
- fn default() -> Self {
42
- Self {
43
- extract_pages: false,
44
- insert_page_markers: false,
45
- marker_format: "\n\n<!-- PAGE {page_num} -->\n\n".to_string(),
46
- }
47
- }
48
- }
49
-
50
- /// Main extraction configuration.
51
- ///
52
- /// This struct contains all configuration options for the extraction process.
53
- /// It can be loaded from TOML, YAML, or JSON files, or created programmatically.
54
- ///
55
- /// # Example
56
- ///
57
- /// ```rust
58
- /// use kreuzberg::core::config::ExtractionConfig;
59
- ///
60
- /// // Create with defaults
61
- /// let config = ExtractionConfig::default();
62
- ///
63
- /// // Load from TOML file
64
- /// // let config = ExtractionConfig::from_toml_file("kreuzberg.toml")?;
65
- /// ```
66
- #[derive(Debug, Clone, Serialize, Deserialize)]
67
- pub struct ExtractionConfig {
68
- /// Enable caching of extraction results
69
- #[serde(default = "default_true")]
70
- pub use_cache: bool,
71
-
72
- /// Enable quality post-processing
73
- #[serde(default = "default_true")]
74
- pub enable_quality_processing: bool,
75
-
76
- /// OCR configuration (None = OCR disabled)
77
- #[serde(default)]
78
- pub ocr: Option<OcrConfig>,
79
-
80
- /// Force OCR even for searchable PDFs
81
- #[serde(default)]
82
- pub force_ocr: bool,
83
-
84
- /// Text chunking configuration (None = chunking disabled)
85
- #[serde(default)]
86
- pub chunking: Option<ChunkingConfig>,
87
-
88
- /// Image extraction configuration (None = no image extraction)
89
- #[serde(default)]
90
- pub images: Option<ImageExtractionConfig>,
91
-
92
- /// PDF-specific options (None = use defaults)
93
- #[cfg(feature = "pdf")]
94
- #[serde(default)]
95
- pub pdf_options: Option<PdfConfig>,
96
-
97
- /// Token reduction configuration (None = no token reduction)
98
- #[serde(default)]
99
- pub token_reduction: Option<TokenReductionConfig>,
100
-
101
- /// Language detection configuration (None = no language detection)
102
- #[serde(default)]
103
- pub language_detection: Option<LanguageDetectionConfig>,
104
-
105
- /// Page extraction configuration (None = no page tracking)
106
- #[serde(default)]
107
- pub pages: Option<PageConfig>,
108
-
109
- /// Keyword extraction configuration (None = no keyword extraction)
110
- #[cfg(any(feature = "keywords-yake", feature = "keywords-rake"))]
111
- #[serde(default)]
112
- pub keywords: Option<crate::keywords::KeywordConfig>,
113
-
114
- /// Post-processor configuration (None = use defaults)
115
- #[serde(default)]
116
- pub postprocessor: Option<PostProcessorConfig>,
117
-
118
- /// HTML to Markdown conversion options (None = use defaults)
119
- ///
120
- /// Configure how HTML documents are converted to Markdown, including heading styles,
121
- /// list formatting, code block styles, and preprocessing options.
122
- #[cfg(feature = "html")]
123
- #[serde(default)]
124
- pub html_options: Option<html_to_markdown_rs::ConversionOptions>,
125
-
126
- /// Maximum concurrent extractions in batch operations (None = num_cpus * 2).
127
- ///
128
- /// Limits parallelism to prevent resource exhaustion when processing
129
- /// large batches. Defaults to twice the number of CPU cores.
130
- #[serde(default)]
131
- pub max_concurrent_extractions: Option<usize>,
132
- }
133
-
134
- /// Post-processor configuration.
135
- #[derive(Debug, Clone, Serialize, Deserialize)]
136
- pub struct PostProcessorConfig {
137
- /// Enable post-processors
138
- #[serde(default = "default_true")]
139
- pub enabled: bool,
140
-
141
- /// Whitelist of processor names to run (None = all enabled)
142
- #[serde(default)]
143
- pub enabled_processors: Option<Vec<String>>,
144
-
145
- /// Blacklist of processor names to skip (None = none disabled)
146
- #[serde(default)]
147
- pub disabled_processors: Option<Vec<String>>,
148
-
149
- /// Pre-computed HashSet for O(1) enabled processor lookup
150
- #[serde(skip)]
151
- pub enabled_set: Option<HashSet<String>>,
152
-
153
- /// Pre-computed HashSet for O(1) disabled processor lookup
154
- #[serde(skip)]
155
- pub disabled_set: Option<HashSet<String>>,
156
- }
157
-
158
- impl PostProcessorConfig {
159
- /// Pre-compute HashSets for O(1) processor name lookups.
160
- ///
161
- /// This method converts the enabled/disabled processor Vec to HashSet
162
- /// for constant-time lookups in the pipeline.
163
- pub fn build_lookup_sets(&mut self) {
164
- if let Some(ref enabled) = self.enabled_processors {
165
- self.enabled_set = Some(enabled.iter().cloned().collect());
166
- }
167
- if let Some(ref disabled) = self.disabled_processors {
168
- self.disabled_set = Some(disabled.iter().cloned().collect());
169
- }
170
- }
171
- }
172
-
173
- /// OCR configuration.
174
- #[derive(Debug, Clone, Serialize, Deserialize)]
175
- pub struct OcrConfig {
176
- /// OCR backend: tesseract, easyocr, paddleocr
177
- #[serde(default = "default_tesseract_backend")]
178
- pub backend: String,
179
-
180
- /// Language code (e.g., "eng", "deu")
181
- #[serde(default = "default_eng")]
182
- pub language: String,
183
-
184
- /// Tesseract-specific configuration (optional)
185
- #[serde(default)]
186
- pub tesseract_config: Option<crate::types::TesseractConfig>,
187
- }
188
-
189
- impl Default for OcrConfig {
190
- fn default() -> Self {
191
- Self {
192
- backend: default_tesseract_backend(),
193
- language: default_eng(),
194
- tesseract_config: None,
195
- }
196
- }
197
- }
198
-
199
- /// Chunking configuration.
200
- #[derive(Debug, Clone, Serialize, Deserialize)]
201
- pub struct ChunkingConfig {
202
- /// Maximum characters per chunk
203
- #[serde(default = "default_chunk_size")]
204
- pub max_chars: usize,
205
-
206
- /// Overlap between chunks in characters
207
- #[serde(default = "default_chunk_overlap")]
208
- pub max_overlap: usize,
209
-
210
- /// Optional embedding configuration for chunk embeddings
211
- #[serde(skip_serializing_if = "Option::is_none")]
212
- pub embedding: Option<EmbeddingConfig>,
213
-
214
- /// Use a preset configuration (overrides individual settings if provided)
215
- #[serde(skip_serializing_if = "Option::is_none")]
216
- pub preset: Option<String>,
217
- }
218
-
219
- /// Embedding configuration for text chunks.
220
- ///
221
- /// Configures embedding generation using ONNX models via fastembed-rs.
222
- /// Requires the `embeddings` feature to be enabled.
223
- #[derive(Debug, Clone, Serialize, Deserialize)]
224
- pub struct EmbeddingConfig {
225
- /// The embedding model to use
226
- pub model: EmbeddingModelType,
227
-
228
- /// Whether to normalize embedding vectors (recommended for cosine similarity)
229
- #[serde(default = "default_normalize")]
230
- pub normalize: bool,
231
-
232
- /// Batch size for embedding generation
233
- #[serde(default = "default_batch_size")]
234
- pub batch_size: usize,
235
-
236
- /// Show model download progress
237
- #[serde(default)]
238
- pub show_download_progress: bool,
239
-
240
- /// Custom cache directory for model files
241
- ///
242
- /// Defaults to `~/.cache/kreuzberg/embeddings/` if not specified.
243
- /// Allows full customization of model download location.
244
- #[serde(skip_serializing_if = "Option::is_none")]
245
- pub cache_dir: Option<std::path::PathBuf>,
246
- }
247
-
248
- impl Default for EmbeddingConfig {
249
- fn default() -> Self {
250
- Self {
251
- model: EmbeddingModelType::Preset {
252
- name: "balanced".to_string(),
253
- },
254
- normalize: true,
255
- batch_size: 32,
256
- show_download_progress: false,
257
- cache_dir: None,
258
- }
259
- }
260
- }
261
-
262
- /// Embedding model types supported by Kreuzberg.
263
- #[derive(Debug, Clone, Serialize, Deserialize)]
264
- #[serde(tag = "type", rename_all = "snake_case")]
265
- pub enum EmbeddingModelType {
266
- /// Use a preset model configuration (recommended)
267
- Preset { name: String },
268
-
269
- /// Use a specific fastembed model by name
270
- #[cfg(feature = "embeddings")]
271
- FastEmbed { model: String, dimensions: usize },
272
-
273
- /// Use a custom ONNX model from HuggingFace
274
- Custom { model_id: String, dimensions: usize },
275
- }
276
-
277
- /// Image extraction configuration.
278
- #[derive(Debug, Clone, Serialize, Deserialize)]
279
- pub struct ImageExtractionConfig {
280
- /// Extract images from documents
281
- #[serde(default = "default_true")]
282
- pub extract_images: bool,
283
-
284
- /// Target DPI for image normalization
285
- #[serde(default = "default_target_dpi")]
286
- pub target_dpi: i32,
287
-
288
- /// Maximum dimension for images (width or height)
289
- #[serde(default = "default_max_dimension")]
290
- pub max_image_dimension: i32,
291
-
292
- /// Automatically adjust DPI based on image content
293
- #[serde(default = "default_true")]
294
- pub auto_adjust_dpi: bool,
295
-
296
- /// Minimum DPI threshold
297
- #[serde(default = "default_min_dpi")]
298
- pub min_dpi: i32,
299
-
300
- /// Maximum DPI threshold
301
- #[serde(default = "default_max_dpi")]
302
- pub max_dpi: i32,
303
- }
304
-
305
- /// PDF-specific configuration.
306
- #[cfg(feature = "pdf")]
307
- #[derive(Debug, Clone, Serialize, Deserialize)]
308
- pub struct PdfConfig {
309
- /// Extract images from PDF
310
- #[serde(default)]
311
- pub extract_images: bool,
312
-
313
- /// List of passwords to try when opening encrypted PDFs
314
- #[serde(default)]
315
- pub passwords: Option<Vec<String>>,
316
-
317
- /// Extract PDF metadata
318
- #[serde(default = "default_true")]
319
- pub extract_metadata: bool,
320
-
321
- /// Hierarchy extraction configuration (None = hierarchy extraction disabled)
322
- #[serde(default)]
323
- pub hierarchy: Option<HierarchyConfig>,
324
- }
325
-
326
- /// Hierarchy extraction configuration for PDF text structure analysis.
327
- ///
328
- /// Enables extraction of document hierarchy levels (H1-H6) based on font size
329
- /// clustering and semantic analysis. When enabled, hierarchical blocks are
330
- /// included in page content.
331
- #[derive(Debug, Clone, Serialize, Deserialize)]
332
- pub struct HierarchyConfig {
333
- /// Enable hierarchy extraction
334
- #[serde(default = "default_true")]
335
- pub enabled: bool,
336
-
337
- /// Number of font size clusters to use for hierarchy levels (1-7)
338
- ///
339
- /// Default: 6, which provides H1-H6 heading levels with body text.
340
- /// Larger values create more fine-grained hierarchy levels.
341
- #[serde(default = "default_k_clusters")]
342
- pub k_clusters: usize,
343
-
344
- /// Include bounding box information in hierarchy blocks
345
- #[serde(default = "default_true")]
346
- pub include_bbox: bool,
347
-
348
- /// OCR coverage threshold for smart OCR triggering (0.0-1.0)
349
- ///
350
- /// Determines when OCR should be triggered based on text block coverage.
351
- /// OCR is triggered when text blocks cover less than this fraction of the page.
352
- /// Default: 0.5 (trigger OCR if less than 50% of page has text)
353
- #[serde(default = "default_ocr_coverage_threshold")]
354
- pub ocr_coverage_threshold: Option<f32>,
355
- }
356
-
357
- impl Default for HierarchyConfig {
358
- fn default() -> Self {
359
- Self {
360
- enabled: true,
361
- k_clusters: 6,
362
- include_bbox: true,
363
- ocr_coverage_threshold: None,
364
- }
365
- }
366
- }
367
-
368
- fn default_k_clusters() -> usize {
369
- 6
370
- }
371
-
372
- fn default_ocr_coverage_threshold() -> Option<f32> {
373
- None
374
- }
375
-
376
- /// Token reduction configuration.
377
- #[derive(Debug, Clone, Serialize, Deserialize)]
378
- pub struct TokenReductionConfig {
379
- /// Reduction mode: "off", "light", "moderate", "aggressive", "maximum"
380
- #[serde(default = "default_reduction_mode")]
381
- pub mode: String,
382
-
383
- /// Preserve important words (capitalized, technical terms)
384
- #[serde(default = "default_true")]
385
- pub preserve_important_words: bool,
386
- }
387
-
388
- /// Language detection configuration.
389
- #[derive(Debug, Clone, Serialize, Deserialize)]
390
- pub struct LanguageDetectionConfig {
391
- /// Enable language detection
392
- #[serde(default = "default_true")]
393
- pub enabled: bool,
394
-
395
- /// Minimum confidence threshold (0.0-1.0)
396
- #[serde(default = "default_confidence")]
397
- pub min_confidence: f64,
398
-
399
- /// Detect multiple languages in the document
400
- #[serde(default)]
401
- pub detect_multiple: bool,
402
- }
403
-
404
- fn default_true() -> bool {
405
- true
406
- }
407
- fn default_eng() -> String {
408
- "eng".to_string()
409
- }
410
- fn default_tesseract_backend() -> String {
411
- "tesseract".to_string()
412
- }
413
- fn default_page_marker_format() -> String {
414
- "\n\n<!-- PAGE {page_num} -->\n\n".to_string()
415
- }
416
- fn default_chunk_size() -> usize {
417
- 1000
418
- }
419
- fn default_chunk_overlap() -> usize {
420
- 200
421
- }
422
- fn default_normalize() -> bool {
423
- true
424
- }
425
- fn default_batch_size() -> usize {
426
- 32
427
- }
428
- fn default_target_dpi() -> i32 {
429
- 300
430
- }
431
- fn default_max_dimension() -> i32 {
432
- 4096
433
- }
434
- fn default_min_dpi() -> i32 {
435
- 72
436
- }
437
- fn default_max_dpi() -> i32 {
438
- 600
439
- }
440
- fn default_reduction_mode() -> String {
441
- "off".to_string()
442
- }
443
- fn default_confidence() -> f64 {
444
- 0.8
445
- }
446
-
447
- impl Default for ExtractionConfig {
448
- fn default() -> Self {
449
- Self {
450
- use_cache: true,
451
- enable_quality_processing: true,
452
- ocr: None,
453
- force_ocr: false,
454
- chunking: None,
455
- images: None,
456
- #[cfg(feature = "pdf")]
457
- pdf_options: None,
458
- token_reduction: None,
459
- language_detection: None,
460
- pages: None,
461
- #[cfg(any(feature = "keywords-yake", feature = "keywords-rake"))]
462
- keywords: None,
463
- postprocessor: None,
464
- #[cfg(feature = "html")]
465
- html_options: None,
466
- max_concurrent_extractions: None,
467
- }
468
- }
469
- }
470
-
471
- impl Default for PostProcessorConfig {
472
- fn default() -> Self {
473
- Self {
474
- enabled: true,
475
- enabled_processors: None,
476
- disabled_processors: None,
477
- enabled_set: None,
478
- disabled_set: None,
479
- }
480
- }
481
- }
482
-
483
- impl ExtractionConfig {
484
- /// Check if image processing is needed by examining OCR and image extraction settings.
485
- ///
486
- /// Returns `true` if either OCR is enabled or image extraction is configured,
487
- /// indicating that image decompression and processing should occur.
488
- /// Returns `false` if both are disabled, allowing optimization to skip unnecessary
489
- /// image decompression for text-only extraction workflows.
490
- ///
491
- /// # Optimization Impact
492
- /// For text-only extractions (no OCR, no image extraction), skipping image
493
- /// decompression can improve CPU utilization by 5-10% by avoiding wasteful
494
- /// image I/O and processing when results won't be used.
495
- pub fn needs_image_processing(&self) -> bool {
496
- let ocr_enabled = self.ocr.is_some();
497
-
498
- let image_extraction_enabled = self.images.as_ref().map(|i| i.extract_images).unwrap_or(false);
499
-
500
- ocr_enabled || image_extraction_enabled
501
- }
502
-
503
- /// Apply environment variable overrides to configuration.
504
- ///
505
- /// Environment variables have the highest precedence and will override any values
506
- /// loaded from configuration files. This method supports the following environment variables:
507
- ///
508
- /// - `KREUZBERG_OCR_LANGUAGE`: OCR language (ISO 639-1 or 639-3 code, e.g., "eng", "fra", "deu")
509
- /// - `KREUZBERG_OCR_BACKEND`: OCR backend ("tesseract", "easyocr", or "paddleocr")
510
- /// - `KREUZBERG_CHUNKING_MAX_CHARS`: Maximum characters per chunk (positive integer)
511
- /// - `KREUZBERG_CHUNKING_MAX_OVERLAP`: Maximum overlap between chunks (non-negative integer)
512
- /// - `KREUZBERG_CACHE_ENABLED`: Cache enabled flag ("true" or "false")
513
- /// - `KREUZBERG_TOKEN_REDUCTION_MODE`: Token reduction mode ("off", "light", "moderate", "aggressive", or "maximum")
514
- ///
515
- /// # Behavior
516
- ///
517
- /// - If an environment variable is set and valid, it overrides the current configuration value
518
- /// - If a required parent config is `None` (e.g., `self.ocr` is None), it's created with defaults before applying the override
519
- /// - Invalid values return a `KreuzbergError::Validation` with helpful error messages
520
- /// - Missing or unset environment variables are silently ignored
521
- ///
522
- /// # Example
523
- ///
524
- /// ```rust
525
- /// # use kreuzberg::core::config::ExtractionConfig;
526
- /// # fn example() -> kreuzberg::Result<()> {
527
- /// let mut config = ExtractionConfig::from_file("config.toml")?;
528
- /// // Set KREUZBERG_OCR_LANGUAGE=fra before calling
529
- /// config.apply_env_overrides()?; // OCR language is now "fra"
530
- /// # Ok(())
531
- /// # }
532
- /// ```
533
- ///
534
- /// # Errors
535
- ///
536
- /// Returns `KreuzbergError::Validation` if:
537
- /// - An environment variable contains an invalid value
538
- /// - A number cannot be parsed as the expected type
539
- /// - A boolean is not "true" or "false"
540
- pub fn apply_env_overrides(&mut self) -> Result<()> {
541
- use crate::core::config_validation::{
542
- validate_chunking_params, validate_language_code, validate_ocr_backend, validate_token_reduction_level,
543
- };
544
-
545
- // KREUZBERG_OCR_LANGUAGE override
546
- if let Ok(lang) = std::env::var("KREUZBERG_OCR_LANGUAGE") {
547
- validate_language_code(&lang)?;
548
- if self.ocr.is_none() {
549
- self.ocr = Some(OcrConfig::default());
550
- }
551
- if let Some(ref mut ocr) = self.ocr {
552
- ocr.language = lang;
553
- }
554
- }
555
-
556
- // KREUZBERG_OCR_BACKEND override
557
- if let Ok(backend) = std::env::var("KREUZBERG_OCR_BACKEND") {
558
- validate_ocr_backend(&backend)?;
559
- if self.ocr.is_none() {
560
- self.ocr = Some(OcrConfig::default());
561
- }
562
- if let Some(ref mut ocr) = self.ocr {
563
- ocr.backend = backend;
564
- }
565
- }
566
-
567
- // KREUZBERG_CHUNKING_MAX_CHARS override
568
- if let Ok(max_chars_str) = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS") {
569
- let max_chars: usize = max_chars_str.parse().map_err(|_| KreuzbergError::Validation {
570
- message: format!(
571
- "Invalid value for KREUZBERG_CHUNKING_MAX_CHARS: '{}'. Must be a positive integer.",
572
- max_chars_str
573
- ),
574
- source: None,
575
- })?;
576
-
577
- if max_chars == 0 {
578
- return Err(KreuzbergError::Validation {
579
- message: "KREUZBERG_CHUNKING_MAX_CHARS must be greater than 0".to_string(),
580
- source: None,
581
- });
582
- }
583
-
584
- if self.chunking.is_none() {
585
- self.chunking = Some(ChunkingConfig {
586
- max_chars: default_chunk_size(),
587
- max_overlap: default_chunk_overlap(),
588
- embedding: None,
589
- preset: None,
590
- });
591
- }
592
-
593
- if let Some(ref mut chunking) = self.chunking {
594
- // Validate against current overlap before updating
595
- validate_chunking_params(max_chars, chunking.max_overlap)?;
596
- chunking.max_chars = max_chars;
597
- }
598
- }
599
-
600
- // KREUZBERG_CHUNKING_MAX_OVERLAP override
601
- if let Ok(max_overlap_str) = std::env::var("KREUZBERG_CHUNKING_MAX_OVERLAP") {
602
- let max_overlap: usize = max_overlap_str.parse().map_err(|_| KreuzbergError::Validation {
603
- message: format!(
604
- "Invalid value for KREUZBERG_CHUNKING_MAX_OVERLAP: '{}'. Must be a non-negative integer.",
605
- max_overlap_str
606
- ),
607
- source: None,
608
- })?;
609
-
610
- if self.chunking.is_none() {
611
- self.chunking = Some(ChunkingConfig {
612
- max_chars: default_chunk_size(),
613
- max_overlap: default_chunk_overlap(),
614
- embedding: None,
615
- preset: None,
616
- });
617
- }
618
-
619
- if let Some(ref mut chunking) = self.chunking {
620
- // Validate against current max_chars before updating
621
- validate_chunking_params(chunking.max_chars, max_overlap)?;
622
- chunking.max_overlap = max_overlap;
623
- }
624
- }
625
-
626
- // KREUZBERG_CACHE_ENABLED override
627
- if let Ok(cache_str) = std::env::var("KREUZBERG_CACHE_ENABLED") {
628
- let cache_enabled = match cache_str.to_lowercase().as_str() {
629
- "true" => true,
630
- "false" => false,
631
- _ => {
632
- return Err(KreuzbergError::Validation {
633
- message: format!(
634
- "Invalid value for KREUZBERG_CACHE_ENABLED: '{}'. Must be 'true' or 'false'.",
635
- cache_str
636
- ),
637
- source: None,
638
- });
639
- }
640
- };
641
- self.use_cache = cache_enabled;
642
- }
643
-
644
- // KREUZBERG_TOKEN_REDUCTION_MODE override
645
- if let Ok(mode) = std::env::var("KREUZBERG_TOKEN_REDUCTION_MODE") {
646
- validate_token_reduction_level(&mode)?;
647
- if self.token_reduction.is_none() {
648
- self.token_reduction = Some(TokenReductionConfig {
649
- mode: default_reduction_mode(),
650
- preserve_important_words: default_true(),
651
- });
652
- }
653
- if let Some(ref mut token_reduction) = self.token_reduction {
654
- token_reduction.mode = mode;
655
- }
656
- }
657
-
658
- Ok(())
659
- }
660
-
661
- /// Load configuration from a TOML file.
662
- ///
663
- /// # Arguments
664
- ///
665
- /// * `path` - Path to the TOML file
666
- ///
667
- /// # Errors
668
- ///
669
- /// Returns `KreuzbergError::Validation` if file doesn't exist or is invalid TOML.
670
- pub fn from_toml_file(path: impl AsRef<Path>) -> Result<Self> {
671
- let path = path.as_ref();
672
-
673
- let metadata = std::fs::metadata(path)
674
- .map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
675
- let mtime = metadata.modified().map_err(|e| {
676
- KreuzbergError::validation(format!("Failed to get modification time for {}: {}", path.display(), e))
677
- })?;
678
-
679
- if let Some(entry) = CONFIG_CACHE.get(path)
680
- && entry.0 == mtime
681
- {
682
- return Ok((*entry.1).clone());
683
- }
684
-
685
- let content = std::fs::read_to_string(path)
686
- .map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
687
-
688
- let config: Self = toml::from_str(&content)
689
- .map_err(|e| KreuzbergError::validation(format!("Invalid TOML in {}: {}", path.display(), e)))?;
690
-
691
- let config_arc = Arc::new(config.clone());
692
- CONFIG_CACHE.insert(path.to_path_buf(), (mtime, config_arc));
693
-
694
- Ok(config)
695
- }
696
-
697
- /// Load configuration from a YAML file.
698
- pub fn from_yaml_file(path: impl AsRef<Path>) -> Result<Self> {
699
- let path = path.as_ref();
700
-
701
- let metadata = std::fs::metadata(path)
702
- .map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
703
- let mtime = metadata.modified().map_err(|e| {
704
- KreuzbergError::validation(format!("Failed to get modification time for {}: {}", path.display(), e))
705
- })?;
706
-
707
- if let Some(entry) = CONFIG_CACHE.get(path)
708
- && entry.0 == mtime
709
- {
710
- return Ok((*entry.1).clone());
711
- }
712
-
713
- let content = std::fs::read_to_string(path)
714
- .map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
715
-
716
- let config: Self = serde_yaml_ng::from_str(&content)
717
- .map_err(|e| KreuzbergError::validation(format!("Invalid YAML in {}: {}", path.display(), e)))?;
718
-
719
- let config_arc = Arc::new(config.clone());
720
- CONFIG_CACHE.insert(path.to_path_buf(), (mtime, config_arc));
721
-
722
- Ok(config)
723
- }
724
-
725
- /// Load configuration from a JSON file.
726
- pub fn from_json_file(path: impl AsRef<Path>) -> Result<Self> {
727
- let path = path.as_ref();
728
-
729
- let metadata = std::fs::metadata(path)
730
- .map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
731
- let mtime = metadata.modified().map_err(|e| {
732
- KreuzbergError::validation(format!("Failed to get modification time for {}: {}", path.display(), e))
733
- })?;
734
-
735
- if let Some(entry) = CONFIG_CACHE.get(path)
736
- && entry.0 == mtime
737
- {
738
- return Ok((*entry.1).clone());
739
- }
740
-
741
- let content = std::fs::read_to_string(path)
742
- .map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
743
-
744
- let config: Self = serde_json::from_str(&content)
745
- .map_err(|e| KreuzbergError::validation(format!("Invalid JSON in {}: {}", path.display(), e)))?;
746
-
747
- let config_arc = Arc::new(config.clone());
748
- CONFIG_CACHE.insert(path.to_path_buf(), (mtime, config_arc));
749
-
750
- Ok(config)
751
- }
752
-
753
- /// Load configuration from a file, auto-detecting format by extension.
754
- ///
755
- /// Supported formats:
756
- /// - `.toml` - TOML format
757
- /// - `.yaml` - YAML format
758
- /// - `.json` - JSON format
759
- ///
760
- /// # Arguments
761
- ///
762
- /// * `path` - Path to the configuration file
763
- ///
764
- /// # Errors
765
- ///
766
- /// Returns `KreuzbergError::Validation` if:
767
- /// - File doesn't exist
768
- /// - File extension is not supported
769
- /// - File content is invalid for the detected format
770
- ///
771
- /// # Example
772
- ///
773
- /// ```rust
774
- /// use kreuzberg::core::config::ExtractionConfig;
775
- ///
776
- /// // Auto-detects TOML format
777
- /// // let config = ExtractionConfig::from_file("kreuzberg.toml")?;
778
- ///
779
- /// // Auto-detects YAML format
780
- /// // let config = ExtractionConfig::from_file("kreuzberg.yaml")?;
781
- /// ```
782
- pub fn from_file(path: impl AsRef<Path>) -> Result<Self> {
783
- let path = path.as_ref();
784
-
785
- let metadata = std::fs::metadata(path)
786
- .map_err(|e| KreuzbergError::validation(format!("Failed to read config file {}: {}", path.display(), e)))?;
787
- let mtime = metadata.modified().map_err(|e| {
788
- KreuzbergError::validation(format!("Failed to get modification time for {}: {}", path.display(), e))
789
- })?;
790
-
791
- if let Some(entry) = CONFIG_CACHE.get(path)
792
- && entry.0 == mtime
793
- {
794
- return Ok((*entry.1).clone());
795
- }
796
-
797
- let extension = path.extension().and_then(|ext| ext.to_str()).ok_or_else(|| {
798
- KreuzbergError::validation(format!(
799
- "Cannot determine file format: no extension found in {}",
800
- path.display()
801
- ))
802
- })?;
803
-
804
- let config = match extension.to_lowercase().as_str() {
805
- "toml" => Self::from_toml_file(path)?,
806
- "yaml" | "yml" => Self::from_yaml_file(path)?,
807
- "json" => Self::from_json_file(path)?,
808
- _ => {
809
- return Err(KreuzbergError::validation(format!(
810
- "Unsupported config file format: .{}. Supported formats: .toml, .yaml, .json",
811
- extension
812
- )));
813
- }
814
- };
815
-
816
- let config_arc = Arc::new(config.clone());
817
- CONFIG_CACHE.insert(path.to_path_buf(), (mtime, config_arc));
818
-
819
- Ok(config)
820
- }
821
-
822
- /// Discover configuration file in parent directories.
823
- ///
824
- /// Searches for `kreuzberg.toml` in current directory and parent directories.
825
- ///
826
- /// # Returns
827
- ///
828
- /// - `Some(config)` if found
829
- /// - `None` if no config file found
830
- pub fn discover() -> Result<Option<Self>> {
831
- let mut current = std::env::current_dir().map_err(KreuzbergError::Io)?;
832
-
833
- loop {
834
- let kreuzberg_toml = current.join("kreuzberg.toml");
835
- if kreuzberg_toml.exists() {
836
- return Ok(Some(Self::from_toml_file(kreuzberg_toml)?));
837
- }
838
-
839
- if let Some(parent) = current.parent() {
840
- current = parent.to_path_buf();
841
- } else {
842
- break;
843
- }
844
- }
845
-
846
- Ok(None)
847
- }
848
- }
849
-
850
- #[cfg(test)]
851
- #[allow(unsafe_code)]
852
- mod tests {
853
- use super::*;
854
- use std::fs;
855
- use tempfile::tempdir;
856
-
857
- fn set_env(key: &str, value: &str) {
858
- unsafe {
859
- std::env::set_var(key, value);
860
- }
861
- }
862
-
863
- #[allow(unsafe_code)]
864
- fn remove_env(key: &str) {
865
- unsafe {
866
- std::env::remove_var(key);
867
- }
868
- }
869
-
870
- #[allow(unsafe_code)]
871
- fn restore_env(key: &str, value: Option<String>) {
872
- unsafe {
873
- if let Some(v) = value {
874
- std::env::set_var(key, v);
875
- } else {
876
- std::env::remove_var(key);
877
- }
878
- }
879
- }
880
-
881
- #[test]
882
- fn test_default_config() {
883
- let config = ExtractionConfig::default();
884
- assert!(config.use_cache);
885
- assert!(config.enable_quality_processing);
886
- assert!(config.ocr.is_none());
887
- }
888
-
889
- #[test]
890
- fn test_from_toml_file() {
891
- let dir = tempdir().unwrap();
892
- let config_path = dir.path().join("kreuzberg.toml");
893
-
894
- fs::write(
895
- &config_path,
896
- r#"
897
- use_cache = false
898
- enable_quality_processing = true
899
- "#,
900
- )
901
- .unwrap();
902
-
903
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
904
- assert!(!config.use_cache);
905
- assert!(config.enable_quality_processing);
906
- }
907
-
908
- #[test]
909
- fn test_from_file_auto_detects_toml() {
910
- let dir = tempdir().unwrap();
911
- let config_path = dir.path().join("kreuzberg.toml");
912
-
913
- fs::write(
914
- &config_path,
915
- r#"
916
- use_cache = false
917
- enable_quality_processing = true
918
- "#,
919
- )
920
- .unwrap();
921
-
922
- let config = ExtractionConfig::from_file(&config_path).unwrap();
923
- assert!(!config.use_cache);
924
- assert!(config.enable_quality_processing);
925
- }
926
-
927
- #[test]
928
- fn test_from_file_auto_detects_yaml() {
929
- let dir = tempdir().unwrap();
930
- let config_path = dir.path().join("kreuzberg.yaml");
931
-
932
- fs::write(
933
- &config_path,
934
- r#"
935
- use_cache: true
936
- enable_quality_processing: false
937
- "#,
938
- )
939
- .unwrap();
940
-
941
- let config = ExtractionConfig::from_file(&config_path).unwrap();
942
- assert!(config.use_cache);
943
- assert!(!config.enable_quality_processing);
944
- }
945
-
946
- #[test]
947
- fn test_from_file_unsupported_extension() {
948
- let dir = tempdir().unwrap();
949
- let config_path = dir.path().join("kreuzberg.txt");
950
-
951
- fs::write(&config_path, "use_cache = false").unwrap();
952
-
953
- let result = ExtractionConfig::from_file(&config_path);
954
- assert!(result.is_err());
955
- let err = result.unwrap_err();
956
- assert!(err.to_string().contains("Unsupported config file format"));
957
- }
958
-
959
- #[test]
960
- fn test_from_file_no_extension() {
961
- let dir = tempdir().unwrap();
962
- let config_path = dir.path().join("kreuzberg");
963
-
964
- fs::write(&config_path, "use_cache = false").unwrap();
965
-
966
- let result = ExtractionConfig::from_file(&config_path);
967
- assert!(result.is_err());
968
- let err = result.unwrap_err();
969
- assert!(err.to_string().contains("no extension found"));
970
- }
971
-
972
- #[test]
973
- fn test_discover_kreuzberg_toml() {
974
- let dir = tempdir().unwrap();
975
- let config_path = dir.path().join("kreuzberg.toml");
976
-
977
- fs::write(
978
- &config_path,
979
- r#"
980
- use_cache = false
981
- enable_quality_processing = true
982
- "#,
983
- )
984
- .unwrap();
985
-
986
- let original_dir = std::env::current_dir().unwrap();
987
- std::env::set_current_dir(&dir).unwrap();
988
-
989
- let result = std::panic::catch_unwind(|| {
990
- let config = ExtractionConfig::discover().unwrap();
991
- assert!(config.is_some());
992
- assert!(!config.unwrap().use_cache);
993
- });
994
-
995
- std::env::set_current_dir(&original_dir).unwrap();
996
-
997
- if let Err(e) = result {
998
- std::panic::resume_unwind(e);
999
- }
1000
- }
1001
-
1002
- #[test]
1003
- fn test_v4_config_with_ocr_and_chunking() {
1004
- let dir = tempdir().unwrap();
1005
- let config_path = dir.path().join("kreuzberg.toml");
1006
-
1007
- fs::write(
1008
- &config_path,
1009
- r#"
1010
- use_cache = true
1011
- enable_quality_processing = false
1012
-
1013
- [ocr]
1014
- backend = "tesseract"
1015
- language = "eng"
1016
-
1017
- [chunking]
1018
- max_chars = 2000
1019
- max_overlap = 300
1020
- "#,
1021
- )
1022
- .unwrap();
1023
-
1024
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1025
- assert!(config.use_cache);
1026
- assert!(!config.enable_quality_processing);
1027
- assert!(config.ocr.is_some());
1028
- assert_eq!(config.ocr.unwrap().backend, "tesseract");
1029
- assert!(config.chunking.is_some());
1030
- assert_eq!(config.chunking.unwrap().max_chars, 2000);
1031
- }
1032
-
1033
- #[test]
1034
- fn test_config_with_image_extraction() {
1035
- let dir = tempdir().unwrap();
1036
- let config_path = dir.path().join("kreuzberg.toml");
1037
-
1038
- fs::write(
1039
- &config_path,
1040
- r#"
1041
- use_cache = true
1042
-
1043
- [images]
1044
- extract_images = true
1045
- target_dpi = 300
1046
- max_image_dimension = 4096
1047
- auto_adjust_dpi = true
1048
- min_dpi = 72
1049
- max_dpi = 600
1050
- "#,
1051
- )
1052
- .unwrap();
1053
-
1054
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1055
- assert!(config.images.is_some());
1056
- let images = config.images.unwrap();
1057
- assert!(images.extract_images);
1058
- assert_eq!(images.target_dpi, 300);
1059
- assert_eq!(images.max_image_dimension, 4096);
1060
- assert!(images.auto_adjust_dpi);
1061
- assert_eq!(images.min_dpi, 72);
1062
- assert_eq!(images.max_dpi, 600);
1063
- }
1064
-
1065
- #[test]
1066
- #[cfg(feature = "pdf")]
1067
- fn test_config_with_pdf_options() {
1068
- let dir = tempdir().unwrap();
1069
- let config_path = dir.path().join("kreuzberg.toml");
1070
-
1071
- fs::write(
1072
- &config_path,
1073
- r#"
1074
- use_cache = true
1075
-
1076
- [pdf_options]
1077
- extract_images = true
1078
- passwords = ["password1", "password2"]
1079
- extract_metadata = true
1080
- "#,
1081
- )
1082
- .unwrap();
1083
-
1084
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1085
- assert!(config.pdf_options.is_some());
1086
- let pdf = config.pdf_options.unwrap();
1087
- assert!(pdf.extract_images);
1088
- assert!(pdf.extract_metadata);
1089
- assert!(pdf.passwords.is_some());
1090
- let passwords = pdf.passwords.unwrap();
1091
- assert_eq!(passwords.len(), 2);
1092
- assert_eq!(passwords[0], "password1");
1093
- assert_eq!(passwords[1], "password2");
1094
- }
1095
-
1096
- #[test]
1097
- fn test_config_with_token_reduction() {
1098
- let dir = tempdir().unwrap();
1099
- let config_path = dir.path().join("kreuzberg.toml");
1100
-
1101
- fs::write(
1102
- &config_path,
1103
- r#"
1104
- use_cache = true
1105
-
1106
- [token_reduction]
1107
- mode = "aggressive"
1108
- preserve_important_words = true
1109
- "#,
1110
- )
1111
- .unwrap();
1112
-
1113
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1114
- assert!(config.token_reduction.is_some());
1115
- let token = config.token_reduction.unwrap();
1116
- assert_eq!(token.mode, "aggressive");
1117
- assert!(token.preserve_important_words);
1118
- }
1119
-
1120
- #[test]
1121
- fn test_config_with_language_detection() {
1122
- let dir = tempdir().unwrap();
1123
- let config_path = dir.path().join("kreuzberg.toml");
1124
-
1125
- fs::write(
1126
- &config_path,
1127
- r#"
1128
- use_cache = true
1129
-
1130
- [language_detection]
1131
- enabled = true
1132
- min_confidence = 0.9
1133
- detect_multiple = true
1134
- "#,
1135
- )
1136
- .unwrap();
1137
-
1138
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1139
- assert!(config.language_detection.is_some());
1140
- let lang = config.language_detection.unwrap();
1141
- assert!(lang.enabled);
1142
- assert_eq!(lang.min_confidence, 0.9);
1143
- assert!(lang.detect_multiple);
1144
- }
1145
-
1146
- #[test]
1147
- fn test_config_with_all_optional_fields() {
1148
- let dir = tempdir().unwrap();
1149
- let config_path = dir.path().join("kreuzberg.toml");
1150
-
1151
- fs::write(
1152
- &config_path,
1153
- r#"
1154
- use_cache = true
1155
- enable_quality_processing = true
1156
- force_ocr = false
1157
-
1158
- [ocr]
1159
- backend = "tesseract"
1160
- language = "eng"
1161
-
1162
- [chunking]
1163
- max_chars = 1500
1164
- max_overlap = 250
1165
-
1166
- [images]
1167
- extract_images = true
1168
- target_dpi = 300
1169
-
1170
- [pdf_options]
1171
- extract_images = false
1172
- extract_metadata = true
1173
-
1174
- [token_reduction]
1175
- mode = "moderate"
1176
-
1177
- [language_detection]
1178
- enabled = true
1179
- "#,
1180
- )
1181
- .unwrap();
1182
-
1183
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1184
- assert!(config.use_cache);
1185
- assert!(config.enable_quality_processing);
1186
- assert!(!config.force_ocr);
1187
- assert!(config.ocr.is_some());
1188
- assert!(config.chunking.is_some());
1189
- assert!(config.images.is_some());
1190
- assert!(config.token_reduction.is_some());
1191
- assert!(config.language_detection.is_some());
1192
- #[cfg(feature = "pdf")]
1193
- assert!(config.pdf_options.is_some());
1194
- }
1195
-
1196
- #[test]
1197
- fn test_image_config_defaults() {
1198
- let dir = tempdir().unwrap();
1199
- let config_path = dir.path().join("kreuzberg.toml");
1200
-
1201
- fs::write(
1202
- &config_path,
1203
- r#"
1204
- [images]
1205
- "#,
1206
- )
1207
- .unwrap();
1208
-
1209
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1210
- let images = config.images.unwrap();
1211
- assert!(images.extract_images);
1212
- assert_eq!(images.target_dpi, 300);
1213
- assert_eq!(images.max_image_dimension, 4096);
1214
- assert!(images.auto_adjust_dpi);
1215
- assert_eq!(images.min_dpi, 72);
1216
- assert_eq!(images.max_dpi, 600);
1217
- }
1218
-
1219
- #[test]
1220
- fn test_token_reduction_defaults() {
1221
- let dir = tempdir().unwrap();
1222
- let config_path = dir.path().join("kreuzberg.toml");
1223
-
1224
- fs::write(
1225
- &config_path,
1226
- r#"
1227
- [token_reduction]
1228
- "#,
1229
- )
1230
- .unwrap();
1231
-
1232
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1233
- let token = config.token_reduction.unwrap();
1234
- assert_eq!(token.mode, "off");
1235
- assert!(token.preserve_important_words);
1236
- }
1237
-
1238
- #[test]
1239
- fn test_language_detection_defaults() {
1240
- let dir = tempdir().unwrap();
1241
- let config_path = dir.path().join("kreuzberg.toml");
1242
-
1243
- fs::write(
1244
- &config_path,
1245
- r#"
1246
- [language_detection]
1247
- "#,
1248
- )
1249
- .unwrap();
1250
-
1251
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1252
- let lang = config.language_detection.unwrap();
1253
- assert!(lang.enabled);
1254
- assert_eq!(lang.min_confidence, 0.8);
1255
- assert!(!lang.detect_multiple);
1256
- }
1257
-
1258
- #[test]
1259
- #[cfg(feature = "pdf")]
1260
- fn test_pdf_config_defaults() {
1261
- let dir = tempdir().unwrap();
1262
- let config_path = dir.path().join("kreuzberg.toml");
1263
-
1264
- fs::write(
1265
- &config_path,
1266
- r#"
1267
- [pdf_options]
1268
- "#,
1269
- )
1270
- .unwrap();
1271
-
1272
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1273
- let pdf = config.pdf_options.unwrap();
1274
- assert!(!pdf.extract_images);
1275
- assert!(pdf.extract_metadata);
1276
- assert!(pdf.passwords.is_none());
1277
- }
1278
-
1279
- #[test]
1280
- fn test_config_with_postprocessor() {
1281
- let dir = tempdir().unwrap();
1282
- let config_path = dir.path().join("kreuzberg.toml");
1283
-
1284
- fs::write(
1285
- &config_path,
1286
- r#"
1287
- use_cache = true
1288
-
1289
- [postprocessor]
1290
- enabled = true
1291
- enabled_processors = ["entity_extraction", "keyword_extraction"]
1292
- "#,
1293
- )
1294
- .unwrap();
1295
-
1296
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1297
- assert!(config.postprocessor.is_some());
1298
- let pp = config.postprocessor.unwrap();
1299
- assert!(pp.enabled);
1300
- assert!(pp.enabled_processors.is_some());
1301
- let enabled = pp.enabled_processors.unwrap();
1302
- assert_eq!(enabled.len(), 2);
1303
- assert_eq!(enabled[0], "entity_extraction");
1304
- assert_eq!(enabled[1], "keyword_extraction");
1305
- assert!(pp.disabled_processors.is_none());
1306
- }
1307
-
1308
- #[test]
1309
- fn test_postprocessor_config_defaults() {
1310
- let dir = tempdir().unwrap();
1311
- let config_path = dir.path().join("kreuzberg.toml");
1312
-
1313
- fs::write(
1314
- &config_path,
1315
- r#"
1316
- [postprocessor]
1317
- "#,
1318
- )
1319
- .unwrap();
1320
-
1321
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1322
- let pp = config.postprocessor.unwrap();
1323
- assert!(pp.enabled);
1324
- assert!(pp.enabled_processors.is_none());
1325
- assert!(pp.disabled_processors.is_none());
1326
- }
1327
-
1328
- #[test]
1329
- fn test_postprocessor_config_disabled() {
1330
- let dir = tempdir().unwrap();
1331
- let config_path = dir.path().join("kreuzberg.toml");
1332
-
1333
- fs::write(
1334
- &config_path,
1335
- r#"
1336
- [postprocessor]
1337
- enabled = false
1338
- "#,
1339
- )
1340
- .unwrap();
1341
-
1342
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1343
- let pp = config.postprocessor.unwrap();
1344
- assert!(!pp.enabled);
1345
- }
1346
-
1347
- #[test]
1348
- fn test_postprocessor_config_blacklist() {
1349
- let dir = tempdir().unwrap();
1350
- let config_path = dir.path().join("kreuzberg.toml");
1351
-
1352
- fs::write(
1353
- &config_path,
1354
- r#"
1355
- [postprocessor]
1356
- disabled_processors = ["category_extraction"]
1357
- "#,
1358
- )
1359
- .unwrap();
1360
-
1361
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1362
- let pp = config.postprocessor.unwrap();
1363
- assert!(pp.enabled);
1364
- assert!(pp.disabled_processors.is_some());
1365
- let disabled = pp.disabled_processors.unwrap();
1366
- assert_eq!(disabled.len(), 1);
1367
- assert_eq!(disabled[0], "category_extraction");
1368
- }
1369
-
1370
- #[test]
1371
- fn test_config_with_tesseract_config() {
1372
- let dir = tempdir().unwrap();
1373
- let config_path = dir.path().join("kreuzberg.toml");
1374
-
1375
- fs::write(
1376
- &config_path,
1377
- r#"
1378
- use_cache = true
1379
-
1380
- [ocr]
1381
- backend = "tesseract"
1382
- language = "eng"
1383
-
1384
- [ocr.tesseract_config]
1385
- psm = 6
1386
- output_format = "text"
1387
- enable_table_detection = false
1388
- tessedit_char_whitelist = "0123456789"
1389
- "#,
1390
- )
1391
- .unwrap();
1392
-
1393
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1394
- assert!(config.ocr.is_some());
1395
- let ocr = config.ocr.unwrap();
1396
- assert_eq!(ocr.backend, "tesseract");
1397
- assert_eq!(ocr.language, "eng");
1398
- assert!(ocr.tesseract_config.is_some());
1399
-
1400
- let tess = ocr.tesseract_config.unwrap();
1401
- assert_eq!(tess.psm, 6);
1402
- assert_eq!(tess.output_format, "text");
1403
- assert!(!tess.enable_table_detection);
1404
- assert_eq!(tess.tessedit_char_whitelist, "0123456789");
1405
- }
1406
-
1407
- #[test]
1408
- fn test_ocr_config_without_tesseract_config() {
1409
- let dir = tempdir().unwrap();
1410
- let config_path = dir.path().join("kreuzberg.toml");
1411
-
1412
- fs::write(
1413
- &config_path,
1414
- r#"
1415
- [ocr]
1416
- backend = "easyocr"
1417
- language = "eng"
1418
- "#,
1419
- )
1420
- .unwrap();
1421
-
1422
- let config = ExtractionConfig::from_toml_file(&config_path).unwrap();
1423
- assert!(config.ocr.is_some());
1424
- let ocr = config.ocr.unwrap();
1425
- assert_eq!(ocr.backend, "easyocr");
1426
- assert_eq!(ocr.language, "eng");
1427
- assert!(ocr.tesseract_config.is_none());
1428
- }
1429
-
1430
- #[test]
1431
- fn test_tesseract_config_defaults() {
1432
- let tess = crate::types::TesseractConfig::default();
1433
- assert_eq!(tess.language, "eng");
1434
- assert_eq!(tess.psm, 3);
1435
- assert_eq!(tess.output_format, "markdown");
1436
- assert!(tess.enable_table_detection);
1437
- assert_eq!(tess.table_min_confidence, 0.0);
1438
- assert_eq!(tess.table_column_threshold, 50);
1439
- assert_eq!(tess.table_row_threshold_ratio, 0.5);
1440
- assert!(tess.use_cache);
1441
- assert!(tess.classify_use_pre_adapted_templates);
1442
- assert!(!tess.language_model_ngram_on);
1443
- assert!(tess.tessedit_dont_blkrej_good_wds);
1444
- assert!(tess.tessedit_dont_rowrej_good_wds);
1445
- assert!(tess.tessedit_enable_dict_correction);
1446
- assert_eq!(tess.tessedit_char_whitelist, "");
1447
- assert!(tess.tessedit_use_primary_params_model);
1448
- assert!(tess.textord_space_size_is_variable);
1449
- assert!(!tess.thresholding_method);
1450
- }
1451
-
1452
- #[test]
1453
- fn test_config_cache_hit_same_mtime() {
1454
- let dir = tempdir().unwrap();
1455
- let config_path = dir.path().join("kreuzberg.toml");
1456
-
1457
- fs::write(
1458
- &config_path,
1459
- r#"
1460
- use_cache = false
1461
- enable_quality_processing = true
1462
- "#,
1463
- )
1464
- .unwrap();
1465
-
1466
- let config1 = ExtractionConfig::from_toml_file(&config_path).unwrap();
1467
- let config2 = ExtractionConfig::from_toml_file(&config_path).unwrap();
1468
-
1469
- assert_eq!(config1.use_cache, config2.use_cache);
1470
- assert_eq!(config1.enable_quality_processing, config2.enable_quality_processing);
1471
- }
1472
-
1473
- #[test]
1474
- fn test_config_cache_invalidation_on_mtime_change() {
1475
- let dir = tempdir().unwrap();
1476
- let config_path = dir.path().join("kreuzberg.toml");
1477
-
1478
- fs::write(
1479
- &config_path,
1480
- r#"
1481
- use_cache = false
1482
- enable_quality_processing = true
1483
- "#,
1484
- )
1485
- .unwrap();
1486
-
1487
- let config1 = ExtractionConfig::from_toml_file(&config_path).unwrap();
1488
- assert!(!config1.use_cache);
1489
-
1490
- std::thread::sleep(std::time::Duration::from_secs(1));
1491
-
1492
- fs::write(
1493
- &config_path,
1494
- r#"
1495
- use_cache = true
1496
- enable_quality_processing = false
1497
- "#,
1498
- )
1499
- .unwrap();
1500
-
1501
- let config2 = ExtractionConfig::from_toml_file(&config_path).unwrap();
1502
- assert!(config2.use_cache);
1503
- assert!(!config2.enable_quality_processing);
1504
- }
1505
-
1506
- #[test]
1507
- fn test_config_cache_works_with_json() {
1508
- let dir = tempdir().unwrap();
1509
- let config_path = dir.path().join("kreuzberg.json");
1510
-
1511
- fs::write(
1512
- &config_path,
1513
- r#"{"use_cache": true, "enable_quality_processing": false}"#,
1514
- )
1515
- .unwrap();
1516
-
1517
- let config1 = ExtractionConfig::from_json_file(&config_path).unwrap();
1518
- let config2 = ExtractionConfig::from_json_file(&config_path).unwrap();
1519
-
1520
- assert!(config1.use_cache);
1521
- assert!(!config1.enable_quality_processing);
1522
- assert_eq!(config1.use_cache, config2.use_cache);
1523
- }
1524
-
1525
- #[test]
1526
- fn test_config_cache_works_with_yaml() {
1527
- let dir = tempdir().unwrap();
1528
- let config_path = dir.path().join("kreuzberg.yaml");
1529
-
1530
- fs::write(
1531
- &config_path,
1532
- r#"
1533
- use_cache: true
1534
- enable_quality_processing: false
1535
- "#,
1536
- )
1537
- .unwrap();
1538
-
1539
- let config1 = ExtractionConfig::from_yaml_file(&config_path).unwrap();
1540
- let config2 = ExtractionConfig::from_yaml_file(&config_path).unwrap();
1541
-
1542
- assert!(config1.use_cache);
1543
- assert!(!config1.enable_quality_processing);
1544
- assert_eq!(config1.use_cache, config2.use_cache);
1545
- }
1546
-
1547
- #[test]
1548
- #[serial_test::serial]
1549
- fn test_apply_env_overrides_ocr_language() {
1550
- let original_lang = std::env::var("KREUZBERG_OCR_LANGUAGE").ok();
1551
-
1552
- set_env("KREUZBERG_OCR_LANGUAGE", "fra");
1553
- let mut config = ExtractionConfig::default();
1554
- assert!(config.ocr.is_none());
1555
-
1556
- config.apply_env_overrides().unwrap();
1557
-
1558
- assert!(config.ocr.is_some());
1559
- assert_eq!(config.ocr.unwrap().language, "fra");
1560
-
1561
- restore_env("KREUZBERG_OCR_LANGUAGE", original_lang);
1562
- }
1563
-
1564
- #[test]
1565
- #[serial_test::serial]
1566
- fn test_apply_env_overrides_ocr_backend() {
1567
- let original_backend = std::env::var("KREUZBERG_OCR_BACKEND").ok();
1568
-
1569
- set_env("KREUZBERG_OCR_BACKEND", "easyocr");
1570
- let mut config = ExtractionConfig::default();
1571
- assert!(config.ocr.is_none());
1572
-
1573
- config.apply_env_overrides().unwrap();
1574
-
1575
- assert!(config.ocr.is_some());
1576
- assert_eq!(config.ocr.unwrap().backend, "easyocr");
1577
-
1578
- restore_env("KREUZBERG_OCR_BACKEND", original_backend);
1579
- }
1580
-
1581
- #[test]
1582
- #[serial_test::serial]
1583
- fn test_apply_env_overrides_invalid_ocr_language() {
1584
- let original_lang = std::env::var("KREUZBERG_OCR_LANGUAGE").ok();
1585
-
1586
- set_env("KREUZBERG_OCR_LANGUAGE", "invalid_lang");
1587
- let mut config = ExtractionConfig::default();
1588
-
1589
- let result = config.apply_env_overrides();
1590
- assert!(result.is_err());
1591
- let err_msg = result.unwrap_err().to_string();
1592
- assert!(err_msg.contains("Invalid language code"));
1593
-
1594
- restore_env("KREUZBERG_OCR_LANGUAGE", original_lang);
1595
- }
1596
-
1597
- #[test]
1598
- #[serial_test::serial]
1599
- fn test_apply_env_overrides_invalid_ocr_backend() {
1600
- let original_backend = std::env::var("KREUZBERG_OCR_BACKEND").ok();
1601
-
1602
- set_env("KREUZBERG_OCR_BACKEND", "invalid_backend");
1603
- let mut config = ExtractionConfig::default();
1604
-
1605
- let result = config.apply_env_overrides();
1606
- assert!(result.is_err());
1607
- let err_msg = result.unwrap_err().to_string();
1608
- assert!(err_msg.contains("Invalid OCR backend"));
1609
-
1610
- restore_env("KREUZBERG_OCR_BACKEND", original_backend);
1611
- }
1612
-
1613
- #[test]
1614
- #[serial_test::serial]
1615
- fn test_apply_env_overrides_chunking_max_chars() {
1616
- let original_max_chars = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS").ok();
1617
-
1618
- set_env("KREUZBERG_CHUNKING_MAX_CHARS", "2000");
1619
- let mut config = ExtractionConfig::default();
1620
- assert!(config.chunking.is_none());
1621
-
1622
- config.apply_env_overrides().unwrap();
1623
-
1624
- assert!(config.chunking.is_some());
1625
- assert_eq!(config.chunking.unwrap().max_chars, 2000);
1626
-
1627
- restore_env("KREUZBERG_CHUNKING_MAX_CHARS", original_max_chars);
1628
- }
1629
-
1630
- #[test]
1631
- #[serial_test::serial]
1632
- fn test_apply_env_overrides_chunking_max_overlap() {
1633
- let original_overlap = std::env::var("KREUZBERG_CHUNKING_MAX_OVERLAP").ok();
1634
-
1635
- set_env("KREUZBERG_CHUNKING_MAX_OVERLAP", "300");
1636
- let mut config = ExtractionConfig::default();
1637
- assert!(config.chunking.is_none());
1638
-
1639
- config.apply_env_overrides().unwrap();
1640
-
1641
- assert!(config.chunking.is_some());
1642
- assert_eq!(config.chunking.unwrap().max_overlap, 300);
1643
-
1644
- restore_env("KREUZBERG_CHUNKING_MAX_OVERLAP", original_overlap);
1645
- }
1646
-
1647
- #[test]
1648
- #[serial_test::serial]
1649
- fn test_apply_env_overrides_invalid_chunking_max_chars() {
1650
- let original_max_chars = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS").ok();
1651
-
1652
- set_env("KREUZBERG_CHUNKING_MAX_CHARS", "not_a_number");
1653
- let mut config = ExtractionConfig::default();
1654
-
1655
- let result = config.apply_env_overrides();
1656
- assert!(result.is_err());
1657
- let err_msg = result.unwrap_err().to_string();
1658
- assert!(err_msg.contains("KREUZBERG_CHUNKING_MAX_CHARS"));
1659
-
1660
- restore_env("KREUZBERG_CHUNKING_MAX_CHARS", original_max_chars);
1661
- }
1662
-
1663
- #[test]
1664
- #[serial_test::serial]
1665
- fn test_apply_env_overrides_chunking_max_chars_zero() {
1666
- let original_max_chars = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS").ok();
1667
-
1668
- set_env("KREUZBERG_CHUNKING_MAX_CHARS", "0");
1669
- let mut config = ExtractionConfig::default();
1670
-
1671
- let result = config.apply_env_overrides();
1672
- assert!(result.is_err());
1673
- let err_msg = result.unwrap_err().to_string();
1674
- assert!(err_msg.contains("greater than 0"));
1675
-
1676
- restore_env("KREUZBERG_CHUNKING_MAX_CHARS", original_max_chars);
1677
- }
1678
-
1679
- #[test]
1680
- #[serial_test::serial]
1681
- fn test_apply_env_overrides_chunking_overlap_validation() {
1682
- let original_overlap = std::env::var("KREUZBERG_CHUNKING_MAX_OVERLAP").ok();
1683
-
1684
- set_env("KREUZBERG_CHUNKING_MAX_OVERLAP", "1500");
1685
- // Set chunking with max_chars = 1000
1686
- let mut config = ExtractionConfig {
1687
- chunking: Some(ChunkingConfig {
1688
- max_chars: 1000,
1689
- max_overlap: 200,
1690
- embedding: None,
1691
- preset: None,
1692
- }),
1693
- ..Default::default()
1694
- };
1695
-
1696
- let result = config.apply_env_overrides();
1697
- assert!(result.is_err());
1698
- let err_msg = result.unwrap_err().to_string();
1699
- assert!(err_msg.contains("overlap"));
1700
-
1701
- restore_env("KREUZBERG_CHUNKING_MAX_OVERLAP", original_overlap);
1702
- }
1703
-
1704
- #[test]
1705
- #[serial_test::serial]
1706
- fn test_apply_env_overrides_cache_enabled_true() {
1707
- let original_cache = std::env::var("KREUZBERG_CACHE_ENABLED").ok();
1708
-
1709
- set_env("KREUZBERG_CACHE_ENABLED", "true");
1710
- let mut config = ExtractionConfig {
1711
- use_cache: false,
1712
- ..Default::default()
1713
- };
1714
-
1715
- config.apply_env_overrides().unwrap();
1716
- assert!(config.use_cache);
1717
-
1718
- restore_env("KREUZBERG_CACHE_ENABLED", original_cache);
1719
- }
1720
-
1721
- #[test]
1722
- #[serial_test::serial]
1723
- fn test_apply_env_overrides_cache_enabled_false() {
1724
- let original_cache = std::env::var("KREUZBERG_CACHE_ENABLED").ok();
1725
-
1726
- set_env("KREUZBERG_CACHE_ENABLED", "false");
1727
- let mut config = ExtractionConfig {
1728
- use_cache: true,
1729
- ..Default::default()
1730
- };
1731
-
1732
- config.apply_env_overrides().unwrap();
1733
- assert!(!config.use_cache);
1734
-
1735
- restore_env("KREUZBERG_CACHE_ENABLED", original_cache);
1736
- }
1737
-
1738
- #[test]
1739
- #[serial_test::serial]
1740
- fn test_apply_env_overrides_cache_enabled_case_insensitive() {
1741
- let original_cache = std::env::var("KREUZBERG_CACHE_ENABLED").ok();
1742
-
1743
- set_env("KREUZBERG_CACHE_ENABLED", "TRUE");
1744
- let mut config = ExtractionConfig {
1745
- use_cache: false,
1746
- ..Default::default()
1747
- };
1748
-
1749
- config.apply_env_overrides().unwrap();
1750
- assert!(config.use_cache);
1751
-
1752
- set_env("KREUZBERG_CACHE_ENABLED", "FaLsE");
1753
- config.use_cache = true;
1754
- config.apply_env_overrides().unwrap();
1755
- assert!(!config.use_cache);
1756
-
1757
- restore_env("KREUZBERG_CACHE_ENABLED", original_cache);
1758
- }
1759
-
1760
- #[test]
1761
- #[serial_test::serial]
1762
- fn test_apply_env_overrides_invalid_cache_enabled() {
1763
- let original_cache = std::env::var("KREUZBERG_CACHE_ENABLED").ok();
1764
-
1765
- set_env("KREUZBERG_CACHE_ENABLED", "maybe");
1766
- let mut config = ExtractionConfig::default();
1767
-
1768
- let result = config.apply_env_overrides();
1769
- assert!(result.is_err());
1770
- let err_msg = result.unwrap_err().to_string();
1771
- assert!(err_msg.contains("KREUZBERG_CACHE_ENABLED"));
1772
- assert!(err_msg.contains("'true' or 'false'"));
1773
-
1774
- restore_env("KREUZBERG_CACHE_ENABLED", original_cache);
1775
- }
1776
-
1777
- #[test]
1778
- #[serial_test::serial]
1779
- fn test_apply_env_overrides_token_reduction_mode() {
1780
- let original_mode = std::env::var("KREUZBERG_TOKEN_REDUCTION_MODE").ok();
1781
-
1782
- set_env("KREUZBERG_TOKEN_REDUCTION_MODE", "moderate");
1783
- let mut config = ExtractionConfig::default();
1784
- assert!(config.token_reduction.is_none());
1785
-
1786
- config.apply_env_overrides().unwrap();
1787
-
1788
- assert!(config.token_reduction.is_some());
1789
- assert_eq!(config.token_reduction.unwrap().mode, "moderate");
1790
-
1791
- restore_env("KREUZBERG_TOKEN_REDUCTION_MODE", original_mode);
1792
- }
1793
-
1794
- #[test]
1795
- #[serial_test::serial]
1796
- fn test_apply_env_overrides_invalid_token_reduction_mode() {
1797
- let original_mode = std::env::var("KREUZBERG_TOKEN_REDUCTION_MODE").ok();
1798
-
1799
- set_env("KREUZBERG_TOKEN_REDUCTION_MODE", "extreme");
1800
- let mut config = ExtractionConfig::default();
1801
-
1802
- let result = config.apply_env_overrides();
1803
- assert!(result.is_err());
1804
- let err_msg = result.unwrap_err().to_string();
1805
- assert!(err_msg.contains("Invalid token reduction level"));
1806
-
1807
- restore_env("KREUZBERG_TOKEN_REDUCTION_MODE", original_mode);
1808
- }
1809
-
1810
- #[test]
1811
- #[serial_test::serial]
1812
- fn test_apply_env_overrides_multiple_overrides() {
1813
- let original_lang = std::env::var("KREUZBERG_OCR_LANGUAGE").ok();
1814
- let original_backend = std::env::var("KREUZBERG_OCR_BACKEND").ok();
1815
- let original_max_chars = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS").ok();
1816
- let original_cache = std::env::var("KREUZBERG_CACHE_ENABLED").ok();
1817
-
1818
- set_env("KREUZBERG_OCR_LANGUAGE", "deu");
1819
- set_env("KREUZBERG_OCR_BACKEND", "paddleocr");
1820
- set_env("KREUZBERG_CHUNKING_MAX_CHARS", "3000");
1821
- set_env("KREUZBERG_CACHE_ENABLED", "false");
1822
-
1823
- let mut config = ExtractionConfig::default();
1824
- config.apply_env_overrides().unwrap();
1825
-
1826
- let ocr = config.ocr.as_ref().unwrap();
1827
- assert_eq!(ocr.language, "deu");
1828
- assert_eq!(ocr.backend, "paddleocr");
1829
- let chunking = config.chunking.as_ref().unwrap();
1830
- assert_eq!(chunking.max_chars, 3000);
1831
- assert!(!config.use_cache);
1832
-
1833
- restore_env("KREUZBERG_OCR_LANGUAGE", original_lang);
1834
- restore_env("KREUZBERG_OCR_BACKEND", original_backend);
1835
- restore_env("KREUZBERG_CHUNKING_MAX_CHARS", original_max_chars);
1836
- restore_env("KREUZBERG_CACHE_ENABLED", original_cache);
1837
- }
1838
-
1839
- #[test]
1840
- #[serial_test::serial]
1841
- fn test_apply_env_overrides_preserves_existing_ocr_config() {
1842
- let original_lang = std::env::var("KREUZBERG_OCR_LANGUAGE").ok();
1843
-
1844
- set_env("KREUZBERG_OCR_LANGUAGE", "fra");
1845
-
1846
- let mut config = ExtractionConfig {
1847
- ocr: Some(OcrConfig {
1848
- backend: "tesseract".to_string(),
1849
- language: "eng".to_string(),
1850
- tesseract_config: None,
1851
- }),
1852
- ..Default::default()
1853
- };
1854
-
1855
- config.apply_env_overrides().unwrap();
1856
-
1857
- let ocr = config.ocr.unwrap();
1858
- assert_eq!(ocr.language, "fra");
1859
- assert_eq!(ocr.backend, "tesseract"); // Not overridden
1860
-
1861
- restore_env("KREUZBERG_OCR_LANGUAGE", original_lang);
1862
- }
1863
-
1864
- #[test]
1865
- #[serial_test::serial]
1866
- fn test_apply_env_overrides_no_env_vars() {
1867
- // Ensure no env vars are set
1868
- remove_env("KREUZBERG_OCR_LANGUAGE");
1869
- remove_env("KREUZBERG_OCR_BACKEND");
1870
- remove_env("KREUZBERG_CHUNKING_MAX_CHARS");
1871
- remove_env("KREUZBERG_CHUNKING_MAX_OVERLAP");
1872
- remove_env("KREUZBERG_CACHE_ENABLED");
1873
- remove_env("KREUZBERG_TOKEN_REDUCTION_MODE");
1874
-
1875
- let mut config = ExtractionConfig {
1876
- use_cache: false,
1877
- ocr: Some(OcrConfig {
1878
- backend: "tesseract".to_string(),
1879
- language: "eng".to_string(),
1880
- tesseract_config: None,
1881
- }),
1882
- ..Default::default()
1883
- };
1884
-
1885
- // Should not change anything
1886
- config.apply_env_overrides().unwrap();
1887
-
1888
- assert!(!config.use_cache);
1889
- assert_eq!(config.ocr.unwrap().language, "eng");
1890
- }
1891
-
1892
- #[test]
1893
- #[serial_test::serial]
1894
- fn test_apply_env_overrides_chunking_params_validation() {
1895
- let original_max_chars = std::env::var("KREUZBERG_CHUNKING_MAX_CHARS").ok();
1896
-
1897
- set_env("KREUZBERG_CHUNKING_MAX_CHARS", "500");
1898
- let mut config = ExtractionConfig {
1899
- chunking: Some(ChunkingConfig {
1900
- max_chars: 1000,
1901
- max_overlap: 600, // Valid: < 1000
1902
- embedding: None,
1903
- preset: None,
1904
- }),
1905
- ..Default::default()
1906
- };
1907
-
1908
- let result = config.apply_env_overrides();
1909
- assert!(result.is_err());
1910
- // After override, max_chars = 500, max_overlap = 600 (invalid: >= 500)
1911
-
1912
- restore_env("KREUZBERG_CHUNKING_MAX_CHARS", original_max_chars);
1913
- }
1914
- }