attune-ai 2.0.0__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 (457) hide show
  1. attune/__init__.py +358 -0
  2. attune/adaptive/__init__.py +13 -0
  3. attune/adaptive/task_complexity.py +127 -0
  4. attune/agent_monitoring.py +414 -0
  5. attune/cache/__init__.py +117 -0
  6. attune/cache/base.py +166 -0
  7. attune/cache/dependency_manager.py +256 -0
  8. attune/cache/hash_only.py +251 -0
  9. attune/cache/hybrid.py +457 -0
  10. attune/cache/storage.py +285 -0
  11. attune/cache_monitor.py +356 -0
  12. attune/cache_stats.py +298 -0
  13. attune/cli/__init__.py +152 -0
  14. attune/cli/__main__.py +12 -0
  15. attune/cli/commands/__init__.py +1 -0
  16. attune/cli/commands/batch.py +264 -0
  17. attune/cli/commands/cache.py +248 -0
  18. attune/cli/commands/help.py +331 -0
  19. attune/cli/commands/info.py +140 -0
  20. attune/cli/commands/inspect.py +436 -0
  21. attune/cli/commands/inspection.py +57 -0
  22. attune/cli/commands/memory.py +48 -0
  23. attune/cli/commands/metrics.py +92 -0
  24. attune/cli/commands/orchestrate.py +184 -0
  25. attune/cli/commands/patterns.py +207 -0
  26. attune/cli/commands/profiling.py +202 -0
  27. attune/cli/commands/provider.py +98 -0
  28. attune/cli/commands/routing.py +285 -0
  29. attune/cli/commands/setup.py +96 -0
  30. attune/cli/commands/status.py +235 -0
  31. attune/cli/commands/sync.py +166 -0
  32. attune/cli/commands/tier.py +121 -0
  33. attune/cli/commands/utilities.py +114 -0
  34. attune/cli/commands/workflow.py +579 -0
  35. attune/cli/core.py +32 -0
  36. attune/cli/parsers/__init__.py +68 -0
  37. attune/cli/parsers/batch.py +118 -0
  38. attune/cli/parsers/cache.py +65 -0
  39. attune/cli/parsers/help.py +41 -0
  40. attune/cli/parsers/info.py +26 -0
  41. attune/cli/parsers/inspect.py +66 -0
  42. attune/cli/parsers/metrics.py +42 -0
  43. attune/cli/parsers/orchestrate.py +61 -0
  44. attune/cli/parsers/patterns.py +54 -0
  45. attune/cli/parsers/provider.py +40 -0
  46. attune/cli/parsers/routing.py +110 -0
  47. attune/cli/parsers/setup.py +42 -0
  48. attune/cli/parsers/status.py +47 -0
  49. attune/cli/parsers/sync.py +31 -0
  50. attune/cli/parsers/tier.py +33 -0
  51. attune/cli/parsers/workflow.py +77 -0
  52. attune/cli/utils/__init__.py +1 -0
  53. attune/cli/utils/data.py +242 -0
  54. attune/cli/utils/helpers.py +68 -0
  55. attune/cli_legacy.py +3957 -0
  56. attune/cli_minimal.py +1159 -0
  57. attune/cli_router.py +437 -0
  58. attune/cli_unified.py +814 -0
  59. attune/config/__init__.py +66 -0
  60. attune/config/xml_config.py +286 -0
  61. attune/config.py +545 -0
  62. attune/coordination.py +870 -0
  63. attune/core.py +1511 -0
  64. attune/core_modules/__init__.py +15 -0
  65. attune/cost_tracker.py +626 -0
  66. attune/dashboard/__init__.py +41 -0
  67. attune/dashboard/app.py +512 -0
  68. attune/dashboard/simple_server.py +435 -0
  69. attune/dashboard/standalone_server.py +547 -0
  70. attune/discovery.py +306 -0
  71. attune/emergence.py +306 -0
  72. attune/exceptions.py +123 -0
  73. attune/feedback_loops.py +373 -0
  74. attune/hot_reload/README.md +473 -0
  75. attune/hot_reload/__init__.py +62 -0
  76. attune/hot_reload/config.py +83 -0
  77. attune/hot_reload/integration.py +229 -0
  78. attune/hot_reload/reloader.py +298 -0
  79. attune/hot_reload/watcher.py +183 -0
  80. attune/hot_reload/websocket.py +177 -0
  81. attune/levels.py +577 -0
  82. attune/leverage_points.py +441 -0
  83. attune/logging_config.py +261 -0
  84. attune/mcp/__init__.py +10 -0
  85. attune/mcp/server.py +506 -0
  86. attune/memory/__init__.py +237 -0
  87. attune/memory/claude_memory.py +469 -0
  88. attune/memory/config.py +224 -0
  89. attune/memory/control_panel.py +1290 -0
  90. attune/memory/control_panel_support.py +145 -0
  91. attune/memory/cross_session.py +845 -0
  92. attune/memory/edges.py +179 -0
  93. attune/memory/encryption.py +159 -0
  94. attune/memory/file_session.py +770 -0
  95. attune/memory/graph.py +570 -0
  96. attune/memory/long_term.py +913 -0
  97. attune/memory/long_term_types.py +99 -0
  98. attune/memory/mixins/__init__.py +25 -0
  99. attune/memory/mixins/backend_init_mixin.py +249 -0
  100. attune/memory/mixins/capabilities_mixin.py +208 -0
  101. attune/memory/mixins/handoff_mixin.py +208 -0
  102. attune/memory/mixins/lifecycle_mixin.py +49 -0
  103. attune/memory/mixins/long_term_mixin.py +352 -0
  104. attune/memory/mixins/promotion_mixin.py +109 -0
  105. attune/memory/mixins/short_term_mixin.py +182 -0
  106. attune/memory/nodes.py +179 -0
  107. attune/memory/redis_bootstrap.py +540 -0
  108. attune/memory/security/__init__.py +31 -0
  109. attune/memory/security/audit_logger.py +932 -0
  110. attune/memory/security/pii_scrubber.py +640 -0
  111. attune/memory/security/secrets_detector.py +678 -0
  112. attune/memory/short_term.py +2192 -0
  113. attune/memory/simple_storage.py +302 -0
  114. attune/memory/storage/__init__.py +15 -0
  115. attune/memory/storage_backend.py +167 -0
  116. attune/memory/summary_index.py +583 -0
  117. attune/memory/types.py +446 -0
  118. attune/memory/unified.py +182 -0
  119. attune/meta_workflows/__init__.py +74 -0
  120. attune/meta_workflows/agent_creator.py +248 -0
  121. attune/meta_workflows/builtin_templates.py +567 -0
  122. attune/meta_workflows/cli_commands/__init__.py +56 -0
  123. attune/meta_workflows/cli_commands/agent_commands.py +321 -0
  124. attune/meta_workflows/cli_commands/analytics_commands.py +442 -0
  125. attune/meta_workflows/cli_commands/config_commands.py +232 -0
  126. attune/meta_workflows/cli_commands/memory_commands.py +182 -0
  127. attune/meta_workflows/cli_commands/template_commands.py +354 -0
  128. attune/meta_workflows/cli_commands/workflow_commands.py +382 -0
  129. attune/meta_workflows/cli_meta_workflows.py +59 -0
  130. attune/meta_workflows/form_engine.py +292 -0
  131. attune/meta_workflows/intent_detector.py +409 -0
  132. attune/meta_workflows/models.py +569 -0
  133. attune/meta_workflows/pattern_learner.py +738 -0
  134. attune/meta_workflows/plan_generator.py +384 -0
  135. attune/meta_workflows/session_context.py +397 -0
  136. attune/meta_workflows/template_registry.py +229 -0
  137. attune/meta_workflows/workflow.py +984 -0
  138. attune/metrics/__init__.py +12 -0
  139. attune/metrics/collector.py +31 -0
  140. attune/metrics/prompt_metrics.py +194 -0
  141. attune/models/__init__.py +172 -0
  142. attune/models/__main__.py +13 -0
  143. attune/models/adaptive_routing.py +437 -0
  144. attune/models/auth_cli.py +444 -0
  145. attune/models/auth_strategy.py +450 -0
  146. attune/models/cli.py +655 -0
  147. attune/models/empathy_executor.py +354 -0
  148. attune/models/executor.py +257 -0
  149. attune/models/fallback.py +762 -0
  150. attune/models/provider_config.py +282 -0
  151. attune/models/registry.py +472 -0
  152. attune/models/tasks.py +359 -0
  153. attune/models/telemetry/__init__.py +71 -0
  154. attune/models/telemetry/analytics.py +594 -0
  155. attune/models/telemetry/backend.py +196 -0
  156. attune/models/telemetry/data_models.py +431 -0
  157. attune/models/telemetry/storage.py +489 -0
  158. attune/models/token_estimator.py +420 -0
  159. attune/models/validation.py +280 -0
  160. attune/monitoring/__init__.py +52 -0
  161. attune/monitoring/alerts.py +946 -0
  162. attune/monitoring/alerts_cli.py +448 -0
  163. attune/monitoring/multi_backend.py +271 -0
  164. attune/monitoring/otel_backend.py +362 -0
  165. attune/optimization/__init__.py +19 -0
  166. attune/optimization/context_optimizer.py +272 -0
  167. attune/orchestration/__init__.py +67 -0
  168. attune/orchestration/agent_templates.py +707 -0
  169. attune/orchestration/config_store.py +499 -0
  170. attune/orchestration/execution_strategies.py +2111 -0
  171. attune/orchestration/meta_orchestrator.py +1168 -0
  172. attune/orchestration/pattern_learner.py +696 -0
  173. attune/orchestration/real_tools.py +931 -0
  174. attune/pattern_cache.py +187 -0
  175. attune/pattern_library.py +542 -0
  176. attune/patterns/debugging/all_patterns.json +81 -0
  177. attune/patterns/debugging/workflow_20260107_1770825e.json +77 -0
  178. attune/patterns/refactoring_memory.json +89 -0
  179. attune/persistence.py +564 -0
  180. attune/platform_utils.py +265 -0
  181. attune/plugins/__init__.py +28 -0
  182. attune/plugins/base.py +361 -0
  183. attune/plugins/registry.py +268 -0
  184. attune/project_index/__init__.py +32 -0
  185. attune/project_index/cli.py +335 -0
  186. attune/project_index/index.py +667 -0
  187. attune/project_index/models.py +504 -0
  188. attune/project_index/reports.py +474 -0
  189. attune/project_index/scanner.py +777 -0
  190. attune/project_index/scanner_parallel.py +291 -0
  191. attune/prompts/__init__.py +61 -0
  192. attune/prompts/config.py +77 -0
  193. attune/prompts/context.py +177 -0
  194. attune/prompts/parser.py +285 -0
  195. attune/prompts/registry.py +313 -0
  196. attune/prompts/templates.py +208 -0
  197. attune/redis_config.py +302 -0
  198. attune/redis_memory.py +799 -0
  199. attune/resilience/__init__.py +56 -0
  200. attune/resilience/circuit_breaker.py +256 -0
  201. attune/resilience/fallback.py +179 -0
  202. attune/resilience/health.py +300 -0
  203. attune/resilience/retry.py +209 -0
  204. attune/resilience/timeout.py +135 -0
  205. attune/routing/__init__.py +43 -0
  206. attune/routing/chain_executor.py +433 -0
  207. attune/routing/classifier.py +217 -0
  208. attune/routing/smart_router.py +234 -0
  209. attune/routing/workflow_registry.py +343 -0
  210. attune/scaffolding/README.md +589 -0
  211. attune/scaffolding/__init__.py +35 -0
  212. attune/scaffolding/__main__.py +14 -0
  213. attune/scaffolding/cli.py +240 -0
  214. attune/scaffolding/templates/base_wizard.py.jinja2 +121 -0
  215. attune/scaffolding/templates/coach_wizard.py.jinja2 +321 -0
  216. attune/scaffolding/templates/domain_wizard.py.jinja2 +408 -0
  217. attune/scaffolding/templates/linear_flow_wizard.py.jinja2 +203 -0
  218. attune/socratic/__init__.py +256 -0
  219. attune/socratic/ab_testing.py +958 -0
  220. attune/socratic/blueprint.py +533 -0
  221. attune/socratic/cli.py +703 -0
  222. attune/socratic/collaboration.py +1114 -0
  223. attune/socratic/domain_templates.py +924 -0
  224. attune/socratic/embeddings.py +738 -0
  225. attune/socratic/engine.py +794 -0
  226. attune/socratic/explainer.py +682 -0
  227. attune/socratic/feedback.py +772 -0
  228. attune/socratic/forms.py +629 -0
  229. attune/socratic/generator.py +732 -0
  230. attune/socratic/llm_analyzer.py +637 -0
  231. attune/socratic/mcp_server.py +702 -0
  232. attune/socratic/session.py +312 -0
  233. attune/socratic/storage.py +667 -0
  234. attune/socratic/success.py +730 -0
  235. attune/socratic/visual_editor.py +860 -0
  236. attune/socratic/web_ui.py +958 -0
  237. attune/telemetry/__init__.py +39 -0
  238. attune/telemetry/agent_coordination.py +475 -0
  239. attune/telemetry/agent_tracking.py +367 -0
  240. attune/telemetry/approval_gates.py +545 -0
  241. attune/telemetry/cli.py +1231 -0
  242. attune/telemetry/commands/__init__.py +14 -0
  243. attune/telemetry/commands/dashboard_commands.py +696 -0
  244. attune/telemetry/event_streaming.py +409 -0
  245. attune/telemetry/feedback_loop.py +567 -0
  246. attune/telemetry/usage_tracker.py +591 -0
  247. attune/templates.py +754 -0
  248. attune/test_generator/__init__.py +38 -0
  249. attune/test_generator/__main__.py +14 -0
  250. attune/test_generator/cli.py +234 -0
  251. attune/test_generator/generator.py +355 -0
  252. attune/test_generator/risk_analyzer.py +216 -0
  253. attune/test_generator/templates/unit_test.py.jinja2 +272 -0
  254. attune/tier_recommender.py +384 -0
  255. attune/tools.py +183 -0
  256. attune/trust/__init__.py +28 -0
  257. attune/trust/circuit_breaker.py +579 -0
  258. attune/trust_building.py +527 -0
  259. attune/validation/__init__.py +19 -0
  260. attune/validation/xml_validator.py +281 -0
  261. attune/vscode_bridge.py +173 -0
  262. attune/workflow_commands.py +780 -0
  263. attune/workflow_patterns/__init__.py +33 -0
  264. attune/workflow_patterns/behavior.py +249 -0
  265. attune/workflow_patterns/core.py +76 -0
  266. attune/workflow_patterns/output.py +99 -0
  267. attune/workflow_patterns/registry.py +255 -0
  268. attune/workflow_patterns/structural.py +288 -0
  269. attune/workflows/__init__.py +539 -0
  270. attune/workflows/autonomous_test_gen.py +1268 -0
  271. attune/workflows/base.py +2667 -0
  272. attune/workflows/batch_processing.py +342 -0
  273. attune/workflows/bug_predict.py +1084 -0
  274. attune/workflows/builder.py +273 -0
  275. attune/workflows/caching.py +253 -0
  276. attune/workflows/code_review.py +1048 -0
  277. attune/workflows/code_review_adapters.py +312 -0
  278. attune/workflows/code_review_pipeline.py +722 -0
  279. attune/workflows/config.py +645 -0
  280. attune/workflows/dependency_check.py +644 -0
  281. attune/workflows/document_gen/__init__.py +25 -0
  282. attune/workflows/document_gen/config.py +30 -0
  283. attune/workflows/document_gen/report_formatter.py +162 -0
  284. attune/workflows/document_gen/workflow.py +1426 -0
  285. attune/workflows/document_manager.py +216 -0
  286. attune/workflows/document_manager_README.md +134 -0
  287. attune/workflows/documentation_orchestrator.py +1205 -0
  288. attune/workflows/history.py +510 -0
  289. attune/workflows/keyboard_shortcuts/__init__.py +39 -0
  290. attune/workflows/keyboard_shortcuts/generators.py +391 -0
  291. attune/workflows/keyboard_shortcuts/parsers.py +416 -0
  292. attune/workflows/keyboard_shortcuts/prompts.py +295 -0
  293. attune/workflows/keyboard_shortcuts/schema.py +193 -0
  294. attune/workflows/keyboard_shortcuts/workflow.py +509 -0
  295. attune/workflows/llm_base.py +363 -0
  296. attune/workflows/manage_docs.py +87 -0
  297. attune/workflows/manage_docs_README.md +134 -0
  298. attune/workflows/manage_documentation.py +821 -0
  299. attune/workflows/new_sample_workflow1.py +149 -0
  300. attune/workflows/new_sample_workflow1_README.md +150 -0
  301. attune/workflows/orchestrated_health_check.py +849 -0
  302. attune/workflows/orchestrated_release_prep.py +600 -0
  303. attune/workflows/output.py +413 -0
  304. attune/workflows/perf_audit.py +863 -0
  305. attune/workflows/pr_review.py +762 -0
  306. attune/workflows/progress.py +785 -0
  307. attune/workflows/progress_server.py +322 -0
  308. attune/workflows/progressive/README 2.md +454 -0
  309. attune/workflows/progressive/README.md +454 -0
  310. attune/workflows/progressive/__init__.py +82 -0
  311. attune/workflows/progressive/cli.py +219 -0
  312. attune/workflows/progressive/core.py +488 -0
  313. attune/workflows/progressive/orchestrator.py +723 -0
  314. attune/workflows/progressive/reports.py +520 -0
  315. attune/workflows/progressive/telemetry.py +274 -0
  316. attune/workflows/progressive/test_gen.py +495 -0
  317. attune/workflows/progressive/workflow.py +589 -0
  318. attune/workflows/refactor_plan.py +694 -0
  319. attune/workflows/release_prep.py +895 -0
  320. attune/workflows/release_prep_crew.py +969 -0
  321. attune/workflows/research_synthesis.py +404 -0
  322. attune/workflows/routing.py +168 -0
  323. attune/workflows/secure_release.py +593 -0
  324. attune/workflows/security_adapters.py +297 -0
  325. attune/workflows/security_audit.py +1329 -0
  326. attune/workflows/security_audit_phase3.py +355 -0
  327. attune/workflows/seo_optimization.py +633 -0
  328. attune/workflows/step_config.py +234 -0
  329. attune/workflows/telemetry_mixin.py +269 -0
  330. attune/workflows/test5.py +125 -0
  331. attune/workflows/test5_README.md +158 -0
  332. attune/workflows/test_coverage_boost_crew.py +849 -0
  333. attune/workflows/test_gen/__init__.py +52 -0
  334. attune/workflows/test_gen/ast_analyzer.py +249 -0
  335. attune/workflows/test_gen/config.py +88 -0
  336. attune/workflows/test_gen/data_models.py +38 -0
  337. attune/workflows/test_gen/report_formatter.py +289 -0
  338. attune/workflows/test_gen/test_templates.py +381 -0
  339. attune/workflows/test_gen/workflow.py +655 -0
  340. attune/workflows/test_gen.py +54 -0
  341. attune/workflows/test_gen_behavioral.py +477 -0
  342. attune/workflows/test_gen_parallel.py +341 -0
  343. attune/workflows/test_lifecycle.py +526 -0
  344. attune/workflows/test_maintenance.py +627 -0
  345. attune/workflows/test_maintenance_cli.py +590 -0
  346. attune/workflows/test_maintenance_crew.py +840 -0
  347. attune/workflows/test_runner.py +622 -0
  348. attune/workflows/tier_tracking.py +531 -0
  349. attune/workflows/xml_enhanced_crew.py +285 -0
  350. attune_ai-2.0.0.dist-info/METADATA +1026 -0
  351. attune_ai-2.0.0.dist-info/RECORD +457 -0
  352. attune_ai-2.0.0.dist-info/WHEEL +5 -0
  353. attune_ai-2.0.0.dist-info/entry_points.txt +26 -0
  354. attune_ai-2.0.0.dist-info/licenses/LICENSE +201 -0
  355. attune_ai-2.0.0.dist-info/licenses/LICENSE_CHANGE_ANNOUNCEMENT.md +101 -0
  356. attune_ai-2.0.0.dist-info/top_level.txt +5 -0
  357. attune_healthcare/__init__.py +13 -0
  358. attune_healthcare/monitors/__init__.py +9 -0
  359. attune_healthcare/monitors/clinical_protocol_monitor.py +315 -0
  360. attune_healthcare/monitors/monitoring/__init__.py +44 -0
  361. attune_healthcare/monitors/monitoring/protocol_checker.py +300 -0
  362. attune_healthcare/monitors/monitoring/protocol_loader.py +214 -0
  363. attune_healthcare/monitors/monitoring/sensor_parsers.py +306 -0
  364. attune_healthcare/monitors/monitoring/trajectory_analyzer.py +389 -0
  365. attune_llm/README.md +553 -0
  366. attune_llm/__init__.py +28 -0
  367. attune_llm/agent_factory/__init__.py +53 -0
  368. attune_llm/agent_factory/adapters/__init__.py +85 -0
  369. attune_llm/agent_factory/adapters/autogen_adapter.py +312 -0
  370. attune_llm/agent_factory/adapters/crewai_adapter.py +483 -0
  371. attune_llm/agent_factory/adapters/haystack_adapter.py +298 -0
  372. attune_llm/agent_factory/adapters/langchain_adapter.py +362 -0
  373. attune_llm/agent_factory/adapters/langgraph_adapter.py +333 -0
  374. attune_llm/agent_factory/adapters/native.py +228 -0
  375. attune_llm/agent_factory/adapters/wizard_adapter.py +423 -0
  376. attune_llm/agent_factory/base.py +305 -0
  377. attune_llm/agent_factory/crews/__init__.py +67 -0
  378. attune_llm/agent_factory/crews/code_review.py +1113 -0
  379. attune_llm/agent_factory/crews/health_check.py +1262 -0
  380. attune_llm/agent_factory/crews/refactoring.py +1128 -0
  381. attune_llm/agent_factory/crews/security_audit.py +1018 -0
  382. attune_llm/agent_factory/decorators.py +287 -0
  383. attune_llm/agent_factory/factory.py +558 -0
  384. attune_llm/agent_factory/framework.py +193 -0
  385. attune_llm/agent_factory/memory_integration.py +328 -0
  386. attune_llm/agent_factory/resilient.py +320 -0
  387. attune_llm/agents_md/__init__.py +22 -0
  388. attune_llm/agents_md/loader.py +218 -0
  389. attune_llm/agents_md/parser.py +271 -0
  390. attune_llm/agents_md/registry.py +307 -0
  391. attune_llm/claude_memory.py +466 -0
  392. attune_llm/cli/__init__.py +8 -0
  393. attune_llm/cli/sync_claude.py +487 -0
  394. attune_llm/code_health.py +1313 -0
  395. attune_llm/commands/__init__.py +51 -0
  396. attune_llm/commands/context.py +375 -0
  397. attune_llm/commands/loader.py +301 -0
  398. attune_llm/commands/models.py +231 -0
  399. attune_llm/commands/parser.py +371 -0
  400. attune_llm/commands/registry.py +429 -0
  401. attune_llm/config/__init__.py +29 -0
  402. attune_llm/config/unified.py +291 -0
  403. attune_llm/context/__init__.py +22 -0
  404. attune_llm/context/compaction.py +455 -0
  405. attune_llm/context/manager.py +434 -0
  406. attune_llm/contextual_patterns.py +361 -0
  407. attune_llm/core.py +907 -0
  408. attune_llm/git_pattern_extractor.py +435 -0
  409. attune_llm/hooks/__init__.py +24 -0
  410. attune_llm/hooks/config.py +306 -0
  411. attune_llm/hooks/executor.py +289 -0
  412. attune_llm/hooks/registry.py +302 -0
  413. attune_llm/hooks/scripts/__init__.py +39 -0
  414. attune_llm/hooks/scripts/evaluate_session.py +201 -0
  415. attune_llm/hooks/scripts/first_time_init.py +285 -0
  416. attune_llm/hooks/scripts/pre_compact.py +207 -0
  417. attune_llm/hooks/scripts/session_end.py +183 -0
  418. attune_llm/hooks/scripts/session_start.py +163 -0
  419. attune_llm/hooks/scripts/suggest_compact.py +225 -0
  420. attune_llm/learning/__init__.py +30 -0
  421. attune_llm/learning/evaluator.py +438 -0
  422. attune_llm/learning/extractor.py +514 -0
  423. attune_llm/learning/storage.py +560 -0
  424. attune_llm/levels.py +227 -0
  425. attune_llm/pattern_confidence.py +414 -0
  426. attune_llm/pattern_resolver.py +272 -0
  427. attune_llm/pattern_summary.py +350 -0
  428. attune_llm/providers.py +967 -0
  429. attune_llm/routing/__init__.py +32 -0
  430. attune_llm/routing/model_router.py +362 -0
  431. attune_llm/security/IMPLEMENTATION_SUMMARY.md +413 -0
  432. attune_llm/security/PHASE2_COMPLETE.md +384 -0
  433. attune_llm/security/PHASE2_SECRETS_DETECTOR_COMPLETE.md +271 -0
  434. attune_llm/security/QUICK_REFERENCE.md +316 -0
  435. attune_llm/security/README.md +262 -0
  436. attune_llm/security/__init__.py +62 -0
  437. attune_llm/security/audit_logger.py +929 -0
  438. attune_llm/security/audit_logger_example.py +152 -0
  439. attune_llm/security/pii_scrubber.py +640 -0
  440. attune_llm/security/secrets_detector.py +678 -0
  441. attune_llm/security/secrets_detector_example.py +304 -0
  442. attune_llm/security/secure_memdocs.py +1192 -0
  443. attune_llm/security/secure_memdocs_example.py +278 -0
  444. attune_llm/session_status.py +745 -0
  445. attune_llm/state.py +246 -0
  446. attune_llm/utils/__init__.py +5 -0
  447. attune_llm/utils/tokens.py +349 -0
  448. attune_software/SOFTWARE_PLUGIN_README.md +57 -0
  449. attune_software/__init__.py +13 -0
  450. attune_software/cli/__init__.py +120 -0
  451. attune_software/cli/inspect.py +362 -0
  452. attune_software/cli.py +574 -0
  453. attune_software/plugin.py +188 -0
  454. workflow_scaffolding/__init__.py +11 -0
  455. workflow_scaffolding/__main__.py +12 -0
  456. workflow_scaffolding/cli.py +206 -0
  457. workflow_scaffolding/generator.py +265 -0
