kreuzberg 4.0.0.rc1 → 4.0.0.rc2

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 (342) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +14 -8
  3. data/.rspec +3 -3
  4. data/.rubocop.yaml +1 -534
  5. data/.rubocop.yml +538 -0
  6. data/Gemfile +8 -9
  7. data/Gemfile.lock +9 -109
  8. data/README.md +426 -421
  9. data/Rakefile +25 -25
  10. data/Steepfile +47 -47
  11. data/examples/async_patterns.rb +341 -340
  12. data/ext/kreuzberg_rb/extconf.rb +45 -35
  13. data/ext/kreuzberg_rb/native/Cargo.lock +6535 -0
  14. data/ext/kreuzberg_rb/native/Cargo.toml +44 -36
  15. data/ext/kreuzberg_rb/native/README.md +425 -425
  16. data/ext/kreuzberg_rb/native/build.rs +15 -17
  17. data/ext/kreuzberg_rb/native/include/ieeefp.h +11 -11
  18. data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +14 -14
  19. data/ext/kreuzberg_rb/native/include/strings.h +20 -20
  20. data/ext/kreuzberg_rb/native/include/unistd.h +47 -47
  21. data/ext/kreuzberg_rb/native/src/lib.rs +2998 -2939
  22. data/extconf.rb +28 -28
  23. data/kreuzberg.gemspec +148 -105
  24. data/lib/kreuzberg/api_proxy.rb +142 -142
  25. data/lib/kreuzberg/cache_api.rb +46 -45
  26. data/lib/kreuzberg/cli.rb +55 -55
  27. data/lib/kreuzberg/cli_proxy.rb +127 -127
  28. data/lib/kreuzberg/config.rb +691 -684
  29. data/lib/kreuzberg/error_context.rb +32 -0
  30. data/lib/kreuzberg/errors.rb +118 -50
  31. data/lib/kreuzberg/extraction_api.rb +85 -84
  32. data/lib/kreuzberg/mcp_proxy.rb +186 -186
  33. data/lib/kreuzberg/ocr_backend_protocol.rb +113 -113
  34. data/lib/kreuzberg/post_processor_protocol.rb +86 -86
  35. data/lib/kreuzberg/result.rb +216 -216
  36. data/lib/kreuzberg/setup_lib_path.rb +80 -79
  37. data/lib/kreuzberg/validator_protocol.rb +89 -89
  38. data/lib/kreuzberg/version.rb +5 -5
  39. data/lib/kreuzberg.rb +103 -82
  40. data/sig/kreuzberg/internal.rbs +184 -184
  41. data/sig/kreuzberg.rbs +520 -468
  42. data/spec/binding/cache_spec.rb +227 -227
  43. data/spec/binding/cli_proxy_spec.rb +85 -87
  44. data/spec/binding/cli_spec.rb +55 -54
  45. data/spec/binding/config_spec.rb +345 -345
  46. data/spec/binding/config_validation_spec.rb +283 -283
  47. data/spec/binding/error_handling_spec.rb +213 -213
  48. data/spec/binding/errors_spec.rb +66 -66
  49. data/spec/binding/plugins/ocr_backend_spec.rb +307 -307
  50. data/spec/binding/plugins/postprocessor_spec.rb +269 -269
  51. data/spec/binding/plugins/validator_spec.rb +274 -274
  52. data/spec/fixtures/config.toml +39 -39
  53. data/spec/fixtures/config.yaml +41 -42
  54. data/spec/fixtures/invalid_config.toml +4 -4
  55. data/spec/smoke/package_spec.rb +178 -178
  56. data/spec/spec_helper.rb +42 -42
  57. data/vendor/kreuzberg/Cargo.toml +204 -134
  58. data/vendor/kreuzberg/README.md +175 -175
  59. data/vendor/kreuzberg/benches/otel_overhead.rs +48 -0
  60. data/vendor/kreuzberg/build.rs +474 -460
  61. data/vendor/kreuzberg/src/api/error.rs +81 -81
  62. data/vendor/kreuzberg/src/api/handlers.rs +199 -199
  63. data/vendor/kreuzberg/src/api/mod.rs +79 -79
  64. data/vendor/kreuzberg/src/api/server.rs +353 -353
  65. data/vendor/kreuzberg/src/api/types.rs +170 -170
  66. data/vendor/kreuzberg/src/cache/mod.rs +1167 -1143
  67. data/vendor/kreuzberg/src/chunking/mod.rs +677 -677
  68. data/vendor/kreuzberg/src/core/batch_mode.rs +95 -35
  69. data/vendor/kreuzberg/src/core/config.rs +1032 -1032
  70. data/vendor/kreuzberg/src/core/extractor.rs +1024 -903
  71. data/vendor/kreuzberg/src/core/io.rs +329 -327
  72. data/vendor/kreuzberg/src/core/mime.rs +605 -615
  73. data/vendor/kreuzberg/src/core/mod.rs +45 -42
  74. data/vendor/kreuzberg/src/core/pipeline.rs +984 -906
  75. data/vendor/kreuzberg/src/embeddings.rs +432 -323
  76. data/vendor/kreuzberg/src/error.rs +431 -431
  77. data/vendor/kreuzberg/src/extraction/archive.rs +954 -954
  78. data/vendor/kreuzberg/src/extraction/docx.rs +40 -40
  79. data/vendor/kreuzberg/src/extraction/email.rs +854 -854
  80. data/vendor/kreuzberg/src/extraction/excel.rs +688 -688
  81. data/vendor/kreuzberg/src/extraction/html.rs +553 -553
  82. data/vendor/kreuzberg/src/extraction/image.rs +368 -368
  83. data/vendor/kreuzberg/src/extraction/libreoffice.rs +563 -564
  84. data/vendor/kreuzberg/src/extraction/markdown.rs +213 -0
  85. data/vendor/kreuzberg/src/extraction/mod.rs +81 -77
  86. data/vendor/kreuzberg/src/extraction/office_metadata/app_properties.rs +398 -398
  87. data/vendor/kreuzberg/src/extraction/office_metadata/core_properties.rs +247 -247
  88. data/vendor/kreuzberg/src/extraction/office_metadata/custom_properties.rs +240 -240
  89. data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +130 -128
  90. data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +287 -0
  91. data/vendor/kreuzberg/src/extraction/pptx.rs +3000 -3000
  92. data/vendor/kreuzberg/src/extraction/structured.rs +490 -490
  93. data/vendor/kreuzberg/src/extraction/table.rs +328 -328
  94. data/vendor/kreuzberg/src/extraction/text.rs +269 -269
  95. data/vendor/kreuzberg/src/extraction/xml.rs +333 -333
  96. data/vendor/kreuzberg/src/extractors/archive.rs +446 -425
  97. data/vendor/kreuzberg/src/extractors/bibtex.rs +469 -0
  98. data/vendor/kreuzberg/src/extractors/docbook.rs +502 -0
  99. data/vendor/kreuzberg/src/extractors/docx.rs +367 -479
  100. data/vendor/kreuzberg/src/extractors/email.rs +143 -129
  101. data/vendor/kreuzberg/src/extractors/epub.rs +707 -0
  102. data/vendor/kreuzberg/src/extractors/excel.rs +343 -344
  103. data/vendor/kreuzberg/src/extractors/fictionbook.rs +491 -0
  104. data/vendor/kreuzberg/src/extractors/fictionbook.rs.backup2 +738 -0
  105. data/vendor/kreuzberg/src/extractors/html.rs +393 -410
  106. data/vendor/kreuzberg/src/extractors/image.rs +198 -195
  107. data/vendor/kreuzberg/src/extractors/jats.rs +1051 -0
  108. data/vendor/kreuzberg/src/extractors/jupyter.rs +367 -0
  109. data/vendor/kreuzberg/src/extractors/latex.rs +652 -0
  110. data/vendor/kreuzberg/src/extractors/markdown.rs +700 -0
  111. data/vendor/kreuzberg/src/extractors/mod.rs +365 -268
  112. data/vendor/kreuzberg/src/extractors/odt.rs +628 -0
  113. data/vendor/kreuzberg/src/extractors/opml.rs +634 -0
  114. data/vendor/kreuzberg/src/extractors/orgmode.rs +528 -0
  115. data/vendor/kreuzberg/src/extractors/pdf.rs +493 -496
  116. data/vendor/kreuzberg/src/extractors/pptx.rs +248 -234
  117. data/vendor/kreuzberg/src/extractors/rst.rs +576 -0
  118. data/vendor/kreuzberg/src/extractors/rtf.rs +810 -0
  119. data/vendor/kreuzberg/src/extractors/security.rs +484 -0
  120. data/vendor/kreuzberg/src/extractors/security_tests.rs +367 -0
  121. data/vendor/kreuzberg/src/extractors/structured.rs +140 -126
  122. data/vendor/kreuzberg/src/extractors/text.rs +260 -242
  123. data/vendor/kreuzberg/src/extractors/typst.rs +650 -0
  124. data/vendor/kreuzberg/src/extractors/xml.rs +135 -128
  125. data/vendor/kreuzberg/src/image/dpi.rs +164 -164
  126. data/vendor/kreuzberg/src/image/mod.rs +6 -6
  127. data/vendor/kreuzberg/src/image/preprocessing.rs +417 -417
  128. data/vendor/kreuzberg/src/image/resize.rs +89 -89
  129. data/vendor/kreuzberg/src/keywords/config.rs +154 -154
  130. data/vendor/kreuzberg/src/keywords/mod.rs +237 -237
  131. data/vendor/kreuzberg/src/keywords/processor.rs +267 -267
  132. data/vendor/kreuzberg/src/keywords/rake.rs +293 -294
  133. data/vendor/kreuzberg/src/keywords/types.rs +68 -68
  134. data/vendor/kreuzberg/src/keywords/yake.rs +163 -163
  135. data/vendor/kreuzberg/src/language_detection/mod.rs +942 -942
  136. data/vendor/kreuzberg/src/lib.rs +105 -102
  137. data/vendor/kreuzberg/src/mcp/mod.rs +32 -32
  138. data/vendor/kreuzberg/src/mcp/server.rs +1968 -1966
  139. data/vendor/kreuzberg/src/ocr/cache.rs +469 -469
  140. data/vendor/kreuzberg/src/ocr/error.rs +37 -37
  141. data/vendor/kreuzberg/src/ocr/hocr.rs +216 -216
  142. data/vendor/kreuzberg/src/ocr/mod.rs +58 -58
  143. data/vendor/kreuzberg/src/ocr/processor.rs +863 -847
  144. data/vendor/kreuzberg/src/ocr/table/mod.rs +4 -4
  145. data/vendor/kreuzberg/src/ocr/table/tsv_parser.rs +144 -144
  146. data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +450 -450
  147. data/vendor/kreuzberg/src/ocr/types.rs +393 -393
  148. data/vendor/kreuzberg/src/ocr/utils.rs +47 -47
  149. data/vendor/kreuzberg/src/ocr/validation.rs +206 -206
  150. data/vendor/kreuzberg/src/panic_context.rs +154 -0
  151. data/vendor/kreuzberg/src/pdf/error.rs +122 -122
  152. data/vendor/kreuzberg/src/pdf/images.rs +139 -139
  153. data/vendor/kreuzberg/src/pdf/metadata.rs +346 -346
  154. data/vendor/kreuzberg/src/pdf/mod.rs +50 -50
  155. data/vendor/kreuzberg/src/pdf/rendering.rs +369 -369
  156. data/vendor/kreuzberg/src/pdf/table.rs +393 -420
  157. data/vendor/kreuzberg/src/pdf/text.rs +158 -161
  158. data/vendor/kreuzberg/src/plugins/extractor.rs +1013 -1010
  159. data/vendor/kreuzberg/src/plugins/mod.rs +209 -209
  160. data/vendor/kreuzberg/src/plugins/ocr.rs +620 -629
  161. data/vendor/kreuzberg/src/plugins/processor.rs +642 -641
  162. data/vendor/kreuzberg/src/plugins/registry.rs +1337 -1324
  163. data/vendor/kreuzberg/src/plugins/traits.rs +258 -258
  164. data/vendor/kreuzberg/src/plugins/validator.rs +956 -955
  165. data/vendor/kreuzberg/src/stopwords/mod.rs +1470 -1470
  166. data/vendor/kreuzberg/src/text/mod.rs +19 -19
  167. data/vendor/kreuzberg/src/text/quality.rs +697 -697
  168. data/vendor/kreuzberg/src/text/string_utils.rs +217 -217
  169. data/vendor/kreuzberg/src/text/token_reduction/cjk_utils.rs +164 -164
  170. data/vendor/kreuzberg/src/text/token_reduction/config.rs +100 -100
  171. data/vendor/kreuzberg/src/text/token_reduction/core.rs +796 -796
  172. data/vendor/kreuzberg/src/text/token_reduction/filters.rs +902 -902
  173. data/vendor/kreuzberg/src/text/token_reduction/mod.rs +160 -160
  174. data/vendor/kreuzberg/src/text/token_reduction/semantic.rs +619 -619
  175. data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +147 -147
  176. data/vendor/kreuzberg/src/types.rs +903 -873
  177. data/vendor/kreuzberg/src/utils/mod.rs +17 -17
  178. data/vendor/kreuzberg/src/utils/quality.rs +959 -959
  179. data/vendor/kreuzberg/src/utils/string_utils.rs +381 -381
  180. data/vendor/kreuzberg/stopwords/af_stopwords.json +53 -53
  181. data/vendor/kreuzberg/stopwords/ar_stopwords.json +482 -482
  182. data/vendor/kreuzberg/stopwords/bg_stopwords.json +261 -261
  183. data/vendor/kreuzberg/stopwords/bn_stopwords.json +400 -400
  184. data/vendor/kreuzberg/stopwords/br_stopwords.json +1205 -1205
  185. data/vendor/kreuzberg/stopwords/ca_stopwords.json +280 -280
  186. data/vendor/kreuzberg/stopwords/cs_stopwords.json +425 -425
  187. data/vendor/kreuzberg/stopwords/da_stopwords.json +172 -172
  188. data/vendor/kreuzberg/stopwords/de_stopwords.json +622 -622
  189. data/vendor/kreuzberg/stopwords/el_stopwords.json +849 -849
  190. data/vendor/kreuzberg/stopwords/en_stopwords.json +1300 -1300
  191. data/vendor/kreuzberg/stopwords/eo_stopwords.json +175 -175
  192. data/vendor/kreuzberg/stopwords/es_stopwords.json +734 -734
  193. data/vendor/kreuzberg/stopwords/et_stopwords.json +37 -37
  194. data/vendor/kreuzberg/stopwords/eu_stopwords.json +100 -100
  195. data/vendor/kreuzberg/stopwords/fa_stopwords.json +801 -801
  196. data/vendor/kreuzberg/stopwords/fi_stopwords.json +849 -849
  197. data/vendor/kreuzberg/stopwords/fr_stopwords.json +693 -693
  198. data/vendor/kreuzberg/stopwords/ga_stopwords.json +111 -111
  199. data/vendor/kreuzberg/stopwords/gl_stopwords.json +162 -162
  200. data/vendor/kreuzberg/stopwords/gu_stopwords.json +226 -226
  201. data/vendor/kreuzberg/stopwords/ha_stopwords.json +41 -41
  202. data/vendor/kreuzberg/stopwords/he_stopwords.json +196 -196
  203. data/vendor/kreuzberg/stopwords/hi_stopwords.json +227 -227
  204. data/vendor/kreuzberg/stopwords/hr_stopwords.json +181 -181
  205. data/vendor/kreuzberg/stopwords/hu_stopwords.json +791 -791
  206. data/vendor/kreuzberg/stopwords/hy_stopwords.json +47 -47
  207. data/vendor/kreuzberg/stopwords/id_stopwords.json +760 -760
  208. data/vendor/kreuzberg/stopwords/it_stopwords.json +634 -634
  209. data/vendor/kreuzberg/stopwords/ja_stopwords.json +136 -136
  210. data/vendor/kreuzberg/stopwords/kn_stopwords.json +84 -84
  211. data/vendor/kreuzberg/stopwords/ko_stopwords.json +681 -681
  212. data/vendor/kreuzberg/stopwords/ku_stopwords.json +64 -64
  213. data/vendor/kreuzberg/stopwords/la_stopwords.json +51 -51
  214. data/vendor/kreuzberg/stopwords/lt_stopwords.json +476 -476
  215. data/vendor/kreuzberg/stopwords/lv_stopwords.json +163 -163
  216. data/vendor/kreuzberg/stopwords/ml_stopwords.json +1 -1
  217. data/vendor/kreuzberg/stopwords/mr_stopwords.json +101 -101
  218. data/vendor/kreuzberg/stopwords/ms_stopwords.json +477 -477
  219. data/vendor/kreuzberg/stopwords/ne_stopwords.json +490 -490
  220. data/vendor/kreuzberg/stopwords/nl_stopwords.json +415 -415
  221. data/vendor/kreuzberg/stopwords/no_stopwords.json +223 -223
  222. data/vendor/kreuzberg/stopwords/pl_stopwords.json +331 -331
  223. data/vendor/kreuzberg/stopwords/pt_stopwords.json +562 -562
  224. data/vendor/kreuzberg/stopwords/ro_stopwords.json +436 -436
  225. data/vendor/kreuzberg/stopwords/ru_stopwords.json +561 -561
  226. data/vendor/kreuzberg/stopwords/si_stopwords.json +193 -193
  227. data/vendor/kreuzberg/stopwords/sk_stopwords.json +420 -420
  228. data/vendor/kreuzberg/stopwords/sl_stopwords.json +448 -448
  229. data/vendor/kreuzberg/stopwords/so_stopwords.json +32 -32
  230. data/vendor/kreuzberg/stopwords/st_stopwords.json +33 -33
  231. data/vendor/kreuzberg/stopwords/sv_stopwords.json +420 -420
  232. data/vendor/kreuzberg/stopwords/sw_stopwords.json +76 -76
  233. data/vendor/kreuzberg/stopwords/ta_stopwords.json +129 -129
  234. data/vendor/kreuzberg/stopwords/te_stopwords.json +54 -54
  235. data/vendor/kreuzberg/stopwords/th_stopwords.json +118 -118
  236. data/vendor/kreuzberg/stopwords/tl_stopwords.json +149 -149
  237. data/vendor/kreuzberg/stopwords/tr_stopwords.json +506 -506
  238. data/vendor/kreuzberg/stopwords/uk_stopwords.json +75 -75
  239. data/vendor/kreuzberg/stopwords/ur_stopwords.json +519 -519
  240. data/vendor/kreuzberg/stopwords/vi_stopwords.json +647 -647
  241. data/vendor/kreuzberg/stopwords/yo_stopwords.json +62 -62
  242. data/vendor/kreuzberg/stopwords/zh_stopwords.json +796 -796
  243. data/vendor/kreuzberg/stopwords/zu_stopwords.json +31 -31
  244. data/vendor/kreuzberg/tests/api_extract_multipart.rs +52 -0
  245. data/vendor/kreuzberg/tests/api_tests.rs +966 -966
  246. data/vendor/kreuzberg/tests/archive_integration.rs +543 -543
  247. data/vendor/kreuzberg/tests/batch_orchestration.rs +556 -542
  248. data/vendor/kreuzberg/tests/batch_processing.rs +316 -304
  249. data/vendor/kreuzberg/tests/bibtex_parity_test.rs +421 -0
  250. data/vendor/kreuzberg/tests/concurrency_stress.rs +525 -509
  251. data/vendor/kreuzberg/tests/config_features.rs +598 -580
  252. data/vendor/kreuzberg/tests/config_loading_tests.rs +415 -439
  253. data/vendor/kreuzberg/tests/core_integration.rs +510 -493
  254. data/vendor/kreuzberg/tests/csv_integration.rs +414 -424
  255. data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +498 -0
  256. data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +122 -124
  257. data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +370 -0
  258. data/vendor/kreuzberg/tests/email_integration.rs +325 -325
  259. data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +275 -0
  260. data/vendor/kreuzberg/tests/error_handling.rs +393 -393
  261. data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +228 -0
  262. data/vendor/kreuzberg/tests/format_integration.rs +159 -159
  263. data/vendor/kreuzberg/tests/helpers/mod.rs +142 -142
  264. data/vendor/kreuzberg/tests/html_table_test.rs +551 -0
  265. data/vendor/kreuzberg/tests/image_integration.rs +253 -253
  266. data/vendor/kreuzberg/tests/instrumentation_test.rs +139 -0
  267. data/vendor/kreuzberg/tests/jats_extractor_tests.rs +639 -0
  268. data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +704 -0
  269. data/vendor/kreuzberg/tests/keywords_integration.rs +479 -479
  270. data/vendor/kreuzberg/tests/keywords_quality.rs +509 -509
  271. data/vendor/kreuzberg/tests/latex_extractor_tests.rs +496 -0
  272. data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +490 -0
  273. data/vendor/kreuzberg/tests/mime_detection.rs +428 -428
  274. data/vendor/kreuzberg/tests/ocr_configuration.rs +510 -510
  275. data/vendor/kreuzberg/tests/ocr_errors.rs +676 -676
  276. data/vendor/kreuzberg/tests/ocr_quality.rs +627 -627
  277. data/vendor/kreuzberg/tests/ocr_stress.rs +469 -469
  278. data/vendor/kreuzberg/tests/odt_extractor_tests.rs +695 -0
  279. data/vendor/kreuzberg/tests/opml_extractor_tests.rs +616 -0
  280. data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +822 -0
  281. data/vendor/kreuzberg/tests/pdf_integration.rs +43 -43
  282. data/vendor/kreuzberg/tests/pipeline_integration.rs +1411 -1412
  283. data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +771 -771
  284. data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +560 -561
  285. data/vendor/kreuzberg/tests/plugin_system.rs +921 -921
  286. data/vendor/kreuzberg/tests/plugin_validator_test.rs +783 -783
  287. data/vendor/kreuzberg/tests/registry_integration_tests.rs +586 -607
  288. data/vendor/kreuzberg/tests/rst_extractor_tests.rs +692 -0
  289. data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +776 -0
  290. data/vendor/kreuzberg/tests/security_validation.rs +415 -404
  291. data/vendor/kreuzberg/tests/stopwords_integration_test.rs +888 -888
  292. data/vendor/kreuzberg/tests/test_fastembed.rs +609 -609
  293. data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +1259 -0
  294. data/vendor/kreuzberg/tests/typst_extractor_tests.rs +647 -0
  295. data/vendor/kreuzberg/tests/xlsx_metadata_extraction_test.rs +87 -87
  296. data/vendor/rb-sys/.cargo-ok +1 -0
  297. data/vendor/rb-sys/.cargo_vcs_info.json +6 -0
  298. data/vendor/rb-sys/Cargo.lock +393 -0
  299. data/vendor/rb-sys/Cargo.toml +70 -0
  300. data/vendor/rb-sys/Cargo.toml.orig +57 -0
  301. data/vendor/rb-sys/LICENSE-APACHE +190 -0
  302. data/vendor/rb-sys/LICENSE-MIT +21 -0
  303. data/vendor/rb-sys/bin/release.sh +21 -0
  304. data/vendor/rb-sys/build/features.rs +108 -0
  305. data/vendor/rb-sys/build/main.rs +246 -0
  306. data/vendor/rb-sys/build/stable_api_config.rs +153 -0
  307. data/vendor/rb-sys/build/version.rs +48 -0
  308. data/vendor/rb-sys/readme.md +36 -0
  309. data/vendor/rb-sys/src/bindings.rs +21 -0
  310. data/vendor/rb-sys/src/hidden.rs +11 -0
  311. data/vendor/rb-sys/src/lib.rs +34 -0
  312. data/vendor/rb-sys/src/macros.rs +371 -0
  313. data/vendor/rb-sys/src/memory.rs +53 -0
  314. data/vendor/rb-sys/src/ruby_abi_version.rs +38 -0
  315. data/vendor/rb-sys/src/special_consts.rs +31 -0
  316. data/vendor/rb-sys/src/stable_api/compiled.c +179 -0
  317. data/vendor/rb-sys/src/stable_api/compiled.rs +257 -0
  318. data/vendor/rb-sys/src/stable_api/ruby_2_6.rs +316 -0
  319. data/vendor/rb-sys/src/stable_api/ruby_2_7.rs +316 -0
  320. data/vendor/rb-sys/src/stable_api/ruby_3_0.rs +324 -0
  321. data/vendor/rb-sys/src/stable_api/ruby_3_1.rs +317 -0
  322. data/vendor/rb-sys/src/stable_api/ruby_3_2.rs +315 -0
  323. data/vendor/rb-sys/src/stable_api/ruby_3_3.rs +326 -0
  324. data/vendor/rb-sys/src/stable_api/ruby_3_4.rs +327 -0
  325. data/vendor/rb-sys/src/stable_api.rs +261 -0
  326. data/vendor/rb-sys/src/symbol.rs +31 -0
  327. data/vendor/rb-sys/src/tracking_allocator.rs +332 -0
  328. data/vendor/rb-sys/src/utils.rs +89 -0
  329. data/vendor/rb-sys/src/value_type.rs +7 -0
  330. metadata +90 -95
  331. data/pkg/kreuzberg-4.0.0.rc1.gem +0 -0
  332. data/spec/examples.txt +0 -104
  333. data/vendor/kreuzberg/src/bin/profile_extract.rs +0 -455
  334. data/vendor/kreuzberg/src/extraction/pandoc/batch.rs +0 -275
  335. data/vendor/kreuzberg/src/extraction/pandoc/mime_types.rs +0 -178
  336. data/vendor/kreuzberg/src/extraction/pandoc/mod.rs +0 -491
  337. data/vendor/kreuzberg/src/extraction/pandoc/server.rs +0 -496
  338. data/vendor/kreuzberg/src/extraction/pandoc/subprocess.rs +0 -1188
  339. data/vendor/kreuzberg/src/extraction/pandoc/version.rs +0 -162
  340. data/vendor/kreuzberg/src/extractors/pandoc.rs +0 -201
  341. data/vendor/kreuzberg/tests/chunking_offset_demo.rs +0 -92
  342. data/vendor/kreuzberg/tests/pandoc_integration.rs +0 -503
