fitz-sage 0.11.0__tar.gz

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 (378) hide show
  1. fitz_sage-0.11.0/LICENSE +21 -0
  2. fitz_sage-0.11.0/PKG-INFO +778 -0
  3. fitz_sage-0.11.0/README.md +694 -0
  4. fitz_sage-0.11.0/fitz_sage/__init__.py +190 -0
  5. fitz_sage-0.11.0/fitz_sage/api/__init__.py +10 -0
  6. fitz_sage-0.11.0/fitz_sage/api/app.py +53 -0
  7. fitz_sage-0.11.0/fitz_sage/api/dependencies.py +24 -0
  8. fitz_sage-0.11.0/fitz_sage/api/error_handlers.py +71 -0
  9. fitz_sage-0.11.0/fitz_sage/api/models/__init__.py +26 -0
  10. fitz_sage-0.11.0/fitz_sage/api/models/schemas.py +95 -0
  11. fitz_sage-0.11.0/fitz_sage/api/routes/__init__.py +12 -0
  12. fitz_sage-0.11.0/fitz_sage/api/routes/collections.py +52 -0
  13. fitz_sage-0.11.0/fitz_sage/api/routes/health.py +28 -0
  14. fitz_sage-0.11.0/fitz_sage/api/routes/query.py +106 -0
  15. fitz_sage-0.11.0/fitz_sage/backends/__init__.py +16 -0
  16. fitz_sage-0.11.0/fitz_sage/backends/local_llm/__init__.py +0 -0
  17. fitz_sage-0.11.0/fitz_sage/backends/local_llm/chat.py +64 -0
  18. fitz_sage-0.11.0/fitz_sage/backends/local_llm/embedding.py +58 -0
  19. fitz_sage-0.11.0/fitz_sage/backends/local_llm/rerank.py +58 -0
  20. fitz_sage-0.11.0/fitz_sage/backends/local_llm/runtime.py +127 -0
  21. fitz_sage-0.11.0/fitz_sage/backends/local_vector_db/__init__.py +7 -0
  22. fitz_sage-0.11.0/fitz_sage/backends/local_vector_db/config.py +36 -0
  23. fitz_sage-0.11.0/fitz_sage/backends/local_vector_db/pgvector.py +732 -0
  24. fitz_sage-0.11.0/fitz_sage/backends/local_vector_db/runtime.py +18 -0
  25. fitz_sage-0.11.0/fitz_sage/cli/__init__.py +16 -0
  26. fitz_sage-0.11.0/fitz_sage/cli/cli.py +113 -0
  27. fitz_sage-0.11.0/fitz_sage/cli/commands/__init__.py +6 -0
  28. fitz_sage-0.11.0/fitz_sage/cli/commands/collections.py +182 -0
  29. fitz_sage-0.11.0/fitz_sage/cli/commands/config.py +394 -0
  30. fitz_sage-0.11.0/fitz_sage/cli/commands/eval.py +960 -0
  31. fitz_sage-0.11.0/fitz_sage/cli/commands/init.py +22 -0
  32. fitz_sage-0.11.0/fitz_sage/cli/commands/init_config.py +198 -0
  33. fitz_sage-0.11.0/fitz_sage/cli/commands/init_detector.py +26 -0
  34. fitz_sage-0.11.0/fitz_sage/cli/commands/init_models.py +101 -0
  35. fitz_sage-0.11.0/fitz_sage/cli/commands/init_wizard.py +354 -0
  36. fitz_sage-0.11.0/fitz_sage/cli/commands/query.py +415 -0
  37. fitz_sage-0.11.0/fitz_sage/cli/commands/reset.py +130 -0
  38. fitz_sage-0.11.0/fitz_sage/cli/commands/serve.py +95 -0
  39. fitz_sage-0.11.0/fitz_sage/cli/context.py +577 -0
  40. fitz_sage-0.11.0/fitz_sage/cli/services/__init__.py +6 -0
  41. fitz_sage-0.11.0/fitz_sage/cli/services/init_service.py +163 -0
  42. fitz_sage-0.11.0/fitz_sage/cli/ui/__init__.py +85 -0
  43. fitz_sage-0.11.0/fitz_sage/cli/ui/console.py +76 -0
  44. fitz_sage-0.11.0/fitz_sage/cli/ui/display.py +206 -0
  45. fitz_sage-0.11.0/fitz_sage/cli/ui/engine_selection.py +133 -0
  46. fitz_sage-0.11.0/fitz_sage/cli/ui/output.py +180 -0
  47. fitz_sage-0.11.0/fitz_sage/cli/ui/progress.py +150 -0
  48. fitz_sage-0.11.0/fitz_sage/cli/ui/prompts.py +230 -0
  49. fitz_sage-0.11.0/fitz_sage/cli/utils.py +59 -0
  50. fitz_sage-0.11.0/fitz_sage/cloud/__init__.py +15 -0
  51. fitz_sage-0.11.0/fitz_sage/cloud/cache_key.py +75 -0
  52. fitz_sage-0.11.0/fitz_sage/cloud/client.py +319 -0
  53. fitz_sage-0.11.0/fitz_sage/cloud/config.py +53 -0
  54. fitz_sage-0.11.0/fitz_sage/cloud/crypto.py +125 -0
  55. fitz_sage-0.11.0/fitz_sage/code/__init__.py +18 -0
  56. fitz_sage-0.11.0/fitz_sage/code/indexer.py +536 -0
  57. fitz_sage-0.11.0/fitz_sage/code/prompts.py +40 -0
  58. fitz_sage-0.11.0/fitz_sage/code/retriever.py +620 -0
  59. fitz_sage-0.11.0/fitz_sage/config/__init__.py +22 -0
  60. fitz_sage-0.11.0/fitz_sage/config/loader.py +201 -0
  61. fitz_sage-0.11.0/fitz_sage/core/__init__.py +107 -0
  62. fitz_sage-0.11.0/fitz_sage/core/answer.py +25 -0
  63. fitz_sage-0.11.0/fitz_sage/core/answer_mode.py +32 -0
  64. fitz_sage-0.11.0/fitz_sage/core/chunk.py +21 -0
  65. fitz_sage-0.11.0/fitz_sage/core/config.py +321 -0
  66. fitz_sage-0.11.0/fitz_sage/core/config_base.py +178 -0
  67. fitz_sage-0.11.0/fitz_sage/core/conflicts.py +46 -0
  68. fitz_sage-0.11.0/fitz_sage/core/constants.py +44 -0
  69. fitz_sage-0.11.0/fitz_sage/core/constraints.py +80 -0
  70. fitz_sage-0.11.0/fitz_sage/core/detect.py +297 -0
  71. fitz_sage-0.11.0/fitz_sage/core/document.py +124 -0
  72. fitz_sage-0.11.0/fitz_sage/core/engine.py +15 -0
  73. fitz_sage-0.11.0/fitz_sage/core/exceptions.py +149 -0
  74. fitz_sage-0.11.0/fitz_sage/core/firstrun.py +345 -0
  75. fitz_sage-0.11.0/fitz_sage/core/http.py +518 -0
  76. fitz_sage-0.11.0/fitz_sage/core/instrumentation.py +462 -0
  77. fitz_sage-0.11.0/fitz_sage/core/knowledge.py +0 -0
  78. fitz_sage-0.11.0/fitz_sage/core/math.py +48 -0
  79. fitz_sage-0.11.0/fitz_sage/core/paths/__init__.py +184 -0
  80. fitz_sage-0.11.0/fitz_sage/core/paths/cache.py +35 -0
  81. fitz_sage-0.11.0/fitz_sage/core/paths/config.py +17 -0
  82. fitz_sage-0.11.0/fitz_sage/core/paths/ingestion.py +39 -0
  83. fitz_sage-0.11.0/fitz_sage/core/paths/plugins.py +75 -0
  84. fitz_sage-0.11.0/fitz_sage/core/paths/storage.py +49 -0
  85. fitz_sage-0.11.0/fitz_sage/core/paths/workspace.py +74 -0
  86. fitz_sage-0.11.0/fitz_sage/core/platform.py +27 -0
  87. fitz_sage-0.11.0/fitz_sage/core/provenance.py +81 -0
  88. fitz_sage-0.11.0/fitz_sage/core/query.py +22 -0
  89. fitz_sage-0.11.0/fitz_sage/core/registry.py +485 -0
  90. fitz_sage-0.11.0/fitz_sage/core/timeouts.py +153 -0
  91. fitz_sage-0.11.0/fitz_sage/core/utils.py +103 -0
  92. fitz_sage-0.11.0/fitz_sage/engines/__init__.py +0 -0
  93. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/__init__.py +1 -0
  94. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/config/__init__.py +17 -0
  95. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/config/default.yaml +57 -0
  96. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/config/schema.py +376 -0
  97. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/context/__init__.py +1 -0
  98. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/context/assembler.py +169 -0
  99. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/context/compressor.py +177 -0
  100. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/engine.py +1433 -0
  101. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/generation/__init__.py +1 -0
  102. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/generation/synthesizer.py +238 -0
  103. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/__init__.py +1 -0
  104. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/enricher.py +120 -0
  105. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/import_graph_store.py +161 -0
  106. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/pipeline.py +1012 -0
  107. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/raw_file_store.py +127 -0
  108. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/schema.py +263 -0
  109. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/section_store.py +318 -0
  110. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/strategies/__init__.py +1 -0
  111. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/strategies/base.py +50 -0
  112. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/strategies/go.py +440 -0
  113. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/strategies/java.py +448 -0
  114. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/strategies/python_code.py +360 -0
  115. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/strategies/technical_doc.py +213 -0
  116. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/strategies/typescript.py +423 -0
  117. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/symbol_store.py +328 -0
  118. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/ingestion/table_store.py +206 -0
  119. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/progressive/__init__.py +18 -0
  120. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/progressive/builder.py +328 -0
  121. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/progressive/manifest.py +272 -0
  122. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/progressive/parsed_cache.py +208 -0
  123. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/progressive/worker.py +803 -0
  124. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/query_analyzer.py +163 -0
  125. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/query_batcher.py +332 -0
  126. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/__init__.py +1 -0
  127. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/expander.py +420 -0
  128. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/multihop.py +164 -0
  129. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/ranker.py +85 -0
  130. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/reader.py +292 -0
  131. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/reranker.py +76 -0
  132. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/router.py +483 -0
  133. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/strategies/__init__.py +1 -0
  134. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/strategies/agentic_search.py +354 -0
  135. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/strategies/base.py +16 -0
  136. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/strategies/chunk_fallback.py +65 -0
  137. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/strategies/code_search.py +250 -0
  138. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/strategies/llm_code_search.py +416 -0
  139. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/strategies/section_search.py +266 -0
  140. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/strategies/table_search.py +127 -0
  141. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval/table_handler.py +262 -0
  142. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/retrieval_profile.py +239 -0
  143. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/runtime.py +187 -0
  144. fitz_sage-0.11.0/fitz_sage/engines/fitz_krag/types.py +62 -0
  145. fitz_sage-0.11.0/fitz_sage/evaluation/__init__.py +143 -0
  146. fitz_sage-0.11.0/fitz_sage/evaluation/benchmarks/__init__.py +102 -0
  147. fitz_sage-0.11.0/fitz_sage/evaluation/benchmarks/beir.py +563 -0
  148. fitz_sage-0.11.0/fitz_sage/evaluation/benchmarks/fitz_gov.py +383 -0
  149. fitz_sage-0.11.0/fitz_sage/evaluation/benchmarks/rgb.py +642 -0
  150. fitz_sage-0.11.0/fitz_sage/evaluation/dashboard.py +366 -0
  151. fitz_sage-0.11.0/fitz_sage/evaluation/logger.py +248 -0
  152. fitz_sage-0.11.0/fitz_sage/evaluation/models.py +223 -0
  153. fitz_sage-0.11.0/fitz_sage/evaluation/schema.py +105 -0
  154. fitz_sage-0.11.0/fitz_sage/evaluation/stats.py +398 -0
  155. fitz_sage-0.11.0/fitz_sage/governance/__init__.py +103 -0
  156. fitz_sage-0.11.0/fitz_sage/governance/constraints/__init__.py +2 -0
  157. fitz_sage-0.11.0/fitz_sage/governance/constraints/aspect_classifier.py +123 -0
  158. fitz_sage-0.11.0/fitz_sage/governance/constraints/base.py +106 -0
  159. fitz_sage-0.11.0/fitz_sage/governance/constraints/feature_extractor.py +1039 -0
  160. fitz_sage-0.11.0/fitz_sage/governance/constraints/numerical_detector.py +405 -0
  161. fitz_sage-0.11.0/fitz_sage/governance/constraints/plugins/__init__.py +19 -0
  162. fitz_sage-0.11.0/fitz_sage/governance/constraints/plugins/answer_verification.py +251 -0
  163. fitz_sage-0.11.0/fitz_sage/governance/constraints/plugins/causal_attribution.py +179 -0
  164. fitz_sage-0.11.0/fitz_sage/governance/constraints/plugins/conflict_aware.py +516 -0
  165. fitz_sage-0.11.0/fitz_sage/governance/constraints/plugins/insufficient_evidence.py +1197 -0
  166. fitz_sage-0.11.0/fitz_sage/governance/constraints/plugins/specific_info_type.py +347 -0
  167. fitz_sage-0.11.0/fitz_sage/governance/constraints/runner.py +48 -0
  168. fitz_sage-0.11.0/fitz_sage/governance/constraints/semantic.py +731 -0
  169. fitz_sage-0.11.0/fitz_sage/governance/constraints/staged.py +323 -0
  170. fitz_sage-0.11.0/fitz_sage/governance/data/model_v6_cascade.joblib +0 -0
  171. fitz_sage-0.11.0/fitz_sage/governance/decider.py +515 -0
  172. fitz_sage-0.11.0/fitz_sage/governance/governor.py +354 -0
  173. fitz_sage-0.11.0/fitz_sage/governance/instructions.py +37 -0
  174. fitz_sage-0.11.0/fitz_sage/governance/protocol.py +21 -0
  175. fitz_sage-0.11.0/fitz_sage/ingestion/__init__.py +0 -0
  176. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/__init__.py +30 -0
  177. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/base.py +85 -0
  178. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/config.py +52 -0
  179. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/plugins/__init__.py +26 -0
  180. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/plugins/default/__init__.py +7 -0
  181. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/plugins/default/recursive.py +207 -0
  182. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/plugins/default/simple.py +120 -0
  183. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/plugins/markdown.py +372 -0
  184. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/plugins/pdf_sections.py +312 -0
  185. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/plugins/python_code.py +364 -0
  186. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/plugins/table.py +124 -0
  187. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/registry.py +97 -0
  188. fitz_sage-0.11.0/fitz_sage/ingestion/chunking/router.py +239 -0
  189. fitz_sage-0.11.0/fitz_sage/ingestion/detection.py +133 -0
  190. fitz_sage-0.11.0/fitz_sage/ingestion/diff/__init__.py +80 -0
  191. fitz_sage-0.11.0/fitz_sage/ingestion/diff/differ.py +385 -0
  192. fitz_sage-0.11.0/fitz_sage/ingestion/diff/executor.py +851 -0
  193. fitz_sage-0.11.0/fitz_sage/ingestion/diff/scanner.py +241 -0
  194. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/__init__.py +157 -0
  195. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/__init__.py +70 -0
  196. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/analyzer.py +319 -0
  197. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/base.py +181 -0
  198. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/generators.py +395 -0
  199. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/orchestrator.py +158 -0
  200. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/plugins/__init__.py +11 -0
  201. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/plugins/architecture_narrative.py +126 -0
  202. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/plugins/data_model_reference.py +67 -0
  203. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/plugins/dependency_summary.py +106 -0
  204. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/plugins/interface_catalog.py +86 -0
  205. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/plugins/navigation_index.py +94 -0
  206. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/artifacts/registry.py +184 -0
  207. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/base.py +25 -0
  208. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/bus.py +317 -0
  209. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/config.py +153 -0
  210. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/hierarchy/__init__.py +43 -0
  211. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/hierarchy/embedding_provider.py +68 -0
  212. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/hierarchy/enricher.py +714 -0
  213. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/hierarchy/grouper.py +80 -0
  214. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/hierarchy/matcher.py +95 -0
  215. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/hierarchy/semantic_grouper.py +172 -0
  216. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/models.py +37 -0
  217. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/modules/__init__.py +23 -0
  218. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/modules/base.py +88 -0
  219. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/modules/chunk/__init__.py +16 -0
  220. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/modules/chunk/entities.py +49 -0
  221. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/modules/chunk/keywords.py +39 -0
  222. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/modules/chunk/summary.py +39 -0
  223. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/pipeline.py +471 -0
  224. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/prompts/__init__.py +27 -0
  225. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/prompts/chunk/__init__.py +2 -0
  226. fitz_sage-0.11.0/fitz_sage/ingestion/enrichment/registry.py +60 -0
  227. fitz_sage-0.11.0/fitz_sage/ingestion/exceptions/__init__.py +15 -0
  228. fitz_sage-0.11.0/fitz_sage/ingestion/exceptions/base.py +13 -0
  229. fitz_sage-0.11.0/fitz_sage/ingestion/exceptions/chunking.py +13 -0
  230. fitz_sage-0.11.0/fitz_sage/ingestion/exceptions/config.py +13 -0
  231. fitz_sage-0.11.0/fitz_sage/ingestion/exceptions/vector.py +14 -0
  232. fitz_sage-0.11.0/fitz_sage/ingestion/hashing.py +127 -0
  233. fitz_sage-0.11.0/fitz_sage/ingestion/parser/__init__.py +16 -0
  234. fitz_sage-0.11.0/fitz_sage/ingestion/parser/base.py +81 -0
  235. fitz_sage-0.11.0/fitz_sage/ingestion/parser/plugins/__init__.py +9 -0
  236. fitz_sage-0.11.0/fitz_sage/ingestion/parser/plugins/base_parser.py +186 -0
  237. fitz_sage-0.11.0/fitz_sage/ingestion/parser/plugins/csv.py +158 -0
  238. fitz_sage-0.11.0/fitz_sage/ingestion/parser/plugins/docling.py +502 -0
  239. fitz_sage-0.11.0/fitz_sage/ingestion/parser/plugins/docling_vision.py +88 -0
  240. fitz_sage-0.11.0/fitz_sage/ingestion/parser/plugins/glm_ocr.py +489 -0
  241. fitz_sage-0.11.0/fitz_sage/ingestion/parser/plugins/lightweight.py +255 -0
  242. fitz_sage-0.11.0/fitz_sage/ingestion/parser/plugins/plaintext.py +298 -0
  243. fitz_sage-0.11.0/fitz_sage/ingestion/parser/router.py +243 -0
  244. fitz_sage-0.11.0/fitz_sage/ingestion/pipeline/__init__.py +2 -0
  245. fitz_sage-0.11.0/fitz_sage/ingestion/source/__init__.py +16 -0
  246. fitz_sage-0.11.0/fitz_sage/ingestion/source/base.py +91 -0
  247. fitz_sage-0.11.0/fitz_sage/ingestion/source/plugins/__init__.py +6 -0
  248. fitz_sage-0.11.0/fitz_sage/ingestion/source/plugins/filesystem.py +172 -0
  249. fitz_sage-0.11.0/fitz_sage/ingestion/state/__init__.py +35 -0
  250. fitz_sage-0.11.0/fitz_sage/ingestion/state/manager.py +452 -0
  251. fitz_sage-0.11.0/fitz_sage/ingestion/state/schema.py +163 -0
  252. fitz_sage-0.11.0/fitz_sage/ingestion/validation/__init__.py +0 -0
  253. fitz_sage-0.11.0/fitz_sage/integrations/__init__.py +17 -0
  254. fitz_sage-0.11.0/fitz_sage/integrations/base.py +317 -0
  255. fitz_sage-0.11.0/fitz_sage/integrations/langchain/__init__.py +45 -0
  256. fitz_sage-0.11.0/fitz_sage/integrations/langchain/runnable.py +283 -0
  257. fitz_sage-0.11.0/fitz_sage/integrations/llamaindex/__init__.py +43 -0
  258. fitz_sage-0.11.0/fitz_sage/integrations/llamaindex/query_engine.py +328 -0
  259. fitz_sage-0.11.0/fitz_sage/llm/__init__.py +46 -0
  260. fitz_sage-0.11.0/fitz_sage/llm/auth/__init__.py +35 -0
  261. fitz_sage-0.11.0/fitz_sage/llm/auth/api_key.py +57 -0
  262. fitz_sage-0.11.0/fitz_sage/llm/auth/base.py +25 -0
  263. fitz_sage-0.11.0/fitz_sage/llm/auth/certificates.py +141 -0
  264. fitz_sage-0.11.0/fitz_sage/llm/auth/composite.py +43 -0
  265. fitz_sage-0.11.0/fitz_sage/llm/auth/httpx_auth.py +37 -0
  266. fitz_sage-0.11.0/fitz_sage/llm/auth/m2m.py +186 -0
  267. fitz_sage-0.11.0/fitz_sage/llm/auth/token_provider.py +38 -0
  268. fitz_sage-0.11.0/fitz_sage/llm/client.py +123 -0
  269. fitz_sage-0.11.0/fitz_sage/llm/config.py +456 -0
  270. fitz_sage-0.11.0/fitz_sage/llm/factory.py +75 -0
  271. fitz_sage-0.11.0/fitz_sage/llm/providers/__init__.py +76 -0
  272. fitz_sage-0.11.0/fitz_sage/llm/providers/anthropic.py +232 -0
  273. fitz_sage-0.11.0/fitz_sage/llm/providers/base.py +148 -0
  274. fitz_sage-0.11.0/fitz_sage/llm/providers/cohere.py +269 -0
  275. fitz_sage-0.11.0/fitz_sage/llm/providers/enterprise.py +194 -0
  276. fitz_sage-0.11.0/fitz_sage/llm/providers/ollama.py +456 -0
  277. fitz_sage-0.11.0/fitz_sage/llm/providers/openai.py +276 -0
  278. fitz_sage-0.11.0/fitz_sage/llm/transforms.py +459 -0
  279. fitz_sage-0.11.0/fitz_sage/llm/types.py +145 -0
  280. fitz_sage-0.11.0/fitz_sage/logging/__init__.py +22 -0
  281. fitz_sage-0.11.0/fitz_sage/logging/logger.py +82 -0
  282. fitz_sage-0.11.0/fitz_sage/logging/tags.py +26 -0
  283. fitz_sage-0.11.0/fitz_sage/plugin_gen/__init__.py +38 -0
  284. fitz_sage-0.11.0/fitz_sage/plugin_gen/context.py +552 -0
  285. fitz_sage-0.11.0/fitz_sage/plugin_gen/generator.py +283 -0
  286. fitz_sage-0.11.0/fitz_sage/plugin_gen/library_context.py +512 -0
  287. fitz_sage-0.11.0/fitz_sage/plugin_gen/templates/__init__.py +9 -0
  288. fitz_sage-0.11.0/fitz_sage/plugin_gen/test_inputs.py +199 -0
  289. fitz_sage-0.11.0/fitz_sage/plugin_gen/types.py +213 -0
  290. fitz_sage-0.11.0/fitz_sage/plugin_gen/validators.py +589 -0
  291. fitz_sage-0.11.0/fitz_sage/prompts/__init__.py +16 -0
  292. fitz_sage-0.11.0/fitz_sage/prompts/entities.py +70 -0
  293. fitz_sage-0.11.0/fitz_sage/prompts/hierarchy/__init__.py +26 -0
  294. fitz_sage-0.11.0/fitz_sage/prompts/hierarchy/corpus_summary.py +31 -0
  295. fitz_sage-0.11.0/fitz_sage/prompts/hierarchy/corpus_summary_epistemic.py +186 -0
  296. fitz_sage-0.11.0/fitz_sage/prompts/hierarchy/group_summary.py +31 -0
  297. fitz_sage-0.11.0/fitz_sage/prompts/hierarchy/group_summary_epistemic.py +112 -0
  298. fitz_sage-0.11.0/fitz_sage/retrieval/__init__.py +47 -0
  299. fitz_sage-0.11.0/fitz_sage/retrieval/detection/__init__.py +68 -0
  300. fitz_sage-0.11.0/fitz_sage/retrieval/detection/classifier.py +192 -0
  301. fitz_sage-0.11.0/fitz_sage/retrieval/detection/concept_detector.py +206 -0
  302. fitz_sage-0.11.0/fitz_sage/retrieval/detection/data/model_v1_detection.joblib +0 -0
  303. fitz_sage-0.11.0/fitz_sage/retrieval/detection/detectors/__init__.py +8 -0
  304. fitz_sage-0.11.0/fitz_sage/retrieval/detection/detectors/expansion.py +243 -0
  305. fitz_sage-0.11.0/fitz_sage/retrieval/detection/llm_classifier.py +172 -0
  306. fitz_sage-0.11.0/fitz_sage/retrieval/detection/modules/__init__.py +44 -0
  307. fitz_sage-0.11.0/fitz_sage/retrieval/detection/modules/aggregation.py +74 -0
  308. fitz_sage-0.11.0/fitz_sage/retrieval/detection/modules/base.py +75 -0
  309. fitz_sage-0.11.0/fitz_sage/retrieval/detection/modules/comparison.py +48 -0
  310. fitz_sage-0.11.0/fitz_sage/retrieval/detection/modules/freshness.py +51 -0
  311. fitz_sage-0.11.0/fitz_sage/retrieval/detection/modules/rewriter.py +51 -0
  312. fitz_sage-0.11.0/fitz_sage/retrieval/detection/modules/temporal.py +79 -0
  313. fitz_sage-0.11.0/fitz_sage/retrieval/detection/protocol.py +78 -0
  314. fitz_sage-0.11.0/fitz_sage/retrieval/detection/registry.py +259 -0
  315. fitz_sage-0.11.0/fitz_sage/retrieval/entity_graph/__init__.py +6 -0
  316. fitz_sage-0.11.0/fitz_sage/retrieval/entity_graph/store.py +447 -0
  317. fitz_sage-0.11.0/fitz_sage/retrieval/hyde/__init__.py +6 -0
  318. fitz_sage-0.11.0/fitz_sage/retrieval/hyde/generator.py +121 -0
  319. fitz_sage-0.11.0/fitz_sage/retrieval/rewriter/__init__.py +13 -0
  320. fitz_sage-0.11.0/fitz_sage/retrieval/rewriter/rewriter.py +297 -0
  321. fitz_sage-0.11.0/fitz_sage/retrieval/rewriter/types.py +95 -0
  322. fitz_sage-0.11.0/fitz_sage/retrieval/sparse/__init__.py +11 -0
  323. fitz_sage-0.11.0/fitz_sage/retrieval/sparse/index.py +297 -0
  324. fitz_sage-0.11.0/fitz_sage/retrieval/vocabulary/__init__.py +43 -0
  325. fitz_sage-0.11.0/fitz_sage/retrieval/vocabulary/detector.py +269 -0
  326. fitz_sage-0.11.0/fitz_sage/retrieval/vocabulary/matcher.py +218 -0
  327. fitz_sage-0.11.0/fitz_sage/retrieval/vocabulary/models.py +144 -0
  328. fitz_sage-0.11.0/fitz_sage/retrieval/vocabulary/store.py +481 -0
  329. fitz_sage-0.11.0/fitz_sage/retrieval/vocabulary/variations.py +168 -0
  330. fitz_sage-0.11.0/fitz_sage/runtime/__init__.py +128 -0
  331. fitz_sage-0.11.0/fitz_sage/runtime/registry.py +458 -0
  332. fitz_sage-0.11.0/fitz_sage/runtime/runner.py +222 -0
  333. fitz_sage-0.11.0/fitz_sage/sdk/__init__.py +16 -0
  334. fitz_sage-0.11.0/fitz_sage/sdk/fitz.py +146 -0
  335. fitz_sage-0.11.0/fitz_sage/services/__init__.py +11 -0
  336. fitz_sage-0.11.0/fitz_sage/services/fitz_service.py +565 -0
  337. fitz_sage-0.11.0/fitz_sage/storage/__init__.py +33 -0
  338. fitz_sage-0.11.0/fitz_sage/storage/config.py +64 -0
  339. fitz_sage-0.11.0/fitz_sage/storage/postgres.py +864 -0
  340. fitz_sage-0.11.0/fitz_sage/structured/__init__.py +142 -0
  341. fitz_sage-0.11.0/fitz_sage/structured/constants.py +64 -0
  342. fitz_sage-0.11.0/fitz_sage/structured/derived.py +432 -0
  343. fitz_sage-0.11.0/fitz_sage/structured/executor.py +396 -0
  344. fitz_sage-0.11.0/fitz_sage/structured/formatter.py +203 -0
  345. fitz_sage-0.11.0/fitz_sage/structured/ingestion.py +299 -0
  346. fitz_sage-0.11.0/fitz_sage/structured/router.py +226 -0
  347. fitz_sage-0.11.0/fitz_sage/structured/schema.py +462 -0
  348. fitz_sage-0.11.0/fitz_sage/structured/sql_generator.py +421 -0
  349. fitz_sage-0.11.0/fitz_sage/structured/types.py +366 -0
  350. fitz_sage-0.11.0/fitz_sage/tabular/__init__.py +86 -0
  351. fitz_sage-0.11.0/fitz_sage/tabular/direct_query.py +711 -0
  352. fitz_sage-0.11.0/fitz_sage/tabular/extractor.py +163 -0
  353. fitz_sage-0.11.0/fitz_sage/tabular/models.py +197 -0
  354. fitz_sage-0.11.0/fitz_sage/tabular/parser/__init__.py +18 -0
  355. fitz_sage-0.11.0/fitz_sage/tabular/parser/csv_parser.py +113 -0
  356. fitz_sage-0.11.0/fitz_sage/tabular/query.py +640 -0
  357. fitz_sage-0.11.0/fitz_sage/tabular/registry.py +58 -0
  358. fitz_sage-0.11.0/fitz_sage/tabular/store/__init__.py +55 -0
  359. fitz_sage-0.11.0/fitz_sage/tabular/store/base.py +123 -0
  360. fitz_sage-0.11.0/fitz_sage/tabular/store/postgres.py +573 -0
  361. fitz_sage-0.11.0/fitz_sage/utils/logging.py +179 -0
  362. fitz_sage-0.11.0/fitz_sage/vector_db/__init__.py +42 -0
  363. fitz_sage-0.11.0/fitz_sage/vector_db/base.py +62 -0
  364. fitz_sage-0.11.0/fitz_sage/vector_db/custom.py +354 -0
  365. fitz_sage-0.11.0/fitz_sage/vector_db/loader.py +794 -0
  366. fitz_sage-0.11.0/fitz_sage/vector_db/plugins/pgvector.yaml +61 -0
  367. fitz_sage-0.11.0/fitz_sage/vector_db/registry.py +64 -0
  368. fitz_sage-0.11.0/fitz_sage/vector_db/schemas/vector_db_plugin_schema.yaml +300 -0
  369. fitz_sage-0.11.0/fitz_sage/vector_db/types.py +165 -0
  370. fitz_sage-0.11.0/fitz_sage/vector_db/writer.py +159 -0
  371. fitz_sage-0.11.0/fitz_sage.egg-info/PKG-INFO +778 -0
  372. fitz_sage-0.11.0/fitz_sage.egg-info/SOURCES.txt +376 -0
  373. fitz_sage-0.11.0/fitz_sage.egg-info/dependency_links.txt +1 -0
  374. fitz_sage-0.11.0/fitz_sage.egg-info/entry_points.txt +2 -0
  375. fitz_sage-0.11.0/fitz_sage.egg-info/requires.txt +62 -0
  376. fitz_sage-0.11.0/fitz_sage.egg-info/top_level.txt +1 -0
  377. fitz_sage-0.11.0/pyproject.toml +204 -0
  378. fitz_sage-0.11.0/setup.cfg +4 -0
fitz_sage-0.11.0/LICENSE ADDED
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2025 Yan Fitzner
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.