qwen-sci 0.2.2__py3-none-any.whl

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 (549) hide show
  1. qwen_sci-0.2.2.dist-info/METADATA +492 -0
  2. qwen_sci-0.2.2.dist-info/RECORD +549 -0
  3. qwen_sci-0.2.2.dist-info/WHEEL +4 -0
  4. qwen_sci-0.2.2.dist-info/entry_points.txt +6 -0
  5. qwen_sci-0.2.2.dist-info/licenses/LICENSE +21 -0
  6. src/1.txt +1 -0
  7. src/__init__.py +1 -0
  8. src/__main__.py +5 -0
  9. src/agents/__init__.py +1 -0
  10. src/agents/experiment_design_agent/__init__.py +268 -0
  11. src/agents/experiment_design_agent/artifacts.py +747 -0
  12. src/agents/experiment_design_agent/cache.py +319 -0
  13. src/agents/experiment_design_agent/completeness.py +130 -0
  14. src/agents/experiment_design_agent/contracts.py +771 -0
  15. src/agents/experiment_design_agent/counterexample_analyzer.py +187 -0
  16. src/agents/experiment_design_agent/design_evidence_paper_screener.py +470 -0
  17. src/agents/experiment_design_agent/discipline_catalog.py +178 -0
  18. src/agents/experiment_design_agent/evidence_cards.py +338 -0
  19. src/agents/experiment_design_agent/evidence_planner.py +452 -0
  20. src/agents/experiment_design_agent/formal_reasoning_planner.py +884 -0
  21. src/agents/experiment_design_agent/fulltext_acquisition.py +863 -0
  22. src/agents/experiment_design_agent/idea_adapter.py +100 -0
  23. src/agents/experiment_design_agent/idea_intake.py +87 -0
  24. src/agents/experiment_design_agent/llm_json.py +328 -0
  25. src/agents/experiment_design_agent/orchestrator.py +1220 -0
  26. src/agents/experiment_design_agent/reasoning_context.py +183 -0
  27. src/agents/experiment_design_agent/reasoning_validation.py +546 -0
  28. src/agents/experiment_design_agent/run.py +481 -0
  29. src/agents/experiment_design_agent/run_logging.py +292 -0
  30. src/agents/experiment_design_agent/scope_gate.py +198 -0
  31. src/agents/experiment_design_agent/study_type_composer.py +1360 -0
  32. src/agents/experiment_design_agent/survey_evidence.py +1381 -0
  33. src/agents/experiment_design_agent/template_router.py +195 -0
  34. src/agents/experiment_design_agent/variable_claim_extractor.py +93 -0
  35. src/agents/idea_agent/__init__.py +1 -0
  36. src/agents/idea_agent/agent/__init__.py +6 -0
  37. src/agents/idea_agent/agent/artifacts.py +308 -0
  38. src/agents/idea_agent/agent/base.py +122 -0
  39. src/agents/idea_agent/agent/ligagent.py +488 -0
  40. src/agents/idea_agent/agent/mcts.py +2060 -0
  41. src/agents/idea_agent/agent/prompts/__init__.py +48 -0
  42. src/agents/idea_agent/agent/prompts/advanced_analysis.py +235 -0
  43. src/agents/idea_agent/agent/prompts/algorithm_alignment.py +55 -0
  44. src/agents/idea_agent/agent/prompts/algorithm_structuring.py +48 -0
  45. src/agents/idea_agent/agent/prompts/component_extraction.py +62 -0
  46. src/agents/idea_agent/agent/prompts/component_novelty_evaluation.py +54 -0
  47. src/agents/idea_agent/agent/prompts/experiment_findings_extraction.py +39 -0
  48. src/agents/idea_agent/agent/prompts/idea_fusion.py +295 -0
  49. src/agents/idea_agent/agent/prompts/idea_introduction.py +30 -0
  50. src/agents/idea_agent/agent/prompts/idea_result_alignment.py +39 -0
  51. src/agents/idea_agent/agent/prompts/input_interpreter.py +52 -0
  52. src/agents/idea_agent/agent/prompts/keynote_ops.py +78 -0
  53. src/agents/idea_agent/agent/prompts/mcts_evaluation.py +235 -0
  54. src/agents/idea_agent/agent/prompts/mcts_generation.py +140 -0
  55. src/agents/idea_agent/agent/prompts/mechanism_commit_query.py +99 -0
  56. src/agents/idea_agent/agent/prompts/memory_selection.py +0 -0
  57. src/agents/idea_agent/agent/prompts/prompt_modes.py +16 -0
  58. src/agents/idea_agent/agent/prompts/rag_query.py +34 -0
  59. src/agents/idea_agent/agent/prompts/re_analysis_replan.py +48 -0
  60. src/agents/idea_agent/agent/prompts/reference_grounding.py +34 -0
  61. src/agents/idea_agent/agent/prompts/root_domain_classification.py +32 -0
  62. src/agents/idea_agent/agent/prompts/scientific_debate.py +88 -0
  63. src/agents/idea_agent/agent/prompts/scientific_materialization.py +45 -0
  64. src/agents/idea_agent/agent/prompts/skill_instantiation.py +193 -0
  65. src/agents/idea_agent/agent/prompts/theory_transfer_query.py +93 -0
  66. src/agents/idea_agent/agent/prompts/topic_background.py +15 -0
  67. src/agents/idea_agent/agent/skills/DEFAULT_SKILL_TEMPLATES.json +263 -0
  68. src/agents/idea_agent/agent/skills/edit_operator_skills/alternative-path-contrast/SKILL.md +40 -0
  69. src/agents/idea_agent/agent/skills/edit_operator_skills/alternative-path-contrast/references/contrast_patterns.md +22 -0
  70. src/agents/idea_agent/agent/skills/edit_operator_skills/feedback-closed-loop/SKILL.md +40 -0
  71. src/agents/idea_agent/agent/skills/edit_operator_skills/feedback-closed-loop/references/mechanism_cards.md +24 -0
  72. src/agents/idea_agent/agent/skills/edit_operator_skills/hierarchical-decomposition/SKILL.md +39 -0
  73. src/agents/idea_agent/agent/skills/edit_operator_skills/hierarchical-decomposition/references/hierarchy_patterns.md +22 -0
  74. src/agents/idea_agent/agent/skills/edit_operator_skills/mechanism-commit-innovation/SKILL.md +39 -0
  75. src/agents/idea_agent/agent/skills/edit_operator_skills/mechanism-commit-innovation/references/mechanism_commit_patterns.md +24 -0
  76. src/agents/idea_agent/agent/skills/edit_operator_skills/multi-scale-coordinator/SKILL.md +39 -0
  77. src/agents/idea_agent/agent/skills/edit_operator_skills/multi-scale-coordinator/references/coordination_patterns.md +22 -0
  78. src/agents/idea_agent/agent/skills/edit_operator_skills/speculative-execution-with-repair/SKILL.md +41 -0
  79. src/agents/idea_agent/agent/skills/edit_operator_skills/speculative-execution-with-repair/references/speculation_patterns.md +22 -0
  80. src/agents/idea_agent/agent/skills/edit_operator_skills/surgical-modularity/SKILL.md +39 -0
  81. src/agents/idea_agent/agent/skills/edit_operator_skills/surgical-modularity/references/modularity_patterns.md +22 -0
  82. src/agents/idea_agent/agent/skills/edit_operator_skills/theory-transfer-injection/SKILL.md +40 -0
  83. src/agents/idea_agent/agent/skills/edit_operator_skills/theory-transfer-injection/references/transfer_patterns.md +24 -0
  84. src/agents/idea_agent/example/idea_result.json +900 -0
  85. src/agents/idea_agent/run.py +276 -0
  86. src/agents/idea_agent/tests/test_advanced_analysis_contract.py +90 -0
  87. src/agents/idea_agent/tests/test_gap_hypothesis_bridge.py +166 -0
  88. src/agents/idea_agent/tests/test_idea_debate.py +480 -0
  89. src/agents/idea_agent/tests/test_idea_result_v5.py +99 -0
  90. src/agents/idea_agent/tests/test_idea_synthesis.py +92 -0
  91. src/agents/idea_agent/tests/test_mature_idea_sources.py +27 -0
  92. src/agents/idea_agent/tests/test_multimodal_idea_integration.py +386 -0
  93. src/agents/idea_agent/tests/test_profile_materialization_and_scoring.py +219 -0
  94. src/agents/idea_agent/tests/test_profile_native_skill_rendering.py +122 -0
  95. src/agents/idea_agent/tests/test_result_persistence.py +183 -0
  96. src/agents/idea_agent/tests/test_scientific_intervention_profiles.py +241 -0
  97. src/agents/idea_agent/tests/test_scientific_rubric_and_prompts.py +559 -0
  98. src/agents/idea_agent/tests/test_strict_json_object_responses.py +48 -0
  99. src/agents/idea_agent/utils/core/__init__.py +1 -0
  100. src/agents/idea_agent/utils/core/ablation_inputs.py +60 -0
  101. src/agents/idea_agent/utils/core/chat_errors.py +95 -0
  102. src/agents/idea_agent/utils/core/chat_router.py +130 -0
  103. src/agents/idea_agent/utils/core/chat_transport.py +188 -0
  104. src/agents/idea_agent/utils/core/config_loader.py +166 -0
  105. src/agents/idea_agent/utils/core/json_utils.py +28 -0
  106. src/agents/idea_agent/utils/core/logger.py +287 -0
  107. src/agents/idea_agent/utils/core/progress.py +58 -0
  108. src/agents/idea_agent/utils/core/response_parsing.py +57 -0
  109. src/agents/idea_agent/utils/core/run_inputs.py +198 -0
  110. src/agents/idea_agent/utils/mcts/__init__.py +1 -0
  111. src/agents/idea_agent/utils/mcts/component_novelty.py +342 -0
  112. src/agents/idea_agent/utils/mcts/defect_registry.py +302 -0
  113. src/agents/idea_agent/utils/mcts/idea_routes.py +61 -0
  114. src/agents/idea_agent/utils/mcts/idea_taste_presets.py +312 -0
  115. src/agents/idea_agent/utils/mcts/mcts_helpers.py +738 -0
  116. src/agents/idea_agent/utils/mcts/mcts_runtime.py +3000 -0
  117. src/agents/idea_agent/utils/mcts/scientific_intervention_ontology.py +1561 -0
  118. src/agents/idea_agent/utils/mcts/scientific_rubric.py +211 -0
  119. src/agents/idea_agent/utils/mcts/skill_parsing.py +92 -0
  120. src/agents/idea_agent/utils/papers/__init__.py +1 -0
  121. src/agents/idea_agent/utils/papers/paper_graph_vector_store.py +499 -0
  122. src/agents/idea_agent/utils/papers/paper_repository.py +372 -0
  123. src/agents/idea_agent/utils/prompting/__init__.py +1 -0
  124. src/agents/idea_agent/utils/prompting/prompt_views.py +637 -0
  125. src/agents/idea_agent/utils/workflow/__init__.py +1 -0
  126. src/agents/idea_agent/utils/workflow/advanced_analysis_contract.py +244 -0
  127. src/agents/idea_agent/utils/workflow/gap_hypothesis_bridge.py +461 -0
  128. src/agents/idea_agent/utils/workflow/idea_contract.py +426 -0
  129. src/agents/idea_agent/utils/workflow/idea_debate.py +1238 -0
  130. src/agents/idea_agent/utils/workflow/idea_diversity.py +163 -0
  131. src/agents/idea_agent/utils/workflow/idea_fusion.py +889 -0
  132. src/agents/idea_agent/utils/workflow/idea_helpers.py +478 -0
  133. src/agents/idea_agent/utils/workflow/idea_portfolio.py +422 -0
  134. src/agents/idea_agent/utils/workflow/idea_synthesis.py +482 -0
  135. src/agents/idea_agent/utils/workflow/ligagent_flow.py +581 -0
  136. src/agents/idea_agent/utils/workflow/ligagent_handlers.py +1836 -0
  137. src/agents/idea_agent/utils/workflow/ligagent_helpers.py +1331 -0
  138. src/agents/idea_agent/utils/workflow/ligagent_utils.py +322 -0
  139. src/agents/idea_agent/utils/workflow/mature_idea_sources.py +648 -0
  140. src/agents/idea_agent/utils/workflow/multimodal_data_anchoring.py +578 -0
  141. src/agents/idea_agent/utils/workflow/result_persistence.py +774 -0
  142. src/agents/idea_agent/utils/workflow/stage_contract.py +76 -0
  143. src/agents/idea_agent/utils/workflow/workflow_runtime.py +288 -0
  144. src/agents/research_plan_author/__init__.py +50 -0
  145. src/agents/research_plan_author/artifacts.py +270 -0
  146. src/agents/research_plan_author/authoring_blueprint.py +1246 -0
  147. src/agents/research_plan_author/bibtex_renderer.py +252 -0
  148. src/agents/research_plan_author/contract_repair.py +131 -0
  149. src/agents/research_plan_author/contracts.py +745 -0
  150. src/agents/research_plan_author/cross_section_editor.py +232 -0
  151. src/agents/research_plan_author/document_quality.py +987 -0
  152. src/agents/research_plan_author/idea_evolution.py +223 -0
  153. src/agents/research_plan_author/input_loader.py +76 -0
  154. src/agents/research_plan_author/latex_compiler.py +260 -0
  155. src/agents/research_plan_author/latex_safety.py +159 -0
  156. src/agents/research_plan_author/llm_json.py +74 -0
  157. src/agents/research_plan_author/markdown_renderer.py +275 -0
  158. src/agents/research_plan_author/pdf_validator.py +149 -0
  159. src/agents/research_plan_author/render.py +166 -0
  160. src/agents/research_plan_author/render_artifacts.py +187 -0
  161. src/agents/research_plan_author/run.py +765 -0
  162. src/agents/research_plan_author/run_logging.py +209 -0
  163. src/agents/research_plan_author/section_cache.py +184 -0
  164. src/agents/research_plan_author/section_composer.py +1764 -0
  165. src/agents/research_plan_author/section_router.py +209 -0
  166. src/agents/research_plan_author/semantic_validator.py +254 -0
  167. src/agents/research_plan_author/source_bundle.py +46 -0
  168. src/agents/research_plan_author/source_registry.py +208 -0
  169. src/agents/research_plan_author/survey_source_loader.py +137 -0
  170. src/agents/research_plan_author/template_adapter.py +173 -0
  171. src/agents/research_plan_author/template_profile.py +204 -0
  172. src/agents/research_plan_author/tex_renderer.py +496 -0
  173. src/agents/research_plan_author/theory_presentation.py +134 -0
  174. src/agents/research_plan_author/theory_spine.py +703 -0
  175. src/agents/survey_agent/Dockerfile +54 -0
  176. src/agents/survey_agent/README.md +350 -0
  177. src/agents/survey_agent/config/deep_survey.yaml +217 -0
  178. src/agents/survey_agent/config/deep_survey_batch.yaml +176 -0
  179. src/agents/survey_agent/config/deep_survey_fast.yaml +183 -0
  180. src/agents/survey_agent/config/evaluate_baseline.yaml +197 -0
  181. src/agents/survey_agent/config/outcomeRAG.yaml +159 -0
  182. src/agents/survey_agent/env_report.md +348 -0
  183. src/agents/survey_agent/modules/__init__.py +0 -0
  184. src/agents/survey_agent/modules/code_collector.py +2194 -0
  185. src/agents/survey_agent/modules/code_report_generator.py +1282 -0
  186. src/agents/survey_agent/modules/data_manager.py +1713 -0
  187. src/agents/survey_agent/modules/database.py +167 -0
  188. src/agents/survey_agent/modules/fulltext_download_cache.py +462 -0
  189. src/agents/survey_agent/modules/fulltext_http.py +134 -0
  190. src/agents/survey_agent/modules/fulltext_resolution.py +749 -0
  191. src/agents/survey_agent/modules/judge.py +434 -0
  192. src/agents/survey_agent/modules/outcome_RAG.py +236 -0
  193. src/agents/survey_agent/modules/paper_graph_retriever.py +1075 -0
  194. src/agents/survey_agent/modules/pe.py +1729 -0
  195. src/agents/survey_agent/modules/pseudo_creator.py +782 -0
  196. src/agents/survey_agent/modules/pseudo_reviser.py +537 -0
  197. src/agents/survey_agent/modules/refine_agent.py +1952 -0
  198. src/agents/survey_agent/modules/survey_generator.py +5691 -0
  199. src/agents/survey_agent/modules/survey_visualizer.py +1817 -0
  200. src/agents/survey_agent/modules/visual_prompts.py +327 -0
  201. src/agents/survey_agent/modules/work_analyzer.py +2750 -0
  202. src/agents/survey_agent/modules/work_collector.py +5649 -0
  203. src/agents/survey_agent/scripts/baseline_evaluation.py +411 -0
  204. src/agents/survey_agent/scripts/run_deep_survey.py +224 -0
  205. src/agents/survey_agent/scripts/run_deep_survey_adapter.py +407 -0
  206. src/agents/survey_agent/scripts/run_deep_survey_adapter_with_input.py +409 -0
  207. src/agents/survey_agent/scripts/run_deep_survey_batch.py +228 -0
  208. src/agents/survey_agent/scripts/run_deep_survey_batch_arg.py +243 -0
  209. src/agents/survey_agent/scripts/run_deep_survey_batch_settings.py +218 -0
  210. src/agents/survey_agent/scripts/run_keynotes_gen.py +128 -0
  211. src/agents/survey_agent/scripts/run_work_clusters.py +140 -0
  212. src/agents/survey_agent/utils/__init__.py +0 -0
  213. src/agents/survey_agent/utils/ablation_utils.py +317 -0
  214. src/agents/survey_agent/utils/api_call.py +2518 -0
  215. src/agents/survey_agent/utils/chat_utils.py +82 -0
  216. src/agents/survey_agent/utils/config_utils.py +70 -0
  217. src/agents/survey_agent/utils/convert_to_md.py +29 -0
  218. src/agents/survey_agent/utils/err_info.py +24 -0
  219. src/agents/survey_agent/utils/file_utils.py +176 -0
  220. src/agents/survey_agent/utils/gpu_utils.py +145 -0
  221. src/agents/survey_agent/utils/graph_visualizer.py +279 -0
  222. src/agents/survey_agent/utils/html_utils.py +181 -0
  223. src/agents/survey_agent/utils/latex_generator_intra_cluster.py +529 -0
  224. src/agents/survey_agent/utils/latex_generator_relation_table.py +193 -0
  225. src/agents/survey_agent/utils/mineru_section_packer.py +334 -0
  226. src/agents/survey_agent/utils/mineru_utils.py +396 -0
  227. src/agents/survey_agent/utils/paper.py +144 -0
  228. src/agents/survey_agent/utils/read_yaml.py +15 -0
  229. src/agents/survey_agent/utils/repo_utils.py +167 -0
  230. src/agents/survey_agent/utils/rich_logger.py +77 -0
  231. src/agents/survey_agent/utils/step2v2.py +874 -0
  232. src/agents/survey_agent/utils/step2v2_extractor.py +856 -0
  233. src/agents/survey_agent/utils/topic_survey_storage.py +309 -0
  234. src/agents/survey_agent/utils/utils.py +118 -0
  235. src/agents/survey_agent/utils/work.py +2 -0
  236. src/cli.py +1611 -0
  237. src/config/__init__.py +213 -0
  238. src/config/default.yaml +1122 -0
  239. src/harness/LICENSE +21 -0
  240. src/harness/VENDORED.md +14 -0
  241. src/harness/src/__init__.py +0 -0
  242. src/harness/src/openharness/__init__.py +0 -0
  243. src/harness/src/openharness/__main__.py +6 -0
  244. src/harness/src/openharness/api/__init__.py +21 -0
  245. src/harness/src/openharness/api/client.py +281 -0
  246. src/harness/src/openharness/api/codex_client.py +407 -0
  247. src/harness/src/openharness/api/copilot_auth.py +240 -0
  248. src/harness/src/openharness/api/copilot_client.py +134 -0
  249. src/harness/src/openharness/api/errors.py +19 -0
  250. src/harness/src/openharness/api/openai_client.py +577 -0
  251. src/harness/src/openharness/api/provider.py +186 -0
  252. src/harness/src/openharness/api/registry.py +437 -0
  253. src/harness/src/openharness/api/usage.py +17 -0
  254. src/harness/src/openharness/auth/__init__.py +29 -0
  255. src/harness/src/openharness/auth/external.py +610 -0
  256. src/harness/src/openharness/auth/flows.py +189 -0
  257. src/harness/src/openharness/auth/manager.py +482 -0
  258. src/harness/src/openharness/auth/storage.py +269 -0
  259. src/harness/src/openharness/autopilot/__init__.py +23 -0
  260. src/harness/src/openharness/autopilot/service.py +2239 -0
  261. src/harness/src/openharness/autopilot/types.py +91 -0
  262. src/harness/src/openharness/bridge/__init__.py +20 -0
  263. src/harness/src/openharness/bridge/manager.py +106 -0
  264. src/harness/src/openharness/bridge/session_runner.py +46 -0
  265. src/harness/src/openharness/bridge/types.py +37 -0
  266. src/harness/src/openharness/bridge/work_secret.py +41 -0
  267. src/harness/src/openharness/channels/UPSTREAM +6 -0
  268. src/harness/src/openharness/channels/__init__.py +22 -0
  269. src/harness/src/openharness/channels/adapter.py +131 -0
  270. src/harness/src/openharness/channels/bus/__init__.py +6 -0
  271. src/harness/src/openharness/channels/bus/events.py +38 -0
  272. src/harness/src/openharness/channels/bus/queue.py +44 -0
  273. src/harness/src/openharness/channels/impl/__init__.py +6 -0
  274. src/harness/src/openharness/channels/impl/base.py +141 -0
  275. src/harness/src/openharness/channels/impl/dingtalk.py +445 -0
  276. src/harness/src/openharness/channels/impl/discord.py +311 -0
  277. src/harness/src/openharness/channels/impl/email.py +410 -0
  278. src/harness/src/openharness/channels/impl/feishu.py +1342 -0
  279. src/harness/src/openharness/channels/impl/manager.py +257 -0
  280. src/harness/src/openharness/channels/impl/matrix.py +700 -0
  281. src/harness/src/openharness/channels/impl/mochat.py +897 -0
  282. src/harness/src/openharness/channels/impl/qq.py +141 -0
  283. src/harness/src/openharness/channels/impl/slack.py +284 -0
  284. src/harness/src/openharness/channels/impl/telegram.py +525 -0
  285. src/harness/src/openharness/channels/impl/whatsapp.py +159 -0
  286. src/harness/src/openharness/cli.py +2551 -0
  287. src/harness/src/openharness/commands/__init__.py +21 -0
  288. src/harness/src/openharness/commands/registry.py +2783 -0
  289. src/harness/src/openharness/config/__init__.py +34 -0
  290. src/harness/src/openharness/config/paths.py +160 -0
  291. src/harness/src/openharness/config/schema.py +119 -0
  292. src/harness/src/openharness/config/settings.py +1104 -0
  293. src/harness/src/openharness/coordinator/__init__.py +12 -0
  294. src/harness/src/openharness/coordinator/agent_definitions.py +975 -0
  295. src/harness/src/openharness/coordinator/coordinator_mode.py +520 -0
  296. src/harness/src/openharness/engine/__init__.py +80 -0
  297. src/harness/src/openharness/engine/cost_tracker.py +24 -0
  298. src/harness/src/openharness/engine/messages.py +222 -0
  299. src/harness/src/openharness/engine/query.py +1188 -0
  300. src/harness/src/openharness/engine/query_engine.py +313 -0
  301. src/harness/src/openharness/engine/stream_events.py +90 -0
  302. src/harness/src/openharness/hooks/__init__.py +50 -0
  303. src/harness/src/openharness/hooks/events.py +20 -0
  304. src/harness/src/openharness/hooks/executor.py +242 -0
  305. src/harness/src/openharness/hooks/hot_reload.py +31 -0
  306. src/harness/src/openharness/hooks/loader.py +68 -0
  307. src/harness/src/openharness/hooks/schemas.py +66 -0
  308. src/harness/src/openharness/hooks/types.py +38 -0
  309. src/harness/src/openharness/keybindings/__init__.py +14 -0
  310. src/harness/src/openharness/keybindings/default_bindings.py +11 -0
  311. src/harness/src/openharness/keybindings/loader.py +22 -0
  312. src/harness/src/openharness/keybindings/parser.py +18 -0
  313. src/harness/src/openharness/keybindings/resolver.py +13 -0
  314. src/harness/src/openharness/mcp/__init__.py +79 -0
  315. src/harness/src/openharness/mcp/client.py +298 -0
  316. src/harness/src/openharness/mcp/config.py +16 -0
  317. src/harness/src/openharness/mcp/types.py +76 -0
  318. src/harness/src/openharness/memory/__init__.py +25 -0
  319. src/harness/src/openharness/memory/agent.py +91 -0
  320. src/harness/src/openharness/memory/manager.py +183 -0
  321. src/harness/src/openharness/memory/memdir.py +52 -0
  322. src/harness/src/openharness/memory/migrate.py +137 -0
  323. src/harness/src/openharness/memory/paths.py +22 -0
  324. src/harness/src/openharness/memory/relevance.py +145 -0
  325. src/harness/src/openharness/memory/scan.py +116 -0
  326. src/harness/src/openharness/memory/schema.py +443 -0
  327. src/harness/src/openharness/memory/search.py +71 -0
  328. src/harness/src/openharness/memory/team.py +90 -0
  329. src/harness/src/openharness/memory/types.py +33 -0
  330. src/harness/src/openharness/memory/usage.py +153 -0
  331. src/harness/src/openharness/output_styles/__init__.py +5 -0
  332. src/harness/src/openharness/output_styles/loader.py +42 -0
  333. src/harness/src/openharness/permissions/__init__.py +26 -0
  334. src/harness/src/openharness/permissions/checker.py +200 -0
  335. src/harness/src/openharness/permissions/modes.py +13 -0
  336. src/harness/src/openharness/personalization/__init__.py +6 -0
  337. src/harness/src/openharness/personalization/extractor.py +138 -0
  338. src/harness/src/openharness/personalization/rules.py +64 -0
  339. src/harness/src/openharness/personalization/session_hook.py +65 -0
  340. src/harness/src/openharness/platforms.py +86 -0
  341. src/harness/src/openharness/plugins/__init__.py +53 -0
  342. src/harness/src/openharness/plugins/bundled/__init__.py +0 -0
  343. src/harness/src/openharness/plugins/installer.py +39 -0
  344. src/harness/src/openharness/plugins/loader.py +730 -0
  345. src/harness/src/openharness/plugins/schemas.py +24 -0
  346. src/harness/src/openharness/plugins/types.py +59 -0
  347. src/harness/src/openharness/prompts/__init__.py +14 -0
  348. src/harness/src/openharness/prompts/claudemd.py +48 -0
  349. src/harness/src/openharness/prompts/context.py +187 -0
  350. src/harness/src/openharness/prompts/environment.py +135 -0
  351. src/harness/src/openharness/prompts/system_prompt.py +109 -0
  352. src/harness/src/openharness/sandbox/Dockerfile +6 -0
  353. src/harness/src/openharness/sandbox/__init__.py +33 -0
  354. src/harness/src/openharness/sandbox/adapter.py +148 -0
  355. src/harness/src/openharness/sandbox/docker_backend.py +232 -0
  356. src/harness/src/openharness/sandbox/docker_image.py +103 -0
  357. src/harness/src/openharness/sandbox/path_validator.py +37 -0
  358. src/harness/src/openharness/sandbox/session.py +63 -0
  359. src/harness/src/openharness/services/__init__.py +30 -0
  360. src/harness/src/openharness/services/autodream/__init__.py +34 -0
  361. src/harness/src/openharness/services/autodream/backup.py +104 -0
  362. src/harness/src/openharness/services/autodream/lock.py +138 -0
  363. src/harness/src/openharness/services/autodream/prompt.py +128 -0
  364. src/harness/src/openharness/services/autodream/service.py +313 -0
  365. src/harness/src/openharness/services/compact/__init__.py +1725 -0
  366. src/harness/src/openharness/services/cron.py +137 -0
  367. src/harness/src/openharness/services/cron_scheduler.py +595 -0
  368. src/harness/src/openharness/services/lsp/__init__.py +216 -0
  369. src/harness/src/openharness/services/memory_extract/__init__.py +261 -0
  370. src/harness/src/openharness/services/oauth/__init__.py +0 -0
  371. src/harness/src/openharness/services/session_backend.py +97 -0
  372. src/harness/src/openharness/services/session_memory/__init__.py +139 -0
  373. src/harness/src/openharness/services/session_storage.py +230 -0
  374. src/harness/src/openharness/services/token_estimation.py +15 -0
  375. src/harness/src/openharness/services/tool_outputs.py +55 -0
  376. src/harness/src/openharness/skills/__init__.py +44 -0
  377. src/harness/src/openharness/skills/_frontmatter.py +97 -0
  378. src/harness/src/openharness/skills/bundled/__init__.py +75 -0
  379. src/harness/src/openharness/skills/bundled/content/commit.md +25 -0
  380. src/harness/src/openharness/skills/bundled/content/debug.md +25 -0
  381. src/harness/src/openharness/skills/bundled/content/diagnose.md +35 -0
  382. src/harness/src/openharness/skills/bundled/content/plan.md +39 -0
  383. src/harness/src/openharness/skills/bundled/content/review.md +33 -0
  384. src/harness/src/openharness/skills/bundled/content/simplify.md +26 -0
  385. src/harness/src/openharness/skills/bundled/content/skill-creator.md +175 -0
  386. src/harness/src/openharness/skills/bundled/content/test.md +38 -0
  387. src/harness/src/openharness/skills/loader.py +229 -0
  388. src/harness/src/openharness/skills/registry.py +29 -0
  389. src/harness/src/openharness/skills/types.py +24 -0
  390. src/harness/src/openharness/state/__init__.py +6 -0
  391. src/harness/src/openharness/state/app_state.py +30 -0
  392. src/harness/src/openharness/state/store.py +40 -0
  393. src/harness/src/openharness/swarm/__init__.py +70 -0
  394. src/harness/src/openharness/swarm/in_process.py +693 -0
  395. src/harness/src/openharness/swarm/lockfile.py +24 -0
  396. src/harness/src/openharness/swarm/mailbox.py +522 -0
  397. src/harness/src/openharness/swarm/permission_sync.py +1168 -0
  398. src/harness/src/openharness/swarm/registry.py +410 -0
  399. src/harness/src/openharness/swarm/spawn_utils.py +229 -0
  400. src/harness/src/openharness/swarm/subprocess_backend.py +171 -0
  401. src/harness/src/openharness/swarm/team_lifecycle.py +910 -0
  402. src/harness/src/openharness/swarm/types.py +398 -0
  403. src/harness/src/openharness/swarm/worktree.py +315 -0
  404. src/harness/src/openharness/tasks/__init__.py +18 -0
  405. src/harness/src/openharness/tasks/local_agent_task.py +28 -0
  406. src/harness/src/openharness/tasks/local_shell_task.py +17 -0
  407. src/harness/src/openharness/tasks/manager.py +475 -0
  408. src/harness/src/openharness/tasks/stop_task.py +11 -0
  409. src/harness/src/openharness/tasks/types.py +32 -0
  410. src/harness/src/openharness/themes/__init__.py +21 -0
  411. src/harness/src/openharness/themes/builtin.py +89 -0
  412. src/harness/src/openharness/themes/loader.py +55 -0
  413. src/harness/src/openharness/themes/schema.py +54 -0
  414. src/harness/src/openharness/tools/__init__.py +107 -0
  415. src/harness/src/openharness/tools/agent_tool.py +141 -0
  416. src/harness/src/openharness/tools/ask_user_question_tool.py +46 -0
  417. src/harness/src/openharness/tools/base.py +80 -0
  418. src/harness/src/openharness/tools/bash_tool.py +259 -0
  419. src/harness/src/openharness/tools/brief_tool.py +33 -0
  420. src/harness/src/openharness/tools/config_tool.py +54 -0
  421. src/harness/src/openharness/tools/cron_create_tool.py +105 -0
  422. src/harness/src/openharness/tools/cron_delete_tool.py +32 -0
  423. src/harness/src/openharness/tools/cron_list_tool.py +69 -0
  424. src/harness/src/openharness/tools/cron_toggle_tool.py +37 -0
  425. src/harness/src/openharness/tools/enter_plan_mode_tool.py +28 -0
  426. src/harness/src/openharness/tools/enter_worktree_tool.py +80 -0
  427. src/harness/src/openharness/tools/exit_plan_mode_tool.py +28 -0
  428. src/harness/src/openharness/tools/exit_worktree_tool.py +42 -0
  429. src/harness/src/openharness/tools/file_edit_tool.py +95 -0
  430. src/harness/src/openharness/tools/file_read_tool.py +72 -0
  431. src/harness/src/openharness/tools/file_write_tool.py +87 -0
  432. src/harness/src/openharness/tools/glob_tool.py +175 -0
  433. src/harness/src/openharness/tools/grep_tool.py +374 -0
  434. src/harness/src/openharness/tools/image_generation_tool.py +447 -0
  435. src/harness/src/openharness/tools/image_to_text_tool.py +237 -0
  436. src/harness/src/openharness/tools/list_mcp_resources_tool.py +36 -0
  437. src/harness/src/openharness/tools/lsp_tool.py +154 -0
  438. src/harness/src/openharness/tools/mcp_auth_tool.py +71 -0
  439. src/harness/src/openharness/tools/mcp_tool.py +72 -0
  440. src/harness/src/openharness/tools/notebook_edit_tool.py +97 -0
  441. src/harness/src/openharness/tools/read_mcp_resource_tool.py +38 -0
  442. src/harness/src/openharness/tools/remote_trigger_tool.py +74 -0
  443. src/harness/src/openharness/tools/send_message_tool.py +58 -0
  444. src/harness/src/openharness/tools/skill_tool.py +43 -0
  445. src/harness/src/openharness/tools/sleep_tool.py +32 -0
  446. src/harness/src/openharness/tools/task_create_tool.py +56 -0
  447. src/harness/src/openharness/tools/task_get_tool.py +33 -0
  448. src/harness/src/openharness/tools/task_list_tool.py +35 -0
  449. src/harness/src/openharness/tools/task_output_tool.py +35 -0
  450. src/harness/src/openharness/tools/task_stop_tool.py +30 -0
  451. src/harness/src/openharness/tools/task_update_tool.py +50 -0
  452. src/harness/src/openharness/tools/team_create_tool.py +31 -0
  453. src/harness/src/openharness/tools/team_delete_tool.py +30 -0
  454. src/harness/src/openharness/tools/todo_write_tool.py +46 -0
  455. src/harness/src/openharness/tools/tool_search_tool.py +38 -0
  456. src/harness/src/openharness/tools/web_fetch_tool.py +117 -0
  457. src/harness/src/openharness/tools/web_search_tool.py +119 -0
  458. src/harness/src/openharness/ui/__init__.py +5 -0
  459. src/harness/src/openharness/ui/app.py +320 -0
  460. src/harness/src/openharness/ui/backend_host.py +941 -0
  461. src/harness/src/openharness/ui/coordinator_drain.py +198 -0
  462. src/harness/src/openharness/ui/input.py +32 -0
  463. src/harness/src/openharness/ui/output.py +265 -0
  464. src/harness/src/openharness/ui/permission_dialog.py +14 -0
  465. src/harness/src/openharness/ui/protocol.py +247 -0
  466. src/harness/src/openharness/ui/react_launcher.py +179 -0
  467. src/harness/src/openharness/ui/runtime.py +799 -0
  468. src/harness/src/openharness/ui/textual_app.py +495 -0
  469. src/harness/src/openharness/utils/__init__.py +0 -0
  470. src/harness/src/openharness/utils/file_lock.py +86 -0
  471. src/harness/src/openharness/utils/fs.py +98 -0
  472. src/harness/src/openharness/utils/helpers.py +77 -0
  473. src/harness/src/openharness/utils/network_guard.py +340 -0
  474. src/harness/src/openharness/utils/shell.py +147 -0
  475. src/harness/src/openharness/vim/__init__.py +5 -0
  476. src/harness/src/openharness/vim/transitions.py +8 -0
  477. src/harness/src/openharness/voice/__init__.py +7 -0
  478. src/harness/src/openharness/voice/keyterms.py +10 -0
  479. src/harness/src/openharness/voice/stream_stt.py +8 -0
  480. src/harness/src/openharness/voice/voice_mode.py +44 -0
  481. src/llm/__init__.py +29 -0
  482. src/llm/image_generation.py +402 -0
  483. src/llm/provider_registry.py +419 -0
  484. src/llm/runtime_env.py +64 -0
  485. src/llm/structured_output.py +131 -0
  486. src/llm/vision.py +198 -0
  487. src/memory/__init__.py +2 -0
  488. src/memory/api/__init__.py +1 -0
  489. src/memory/api/base_memory_system_api.py +98 -0
  490. src/memory/api/base_symbolic_memory_system_api.py +190 -0
  491. src/memory/api/base_vector_memory_system_api.py +116 -0
  492. src/memory/api/faiss_memory_system_api.py +228 -0
  493. src/memory/api/slot_process_api.py +827 -0
  494. src/memory/api/symbolic_memory_system_api.py +572 -0
  495. src/memory/memory_system/__init__.py +48 -0
  496. src/memory/memory_system/component_taxonomy.py +430 -0
  497. src/memory/memory_system/llm.py +437 -0
  498. src/memory/memory_system/models.py +168 -0
  499. src/memory/memory_system/user_prompt.py +418 -0
  500. src/memory/memory_system/utils.py +282 -0
  501. src/memory/memory_system/vectorstore.py +311 -0
  502. src/memory/memory_system/working_slot.py +99 -0
  503. src/pipeline/__init__.py +129 -0
  504. src/pipeline/contracts.py +531 -0
  505. src/pipeline/discipline_taxonomy.py +916 -0
  506. src/pipeline/evidence_coverage_ledger.py +1150 -0
  507. src/pipeline/evidence_refinement.py +620 -0
  508. src/pipeline/experiment_to_symbolic.py +137 -0
  509. src/pipeline/multimodal_evidence/__init__.py +28 -0
  510. src/pipeline/multimodal_evidence/capabilities.py +107 -0
  511. src/pipeline/multimodal_evidence/claims.py +135 -0
  512. src/pipeline/multimodal_evidence/contract.py +291 -0
  513. src/pipeline/multimodal_evidence/data_sh_compiler.py +505 -0
  514. src/pipeline/multimodal_evidence/handoff_binding.py +135 -0
  515. src/pipeline/multimodal_evidence/manifest.py +254 -0
  516. src/pipeline/multimodal_evidence/native_analysis.py +479 -0
  517. src/pipeline/multimodal_evidence/perception.py +277 -0
  518. src/pipeline/multimodal_evidence/query_binding.py +65 -0
  519. src/pipeline/multimodal_evidence/rendering.py +120 -0
  520. src/pipeline/multimodal_evidence/safety.py +42 -0
  521. src/pipeline/multimodal_evidence/sampling.py +69 -0
  522. src/pipeline/multimodal_evidence/service.py +148 -0
  523. src/pipeline/multimodal_evidence/survey_integration.py +460 -0
  524. src/pipeline/paper_identity.py +52 -0
  525. src/pipeline/research_design_inventory.py +338 -0
  526. src/pipeline/research_domain_resolution.py +669 -0
  527. src/pipeline/research_identity.py +548 -0
  528. src/pipeline/research_question_contract.py +646 -0
  529. src/pipeline/retrieval_lanes.py +889 -0
  530. src/pipeline/run_loop.py +862 -0
  531. src/pipeline/science_events.py +30 -0
  532. src/pipeline/science_manifests.py +602 -0
  533. src/pipeline/science_run.py +918 -0
  534. src/pipeline/science_stages.py +709 -0
  535. src/pipeline/science_workflow.py +807 -0
  536. src/pipeline/sh_cluster_projection.py +306 -0
  537. src/pipeline/sh_graph_provenance.py +648 -0
  538. src/pipeline/subhypothesis_decomposition.py +392 -0
  539. src/pipeline/subhypothesis_evidence.py +856 -0
  540. src/pipeline/survey_evidence_plan.py +858 -0
  541. src/pipeline/survey_gap_adjudication.py +647 -0
  542. src/pipeline/survey_gap_candidates.py +840 -0
  543. src/pipeline/survey_gap_ledger.py +683 -0
  544. src/pipeline/survey_gap_triage.py +401 -0
  545. src/pipeline/survey_handoff_persistence.py +481 -0
  546. src/pipeline/survey_handoff_projection.py +457 -0
  547. src/pipeline/survey_idea_handoff.py +1112 -0
  548. src/pipeline/survey_idea_loader.py +289 -0
  549. src/research_run_ids.py +11 -0