@@ -1,641 +1,642 @@
1
- //! Post-processor plugin trait.
2
- //!
3
- //! This module defines traits for implementing custom post-processing logic.
4
-
5
- use crate::Result;
6
- use crate::core::config::ExtractionConfig;
7
- use crate::plugins::Plugin;
8
- use crate::types::ExtractionResult;
9
- use async_trait::async_trait;
10
-
11
- /// Processing stages for post-processors.
12
- ///
13
- /// Post-processors are executed in stage order (Early → Middle → Late).
14
- /// Use stages to control the order of post-processing operations.
15
- #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
16
- pub enum ProcessingStage {
17
- /// Early stage - foundational processing.
18
- ///
19
- /// Use for:
20
- /// - Language detection
21
- /// - Character encoding normalization
22
- /// - Entity extraction (NER)
23
- /// - Text quality scoring
24
- Early,
25
-
26
- /// Middle stage - content transformation.
27
- ///
28
- /// Use for:
29
- /// - Keyword extraction
30
- /// - Token reduction
31
- /// - Text summarization
32
- /// - Semantic analysis
33
- Middle,
34
-
35
- /// Late stage - final enrichment.
36
- ///
37
- /// Use for:
38
- /// - Custom user hooks
39
- /// - Analytics/logging
40
- /// - Final validation
41
- /// - Output formatting
42
- Late,
43
- }
44
-
45
- /// Trait for post-processor plugins.
46
- ///
47
- /// Post-processors transform or enrich extraction results after the initial
48
- /// extraction is complete. They can:
49
- /// - Clean and normalize text
50
- /// - Add metadata (language, keywords, entities)
51
- /// - Split content into chunks
52
- /// - Score quality
53
- /// - Apply custom transformations
54
- ///
55
- /// # Processing Order
56
- ///
57
- /// Post-processors are executed in stage order:
58
- /// 1. **Early** - Language detection, entity extraction
59
- /// 2. **Middle** - Keyword extraction, token reduction
60
- /// 3. **Late** - Custom hooks, final validation
61
- ///
62
- /// Within each stage, processors are executed in registration order.
63
- ///
64
- /// # Error Handling
65
- ///
66
- /// Post-processor errors are non-fatal by default - they're captured in metadata
67
- /// and execution continues. To make errors fatal, return an error from `process()`.
68
- ///
69
- /// # Thread Safety
70
- ///
71
- /// Post-processors must be thread-safe (`Send + Sync`).
72
- ///
73
- /// # Example
74
- ///
75
- /// ```rust
76
- /// use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
77
- /// use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
78
- /// use async_trait::async_trait;
79
- ///
80
- /// /// Add word count metadata to extraction results
81
- /// struct WordCountProcessor;
82
- ///
83
- /// impl Plugin for WordCountProcessor {
84
- /// fn name(&self) -> &str { "word-count" }
85
- /// fn version(&self) -> String { "1.0.0".to_string() }
86
- /// fn initialize(&self) -> Result<()> { Ok(()) }
87
- /// fn shutdown(&self) -> Result<()> { Ok(()) }
88
- /// }
89
- ///
90
- /// #[async_trait]
91
- /// impl PostProcessor for WordCountProcessor {
92
- /// async fn process(&self, result: &mut ExtractionResult, config: &ExtractionConfig)
93
- /// -> Result<()> {
94
- /// // Count words
95
- /// let word_count = result.content.split_whitespace().count();
96
- ///
97
- /// // Add to metadata
98
- /// result.metadata.additional.insert("word_count".to_string(), serde_json::json!(word_count));
99
- ///
100
- /// Ok(())
101
- /// }
102
- ///
103
- /// fn processing_stage(&self) -> ProcessingStage {
104
- /// ProcessingStage::Early
105
- /// }
106
- /// }
107
- /// ```
108
- #[async_trait]
109
- pub trait PostProcessor: Plugin {
110
- /// Process an extraction result.
111
- ///
112
- /// Transform or enrich the extraction result. Can modify:
113
- /// - `content` - The extracted text
114
- /// - `metadata` - Add or update metadata fields
115
- /// - `tables` - Modify or enhance table data
116
- ///
117
- /// # Arguments
118
- ///
119
- /// * `result` - Mutable reference to the extraction result to process
120
- /// * `config` - Extraction configuration
121
- ///
122
- /// # Returns
123
- ///
124
- /// `Ok(())` if processing succeeded, `Err(...)` for fatal failures.
125
- ///
126
- /// # Errors
127
- ///
128
- /// Return errors for fatal processing failures. Non-fatal errors should be
129
- /// captured in metadata directly on the result.
130
- ///
131
- /// # Performance
132
- ///
133
- /// This signature avoids unnecessary cloning of large extraction results by
134
- /// taking a mutable reference instead of ownership. Processors modify the
135
- /// result in place.
136
- ///
137
- /// # Example - Language Detection
138
- ///
139
- /// ```rust
140
- /// # use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
141
- /// # use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
142
- /// # use async_trait::async_trait;
143
- /// # struct LanguageDetector;
144
- /// # impl Plugin for LanguageDetector {
145
- /// # fn name(&self) -> &str { "language-detector" }
146
- /// # fn version(&self) -> String { "1.0.0".to_string() }
147
- /// # fn initialize(&self) -> Result<()> { Ok(()) }
148
- /// # fn shutdown(&self) -> Result<()> { Ok(()) }
149
- /// # }
150
- /// # #[async_trait]
151
- /// # impl PostProcessor for LanguageDetector {
152
- /// # fn processing_stage(&self) -> ProcessingStage { ProcessingStage::Early }
153
- /// async fn process(&self, result: &mut ExtractionResult, config: &ExtractionConfig)
154
- /// -> Result<()> {
155
- /// // Detect language (simplified - use real detection library in practice)
156
- /// let language = "en"; // Placeholder detection
157
- ///
158
- /// // Add to metadata
159
- /// result.metadata.additional.insert("detected_language".to_string(), serde_json::json!(language));
160
- ///
161
- /// Ok(())
162
- /// }
163
- /// # }
164
- /// ```
165
- ///
166
- /// # Example - Text Cleaning
167
- ///
168
- /// ```rust
169
- /// # use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
170
- /// # use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
171
- /// # use async_trait::async_trait;
172
- /// # struct TextCleaner;
173
- /// # impl Plugin for TextCleaner {
174
- /// # fn name(&self) -> &str { "text-cleaner" }
175
- /// # fn version(&self) -> String { "1.0.0".to_string() }
176
- /// # fn initialize(&self) -> Result<()> { Ok(()) }
177
- /// # fn shutdown(&self) -> Result<()> { Ok(()) }
178
- /// # }
179
- /// # #[async_trait]
180
- /// # impl PostProcessor for TextCleaner {
181
- /// # fn processing_stage(&self) -> ProcessingStage { ProcessingStage::Middle }
182
- /// async fn process(&self, result: &mut ExtractionResult, config: &ExtractionConfig)
183
- /// -> Result<()> {
184
- /// // Remove excessive whitespace
185
- /// result.content = result
186
- /// .content
187
- /// .split_whitespace()
188
- /// .collect::<Vec<_>>()
189
- /// .join(" ");
190
- ///
191
- /// Ok(())
192
- /// }
193
- /// # }
194
- /// ```
195
- async fn process(&self, result: &mut ExtractionResult, config: &ExtractionConfig) -> Result<()>;
196
-
197
- /// Get the processing stage for this post-processor.
198
- ///
199
- /// Determines when this processor runs in the pipeline.
200
- ///
201
- /// # Returns
202
- ///
203
- /// The `ProcessingStage` (Early, Middle, or Late).
204
- ///
205
- /// # Example
206
- ///
207
- /// ```rust
208
- /// # use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
209
- /// # use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
210
- /// # use async_trait::async_trait;
211
- /// # struct MyProcessor;
212
- /// # impl Plugin for MyProcessor {
213
- /// # fn name(&self) -> &str { "my-processor" }
214
- /// # fn version(&self) -> String { "1.0.0".to_string() }
215
- /// # fn initialize(&self) -> Result<()> { Ok(()) }
216
- /// # fn shutdown(&self) -> Result<()> { Ok(()) }
217
- /// # }
218
- /// # #[async_trait]
219
- /// # impl PostProcessor for MyProcessor {
220
- /// # async fn process(&self, result: &mut ExtractionResult, _: &ExtractionConfig) -> Result<()> { Ok(()) }
221
- /// fn processing_stage(&self) -> ProcessingStage {
222
- /// ProcessingStage::Early // Run before other processors
223
- /// }
224
- /// # }
225
- /// ```
226
- fn processing_stage(&self) -> ProcessingStage;
227
-
228
- /// Optional: Check if this processor should run for a given result.
229
- ///
230
- /// Allows conditional processing based on MIME type, metadata, or content.
231
- /// Defaults to `true` (always run).
232
- ///
233
- /// # Arguments
234
- ///
235
- /// * `result` - The extraction result to check
236
- /// * `config` - Extraction configuration
237
- ///
238
- /// # Returns
239
- ///
240
- /// `true` if the processor should run, `false` to skip.
241
- ///
242
- /// # Example
243
- ///
244
- /// ```rust
245
- /// # use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
246
- /// # use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
247
- /// # use async_trait::async_trait;
248
- /// # struct PdfOnlyProcessor;
249
- /// # impl Plugin for PdfOnlyProcessor {
250
- /// # fn name(&self) -> &str { "pdf-only" }
251
- /// # fn version(&self) -> String { "1.0.0".to_string() }
252
- /// # fn initialize(&self) -> Result<()> { Ok(()) }
253
- /// # fn shutdown(&self) -> Result<()> { Ok(()) }
254
- /// # }
255
- /// # #[async_trait]
256
- /// # impl PostProcessor for PdfOnlyProcessor {
257
- /// # fn processing_stage(&self) -> ProcessingStage { ProcessingStage::Middle }
258
- /// # async fn process(&self, result: &mut ExtractionResult, _: &ExtractionConfig) -> Result<()> { Ok(()) }
259
- /// /// Only process PDF documents
260
- /// fn should_process(&self, result: &ExtractionResult, config: &ExtractionConfig) -> bool {
261
- /// result.mime_type == "application/pdf"
262
- /// }
263
- /// # }
264
- /// ```
265
- fn should_process(&self, _result: &ExtractionResult, _config: &ExtractionConfig) -> bool {
266
- true
267
- }
268
-
269
- /// Optional: Estimate processing time in milliseconds.
270
- ///
271
- /// Used for logging and debugging. Defaults to 0 (unknown).
272
- ///
273
- /// # Arguments
274
- ///
275
- /// * `result` - The extraction result to estimate for
276
- ///
277
- /// # Returns
278
- ///
279
- /// Estimated processing time in milliseconds.
280
- fn estimated_duration_ms(&self, _result: &ExtractionResult) -> u64 {
281
- 0
282
- }
283
- }
284
-
285
- /// List all registered post-processor names.
286
- ///
287
- /// Returns a vector of all post-processor names currently registered in the
288
- /// global registry.
289
- ///
290
- /// # Returns
291
- ///
292
- /// - `Ok(Vec<String>)` - Vector of post-processor names
293
- /// - `Err(...)` if the registry lock is poisoned
294
- ///
295
- /// # Example
296
- ///
297
- /// ```rust
298
- /// use kreuzberg::plugins::list_post_processors;
299
- ///
300
- /// # tokio_test::block_on(async {
301
- /// let processors = list_post_processors()?;
302
- /// for name in processors {
303
- /// println!("Registered post-processor: {}", name);
304
- /// }
305
- /// # Ok::<(), kreuzberg::KreuzbergError>(())
306
- /// # });
307
- /// ```
308
- pub fn list_post_processors() -> crate::Result<Vec<String>> {
309
- use crate::plugins::registry::get_post_processor_registry;
310
-
311
- let registry = get_post_processor_registry();
312
- let registry = registry
313
- .read()
314
- .expect("~keep Failed to acquire read lock on post-processor registry"); // ~keep
315
-
316
- Ok(registry.list())
317
- }
318
-
319
- #[cfg(test)]
320
- mod tests {
321
- use super::*;
322
- use std::collections::HashMap;
323
-
324
- struct MockPostProcessor {
325
- stage: ProcessingStage,
326
- }
327
-
328
- impl Plugin for MockPostProcessor {
329
- fn name(&self) -> &str {
330
- "mock-processor"
331
- }
332
-
333
- fn version(&self) -> String {
334
- "1.0.0".to_string()
335
- }
336
-
337
- fn initialize(&self) -> Result<()> {
338
- Ok(())
339
- }
340
-
341
- fn shutdown(&self) -> Result<()> {
342
- Ok(())
343
- }
344
- }
345
-
346
- #[async_trait]
347
- impl PostProcessor for MockPostProcessor {
348
- async fn process(&self, result: &mut ExtractionResult, _config: &ExtractionConfig) -> Result<()> {
349
- result
350
- .metadata
351
- .additional
352
- .insert("processed_by".to_string(), serde_json::json!(self.name()));
353
- Ok(())
354
- }
355
-
356
- fn processing_stage(&self) -> ProcessingStage {
357
- self.stage
358
- }
359
- }
360
-
361
- #[tokio::test]
362
- async fn test_post_processor_process() {
363
- let processor = MockPostProcessor {
364
- stage: ProcessingStage::Early,
365
- };
366
-
367
- let mut result = ExtractionResult {
368
- content: "test content".to_string(),
369
- mime_type: "text/plain".to_string(),
370
- metadata: crate::types::Metadata::default(),
371
- tables: vec![],
372
- detected_languages: None,
373
- chunks: None,
374
- images: None,
375
- };
376
-
377
- let config = ExtractionConfig::default();
378
- processor.process(&mut result, &config).await.unwrap();
379
-
380
- assert_eq!(result.content, "test content");
381
- assert_eq!(
382
- result.metadata.additional.get("processed_by").unwrap(),
383
- &serde_json::json!("mock-processor")
384
- );
385
- }
386
-
387
- #[test]
388
- fn test_processing_stage_order() {
389
- assert!(ProcessingStage::Early < ProcessingStage::Middle);
390
- assert!(ProcessingStage::Middle < ProcessingStage::Late);
391
- }
392
-
393
- #[test]
394
- fn test_post_processor_stage() {
395
- let early = MockPostProcessor {
396
- stage: ProcessingStage::Early,
397
- };
398
- let middle = MockPostProcessor {
399
- stage: ProcessingStage::Middle,
400
- };
401
- let late = MockPostProcessor {
402
- stage: ProcessingStage::Late,
403
- };
404
-
405
- assert_eq!(early.processing_stage(), ProcessingStage::Early);
406
- assert_eq!(middle.processing_stage(), ProcessingStage::Middle);
407
- assert_eq!(late.processing_stage(), ProcessingStage::Late);
408
- }
409
-
410
- #[test]
411
- fn test_post_processor_should_process_default() {
412
- let processor = MockPostProcessor {
413
- stage: ProcessingStage::Early,
414
- };
415
-
416
- let result = ExtractionResult {
417
- content: "test".to_string(),
418
- mime_type: "text/plain".to_string(),
419
- metadata: crate::types::Metadata::default(),
420
- tables: vec![],
421
- detected_languages: None,
422
- chunks: None,
423
- images: None,
424
- };
425
-
426
- let config = ExtractionConfig::default();
427
-
428
- assert!(processor.should_process(&result, &config));
429
- }
430
-
431
- #[test]
432
- fn test_processing_stage_equality() {
433
- assert_eq!(ProcessingStage::Early, ProcessingStage::Early);
434
- assert_ne!(ProcessingStage::Early, ProcessingStage::Middle);
435
- assert_ne!(ProcessingStage::Middle, ProcessingStage::Late);
436
- }
437
-
438
- #[test]
439
- fn test_processing_stage_clone() {
440
- let stage = ProcessingStage::Middle;
441
- let cloned = stage;
442
- assert_eq!(stage, cloned);
443
- }
444
-
445
- #[test]
446
- fn test_processing_stage_debug() {
447
- let stage = ProcessingStage::Early;
448
- let debug_str = format!("{:?}", stage);
449
- assert!(debug_str.contains("Early"));
450
- }
451
-
452
- #[test]
453
- fn test_processing_stage_hash() {
454
- use std::collections::HashSet;
455
- let mut set = HashSet::new();
456
- set.insert(ProcessingStage::Early);
457
- set.insert(ProcessingStage::Middle);
458
- set.insert(ProcessingStage::Late);
459
-
460
- assert_eq!(set.len(), 3);
461
- assert!(set.contains(&ProcessingStage::Early));
462
- }
463
-
464
- #[tokio::test]
465
- async fn test_post_processor_plugin_interface() {
466
- let processor = MockPostProcessor {
467
- stage: ProcessingStage::Middle,
468
- };
469
-
470
- assert_eq!(processor.name(), "mock-processor");
471
- assert_eq!(processor.version(), "1.0.0");
472
- assert!(processor.initialize().is_ok());
473
- assert!(processor.shutdown().is_ok());
474
- }
475
-
476
- #[tokio::test]
477
- async fn test_post_processor_empty_content() {
478
- let processor = MockPostProcessor {
479
- stage: ProcessingStage::Early,
480
- };
481
-
482
- let mut result = ExtractionResult {
483
- content: String::new(),
484
- mime_type: "text/plain".to_string(),
485
- metadata: crate::types::Metadata::default(),
486
- tables: vec![],
487
- detected_languages: None,
488
- chunks: None,
489
- images: None,
490
- };
491
-
492
- let config = ExtractionConfig::default();
493
- processor.process(&mut result, &config).await.unwrap();
494
-
495
- assert_eq!(result.content, "");
496
- assert!(result.metadata.additional.contains_key("processed_by"));
497
- }
498
-
499
- #[tokio::test]
500
- async fn test_post_processor_preserves_metadata() {
501
- let processor = MockPostProcessor {
502
- stage: ProcessingStage::Early,
503
- };
504
-
505
- let mut additional = HashMap::new();
506
- additional.insert("existing_key".to_string(), serde_json::json!("existing_value"));
507
-
508
- let mut result = ExtractionResult {
509
- content: "test".to_string(),
510
- mime_type: "text/plain".to_string(),
511
- metadata: crate::types::Metadata {
512
- additional,
513
- ..Default::default()
514
- },
515
- tables: vec![],
516
- detected_languages: None,
517
- chunks: None,
518
- images: None,
519
- };
520
-
521
- let config = ExtractionConfig::default();
522
- processor.process(&mut result, &config).await.unwrap();
523
-
524
- assert_eq!(
525
- result.metadata.additional.get("existing_key").unwrap(),
526
- &serde_json::json!("existing_value")
527
- );
528
- assert!(result.metadata.additional.contains_key("processed_by"));
529
- }
530
-
531
- #[test]
532
- fn test_post_processor_estimated_duration_default() {
533
- let processor = MockPostProcessor {
534
- stage: ProcessingStage::Early,
535
- };
536
-
537
- let result = ExtractionResult {
538
- content: "test".to_string(),
539
- mime_type: "text/plain".to_string(),
540
- metadata: crate::types::Metadata::default(),
541
- tables: vec![],
542
- detected_languages: None,
543
- chunks: None,
544
- images: None,
545
- };
546
-
547
- assert_eq!(processor.estimated_duration_ms(&result), 0);
548
- }
549
-
550
- #[test]
551
- fn test_post_processor_should_process_conditional() {
552
- struct PdfOnlyProcessor;
553
-
554
- impl Plugin for PdfOnlyProcessor {
555
- fn name(&self) -> &str {
556
- "pdf-only"
557
- }
558
- fn version(&self) -> String {
559
- "1.0.0".to_string()
560
- }
561
- fn initialize(&self) -> Result<()> {
562
- Ok(())
563
- }
564
- fn shutdown(&self) -> Result<()> {
565
- Ok(())
566
- }
567
- }
568
-
569
- #[async_trait]
570
- impl PostProcessor for PdfOnlyProcessor {
571
- async fn process(&self, _result: &mut ExtractionResult, _config: &ExtractionConfig) -> Result<()> {
572
- Ok(())
573
- }
574
-
575
- fn processing_stage(&self) -> ProcessingStage {
576
- ProcessingStage::Middle
577
- }
578
-
579
- fn should_process(&self, result: &ExtractionResult, _config: &ExtractionConfig) -> bool {
580
- result.mime_type == "application/pdf"
581
- }
582
- }
583
-
584
- let processor = PdfOnlyProcessor;
585
- let config = ExtractionConfig::default();
586
-
587
- let pdf_result = ExtractionResult {
588
- content: "test".to_string(),
589
- mime_type: "application/pdf".to_string(),
590
- metadata: crate::types::Metadata::default(),
591
- tables: vec![],
592
- detected_languages: None,
593
- chunks: None,
594
- images: None,
595
- };
596
-
597
- let txt_result = ExtractionResult {
598
- content: "test".to_string(),
599
- mime_type: "text/plain".to_string(),
600
- metadata: crate::types::Metadata::default(),
601
- tables: vec![],
602
- detected_languages: None,
603
- chunks: None,
604
- images: None,
605
- };
606
-
607
- assert!(processor.should_process(&pdf_result, &config));
608
- assert!(!processor.should_process(&txt_result, &config));
609
- }
610
-
611
- #[tokio::test]
612
- async fn test_post_processor_preserves_tables() {
613
- use crate::types::Table;
614
-
615
- let processor = MockPostProcessor {
616
- stage: ProcessingStage::Early,
617
- };
618
-
619
- let table = Table {
620
- cells: vec![vec!["A".to_string(), "B".to_string()]],
621
- markdown: "| A | B |".to_string(),
622
- page_number: 0,
623
- };
624
-
625
- let mut result = ExtractionResult {
626
- content: "test".to_string(),
627
- mime_type: "text/plain".to_string(),
628
- metadata: crate::types::Metadata::default(),
629
- tables: vec![table],
630
- detected_languages: None,
631
- chunks: None,
632
- images: None,
633
- };
634
-
635
- let config = ExtractionConfig::default();
636
- processor.process(&mut result, &config).await.unwrap();
637
-
638
- assert_eq!(result.tables.len(), 1);
639
- assert_eq!(result.tables[0].cells.len(), 1);
640
- }
641
- }
1
+ //! Post-processor plugin trait.
2
+ //!
3
+ //! This module defines traits for implementing custom post-processing logic.
4
+
5
+ use crate::Result;
6
+ use crate::core::config::ExtractionConfig;
7
+ use crate::plugins::Plugin;
8
+ use crate::types::ExtractionResult;
9
+ use async_trait::async_trait;
10
+
11
+ /// Processing stages for post-processors.
12
+ ///
13
+ /// Post-processors are executed in stage order (Early → Middle → Late).
14
+ /// Use stages to control the order of post-processing operations.
15
+ #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
16
+ pub enum ProcessingStage {
17
+ /// Early stage - foundational processing.
18
+ ///
19
+ /// Use for:
20
+ /// - Language detection
21
+ /// - Character encoding normalization
22
+ /// - Entity extraction (NER)
23
+ /// - Text quality scoring
24
+ Early,
25
+
26
+ /// Middle stage - content transformation.
27
+ ///
28
+ /// Use for:
29
+ /// - Keyword extraction
30
+ /// - Token reduction
31
+ /// - Text summarization
32
+ /// - Semantic analysis
33
+ Middle,
34
+
35
+ /// Late stage - final enrichment.
36
+ ///
37
+ /// Use for:
38
+ /// - Custom user hooks
39
+ /// - Analytics/logging
40
+ /// - Final validation
41
+ /// - Output formatting
42
+ Late,
43
+ }
44
+
45
+ /// Trait for post-processor plugins.
46
+ ///
47
+ /// Post-processors transform or enrich extraction results after the initial
48
+ /// extraction is complete. They can:
49
+ /// - Clean and normalize text
50
+ /// - Add metadata (language, keywords, entities)
51
+ /// - Split content into chunks
52
+ /// - Score quality
53
+ /// - Apply custom transformations
54
+ ///
55
+ /// # Processing Order
56
+ ///
57
+ /// Post-processors are executed in stage order:
58
+ /// 1. **Early** - Language detection, entity extraction
59
+ /// 2. **Middle** - Keyword extraction, token reduction
60
+ /// 3. **Late** - Custom hooks, final validation
61
+ ///
62
+ /// Within each stage, processors are executed in registration order.
63
+ ///
64
+ /// # Error Handling
65
+ ///
66
+ /// Post-processor errors are non-fatal by default - they're captured in metadata
67
+ /// and execution continues. To make errors fatal, return an error from `process()`.
68
+ ///
69
+ /// # Thread Safety
70
+ ///
71
+ /// Post-processors must be thread-safe (`Send + Sync`).
72
+ ///
73
+ /// # Example
74
+ ///
75
+ /// ```rust
76
+ /// use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
77
+ /// use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
78
+ /// use async_trait::async_trait;
79
+ ///
80
+ /// /// Add word count metadata to extraction results
81
+ /// struct WordCountProcessor;
82
+ ///
83
+ /// impl Plugin for WordCountProcessor {
84
+ /// fn name(&self) -> &str { "word-count" }
85
+ /// fn version(&self) -> String { "1.0.0".to_string() }
86
+ /// fn initialize(&self) -> Result<()> { Ok(()) }
87
+ /// fn shutdown(&self) -> Result<()> { Ok(()) }
88
+ /// }
89
+ ///
90
+ /// #[async_trait]
91
+ /// impl PostProcessor for WordCountProcessor {
92
+ /// async fn process(&self, result: &mut ExtractionResult, config: &ExtractionConfig)
93
+ /// -> Result<()> {
94
+ /// // Count words
95
+ /// let word_count = result.content.split_whitespace().count();
96
+ ///
97
+ /// // Add to metadata
98
+ /// result.metadata.additional.insert("word_count".to_string(), serde_json::json!(word_count));
99
+ ///
100
+ /// Ok(())
101
+ /// }
102
+ ///
103
+ /// fn processing_stage(&self) -> ProcessingStage {
104
+ /// ProcessingStage::Early
105
+ /// }
106
+ /// }
107
+ /// ```
108
+ #[cfg_attr(not(target_arch = "wasm32"), async_trait)]
109
+ #[cfg_attr(target_arch = "wasm32", async_trait(?Send))]
110
+ pub trait PostProcessor: Plugin {
111
+ /// Process an extraction result.
112
+ ///
113
+ /// Transform or enrich the extraction result. Can modify:
114
+ /// - `content` - The extracted text
115
+ /// - `metadata` - Add or update metadata fields
116
+ /// - `tables` - Modify or enhance table data
117
+ ///
118
+ /// # Arguments
119
+ ///
120
+ /// * `result` - Mutable reference to the extraction result to process
121
+ /// * `config` - Extraction configuration
122
+ ///
123
+ /// # Returns
124
+ ///
125
+ /// `Ok(())` if processing succeeded, `Err(...)` for fatal failures.
126
+ ///
127
+ /// # Errors
128
+ ///
129
+ /// Return errors for fatal processing failures. Non-fatal errors should be
130
+ /// captured in metadata directly on the result.
131
+ ///
132
+ /// # Performance
133
+ ///
134
+ /// This signature avoids unnecessary cloning of large extraction results by
135
+ /// taking a mutable reference instead of ownership. Processors modify the
136
+ /// result in place.
137
+ ///
138
+ /// # Example - Language Detection
139
+ ///
140
+ /// ```rust
141
+ /// # use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
142
+ /// # use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
143
+ /// # use async_trait::async_trait;
144
+ /// # struct LanguageDetector;
145
+ /// # impl Plugin for LanguageDetector {
146
+ /// # fn name(&self) -> &str { "language-detector" }
147
+ /// # fn version(&self) -> String { "1.0.0".to_string() }
148
+ /// # fn initialize(&self) -> Result<()> { Ok(()) }
149
+ /// # fn shutdown(&self) -> Result<()> { Ok(()) }
150
+ /// # }
151
+ /// # #[async_trait]
152
+ /// # impl PostProcessor for LanguageDetector {
153
+ /// # fn processing_stage(&self) -> ProcessingStage { ProcessingStage::Early }
154
+ /// async fn process(&self, result: &mut ExtractionResult, config: &ExtractionConfig)
155
+ /// -> Result<()> {
156
+ /// // Detect language (simplified - use real detection library in practice)
157
+ /// let language = "en"; // Placeholder detection
158
+ ///
159
+ /// // Add to metadata
160
+ /// result.metadata.additional.insert("detected_language".to_string(), serde_json::json!(language));
161
+ ///
162
+ /// Ok(())
163
+ /// }
164
+ /// # }
165
+ /// ```
166
+ ///
167
+ /// # Example - Text Cleaning
168
+ ///
169
+ /// ```rust
170
+ /// # use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
171
+ /// # use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
172
+ /// # use async_trait::async_trait;
173
+ /// # struct TextCleaner;
174
+ /// # impl Plugin for TextCleaner {
175
+ /// # fn name(&self) -> &str { "text-cleaner" }
176
+ /// # fn version(&self) -> String { "1.0.0".to_string() }
177
+ /// # fn initialize(&self) -> Result<()> { Ok(()) }
178
+ /// # fn shutdown(&self) -> Result<()> { Ok(()) }
179
+ /// # }
180
+ /// # #[async_trait]
181
+ /// # impl PostProcessor for TextCleaner {
182
+ /// # fn processing_stage(&self) -> ProcessingStage { ProcessingStage::Middle }
183
+ /// async fn process(&self, result: &mut ExtractionResult, config: &ExtractionConfig)
184
+ /// -> Result<()> {
185
+ /// // Remove excessive whitespace
186
+ /// result.content = result
187
+ /// .content
188
+ /// .split_whitespace()
189
+ /// .collect::<Vec<_>>()
190
+ /// .join(" ");
191
+ ///
192
+ /// Ok(())
193
+ /// }
194
+ /// # }
195
+ /// ```
196
+ async fn process(&self, result: &mut ExtractionResult, config: &ExtractionConfig) -> Result<()>;
197
+
198
+ /// Get the processing stage for this post-processor.
199
+ ///
200
+ /// Determines when this processor runs in the pipeline.
201
+ ///
202
+ /// # Returns
203
+ ///
204
+ /// The `ProcessingStage` (Early, Middle, or Late).
205
+ ///
206
+ /// # Example
207
+ ///
208
+ /// ```rust
209
+ /// # use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
210
+ /// # use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
211
+ /// # use async_trait::async_trait;
212
+ /// # struct MyProcessor;
213
+ /// # impl Plugin for MyProcessor {
214
+ /// # fn name(&self) -> &str { "my-processor" }
215
+ /// # fn version(&self) -> String { "1.0.0".to_string() }
216
+ /// # fn initialize(&self) -> Result<()> { Ok(()) }
217
+ /// # fn shutdown(&self) -> Result<()> { Ok(()) }
218
+ /// # }
219
+ /// # #[async_trait]
220
+ /// # impl PostProcessor for MyProcessor {
221
+ /// # async fn process(&self, result: &mut ExtractionResult, _: &ExtractionConfig) -> Result<()> { Ok(()) }
222
+ /// fn processing_stage(&self) -> ProcessingStage {
223
+ /// ProcessingStage::Early // Run before other processors
224
+ /// }
225
+ /// # }
226
+ /// ```
227
+ fn processing_stage(&self) -> ProcessingStage;
228
+
229
+ /// Optional: Check if this processor should run for a given result.
230
+ ///
231
+ /// Allows conditional processing based on MIME type, metadata, or content.
232
+ /// Defaults to `true` (always run).
233
+ ///
234
+ /// # Arguments
235
+ ///
236
+ /// * `result` - The extraction result to check
237
+ /// * `config` - Extraction configuration
238
+ ///
239
+ /// # Returns
240
+ ///
241
+ /// `true` if the processor should run, `false` to skip.
242
+ ///
243
+ /// # Example
244
+ ///
245
+ /// ```rust
246
+ /// # use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
247
+ /// # use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
248
+ /// # use async_trait::async_trait;
249
+ /// # struct PdfOnlyProcessor;
250
+ /// # impl Plugin for PdfOnlyProcessor {
251
+ /// # fn name(&self) -> &str { "pdf-only" }
252
+ /// # fn version(&self) -> String { "1.0.0".to_string() }
253
+ /// # fn initialize(&self) -> Result<()> { Ok(()) }
254
+ /// # fn shutdown(&self) -> Result<()> { Ok(()) }
255
+ /// # }
256
+ /// # #[async_trait]
257
+ /// # impl PostProcessor for PdfOnlyProcessor {
258
+ /// # fn processing_stage(&self) -> ProcessingStage { ProcessingStage::Middle }
259
+ /// # async fn process(&self, result: &mut ExtractionResult, _: &ExtractionConfig) -> Result<()> { Ok(()) }
260
+ /// /// Only process PDF documents
261
+ /// fn should_process(&self, result: &ExtractionResult, config: &ExtractionConfig) -> bool {
262
+ /// result.mime_type == "application/pdf"
263
+ /// }
264
+ /// # }
265
+ /// ```
266
+ fn should_process(&self, _result: &ExtractionResult, _config: &ExtractionConfig) -> bool {
267
+ true
268
+ }
269
+
270
+ /// Optional: Estimate processing time in milliseconds.
271
+ ///
272
+ /// Used for logging and debugging. Defaults to 0 (unknown).
273
+ ///
274
+ /// # Arguments
275
+ ///
276
+ /// * `result` - The extraction result to estimate for
277
+ ///
278
+ /// # Returns
279
+ ///
280
+ /// Estimated processing time in milliseconds.
281
+ fn estimated_duration_ms(&self, _result: &ExtractionResult) -> u64 {
282
+ 0
283
+ }
284
+ }
285
+
286
+ /// List all registered post-processor names.
287
+ ///
288
+ /// Returns a vector of all post-processor names currently registered in the
289
+ /// global registry.
290
+ ///
291
+ /// # Returns
292
+ ///
293
+ /// - `Ok(Vec<String>)` - Vector of post-processor names
294
+ /// - `Err(...)` if the registry lock is poisoned
295
+ ///
296
+ /// # Example
297
+ ///
298
+ /// ```rust
299
+ /// use kreuzberg::plugins::list_post_processors;
300
+ ///
301
+ /// # tokio_test::block_on(async {
302
+ /// let processors = list_post_processors()?;
303
+ /// for name in processors {
304
+ /// println!("Registered post-processor: {}", name);
305
+ /// }
306
+ /// # Ok::<(), kreuzberg::KreuzbergError>(())
307
+ /// # });
308
+ /// ```
309
+ pub fn list_post_processors() -> crate::Result<Vec<String>> {
310
+ use crate::plugins::registry::get_post_processor_registry;
311
+
312
+ let registry = get_post_processor_registry();
313
+ let registry = registry
314
+ .read()
315
+ .expect("~keep Failed to acquire read lock on post-processor registry"); // ~keep
316
+
317
+ Ok(registry.list())
318
+ }
319
+
320
+ #[cfg(test)]
321
+ mod tests {
322
+ use super::*;
323
+ use std::collections::HashMap;
324
+
325
+ struct MockPostProcessor {
326
+ stage: ProcessingStage,
327
+ }
328
+
329
+ impl Plugin for MockPostProcessor {
330
+ fn name(&self) -> &str {
331
+ "mock-processor"
332
+ }
333
+
334
+ fn version(&self) -> String {
335
+ "1.0.0".to_string()
336
+ }
337
+
338
+ fn initialize(&self) -> Result<()> {
339
+ Ok(())
340
+ }
341
+
342
+ fn shutdown(&self) -> Result<()> {
343
+ Ok(())
344
+ }
345
+ }
346
+
347
+ #[async_trait]
348
+ impl PostProcessor for MockPostProcessor {
349
+ async fn process(&self, result: &mut ExtractionResult, _config: &ExtractionConfig) -> Result<()> {
350
+ result
351
+ .metadata
352
+ .additional
353
+ .insert("processed_by".to_string(), serde_json::json!(self.name()));
354
+ Ok(())
355
+ }
356
+
357
+ fn processing_stage(&self) -> ProcessingStage {
358
+ self.stage
359
+ }
360
+ }
361
+
362
+ #[tokio::test]
363
+ async fn test_post_processor_process() {
364
+ let processor = MockPostProcessor {
365
+ stage: ProcessingStage::Early,
366
+ };
367
+
368
+ let mut result = ExtractionResult {
369
+ content: "test content".to_string(),
370
+ mime_type: "text/plain".to_string(),
371
+ metadata: crate::types::Metadata::default(),
372
+ tables: vec![],
373
+ detected_languages: None,
374
+ chunks: None,
375
+ images: None,
376
+ };
377
+
378
+ let config = ExtractionConfig::default();
379
+ processor.process(&mut result, &config).await.unwrap();
380
+
381
+ assert_eq!(result.content, "test content");
382
+ assert_eq!(
383
+ result.metadata.additional.get("processed_by").unwrap(),
384
+ &serde_json::json!("mock-processor")
385
+ );
386
+ }
387
+
388
+ #[test]
389
+ fn test_processing_stage_order() {
390
+ assert!(ProcessingStage::Early < ProcessingStage::Middle);
391
+ assert!(ProcessingStage::Middle < ProcessingStage::Late);
392
+ }
393
+
394
+ #[test]
395
+ fn test_post_processor_stage() {
396
+ let early = MockPostProcessor {
397
+ stage: ProcessingStage::Early,
398
+ };
399
+ let middle = MockPostProcessor {
400
+ stage: ProcessingStage::Middle,
401
+ };
402
+ let late = MockPostProcessor {
403
+ stage: ProcessingStage::Late,
404
+ };
405
+
406
+ assert_eq!(early.processing_stage(), ProcessingStage::Early);
407
+ assert_eq!(middle.processing_stage(), ProcessingStage::Middle);
408
+ assert_eq!(late.processing_stage(), ProcessingStage::Late);
409
+ }
410
+
411
+ #[test]
412
+ fn test_post_processor_should_process_default() {
413
+ let processor = MockPostProcessor {
414
+ stage: ProcessingStage::Early,
415
+ };
416
+
417
+ let result = ExtractionResult {
418
+ content: "test".to_string(),
419
+ mime_type: "text/plain".to_string(),
420
+ metadata: crate::types::Metadata::default(),
421
+ tables: vec![],
422
+ detected_languages: None,
423
+ chunks: None,
424
+ images: None,
425
+ };
426
+
427
+ let config = ExtractionConfig::default();
428
+
429
+ assert!(processor.should_process(&result, &config));
430
+ }
431
+
432
+ #[test]
433
+ fn test_processing_stage_equality() {
434
+ assert_eq!(ProcessingStage::Early, ProcessingStage::Early);
435
+ assert_ne!(ProcessingStage::Early, ProcessingStage::Middle);
436
+ assert_ne!(ProcessingStage::Middle, ProcessingStage::Late);
437
+ }
438
+
439
+ #[test]
440
+ fn test_processing_stage_clone() {
441
+ let stage = ProcessingStage::Middle;
442
+ let cloned = stage;
443
+ assert_eq!(stage, cloned);
444
+ }
445
+
446
+ #[test]
447
+ fn test_processing_stage_debug() {
448
+ let stage = ProcessingStage::Early;
449
+ let debug_str = format!("{:?}", stage);
450
+ assert!(debug_str.contains("Early"));
451
+ }
452
+
453
+ #[test]
454
+ fn test_processing_stage_hash() {
455
+ use std::collections::HashSet;
456
+ let mut set = HashSet::new();
457
+ set.insert(ProcessingStage::Early);
458
+ set.insert(ProcessingStage::Middle);
459
+ set.insert(ProcessingStage::Late);
460
+
461
+ assert_eq!(set.len(), 3);
462
+ assert!(set.contains(&ProcessingStage::Early));
463
+ }
464
+
465
+ #[tokio::test]
466
+ async fn test_post_processor_plugin_interface() {
467
+ let processor = MockPostProcessor {
468
+ stage: ProcessingStage::Middle,
469
+ };
470
+
471
+ assert_eq!(processor.name(), "mock-processor");
472
+ assert_eq!(processor.version(), "1.0.0");
473
+ assert!(processor.initialize().is_ok());
474
+ assert!(processor.shutdown().is_ok());
475
+ }
476
+
477
+ #[tokio::test]
478
+ async fn test_post_processor_empty_content() {
479
+ let processor = MockPostProcessor {
480
+ stage: ProcessingStage::Early,
481
+ };
482
+
483
+ let mut result = ExtractionResult {
484
+ content: String::new(),
485
+ mime_type: "text/plain".to_string(),
486
+ metadata: crate::types::Metadata::default(),
487
+ tables: vec![],
488
+ detected_languages: None,
489
+ chunks: None,
490
+ images: None,
491
+ };
492
+
493
+ let config = ExtractionConfig::default();
494
+ processor.process(&mut result, &config).await.unwrap();
495
+
496
+ assert_eq!(result.content, "");
497
+ assert!(result.metadata.additional.contains_key("processed_by"));
498
+ }
499
+
500
+ #[tokio::test]
501
+ async fn test_post_processor_preserves_metadata() {
502
+ let processor = MockPostProcessor {
503
+ stage: ProcessingStage::Early,
504
+ };
505
+
506
+ let mut additional = HashMap::new();
507
+ additional.insert("existing_key".to_string(), serde_json::json!("existing_value"));
508
+
509
+ let mut result = ExtractionResult {
510
+ content: "test".to_string(),
511
+ mime_type: "text/plain".to_string(),
512
+ metadata: crate::types::Metadata {
513
+ additional,
514
+ ..Default::default()
515
+ },
516
+ tables: vec![],
517
+ detected_languages: None,
518
+ chunks: None,
519
+ images: None,
520
+ };
521
+
522
+ let config = ExtractionConfig::default();
523
+ processor.process(&mut result, &config).await.unwrap();
524
+
525
+ assert_eq!(
526
+ result.metadata.additional.get("existing_key").unwrap(),
527
+ &serde_json::json!("existing_value")
528
+ );
529
+ assert!(result.metadata.additional.contains_key("processed_by"));
530
+ }
531
+
532
+ #[test]
533
+ fn test_post_processor_estimated_duration_default() {
534
+ let processor = MockPostProcessor {
535
+ stage: ProcessingStage::Early,
536
+ };
537
+
538
+ let result = ExtractionResult {
539
+ content: "test".to_string(),
540
+ mime_type: "text/plain".to_string(),
541
+ metadata: crate::types::Metadata::default(),
542
+ tables: vec![],
543
+ detected_languages: None,
544
+ chunks: None,
545
+ images: None,
546
+ };
547
+
548
+ assert_eq!(processor.estimated_duration_ms(&result), 0);
549
+ }
550
+
551
+ #[test]
552
+ fn test_post_processor_should_process_conditional() {
553
+ struct PdfOnlyProcessor;
554
+
555
+ impl Plugin for PdfOnlyProcessor {
556
+ fn name(&self) -> &str {
557
+ "pdf-only"
558
+ }
559
+ fn version(&self) -> String {
560
+ "1.0.0".to_string()
561
+ }
562
+ fn initialize(&self) -> Result<()> {
563
+ Ok(())
564
+ }
565
+ fn shutdown(&self) -> Result<()> {
566
+ Ok(())
567
+ }
568
+ }
569
+
570
+ #[async_trait]
571
+ impl PostProcessor for PdfOnlyProcessor {
572
+ async fn process(&self, _result: &mut ExtractionResult, _config: &ExtractionConfig) -> Result<()> {
573
+ Ok(())
574
+ }
575
+
576
+ fn processing_stage(&self) -> ProcessingStage {
577
+ ProcessingStage::Middle
578
+ }
579
+
580
+ fn should_process(&self, result: &ExtractionResult, _config: &ExtractionConfig) -> bool {
581
+ result.mime_type == "application/pdf"
582
+ }
583
+ }
584
+
585
+ let processor = PdfOnlyProcessor;
586
+ let config = ExtractionConfig::default();
587
+
588
+ let pdf_result = ExtractionResult {
589
+ content: "test".to_string(),
590
+ mime_type: "application/pdf".to_string(),
591
+ metadata: crate::types::Metadata::default(),
592
+ tables: vec![],
593
+ detected_languages: None,
594
+ chunks: None,
595
+ images: None,
596
+ };
597
+
598
+ let txt_result = ExtractionResult {
599
+ content: "test".to_string(),
600
+ mime_type: "text/plain".to_string(),
601
+ metadata: crate::types::Metadata::default(),
602
+ tables: vec![],
603
+ detected_languages: None,
604
+ chunks: None,
605
+ images: None,
606
+ };
607
+
608
+ assert!(processor.should_process(&pdf_result, &config));
609
+ assert!(!processor.should_process(&txt_result, &config));
610
+ }
611
+
612
+ #[tokio::test]
613
+ async fn test_post_processor_preserves_tables() {
614
+ use crate::types::Table;
615
+
616
+ let processor = MockPostProcessor {
617
+ stage: ProcessingStage::Early,
618
+ };
619
+
620
+ let table = Table {
621
+ cells: vec![vec!["A".to_string(), "B".to_string()]],
622
+ markdown: "| A | B |".to_string(),
623
+ page_number: 0,
624
+ };
625
+
626
+ let mut result = ExtractionResult {
627
+ content: "test".to_string(),
628
+ mime_type: "text/plain".to_string(),
629
+ metadata: crate::types::Metadata::default(),
630
+ tables: vec![table],
631
+ detected_languages: None,
632
+ chunks: None,
633
+ images: None,
634
+ };
635
+
636
+ let config = ExtractionConfig::default();
637
+ processor.process(&mut result, &config).await.unwrap();
638
+
639
+ assert_eq!(result.tables.len(), 1);
640
+ assert_eq!(result.tables[0].cells.len(), 1);
641
+ }
642
+ }