@@ -0,0 +1,895 @@
1
+ """Release Preparation Workflow
2
+
3
+ Pre-release quality gate combining health checks, security scan,
4
+ and changelog generation.
5
+
6
+ Stages:
7
+ 1. health (CHEAP) - Run health checks (lint, types, tests)
8
+ 2. security (CAPABLE) - Security scan summary
9
+ 3. changelog (CAPABLE) - Generate changelog from commits
10
+ 4. approve (PREMIUM) - Final release readiness assessment (conditional)
11
+
12
+ Copyright 2025 Smart-AI-Memory
13
+ Licensed under Fair Source License 0.9
14
+ """
15
+
16
+ import json
17
+ import subprocess
18
+ from datetime import datetime
19
+ from typing import Any
20
+
21
+ from .base import BaseWorkflow, ModelTier
22
+ from .step_config import WorkflowStepConfig
23
+
24
+ # Define step configurations for executor-based execution
25
+ RELEASE_PREP_STEPS = {
26
+ "approve": WorkflowStepConfig(
27
+ name="approve",
28
+ task_type="final_review", # Premium tier task
29
+ tier_hint="premium",
30
+ description="Assess release readiness and provide go/no-go recommendation",
31
+ max_tokens=2000,
32
+ ),
33
+ }
34
+
35
+
36
+ class ReleasePreparationWorkflow(BaseWorkflow):
37
+ """Pre-release quality gate workflow.
38
+
39
+ Combines multiple checks to determine if the codebase
40
+ is ready for release.
41
+
42
+ When use_security_crew=True, adds an additional crew_security stage
43
+ that runs SecurityAuditCrew for comprehensive security analysis.
44
+ """
45
+
46
+ name = "release-prep"
47
+ description = "Pre-release quality gate with health, security, and changelog"
48
+
49
+ # Default stages (can be modified in __init__)
50
+ stages = ["health", "security", "changelog", "approve"]
51
+ tier_map = {
52
+ "health": ModelTier.CHEAP,
53
+ "security": ModelTier.CAPABLE,
54
+ "changelog": ModelTier.CAPABLE,
55
+ "approve": ModelTier.PREMIUM,
56
+ }
57
+
58
+ def __init__(
59
+ self,
60
+ skip_approve_if_clean: bool = True,
61
+ use_security_crew: bool = False,
62
+ crew_config: dict | None = None,
63
+ enable_auth_strategy: bool = True,
64
+ **kwargs: Any,
65
+ ):
66
+ """Initialize release preparation workflow.
67
+
68
+ Args:
69
+ skip_approve_if_clean: Skip premium approval if all checks pass
70
+ use_security_crew: Enable SecurityAuditCrew for comprehensive security audit
71
+ crew_config: Configuration dict for SecurityAuditCrew
72
+ enable_auth_strategy: Enable intelligent auth routing (default: True)
73
+ **kwargs: Additional arguments passed to BaseWorkflow
74
+
75
+ """
76
+ super().__init__(**kwargs)
77
+ self.skip_approve_if_clean = skip_approve_if_clean
78
+ self.use_security_crew = use_security_crew
79
+ self.crew_config = crew_config or {}
80
+ self.enable_auth_strategy = enable_auth_strategy
81
+ self._has_blockers: bool = False
82
+ self._auth_mode_used: str | None = None
83
+
84
+ # Dynamically configure stages based on security crew setting
85
+ if use_security_crew:
86
+ self.stages = ["health", "security", "crew_security", "changelog", "approve"]
87
+ self.tier_map = {
88
+ "health": ModelTier.CHEAP,
89
+ "security": ModelTier.CAPABLE,
90
+ "crew_security": ModelTier.PREMIUM,
91
+ "changelog": ModelTier.CAPABLE,
92
+ "approve": ModelTier.PREMIUM,
93
+ }
94
+ else:
95
+ self.stages = ["health", "security", "changelog", "approve"]
96
+ self.tier_map = {
97
+ "health": ModelTier.CHEAP,
98
+ "security": ModelTier.CAPABLE,
99
+ "changelog": ModelTier.CAPABLE,
100
+ "approve": ModelTier.PREMIUM,
101
+ }
102
+
103
+ def should_skip_stage(self, stage_name: str, input_data: Any) -> tuple[bool, str | None]:
104
+ """Skip approval if all checks pass cleanly.
105
+
106
+ Args:
107
+ stage_name: Name of the stage to check
108
+ input_data: Current workflow data
109
+
110
+ Returns:
111
+ Tuple of (should_skip, reason)
112
+
113
+ """
114
+ if stage_name == "approve" and self.skip_approve_if_clean:
115
+ if not self._has_blockers:
116
+ return True, "All checks passed - auto-approved"
117
+ return False, None
118
+
119
+ async def run_stage(
120
+ self,
121
+ stage_name: str,
122
+ tier: ModelTier,
123
+ input_data: Any,
124
+ ) -> tuple[Any, int, int]:
125
+ """Route to specific stage implementation."""
126
+ if stage_name == "health":
127
+ return await self._health(input_data, tier)
128
+ if stage_name == "security":
129
+ return await self._security(input_data, tier)
130
+ if stage_name == "crew_security":
131
+ return await self._crew_security(input_data, tier)
132
+ if stage_name == "changelog":
133
+ return await self._changelog(input_data, tier)
134
+ if stage_name == "approve":
135
+ return await self._approve(input_data, tier)
136
+ raise ValueError(f"Unknown stage: {stage_name}")
137
+
138
+ async def _health(self, input_data: dict, tier: ModelTier) -> tuple[dict, int, int]:
139
+ """Run health checks.
140
+
141
+ Executes lint, type checking, and tests.
142
+ """
143
+ target_path = input_data.get("path", ".")
144
+
145
+ # === AUTH STRATEGY INTEGRATION ===
146
+ if self.enable_auth_strategy:
147
+ try:
148
+ import logging
149
+ from pathlib import Path
150
+
151
+ from attune.models import (
152
+ count_lines_of_code,
153
+ get_auth_strategy,
154
+ get_module_size_category,
155
+ )
156
+ logger = logging.getLogger(__name__)
157
+
158
+ # Calculate total LOC for project/directory
159
+ target = Path(target_path)
160
+ total_lines = 0
161
+ if target.is_file():
162
+ total_lines = count_lines_of_code(target)
163
+ elif target.is_dir():
164
+ for py_file in target.rglob("*.py"):
165
+ try:
166
+ total_lines += count_lines_of_code(py_file)
167
+ except Exception:
168
+ pass
169
+
170
+ if total_lines > 0:
171
+ strategy = get_auth_strategy()
172
+ recommended_mode = strategy.get_recommended_mode(total_lines)
173
+ self._auth_mode_used = recommended_mode.value
174
+
175
+ size_category = get_module_size_category(total_lines)
176
+ logger.info(f"Release prep target: {target_path} ({total_lines:,} LOC, {size_category})")
177
+ logger.info(f"Recommended auth mode: {recommended_mode.value}")
178
+
179
+ cost_estimate = strategy.estimate_cost(total_lines, recommended_mode)
180
+ if recommended_mode.value == "subscription":
181
+ logger.info(f"Cost: {cost_estimate['quota_cost']}")
182
+ else:
183
+ logger.info(f"Cost: ~${cost_estimate['monetary_cost']:.4f}")
184
+
185
+ except Exception as e:
186
+ import logging
187
+ logger = logging.getLogger(__name__)
188
+ logger.warning(f"Auth strategy detection failed: {e}")
189
+
190
+ checks: dict[str, dict] = {}
191
+
192
+ # Lint check (ruff)
193
+ try:
194
+ result = subprocess.run(
195
+ ["python", "-m", "ruff", "check", target_path],
196
+ check=False,
197
+ capture_output=True,
198
+ text=True,
199
+ timeout=60,
200
+ )
201
+ lint_errors = result.stdout.count("error") + result.stderr.count("error")
202
+ checks["lint"] = {
203
+ "passed": result.returncode == 0,
204
+ "errors": lint_errors,
205
+ "tool": "ruff",
206
+ }
207
+ except (subprocess.TimeoutExpired, FileNotFoundError):
208
+ checks["lint"] = {"passed": True, "errors": 0, "tool": "ruff", "skipped": True}
209
+
210
+ # Type check (mypy)
211
+ try:
212
+ result = subprocess.run(
213
+ ["python", "-m", "mypy", target_path, "--ignore-missing-imports"],
214
+ check=False,
215
+ capture_output=True,
216
+ text=True,
217
+ timeout=120,
218
+ )
219
+ type_errors = result.stdout.count("error:")
220
+ checks["types"] = {
221
+ "passed": result.returncode == 0,
222
+ "errors": type_errors,
223
+ "tool": "mypy",
224
+ }
225
+ except (subprocess.TimeoutExpired, FileNotFoundError):
226
+ checks["types"] = {"passed": True, "errors": 0, "tool": "mypy", "skipped": True}
227
+
228
+ # Test check (pytest)
229
+ try:
230
+ result = subprocess.run(
231
+ ["python", "-m", "pytest", "--co", "-q"],
232
+ check=False,
233
+ capture_output=True,
234
+ text=True,
235
+ timeout=60,
236
+ cwd=target_path,
237
+ )
238
+ # Count collected tests
239
+ test_count = 0
240
+ for line in result.stdout.splitlines():
241
+ if "test" in line.lower():
242
+ test_count += 1
243
+
244
+ checks["tests"] = {
245
+ "passed": True,
246
+ "test_count": test_count,
247
+ "tool": "pytest",
248
+ }
249
+ except (subprocess.TimeoutExpired, FileNotFoundError):
250
+ checks["tests"] = {"passed": True, "test_count": 0, "tool": "pytest", "skipped": True}
251
+
252
+ # Calculate health score
253
+ failed_checks = [k for k, v in checks.items() if not v.get("passed", True)]
254
+ health_score = 100 - (len(failed_checks) * 20)
255
+
256
+ if failed_checks:
257
+ self._has_blockers = True
258
+
259
+ health_result = {
260
+ "checks": checks,
261
+ "health_score": max(0, health_score),
262
+ "failed_checks": failed_checks,
263
+ "passed": len(failed_checks) == 0,
264
+ }
265
+
266
+ input_tokens = len(str(input_data)) // 4
267
+ output_tokens = len(str(health_result)) // 4
268
+
269
+ return (
270
+ {
271
+ "health": health_result,
272
+ **input_data,
273
+ },
274
+ input_tokens,
275
+ output_tokens,
276
+ )
277
+
278
+ async def _security(self, input_data: dict, tier: ModelTier) -> tuple[dict, int, int]:
279
+ """Run security scan using Bandit.
280
+
281
+ Uses industry-standard Bandit tool for security analysis.
282
+ """
283
+ target_path = input_data.get("path", ".")
284
+
285
+ issues: list[dict] = []
286
+ high_count = 0
287
+ medium_count = 0
288
+
289
+ # Run Bandit security scanner
290
+ try:
291
+ result = subprocess.run(
292
+ [
293
+ "python",
294
+ "-m",
295
+ "bandit",
296
+ "-r",
297
+ target_path,
298
+ "--severity-level",
299
+ "medium",
300
+ "--format",
301
+ "json",
302
+ ],
303
+ check=False,
304
+ capture_output=True,
305
+ text=True,
306
+ timeout=120,
307
+ )
308
+
309
+ # Parse Bandit JSON output
310
+ if result.stdout:
311
+ try:
312
+ bandit_data = json.loads(result.stdout)
313
+ bandit_results = bandit_data.get("results", [])
314
+
315
+ for finding in bandit_results:
316
+ severity = finding.get("issue_severity", "LOW").lower()
317
+ issues.append(
318
+ {
319
+ "type": finding.get("test_id", "unknown"),
320
+ "file": finding.get("filename", "unknown"),
321
+ "line": finding.get("line_number", 0),
322
+ "severity": severity,
323
+ "message": finding.get("issue_text", ""),
324
+ "confidence": finding.get("issue_confidence", ""),
325
+ }
326
+ )
327
+
328
+ if severity == "high":
329
+ high_count += 1
330
+ elif severity == "medium":
331
+ medium_count += 1
332
+
333
+ except json.JSONDecodeError:
334
+ # If JSON parsing fails, fall back to error count from stderr
335
+ pass
336
+
337
+ except (subprocess.TimeoutExpired, FileNotFoundError):
338
+ # Bandit not available or timed out - skip security scan
339
+ pass
340
+
341
+ if high_count > 0:
342
+ self._has_blockers = True
343
+
344
+ security_result = {
345
+ "issues": issues[:20], # Top 20
346
+ "total_issues": len(issues),
347
+ "high_severity": high_count,
348
+ "medium_severity": medium_count,
349
+ "passed": high_count == 0,
350
+ "tool": "bandit",
351
+ }
352
+
353
+ input_tokens = len(str(input_data)) // 4
354
+ output_tokens = len(str(security_result)) // 4
355
+
356
+ return (
357
+ {
358
+ "security": security_result,
359
+ **input_data,
360
+ },
361
+ input_tokens,
362
+ output_tokens,
363
+ )
364
+
365
+ async def _crew_security(self, input_data: dict, tier: ModelTier) -> tuple[dict, int, int]:
366
+ """Run SecurityAuditCrew for comprehensive security analysis.
367
+
368
+ This stage uses the 5-agent SecurityAuditCrew for deep security
369
+ analysis including vulnerability hunting, risk assessment,
370
+ remediation planning, and compliance mapping.
371
+
372
+ Falls back gracefully if SecurityAuditCrew is not available.
373
+ """
374
+ try:
375
+ from .security_adapters import (
376
+ _check_crew_available,
377
+ _get_crew_audit,
378
+ crew_report_to_workflow_format,
379
+ merge_security_results,
380
+ )
381
+ except ImportError:
382
+ # Security adapters removed - return fallback
383
+ return (
384
+ {
385
+ "crew_security": {
386
+ "available": False,
387
+ "fallback": True,
388
+ "reason": "Security adapters not installed",
389
+ },
390
+ **input_data,
391
+ },
392
+ 0,
393
+ 0,
394
+ )
395
+
396
+ target_path = input_data.get("path", ".")
397
+ existing_security = input_data.get("security", {})
398
+
399
+ # Check if crew is available
400
+ if not _check_crew_available():
401
+ return (
402
+ {
403
+ "crew_security": {
404
+ "available": False,
405
+ "fallback": True,
406
+ "reason": "SecurityAuditCrew not installed",
407
+ },
408
+ **input_data,
409
+ },
410
+ 0,
411
+ 0,
412
+ )
413
+
414
+ # Run SecurityAuditCrew
415
+ report = await _get_crew_audit(target_path, self.crew_config)
416
+
417
+ if report is None:
418
+ return (
419
+ {
420
+ "crew_security": {
421
+ "available": True,
422
+ "fallback": True,
423
+ "reason": "SecurityAuditCrew audit failed or timed out",
424
+ },
425
+ **input_data,
426
+ },
427
+ 0,
428
+ 0,
429
+ )
430
+
431
+ # Convert crew report to workflow format
432
+ crew_results = crew_report_to_workflow_format(report)
433
+
434
+ # Merge with existing security stage results
435
+ existing_issues = existing_security.get("issues", [])
436
+ merged = merge_security_results(crew_results, {"findings": existing_issues})
437
+
438
+ # Update blockers based on crew findings
439
+ critical_count = len(crew_results.get("assessment", {}).get("critical_findings", []))
440
+ high_count = len(crew_results.get("assessment", {}).get("high_findings", []))
441
+
442
+ if critical_count > 0 or high_count > 0:
443
+ self._has_blockers = True
444
+
445
+ crew_security_result = {
446
+ "available": True,
447
+ "fallback": False,
448
+ "findings": crew_results.get("findings", []),
449
+ "finding_count": crew_results.get("finding_count", 0),
450
+ "risk_score": crew_results.get("risk_score", 0),
451
+ "risk_level": crew_results.get("risk_level", "none"),
452
+ "critical_count": critical_count,
453
+ "high_count": high_count,
454
+ "summary": crew_results.get("summary", ""),
455
+ "agents_used": crew_results.get("agents_used", []),
456
+ "merged_results": merged,
457
+ }
458
+
459
+ # Estimate tokens (crew uses internal LLM calls)
460
+ input_tokens = len(str(input_data)) // 4
461
+ output_tokens = len(str(crew_security_result)) // 4
462
+
463
+ return (
464
+ {
465
+ "crew_security": crew_security_result,
466
+ **input_data,
467
+ },
468
+ input_tokens,
469
+ output_tokens,
470
+ )
471
+
472
+ async def _changelog(self, input_data: dict, tier: ModelTier) -> tuple[dict, int, int]:
473
+ """Generate changelog from recent commits.
474
+
475
+ Extracts commit messages and organizes by type.
476
+ """
477
+ target_path = input_data.get("path", ".")
478
+ since = input_data.get("since", "1 week ago")
479
+
480
+ commits: list[dict] = []
481
+
482
+ try:
483
+ result = subprocess.run(
484
+ ["git", "log", f"--since={since}", "--oneline", "--no-merges"],
485
+ check=False,
486
+ capture_output=True,
487
+ text=True,
488
+ timeout=30,
489
+ cwd=target_path,
490
+ )
491
+
492
+ for line in result.stdout.splitlines():
493
+ if not line.strip():
494
+ continue
495
+
496
+ parts = line.split(" ", 1)
497
+ if len(parts) >= 2:
498
+ sha = parts[0]
499
+ message = parts[1]
500
+
501
+ # Categorize by conventional commit prefix
502
+ category = "other"
503
+ if message.startswith("feat"):
504
+ category = "features"
505
+ elif message.startswith("fix"):
506
+ category = "fixes"
507
+ elif message.startswith("docs"):
508
+ category = "docs"
509
+ elif message.startswith("refactor"):
510
+ category = "refactor"
511
+ elif message.startswith("test"):
512
+ category = "tests"
513
+ elif message.startswith("chore"):
514
+ category = "chores"
515
+
516
+ commits.append(
517
+ {
518
+ "sha": sha,
519
+ "message": message,
520
+ "category": category,
521
+ },
522
+ )
523
+ except (subprocess.TimeoutExpired, FileNotFoundError):
524
+ pass
525
+
526
+ # Group by category
527
+ by_category: dict[str, list] = {}
528
+ for commit in commits:
529
+ cat = commit["category"]
530
+ if cat not in by_category:
531
+ by_category[cat] = []
532
+ by_category[cat].append(commit)
533
+
534
+ changelog = {
535
+ "commits": commits,
536
+ "total_commits": len(commits),
537
+ "by_category": {k: len(v) for k, v in by_category.items()},
538
+ "generated_at": datetime.now().isoformat(),
539
+ "period": since,
540
+ }
541
+
542
+ input_tokens = len(str(input_data)) // 4
543
+ output_tokens = len(str(changelog)) // 4
544
+
545
+ return (
546
+ {
547
+ "changelog": changelog,
548
+ **input_data,
549
+ },
550
+ input_tokens,
551
+ output_tokens,
552
+ )
553
+
554
+ async def _approve(self, input_data: dict, tier: ModelTier) -> tuple[dict, int, int]:
555
+ """Final release readiness assessment using LLM.
556
+
557
+ Synthesizes all checks into go/no-go recommendation.
558
+
559
+ Supports XML-enhanced prompts when enabled in workflow config.
560
+ """
561
+ health = input_data.get("health", {})
562
+ security = input_data.get("security", {})
563
+ changelog = input_data.get("changelog", {})
564
+ target = input_data.get("path", "")
565
+
566
+ # Gather blockers
567
+ blockers: list[str] = []
568
+
569
+ if not health.get("passed", False):
570
+ for check in health.get("failed_checks", []):
571
+ blockers.append(f"Health check failed: {check}")
572
+
573
+ if not security.get("passed", False):
574
+ blockers.append(f"Security issues: {security.get('high_severity', 0)} high severity")
575
+
576
+ if changelog.get("total_commits", 0) == 0:
577
+ blockers.append("No commits in release period")
578
+
579
+ # Gather warnings
580
+ warnings: list[str] = []
581
+
582
+ if security.get("medium_severity", 0) > 0:
583
+ warnings.append(f"{security.get('medium_severity')} medium security issues")
584
+
585
+ test_count = health.get("checks", {}).get("tests", {}).get("test_count", 0)
586
+ if test_count < 10:
587
+ warnings.append(f"Low test count: {test_count}")
588
+
589
+ # Build input payload for LLM
590
+ input_payload = f"""Target: {target or "codebase"}
591
+
592
+ Health Score: {health.get("health_score", 0)}/100
593
+ Health Checks: {json.dumps(health.get("checks", {}), indent=2)}
594
+
595
+ Security Issues: {security.get("total_issues", 0)}
596
+ High Severity: {security.get("high_severity", 0)}
597
+ Medium Severity: {security.get("medium_severity", 0)}
598
+
599
+ Commit Count: {changelog.get("total_commits", 0)}
600
+ Changes by Category: {json.dumps(changelog.get("by_category", {}), indent=2)}
601
+
602
+ Blockers: {json.dumps(blockers, indent=2)}
603
+ Warnings: {json.dumps(warnings, indent=2)}"""
604
+
605
+ # Check if XML prompts are enabled
606
+ if self._is_xml_enabled():
607
+ # Use XML-enhanced prompt
608
+ user_message = self._render_xml_prompt(
609
+ role="release manager assessing release readiness",
610
+ goal="Provide a comprehensive release readiness assessment",
611
+ instructions=[
612
+ "Evaluate all health checks and their implications",
613
+ "Assess security findings and their risk level",
614
+ "Review the changelog for completeness",
615
+ "Identify any blockers that must be resolved",
616
+ "Provide a clear go/no-go recommendation",
617
+ "Suggest remediation steps for any issues",
618
+ ],
619
+ constraints=[
620
+ "Be conservative - flag potential issues",
621
+ "Provide clear, actionable feedback",
622
+ "Include confidence level in recommendation",
623
+ ],
624
+ input_type="release_checks",
625
+ input_payload=input_payload,
626
+ extra={
627
+ "blocker_count": len(blockers),
628
+ "warning_count": len(warnings),
629
+ },
630
+ )
631
+ system = None # XML prompt includes all context
632
+ else:
633
+ # Use legacy plain text prompts
634
+ system = """You are a release manager assessing release readiness.
635
+ Analyze the health checks, security findings, and changelog to provide
636
+ a clear go/no-go recommendation.
637
+
638
+ Be thorough and flag any potential issues."""
639
+
640
+ user_message = f"""Assess release readiness:
641
+
642
+ {input_payload}
643
+
644
+ Provide a comprehensive release readiness assessment."""
645
+
646
+ # Try executor-based execution first (Phase 3 pattern)
647
+ if self._executor is not None or self._api_key:
648
+ try:
649
+ step = RELEASE_PREP_STEPS["approve"]
650
+ response, input_tokens, output_tokens, cost = await self.run_step_with_executor(
651
+ step=step,
652
+ prompt=user_message,
653
+ system=system,
654
+ )
655
+ except Exception:
656
+ # Fall back to legacy _call_llm if executor fails
657
+ response, input_tokens, output_tokens = await self._call_llm(
658
+ tier,
659
+ system or "",
660
+ user_message,
661
+ max_tokens=2000,
662
+ )
663
+ else:
664
+ # Legacy path for backward compatibility
665
+ response, input_tokens, output_tokens = await self._call_llm(
666
+ tier,
667
+ system or "",
668
+ user_message,
669
+ max_tokens=2000,
670
+ )
671
+
672
+ # Parse XML response if enforcement is enabled
673
+ parsed_data = self._parse_xml_response(response)
674
+
675
+ # Make decision
676
+ approved = len(blockers) == 0
677
+ confidence = "high" if approved and len(warnings) == 0 else "medium" if approved else "low"
678
+
679
+ result = {
680
+ "approved": approved,
681
+ "confidence": confidence,
682
+ "blockers": blockers,
683
+ "warnings": warnings,
684
+ "health_score": health.get("health_score", 0),
685
+ "commit_count": changelog.get("total_commits", 0),
686
+ "assessment": response,
687
+ "recommendation": (
688
+ "Ready for release" if approved else "Address blockers before release"
689
+ ),
690
+ "model_tier_used": tier.value,
691
+ }
692
+
693
+ # Include auth mode used for telemetry
694
+ if self._auth_mode_used:
695
+ result["auth_mode_used"] = self._auth_mode_used
696
+
697
+ # Merge parsed XML data if available
698
+ if parsed_data.get("xml_parsed"):
699
+ result.update(
700
+ {
701
+ "xml_parsed": True,
702
+ "summary": parsed_data.get("summary"),
703
+ "findings": parsed_data.get("findings", []),
704
+ "checklist": parsed_data.get("checklist", []),
705
+ },
706
+ )
707
+
708
+ # Add formatted report for human readability
709
+ result["formatted_report"] = format_release_prep_report(result, input_data)
710
+
711
+ return (result, input_tokens, output_tokens)
712
+
713
+
714
+ def format_release_prep_report(result: dict, input_data: dict) -> str:
715
+ """Format release preparation output as a human-readable report.
716
+
717
+ Args:
718
+ result: The approve stage result
719
+ input_data: Input data from previous stages
720
+
721
+ Returns:
722
+ Formatted report string
723
+
724
+ """
725
+ lines = []
726
+
727
+ # Header with approval status
728
+ approved = result.get("approved", False)
729
+ confidence = result.get("confidence", "unknown").upper()
730
+
731
+ if approved:
732
+ status_icon = "✅"
733
+ status_text = "READY FOR RELEASE"
734
+ else:
735
+ status_icon = "❌"
736
+ status_text = "NOT READY"
737
+
738
+ lines.append("=" * 60)
739
+ lines.append("RELEASE PREPARATION REPORT")
740
+ lines.append("=" * 60)
741
+ lines.append("")
742
+ lines.append(f"Status: {status_icon} {status_text}")
743
+ lines.append(f"Confidence: {confidence}")
744
+ lines.append(f"Recommendation: {result.get('recommendation', 'N/A')}")
745
+ lines.append("")
746
+
747
+ # Health checks
748
+ health = input_data.get("health", {})
749
+ health_score = health.get("health_score", 0)
750
+ checks = health.get("checks", {})
751
+
752
+ lines.append("-" * 60)
753
+ lines.append("HEALTH CHECKS")
754
+ lines.append("-" * 60)
755
+ lines.append(f"Health Score: {health_score}/100")
756
+ lines.append("")
757
+
758
+ for check_name, check_data in checks.items():
759
+ passed = check_data.get("passed", False)
760
+ skipped = check_data.get("skipped", False)
761
+ tool = check_data.get("tool", "unknown")
762
+
763
+ if skipped:
764
+ icon = "⏭️"
765
+ status = "SKIPPED"
766
+ elif passed:
767
+ icon = "✅"
768
+ status = "PASSED"
769
+ else:
770
+ icon = "❌"
771
+ status = "FAILED"
772
+
773
+ errors = check_data.get("errors", 0)
774
+ extra = f" ({errors} errors)" if errors else ""
775
+ lines.append(f" {icon} {check_name.upper()} ({tool}): {status}{extra}")
776
+ lines.append("")
777
+
778
+ # Security summary
779
+ security = input_data.get("security", {})
780
+ if security:
781
+ lines.append("-" * 60)
782
+ lines.append("SECURITY SCAN")
783
+ lines.append("-" * 60)
784
+ total_issues = security.get("total_issues", 0)
785
+ high = security.get("high_severity", 0)
786
+ medium = security.get("medium_severity", 0)
787
+ passed = security.get("passed", True)
788
+
789
+ if passed:
790
+ lines.append("✅ No high severity issues found")
791
+ else:
792
+ lines.append(f"❌ {high} high severity issues found")
793
+
794
+ lines.append(f"Total Issues: {total_issues}")
795
+ lines.append(f" 🔴 High: {high}")
796
+ lines.append(f" 🟡 Medium: {medium}")
797
+ lines.append("")
798
+
799
+ # Changelog summary
800
+ changelog = input_data.get("changelog", {})
801
+ if changelog:
802
+ lines.append("-" * 60)
803
+ lines.append("CHANGELOG")
804
+ lines.append("-" * 60)
805
+ commit_count = changelog.get("total_commits", 0)
806
+ by_category = changelog.get("by_category", {})
807
+ period = changelog.get("period", "unknown")
808
+
809
+ lines.append(f"Period: {period}")
810
+ lines.append(f"Total Commits: {commit_count}")
811
+ if by_category:
812
+ lines.append("By Category:")
813
+ for cat, count in by_category.items():
814
+ lines.append(f" • {cat}: {count}")
815
+ lines.append("")
816
+
817
+ # Blockers
818
+ blockers = result.get("blockers", [])
819
+ if blockers:
820
+ lines.append("-" * 60)
821
+ lines.append("🚫 BLOCKERS")
822
+ lines.append("-" * 60)
823
+ for blocker in blockers:
824
+ lines.append(f" • {blocker}")
825
+ lines.append("")
826
+
827
+ # Warnings
828
+ warnings = result.get("warnings", [])
829
+ if warnings:
830
+ lines.append("-" * 60)
831
+ lines.append("⚠️ WARNINGS")
832
+ lines.append("-" * 60)
833
+ for warning in warnings:
834
+ lines.append(f" • {warning}")
835
+ lines.append("")
836
+
837
+ # LLM Assessment
838
+ assessment = result.get("assessment", "")
839
+ if assessment and not assessment.startswith("[Simulated"):
840
+ lines.append("-" * 60)
841
+ lines.append("DETAILED ASSESSMENT")
842
+ lines.append("-" * 60)
843
+ if len(assessment) > 1500:
844
+ lines.append(assessment[:1500] + "...")
845
+ else:
846
+ lines.append(assessment)
847
+ lines.append("")
848
+
849
+ # Footer
850
+ lines.append("=" * 60)
851
+ model_tier = result.get("model_tier_used", "unknown")
852
+ lines.append(f"Assessed using {model_tier} tier model")
853
+ lines.append("=" * 60)
854
+
855
+ return "\n".join(lines)
856
+
857
+
858
+ def main():
859
+ """CLI entry point for release preparation workflow."""
860
+ import asyncio
861
+
862
+ async def run():
863
+ workflow = ReleasePreparationWorkflow()
864
+ result = await workflow.execute(path=".")
865
+
866
+ print("\nRelease Preparation Results")
867
+ print("=" * 50)
868
+ print(f"Provider: {result.provider}")
869
+ print(f"Success: {result.success}")
870
+
871
+ output = result.final_output
872
+ print(f"Approved: {output.get('approved', False)}")
873
+ print(f"Confidence: {output.get('confidence', 'N/A')}")
874
+
875
+ if output.get("blockers"):
876
+ print("\nBlockers:")
877
+ for b in output["blockers"]:
878
+ print(f" - {b}")
879
+
880
+ if output.get("warnings"):
881
+ print("\nWarnings:")
882
+ for w in output["warnings"]:
883
+ print(f" - {w}")
884
+
885
+ print("\nCost Report:")
886
+ print(f" Total Cost: ${result.cost_report.total_cost:.4f}")
887
+ savings = result.cost_report.savings
888
+ pct = result.cost_report.savings_percent
889
+ print(f" Savings: ${savings:.4f} ({pct:.1f}%)")
890
+
891
+ asyncio.run(run())
892
+
893
+
894
+ if __name__ == "__main__":
895
+ main()