@@ -0,0 +1,496 @@
1
+ """Render a validated English ResearchPlanDocument into a restricted TeX project."""
2
+
3
+ from __future__ import annotations
4
+
5
+ from dataclasses import dataclass
6
+ import os
7
+ from pathlib import Path
8
+ import re
9
+ import tempfile
10
+ from collections.abc import Mapping
11
+ from typing import Any
12
+
13
+ from .bibtex_renderer import BibtexRenderResult, ensure_citation_coverage, render_bibtex
14
+ from .contracts import AUTHORING_LANGUAGE, validate_research_plan_document
15
+ from .latex_safety import (
16
+ LatexSafetyError,
17
+ escape_latex_text,
18
+ normalize_visible_text,
19
+ safe_math_expression,
20
+ split_equation_content,
21
+ )
22
+ from .template_adapter import MaterializedTemplate, TemplateAdapter, TemplateAdapterError
23
+ from .template_profile import TemplateProfile
24
+ from .theory_presentation import theory_block_presentation, theory_unit_registry, visible_theory_text
25
+
26
+
27
+ TEX_RENDER_SCHEMA_VERSION = "research_plan_author_tex_render_v1"
28
+
29
+
30
+ class TexRenderError(RuntimeError):
31
+ """Raised when a document cannot be rendered without violating Author constraints."""
32
+
33
+
34
+ @dataclass(frozen=True)
35
+ class TexRenderResult:
36
+ profile_id: str
37
+ project_dir: Path
38
+ main_tex: Path
39
+ bibtex: Path
40
+ bibliography: BibtexRenderResult
41
+
42
+ def as_dict(self) -> dict[str, Any]:
43
+ return {
44
+ "schema_version": TEX_RENDER_SCHEMA_VERSION,
45
+ "profile_id": self.profile_id,
46
+ "project_dir": str(self.project_dir),
47
+ "main_tex": str(self.main_tex),
48
+ "bibtex": str(self.bibtex),
49
+ "bibliography": self.bibliography.as_dict(),
50
+ }
51
+
52
+
53
+ def _mapping(value: object) -> dict[str, Any]:
54
+ return dict(value) if isinstance(value, Mapping) else {}
55
+
56
+
57
+ def _text(value: object) -> str:
58
+ return str(value or "").strip()
59
+
60
+
61
+ _PRIVATE_PROVENANCE_ANCHOR = re.compile(
62
+ r"\[?(?:survey:survey_markdown(?:#section-[A-Za-z0-9._:+-]+)?|anchor:[A-Za-z0-9_:+-]+(?:\.[A-Za-z0-9_:+-]+)*)\]?",
63
+ re.IGNORECASE,
64
+ )
65
+
66
+
67
+ def _strip_private_survey_anchors(value: object) -> str:
68
+ """Keep Survey provenance private even if a composer leaks an anchor into prose."""
69
+
70
+ text = str(value or "")
71
+ cleaned = _PRIVATE_PROVENANCE_ANCHOR.sub("", text)
72
+ return re.sub(r"[ \t]{2,}", " ", cleaned).replace(" \n", "\n").strip()
73
+
74
+
75
+ def _label_component(value: object) -> str:
76
+ component = re.sub(r"[^A-Za-z0-9:-]+", "-", _text(value)).strip("-:")
77
+ return component or "unnamed"
78
+
79
+
80
+ def _equation_label(section_id: object, block_id: object) -> str:
81
+ return f"eq:{_label_component(section_id)}-{_label_component(block_id)}"
82
+
83
+
84
+ def _table_label(section_id: object, block_id: object) -> str:
85
+ return f"tab:{_label_component(section_id)}-{_label_component(block_id)}"
86
+
87
+
88
+ def _claim_map(document: Mapping[str, Any]) -> dict[str, dict[str, Any]]:
89
+ claims: dict[str, dict[str, Any]] = {}
90
+ for raw in document.get("claim_provenance") or []:
91
+ claim = _mapping(raw)
92
+ claim_id = _text(claim.get("claim_id"))
93
+ if claim_id:
94
+ claims[claim_id] = claim
95
+ return claims
96
+
97
+
98
+ def _block_citations(block: Mapping[str, Any], claims: Mapping[str, Mapping[str, Any]]) -> list[str]:
99
+ keys: list[str] = []
100
+ for claim_id in block.get("claim_ids") or []:
101
+ claim = claims.get(_text(claim_id), {})
102
+ for key in claim.get("citation_keys") or []:
103
+ value = _text(key)
104
+ if value and value not in keys:
105
+ keys.append(value)
106
+ return keys
107
+
108
+
109
+ def _citation_suffix(keys: list[str]) -> str:
110
+ return "" if not keys else "~\\cite{" + ",".join(keys) + "}"
111
+
112
+
113
+ def _paragraphs(text: str, *, label: str) -> str:
114
+ visible_text = _strip_private_survey_anchors(text)
115
+ fragments = [fragment.strip() for fragment in re.split(r"\n\s*\n", visible_text) if fragment.strip()]
116
+ if not fragments:
117
+ return ""
118
+ return "\n\n".join(escape_latex_text(fragment, label=label).replace("\n", " ") for fragment in fragments)
119
+
120
+
121
+ def _list_items(text: str, *, label: str) -> list[str]:
122
+ lines = [line.strip() for line in _strip_private_survey_anchors(text).splitlines() if line.strip()]
123
+ items = [re.sub(r"^(?:[-*+]|\d+[.)])\s*", "", line).strip() for line in lines]
124
+ return [escape_latex_text(item, label=label) for item in items if item]
125
+
126
+
127
+ def _render_table(text: str, *, label: str, caption: str, table_label: str) -> str:
128
+ rows: list[list[str]] = []
129
+ for line in (line.strip() for line in _strip_private_survey_anchors(text).splitlines() if line.strip()):
130
+ stripped = line.strip("|")
131
+ cells = [cell.strip() for cell in (stripped.split("|") if "|" in stripped else stripped.split("\t"))]
132
+ if len(cells) >= 2 and any(cells):
133
+ if all(re.fullmatch(r":?-{3,}:?", cell.replace(" ", "")) for cell in cells):
134
+ continue
135
+ rows.append(cells)
136
+ if not rows or len(rows) > 40:
137
+ return "\\begin{quote}\\small " + _paragraphs(text, label=label) + " \\end{quote}"
138
+ width = max(len(row) for row in rows)
139
+ if width > 6:
140
+ return "\\begin{quote}\\small " + _paragraphs(text, label=label) + " \\end{quote}"
141
+ normalized = [row + [""] * (width - len(row)) for row in rows]
142
+ longest_cell = max((len(cell) for row in normalized for cell in row), default=0)
143
+ use_wide_layout = width >= 4 or longest_cell > 52
144
+ environment = "table*" if use_wide_layout else "table"
145
+ available_width = r"\textwidth" if use_wide_layout else r"\linewidth"
146
+ if use_wide_layout:
147
+ column_width = f"{0.92 / width:.4f}".rstrip("0").rstrip(".")
148
+ else:
149
+ column_width = "0.45" if width == 2 else "0.29"
150
+ rendered_rows = [
151
+ " & ".join(escape_latex_text(cell, label=label) for cell in row) + r" \\\hline"
152
+ for row in normalized
153
+ ]
154
+ return "\n".join(
155
+ [
156
+ "\\begin{" + environment + "}[htbp]",
157
+ "\\centering",
158
+ "\\footnotesize" if use_wide_layout else "\\small",
159
+ "\\setlength{\\tabcolsep}{3pt}" if use_wide_layout else "",
160
+ "\\renewcommand{\\arraystretch}{1.12}" if use_wide_layout else "",
161
+ "\\caption{" + escape_latex_text(caption, label=f"{label} caption") + "}",
162
+ "\\label{" + table_label + "}",
163
+ "\\begin{tabular}{|" + "|".join(
164
+ f"p{{{column_width}{available_width}}}" for _ in range(width)
165
+ ) + "|}",
166
+ "\\hline",
167
+ *rendered_rows,
168
+ "\\end{tabular}",
169
+ "\\end{" + environment + "}",
170
+ ]
171
+ )
172
+
173
+
174
+ def _equation_cross_references(
175
+ block: Mapping[str, Any],
176
+ *,
177
+ equation_labels: Mapping[str, str],
178
+ ) -> str:
179
+ labels = [
180
+ equation_labels[reference_id]
181
+ for raw_reference_id in block.get("reference_block_ids") or []
182
+ if (reference_id := _text(raw_reference_id)) in equation_labels
183
+ ]
184
+ if not labels:
185
+ return ""
186
+ rendered = ", ".join("Eq.~\\eqref{" + label + "}" for label in labels)
187
+ return " See " + rendered + "."
188
+
189
+
190
+ def _render_math_expression(expression: str, *, equation_label: str | None) -> str:
191
+ lines = [line.strip() for line in expression.split(r"\\")]
192
+ if len(lines) == 1:
193
+ if equation_label:
194
+ return "\\begin{equation}\n" + expression + "\n\\label{" + equation_label + "}\n\\end{equation}"
195
+ return "\\begin{equation*}\n" + expression + "\n\\end{equation*}"
196
+ if not all(lines):
197
+ raise LatexSafetyError("equation has an empty align row")
198
+ rendered_lines = [
199
+ line
200
+ + (
201
+ r" \nonumber \\"
202
+ if index < len(lines) - 1
203
+ else (r" \label{" + equation_label + "}" if equation_label else "")
204
+ )
205
+ for index, line in enumerate(lines)
206
+ ]
207
+ environment = "align" if equation_label else "align*"
208
+ return "\\begin{" + environment + "}\n" + "\n".join(rendered_lines) + "\n\\end{" + environment + "}"
209
+
210
+
211
+ def _render_equation(text: str, *, label: str, equation_label: str) -> str:
212
+ fragments = split_equation_content(_strip_private_survey_anchors(text), label=label)
213
+ rendered: list[str] = []
214
+ labeled_expression = False
215
+ for kind, fragment in fragments:
216
+ if kind == "prose":
217
+ prose = _paragraphs(fragment, label=label)
218
+ if prose:
219
+ rendered.append(prose)
220
+ continue
221
+ rendered.append(
222
+ _render_math_expression(
223
+ safe_math_expression(fragment, label=label),
224
+ equation_label=equation_label if not labeled_expression else None,
225
+ )
226
+ )
227
+ labeled_expression = True
228
+ if not labeled_expression:
229
+ raise LatexSafetyError(f"{label} contains no valid mathematical expression")
230
+ return "\n\n".join(rendered)
231
+
232
+
233
+ def _render_block(
234
+ block: Mapping[str, Any],
235
+ *,
236
+ claims: Mapping[str, Mapping[str, Any]],
237
+ document: Mapping[str, Any],
238
+ theory_registry: Mapping[str, Mapping[str, Any]],
239
+ section_id: str,
240
+ equation_labels: Mapping[str, str],
241
+ ) -> str:
242
+ kind = _text(block.get("kind"))
243
+ text = normalize_visible_text(
244
+ _strip_private_survey_anchors(visible_theory_text(document, block.get("text"))),
245
+ label=f"block {block.get('block_id') or kind}",
246
+ )
247
+ citations = _citation_suffix(_block_citations(block, claims))
248
+ label = f"block {block.get('block_id') or kind}"
249
+ heading_text = _strip_private_survey_anchors(visible_theory_text(document, block.get("heading")))
250
+ heading = "" if not heading_text else "\\subsection{" + escape_latex_text(heading_text, label=f"{label} heading") + "}\n"
251
+ cross_references = _equation_cross_references(block, equation_labels=equation_labels)
252
+ theory_prefix, theory_status = theory_block_presentation(
253
+ block,
254
+ claims=claims,
255
+ registry=theory_registry,
256
+ )
257
+ if kind == "equation":
258
+ equation_label = _equation_label(section_id, block.get("block_id"))
259
+ return heading + _render_equation(text, label=label, equation_label=equation_label) + citations
260
+ if kind == "list":
261
+ items = _list_items(text, label=label)
262
+ if not items:
263
+ return ""
264
+ return heading + "\\begin{itemize}\n" + "\n".join(f"\\item {item}" for item in items) + "\n\\end{itemize}" + citations
265
+ if kind == "table":
266
+ caption = heading_text or theory_prefix.rstrip(".") or (
267
+ f"Decision matrix ({theory_status})" if theory_status else "Decision matrix"
268
+ )
269
+ return _render_table(
270
+ text,
271
+ label=label,
272
+ caption=caption,
273
+ table_label=_table_label(section_id, block.get("block_id")),
274
+ ) + cross_references + citations
275
+ paragraph = _paragraphs(text, label=label)
276
+ if theory_prefix:
277
+ return heading + "\\paragraph{" + escape_latex_text(theory_prefix, label=f"{label} theory presentation") + "} " + paragraph + cross_references + citations
278
+ if kind == "definition":
279
+ return heading + "\\paragraph{Definition.} " + paragraph + cross_references + citations
280
+ if kind == "proposition":
281
+ return heading + "\\paragraph{Proposition (proposed).} " + paragraph + cross_references + citations
282
+ if kind == "protocol":
283
+ return heading + "\\paragraph{Planned protocol.} " + paragraph + cross_references + citations
284
+ if kind == "outcome_branch":
285
+ return heading + "\\paragraph{Conditional outcome branch.} " + paragraph + cross_references + citations
286
+ if kind == "review_checklist":
287
+ return heading + "\\paragraph{Human-review checklist.} " + paragraph + cross_references + citations
288
+ return heading + paragraph + cross_references + citations
289
+
290
+
291
+ def _render_section(
292
+ section: Mapping[str, Any],
293
+ *,
294
+ claims: Mapping[str, Mapping[str, Any]],
295
+ document: Mapping[str, Any],
296
+ theory_registry: Mapping[str, Mapping[str, Any]],
297
+ document_equation_labels: Mapping[str, str],
298
+ appendix: bool,
299
+ ) -> str:
300
+ section_id = _text(section.get("section_id"))
301
+ title = escape_latex_text(
302
+ _strip_private_survey_anchors(visible_theory_text(document, section.get("title"))),
303
+ label=f"section {section_id} title",
304
+ )
305
+ heading = "\\section" if appendix else "\\section"
306
+ equation_labels = {
307
+ _text(block.get("block_id")): _equation_label(section_id, block.get("block_id"))
308
+ for block in section.get("blocks") or []
309
+ if isinstance(block, Mapping) and _text(block.get("kind")) == "equation"
310
+ }
311
+ equation_labels = {**document_equation_labels, **equation_labels}
312
+ blocks = [
313
+ _render_block(
314
+ _mapping(block),
315
+ claims=claims,
316
+ document=document,
317
+ theory_registry=theory_registry,
318
+ section_id=section_id,
319
+ equation_labels=equation_labels,
320
+ )
321
+ for block in section.get("blocks") or []
322
+ if isinstance(block, Mapping)
323
+ ]
324
+ blocks = [block for block in blocks if block]
325
+ if _text(section.get("applicability")) == "not_applicable":
326
+ blocks.insert(0, "\\emph{Not applicable to this proposal.}")
327
+ if not blocks:
328
+ blocks.append("\\emph{No additional prose is available for this source-bounded proposal section.}")
329
+ return heading + "{" + title + "}\n" + "\n\n".join(blocks)
330
+
331
+
332
+ def _render_body(document: Mapping[str, Any], *, profile: TemplateProfile) -> str:
333
+ claims = _claim_map(document)
334
+ theory_registry = theory_unit_registry(document)
335
+ document_equation_labels = {
336
+ f"{_text(section.get('section_id'))}:{_text(block.get('block_id'))}": _equation_label(
337
+ section.get("section_id"),
338
+ block.get("block_id"),
339
+ )
340
+ for section in [*(document.get("sections") or []), *(document.get("appendices") or [])]
341
+ if isinstance(section, Mapping)
342
+ for block in section.get("blocks") or []
343
+ if isinstance(block, Mapping)
344
+ and _text(section.get("section_id"))
345
+ and _text(block.get("block_id"))
346
+ and _text(block.get("kind")) == "equation"
347
+ }
348
+ sections = [
349
+ _render_section(
350
+ _mapping(section),
351
+ claims=claims,
352
+ document=document,
353
+ theory_registry=theory_registry,
354
+ document_equation_labels=document_equation_labels,
355
+ appendix=False,
356
+ )
357
+ for section in document.get("sections") or []
358
+ if isinstance(section, Mapping) and _text(section.get("section_id")) != "references"
359
+ ]
360
+ appendices = [
361
+ _render_section(
362
+ _mapping(section),
363
+ claims=claims,
364
+ document=document,
365
+ theory_registry=theory_registry,
366
+ document_equation_labels=document_equation_labels,
367
+ appendix=True,
368
+ )
369
+ for section in document.get("appendices") or []
370
+ if isinstance(section, Mapping)
371
+ ]
372
+ if appendices:
373
+ appendix_command = "\\appendices" if profile.profile_id == "ieee_conference_v1" else "\\appendix"
374
+ sections.append(appendix_command + "\n" + "\n\n".join(appendices))
375
+ return "\n\n".join(sections)
376
+
377
+
378
+ def _render_abstract(document: Mapping[str, Any], *, profile: TemplateProfile) -> str:
379
+ abstract = _mapping(document.get("abstract"))
380
+ text = _paragraphs(
381
+ _strip_private_survey_anchors(visible_theory_text(document, abstract.get("text"))),
382
+ label="abstract",
383
+ )
384
+ if not text:
385
+ raise TexRenderError("ResearchPlanDocument has no rendered abstract")
386
+ keywords = [
387
+ escape_latex_text(_strip_private_survey_anchors(visible_theory_text(document, keyword)), label="keyword")
388
+ for keyword in document.get("keywords") or []
389
+ ]
390
+ lines = ["\\begin{abstract}", text, "\\end{abstract}"]
391
+ if keywords and profile.profile_id == "ieee_conference_v1":
392
+ lines.extend(["\\begin{IEEEkeywords}", ", ".join(keywords), "\\end{IEEEkeywords}"])
393
+ elif keywords:
394
+ lines.append("\\paragraph{Keywords.} " + ", ".join(keywords))
395
+ return "\n".join(lines)
396
+
397
+
398
+ def _render_bibliography(result: BibtexRenderResult, *, profile: TemplateProfile) -> str:
399
+ if not result.emitted_keys:
400
+ return (
401
+ "\\section*{References}\n"
402
+ "\\emph{No provenance-backed bibliographic records are available for rendering; see the bibliography completion ledger.}"
403
+ )
404
+ database = Path(profile.generated_bib).with_suffix("").as_posix()
405
+ return "\n".join(
406
+ [
407
+ "\\section*{References}",
408
+ f"\\bibliographystyle{{{profile.bibliography_style}}}",
409
+ f"\\bibliography{{{database}}}",
410
+ ]
411
+ )
412
+
413
+
414
+ def _write_text_atomically(path: Path, content: str) -> None:
415
+ temporary: Path | None = None
416
+ try:
417
+ path.parent.mkdir(parents=True, exist_ok=True)
418
+ descriptor, raw_path = tempfile.mkstemp(
419
+ prefix=f".{path.name}.", suffix=".tmp", dir=path.parent, text=True
420
+ )
421
+ temporary = Path(raw_path)
422
+ with os.fdopen(descriptor, "w", encoding="utf-8", newline="\n") as handle:
423
+ handle.write(content)
424
+ handle.flush()
425
+ os.fsync(handle.fileno())
426
+ os.replace(temporary, path)
427
+ except OSError as error:
428
+ raise TexRenderError(f"cannot write generated file '{path.name}': {error}") from error
429
+ finally:
430
+ if temporary is not None:
431
+ temporary.unlink(missing_ok=True)
432
+
433
+
434
+ def _validate_document(document: Mapping[str, Any]) -> None:
435
+ errors = validate_research_plan_document(document)
436
+ if errors:
437
+ raise TexRenderError("ResearchPlanDocument schema validation failed: " + "; ".join(errors))
438
+ if document.get("language") != AUTHORING_LANGUAGE:
439
+ raise TexRenderError("ResearchPlanDocument language must be English")
440
+ if document.get("document_status") != "PROPOSAL_NO_OBSERVED_RESULTS":
441
+ raise TexRenderError("only PROPOSAL_NO_OBSERVED_RESULTS documents may be rendered")
442
+ metadata = _mapping(document.get("document_metadata"))
443
+ normalize_visible_text(metadata.get("title"), label="document title")
444
+
445
+
446
+ def render_tex_project(
447
+ document: Mapping[str, Any],
448
+ *,
449
+ template_dir: str | Path,
450
+ project_dir: str | Path,
451
+ profile: TemplateProfile,
452
+ author_name: str = "Anonymous Research Plan Author",
453
+ ) -> TexRenderResult:
454
+ """Copy a declared template and render one source-bounded TeX project."""
455
+
456
+ try:
457
+ _validate_document(document)
458
+ bibliography = render_bibtex(document)
459
+ ensure_citation_coverage(document, bibliography)
460
+ title = escape_latex_text(
461
+ visible_theory_text(document, _mapping(document.get("document_metadata")).get("title")),
462
+ label="document title",
463
+ )
464
+ author = escape_latex_text(author_name, label="author name")
465
+ abstract = _render_abstract(document, profile=profile)
466
+ body = _render_body(document, profile=profile)
467
+ bibliography_fragment = _render_bibliography(bibliography, profile=profile)
468
+ except (LatexSafetyError, TemplateAdapterError) as error:
469
+ raise TexRenderError(str(error)) from error
470
+ adapter = TemplateAdapter()
471
+ try:
472
+ materialized: MaterializedTemplate = adapter.materialize(template_dir, project_dir, profile)
473
+ _write_text_atomically(materialized.generated_bib, bibliography.content)
474
+ main_tex = adapter.apply(
475
+ materialized,
476
+ profile,
477
+ {
478
+ "title": "\\title{" + title + "}",
479
+ "author": "\\author{" + author + "}",
480
+ "abstract": abstract,
481
+ "body": body,
482
+ "bibliography": bibliography_fragment,
483
+ },
484
+ )
485
+ except (TemplateAdapterError, LatexSafetyError) as error:
486
+ raise TexRenderError(str(error)) from error
487
+ return TexRenderResult(
488
+ profile_id=profile.profile_id,
489
+ project_dir=materialized.project_dir,
490
+ main_tex=main_tex,
491
+ bibtex=materialized.generated_bib,
492
+ bibliography=bibliography,
493
+ )
494
+
495
+
496
+ __all__ = ["TEX_RENDER_SCHEMA_VERSION", "TexRenderError", "TexRenderResult", "render_tex_project"]
@@ -0,0 +1,134 @@
1
+ """Deterministic public presentation for compiled mathematics-theory units."""
2
+
3
+ from __future__ import annotations
4
+
5
+ from collections.abc import Mapping
6
+ from typing import Any
7
+
8
+ from .theory_spine import replace_theory_spine_internal_ids, theory_spine_internal_ids_in_text
9
+
10
+
11
+ _STATUS_LABELS = {
12
+ "proposed": "Candidate",
13
+ "candidate": "Candidate",
14
+ "unverified": "Unverified",
15
+ "expected_not_observed": "Expected---Not Observed",
16
+ "no_information": "No-information",
17
+ "needs_human_input": "Review-required",
18
+ "review_required": "Review-required",
19
+ }
20
+
21
+
22
+ def _mapping(value: object) -> dict[str, Any]:
23
+ return dict(value) if isinstance(value, Mapping) else {}
24
+
25
+
26
+ def _text(value: object) -> str:
27
+ return str(value or "").strip()
28
+
29
+
30
+ def theory_spine_for_document(document: Mapping[str, Any]) -> dict[str, Any]:
31
+ """Resolve the private spine registry retained with a composed document."""
32
+
33
+ direct = _mapping(document.get("theory_spine"))
34
+ if direct:
35
+ return direct
36
+ return _mapping(_mapping(document.get("authoring_blueprint")).get("theory_spine"))
37
+
38
+
39
+ def visible_theory_text(document: Mapping[str, Any], value: object) -> str:
40
+ """Use public display labels if a private theory ID reaches a renderer."""
41
+
42
+ visible = replace_theory_spine_internal_ids(value, theory_spine_for_document(document))
43
+ for private_identifier in theory_spine_internal_ids_in_text(visible):
44
+ visible = visible.replace(private_identifier, "an internal theory reference")
45
+ return visible
46
+
47
+
48
+ def theory_unit_registry(document: Mapping[str, Any]) -> dict[str, dict[str, Any]]:
49
+ """Index compiled units by private ID for deterministic public labels."""
50
+
51
+ spine = theory_spine_for_document(document)
52
+ registry: dict[str, dict[str, Any]] = {}
53
+ for collection, identifier_field, unit_kind in (
54
+ ("lemma_units", "lemma_id", "lemma"),
55
+ ("proof_obligations", "proof_obligation_id", "proof_obligation"),
56
+ ("falsifiers", "falsifier_id", "falsifier"),
57
+ ("decision_branches", "branch_id", "decision_branch"),
58
+ ):
59
+ for raw_record in spine.get(collection) or []:
60
+ record = _mapping(raw_record)
61
+ identifier = _text(record.get(identifier_field))
62
+ if identifier:
63
+ registry[identifier] = {"unit_kind": unit_kind, **record}
64
+ return registry
65
+
66
+
67
+ def _status_label(
68
+ block: Mapping[str, Any],
69
+ claims: Mapping[str, Mapping[str, Any]],
70
+ units: list[Mapping[str, Any]],
71
+ ) -> str:
72
+ if any(_text(unit.get("branch_kind")) in {"no_information", "compiler_no_information"} for unit in units):
73
+ return "No-information"
74
+ statuses = {_text(unit.get("status")) for unit in units if _text(unit.get("status"))}
75
+ for status in ("no_information", "expected_not_observed", "candidate", "unverified"):
76
+ if status in statuses:
77
+ return _STATUS_LABELS[status]
78
+ qualifications = {
79
+ _text(_mapping(claims.get(_text(claim_id))).get("qualification"))
80
+ for claim_id in block.get("claim_ids") or []
81
+ if _text(claim_id)
82
+ }
83
+ for qualification in (
84
+ "expected_not_observed",
85
+ "needs_human_input",
86
+ "review_required",
87
+ "proposed",
88
+ "unverified",
89
+ ):
90
+ if qualification in qualifications:
91
+ return _STATUS_LABELS[qualification]
92
+ return ""
93
+
94
+
95
+ def theory_block_presentation(
96
+ block: Mapping[str, Any],
97
+ *,
98
+ claims: Mapping[str, Mapping[str, Any]],
99
+ registry: Mapping[str, Mapping[str, Any]],
100
+ ) -> tuple[str, str]:
101
+ """Return the public prefix and status label for one structured block."""
102
+
103
+ units = [
104
+ _mapping(registry.get(_text(unit_id)))
105
+ for unit_id in block.get("theory_unit_ids") or []
106
+ if _mapping(registry.get(_text(unit_id)))
107
+ ]
108
+ status = _status_label(block, claims, units)
109
+ unit_kinds = {_text(unit.get("unit_kind")) for unit in units}
110
+ labels = [_text(unit.get("display_label")) for unit in units if _text(unit.get("display_label"))]
111
+ label_text = ", ".join(dict.fromkeys(labels))
112
+ kind = _text(block.get("kind"))
113
+ if kind == "lemma":
114
+ noun = "Lemma" if len(labels) == 1 else "Lemma Registry"
115
+ return f"{noun}{(' ' + label_text) if label_text else ''}{(' (' + status + ')') if status else ''}.", status
116
+ if "proof_obligation" in unit_kinds:
117
+ noun = "Proof Obligation" if len(labels) == 1 else "Proof Obligation Registry"
118
+ return f"{noun}{(' ' + label_text) if label_text else ''}{(' (' + (status or 'Unverified') + ')')}.", status or "Unverified"
119
+ if kind == "outcome_branch" or "decision_branch" in unit_kinds:
120
+ branch_status = status or "Expected---Not Observed"
121
+ if branch_status == "No-information":
122
+ return "Decision Status: No-information.", branch_status
123
+ return f"Pre-registered Branch ({branch_status}).", branch_status
124
+ if kind == "proposition":
125
+ return f"Proposition ({status or 'Candidate'}).", status or "Candidate"
126
+ return "", status
127
+
128
+
129
+ __all__ = [
130
+ "theory_block_presentation",
131
+ "theory_spine_for_document",
132
+ "theory_unit_registry",
133
+ "visible_theory_text",
134
+ ]