attune-ai 2.1.4__py3-none-any.whl → 2.2.0__py3-none-any.whl

attune/orchestration/meta_orchestrator.py

@@ -700,9 +700,7 @@ class MetaOrchestrator:
         except ImportError:
             logger.warning("AskUserQuestion not available, using defaults")
             suggested_pattern = self._choose_composition_pattern(requirements, suggested_agents)
-            return self.create_execution_plan(
-                requirements, suggested_agents, suggested_pattern
-            )
+            return self.create_execution_plan(requirements, suggested_agents, suggested_pattern)
 
         # Present all patterns with descriptions
         pattern_response = AskUserQuestion(

attune/project_index/scanner.py

@@ -483,9 +483,7 @@ class ProjectScanner:
                 import re
 
                 test_func_pattern = re.compile(r"^\s*def\s+test_\w+\(")
-                metrics["test_count"] = sum(
-                    1 for line in lines if test_func_pattern.match(line)
-                )
+                metrics["test_count"] = sum(1 for line in lines if test_func_pattern.match(line))
                 # Mark as having test functions (for test file records)
                 if metrics["test_count"] > 0:
                     metrics["lines_of_test"] = metrics["lines_of_code"]

attune/project_index/scanner_parallel.py

@@ -253,9 +253,11 @@ def compare_sequential_vs_parallel(project_root: str = ".", workers: int = 4) ->
         "sequential_time": sequential_time,
         "parallel_time": parallel_time,
         "speedup": speedup,
-        "improvement_pct": ((sequential_time - parallel_time) / sequential_time * 100)
-        if sequential_time > 0
-        else 0,
+        "improvement_pct": (
+            ((sequential_time - parallel_time) / sequential_time * 100)
+            if sequential_time > 0
+            else 0
+        ),
         "files_scanned": summary_seq.total_files,
         "workers": workers,
     }
@@ -281,9 +283,9 @@ if __name__ == "__main__":
     print(f"\nSpeedup: {results['speedup']:.2f}x")
     print(f"Improvement: {results['improvement_pct']:.1f}%")
 
-    if results['speedup'] >= 2.0:
+    if results["speedup"] >= 2.0:
         print("\n✅ Parallel processing is highly effective!")
-    elif results['speedup'] >= 1.5:
+    elif results["speedup"] >= 1.5:
         print("\n✅ Parallel processing provides moderate benefit")
     else:
         print("\n⚠️  Parallel processing may not be worth the overhead")

attune/socratic_router.py

@@ -720,9 +720,7 @@ class AttuneRouter:
             + (f", args='{args}'" if args else ""),
         }
 
-    def continue_session(
-        self, session_id: str, answers: dict[str, Any]
-    ) -> dict[str, Any]:
+    def continue_session(self, session_id: str, answers: dict[str, Any]) -> dict[str, Any]:
         """Continue a deep discovery session.
 
         Args:

attune/telemetry/agent_coordination.py

@@ -71,7 +71,11 @@ class CoordinationSignal:
             "source_agent": self.source_agent,
             "target_agent": self.target_agent,
             "payload": self.payload,
-            "timestamp": self.timestamp.isoformat() if isinstance(self.timestamp, datetime) else self.timestamp,
+            "timestamp": (
+                self.timestamp.isoformat()
+                if isinstance(self.timestamp, datetime)
+                else self.timestamp
+            ),
             "ttl_seconds": self.ttl_seconds,
         }
 
@@ -303,7 +307,9 @@ class CoordinationSignals:
 
         while time.time() - start_time < timeout:
             # Check for signal
-            signal = self.check_signal(signal_type=signal_type, source_agent=source_agent, consume=True)
+            signal = self.check_signal(
+                signal_type=signal_type, source_agent=source_agent, consume=True
+            )
 
             if signal:
                 return signal
@@ -335,7 +341,7 @@ class CoordinationSignals:
             # Check broadcast signals: empathy:signal:*:{type}:*
             patterns = [
                 f"{self.KEY_PREFIX}{self.agent_id}:{signal_type}:*",
-                f"{self.KEY_PREFIX}{self.BROADCAST_TARGET}:{signal_type}:*"
+                f"{self.KEY_PREFIX}{self.BROADCAST_TARGET}:{signal_type}:*",
             ]
 
             for pattern in patterns:

attune/telemetry/agent_tracking.py

@@ -62,7 +62,11 @@ class AgentHeartbeat:
             "status": self.status,
             "progress": self.progress,
             "current_task": self.current_task,
-            "last_beat": self.last_beat.isoformat() if isinstance(self.last_beat, datetime) else self.last_beat,
+            "last_beat": (
+                self.last_beat.isoformat()
+                if isinstance(self.last_beat, datetime)
+                else self.last_beat
+            ),
             "metadata": self.metadata,
             "display_name": self.display_name,
         }
@@ -210,7 +214,12 @@ class HeartbeatCoordinator:
         self.agent_id = None
 
     def _publish_heartbeat(
-        self, status: str, progress: float, current_task: str, metadata: dict[str, Any], display_name: str | None = None
+        self,
+        status: str,
+        progress: float,
+        current_task: str,
+        metadata: dict[str, Any],
+        display_name: str | None = None,
     ) -> None:
         """Publish heartbeat to Redis with TTL and optionally to event stream."""
         if not self.memory or not self.agent_id:
@@ -361,7 +370,11 @@ class HeartbeatCoordinator:
 
         for agent in active:
             time_since_beat = (now - agent.last_beat).total_seconds()
-            if time_since_beat > threshold_seconds and agent.status not in ("completed", "failed", "cancelled"):
+            if time_since_beat > threshold_seconds and agent.status not in (
+                "completed",
+                "failed",
+                "cancelled",
+            ):
                 stale.append(agent)
 
         return stale

attune/telemetry/approval_gates.py

@@ -71,7 +71,11 @@ class ApprovalRequest:
             "approval_type": self.approval_type,
             "agent_id": self.agent_id,
             "context": self.context,
-            "timestamp": self.timestamp.isoformat() if isinstance(self.timestamp, datetime) else self.timestamp,
+            "timestamp": (
+                self.timestamp.isoformat()
+                if isinstance(self.timestamp, datetime)
+                else self.timestamp
+            ),
             "timeout_seconds": self.timeout_seconds,
             "status": self.status,
         }
@@ -116,7 +120,11 @@ class ApprovalResponse:
             "approved": self.approved,
             "responder": self.responder,
             "reason": self.reason,
-            "timestamp": self.timestamp.isoformat() if isinstance(self.timestamp, datetime) else self.timestamp,
+            "timestamp": (
+                self.timestamp.isoformat()
+                if isinstance(self.timestamp, datetime)
+                else self.timestamp
+            ),
         }
 
     @classmethod
@@ -240,13 +248,18 @@ class ApprovalGate:
             if hasattr(self.memory, "_client") and self.memory._client:
                 import json
 
-                self.memory._client.setex(request_key, int(timeout) + 60, json.dumps(request.to_dict()))
+                self.memory._client.setex(
+                    request_key, int(timeout) + 60, json.dumps(request.to_dict())
+                )
             else:
                 logger.warning("Cannot store approval request: no Redis backend available")
         except Exception as e:
             logger.error(f"Failed to store approval request: {e}")
             return ApprovalResponse(
-                request_id=request_id, approved=False, responder="system", reason=f"Storage error: {e}"
+                request_id=request_id,
+                approved=False,
+                responder="system",
+                reason=f"Storage error: {e}",
             )
 
         # Send approval_request signal (for notifications)
@@ -363,7 +376,11 @@ class ApprovalGate:
             return False
 
         response = ApprovalResponse(
-            request_id=request_id, approved=approved, responder=responder, reason=reason, timestamp=datetime.utcnow()
+            request_id=request_id,
+            approved=approved,
+            responder=responder,
+            reason=reason,
+            timestamp=datetime.utcnow(),
         )
 
         # Store approval response (for workflow to retrieve)

attune/telemetry/cli.py

@@ -25,6 +25,8 @@ except ImportError:
 
 from attune.config import _validate_file_path
 
+# Import dashboard commands for backward compatibility (re-exported)
+from .commands import cmd_file_test_dashboard, cmd_telemetry_dashboard  # noqa: F401
 from .usage_tracker import UsageTracker
 
 # _validate_file_path is now imported from attune.config
@@ -597,7 +599,6 @@ def cmd_telemetry_export(args: Any) -> int:
 # ==============================================================================
 
 
-
 # ==============================================================================
 # Dashboard Commands (Extracted to Separate Module)
 # ==============================================================================
@@ -606,6 +607,7 @@ def cmd_telemetry_export(args: Any) -> int:
 # Imported at top of file for backward compatibility.
 # ==============================================================================
 
+
 def cmd_tier1_status(args: Any) -> int:
     """Show comprehensive Tier 1 automation status.
 
@@ -1227,5 +1229,3 @@ def cmd_file_test_status(args: Any) -> int:
                     print(f"    - {test.get('name', 'unknown')}: {test.get('error', '')[:40]}")
 
     return 0
-
-

attune/telemetry/commands/dashboard_commands.py

@@ -359,7 +359,8 @@ def cmd_file_test_dashboard(args: Any) -> int:
             </tr>
             """
 
-        return """<!DOCTYPE html>
+        return (
+            """<!DOCTYPE html>
 <html lang="en">
 <head>
     <meta charset="UTF-8">
@@ -515,23 +516,33 @@ def cmd_file_test_dashboard(args: Any) -> int:
 
         <div class="stats">
             <div class="stat-card total">
-                <div class="stat-value">""" + str(total) + """</div>
+                <div class="stat-value">"""
+            + str(total)
+            + """</div>
                 <div class="stat-label">Total Files</div>
             </div>
             <div class="stat-card passed">
-                <div class="stat-value">""" + str(passed) + """</div>
+                <div class="stat-value">"""
+            + str(passed)
+            + """</div>
                 <div class="stat-label">Passed</div>
             </div>
             <div class="stat-card failed">
-                <div class="stat-value">""" + str(failed) + """</div>
+                <div class="stat-value">"""
+            + str(failed)
+            + """</div>
                 <div class="stat-label">Failed</div>
             </div>
             <div class="stat-card no-tests">
-                <div class="stat-value">""" + str(no_tests) + """</div>
+                <div class="stat-value">"""
+            + str(no_tests)
+            + """</div>
                 <div class="stat-label">No Tests</div>
             </div>
             <div class="stat-card stale">
-                <div class="stat-value">""" + str(stale) + """</div>
+                <div class="stat-value">"""
+            + str(stale)
+            + """</div>
                 <div class="stat-label">Stale</div>
             </div>
         </div>
@@ -558,12 +569,16 @@ def cmd_file_test_dashboard(args: Any) -> int:
                 </tr>
             </thead>
             <tbody>
-                """ + rows_html + """
+                """
+            + rows_html
+            + """
             </tbody>
         </table>
 
         <div class="last-updated">
-            Last updated: """ + datetime.now().strftime("%Y-%m-%d %H:%M:%S") + """
+            Last updated: """
+            + datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            + """
         </div>
     </div>
 
@@ -618,6 +633,7 @@ def cmd_file_test_dashboard(args: Any) -> int:
     </script>
 </body>
 </html>"""
+        )
 
     def _generate_empty_dashboard() -> str:
         """Generate dashboard HTML when no data available."""

attune/telemetry/event_streaming.py

@@ -55,7 +55,11 @@ class StreamEvent:
         return {
             "event_id": self.event_id,
             "event_type": self.event_type,
-            "timestamp": self.timestamp.isoformat() if isinstance(self.timestamp, datetime) else self.timestamp,
+            "timestamp": (
+                self.timestamp.isoformat()
+                if isinstance(self.timestamp, datetime)
+                else self.timestamp
+            ),
             "data": self.data,
             "source": self.source,
         }
@@ -235,7 +239,9 @@ class EventStreamer:
         else:
             # Subscribe to all event streams (expensive - requires KEYS scan)
             all_streams = self.memory._client.keys(f"{self.STREAM_PREFIX}*")
-            streams = {s.decode("utf-8") if isinstance(s, bytes) else s: start_id for s in all_streams}
+            streams = {
+                s.decode("utf-8") if isinstance(s, bytes) else s: start_id for s in all_streams
+            }
 
         if not streams:
             logger.debug("No streams to consume")

attune/telemetry/feedback_loop.py

@@ -80,7 +80,11 @@ class FeedbackEntry:
             "stage_name": self.stage_name,
             "tier": self.tier,
             "quality_score": self.quality_score,
-            "timestamp": self.timestamp.isoformat() if isinstance(self.timestamp, datetime) else self.timestamp,
+            "timestamp": (
+                self.timestamp.isoformat()
+                if isinstance(self.timestamp, datetime)
+                else self.timestamp
+            ),
             "metadata": self.metadata,
         }
 
@@ -252,7 +256,11 @@ class FeedbackLoop:
         return feedback_id
 
     def get_feedback_history(
-        self, workflow_name: str, stage_name: str, tier: str | ModelTier | None = None, limit: int = 100
+        self,
+        workflow_name: str,
+        stage_name: str,
+        tier: str | ModelTier | None = None,
+        limit: int = 100,
     ) -> list[FeedbackEntry]:
         """Get feedback history for a workflow stage.
 

attune/tools.py

@@ -11,6 +11,7 @@ Integration with Claude Code:
 
 Created: 2026-01-29
 """
+
 import json
 import logging
 import os

attune/workflow_commands.py

@@ -427,9 +427,7 @@ def ship_workflow(
     print("4. Checking git status...")
     success, output = _run_command(["git", "status", "--porcelain"])
     if success:
-        staged = sum(
-            1 for line in output.split("\n") if line.startswith(("A ", "M ", "D ", "R "))
-        )
+        staged = sum(1 for line in output.split("\n") if line.startswith(("A ", "M ", "D ", "R ")))
         unstaged = sum(1 for line in output.split("\n") if line.startswith((" M", " D", "??")))
         if staged > 0:
             print(f"   INFO - {staged} staged, {unstaged} unstaged")

attune/workflows/__init__.py

@@ -64,7 +64,6 @@ if TYPE_CHECKING:
     from .security_audit import SecurityAuditWorkflow
     from .seo_optimization import SEOOptimizationWorkflow
     from .step_config import WorkflowStepConfig
-    from .test5 import Test5Workflow
     from .test_coverage_boost_crew import TestCoverageBoostCrew, TestCoverageBoostCrewResult
     from .test_gen import TestGenerationWorkflow
     from .test_gen_behavioral import BehavioralTestGenerationWorkflow
@@ -89,6 +88,10 @@ from .builder import WorkflowBuilder, workflow_builder
 # Config is small and frequently needed
 from .config import DEFAULT_MODELS, ModelConfig, WorkflowConfig, create_example_config, get_model
 
+# Mixins for workflow composition (extracted for maintainability)
+from .cost_mixin import CostTrackingMixin
+from .parsing_mixin import ResponseParsingMixin
+
 # Routing strategies (small, frequently needed for builder pattern)
 from .routing import (
     BalancedRouting,
@@ -136,12 +139,19 @@ _LAZY_WORKFLOW_IMPORTS: dict[str, tuple[str, str]] = {
     "SecureReleaseResult": (".secure_release", "SecureReleaseResult"),
     "SecurityAuditWorkflow": (".security_audit", "SecurityAuditWorkflow"),
     "SEOOptimizationWorkflow": (".seo_optimization", "SEOOptimizationWorkflow"),
-    "Test5Workflow": (".test5", "Test5Workflow"),
     "TestCoverageBoostCrew": (".test_coverage_boost_crew", "TestCoverageBoostCrew"),
     "TestCoverageBoostCrewResult": (".test_coverage_boost_crew", "TestCoverageBoostCrewResult"),
     "TestGenerationWorkflow": (".test_gen", "TestGenerationWorkflow"),
-    "BehavioralTestGenerationWorkflow": (".test_gen_behavioral", "BehavioralTestGenerationWorkflow"),
+    "BehavioralTestGenerationWorkflow": (
+        ".test_gen_behavioral",
+        "BehavioralTestGenerationWorkflow",
+    ),
     "ParallelTestGenerationWorkflow": (".test_gen_parallel", "ParallelTestGenerationWorkflow"),
+    # Additional workflows (restored to registry)
+    "TestMaintenanceWorkflow": (".test_maintenance", "TestMaintenanceWorkflow"),
+    "BatchProcessingWorkflow": (".batch_processing", "BatchProcessingWorkflow"),
+    "ProgressiveTestGenWorkflow": (".progressive.test_gen", "ProgressiveTestGenWorkflow"),
+    "AutonomousTestGenerator": (".autonomous_test_gen", "AutonomousTestGenerator"),
     "XMLAgent": (".xml_enhanced_crew", "XMLAgent"),
     "XMLTask": (".xml_enhanced_crew", "XMLTask"),
     "parse_xml_response": (".xml_enhanced_crew", "parse_xml_response"),
@@ -226,7 +236,7 @@ _DEFAULT_WORKFLOW_NAMES: dict[str, str] = {
     "refactor-plan": "RefactorPlanWorkflow",
     # Operational workflows
     "dependency-check": "DependencyCheckWorkflow",
-    "release-prep-legacy": "ReleasePreparationWorkflow",
+    # release-prep-legacy removed - handled by migration system (use release-prep instead)
     # Composite security pipeline (v3.0)
     "secure-release": "SecureReleasePipeline",
     # Code review crew integration (v3.1)
@@ -234,19 +244,26 @@ _DEFAULT_WORKFLOW_NAMES: dict[str, str] = {
     "pr-review": "PRReviewWorkflow",
     # Documentation management (v3.5)
     "doc-orchestrator": "DocumentationOrchestrator",
-    "manage-docs": "DocumentationOrchestrator",  # Points to orchestrator (crew deprecated)
+    # manage-docs removed - handled by migration system (was deprecated)
     # Keyboard Conductor (v3.6)
     "keyboard-shortcuts": "KeyboardShortcutWorkflow",
-    # User-generated workflows
+    # User-generated workflows (document-manager kept for backward compat, handled by migration)
     "document-manager": "DocumentManagerWorkflow",
-    "test5": "Test5Workflow",
     # Meta-orchestration workflows (v4.0.0 - CANONICAL)
     "orchestrated-health-check": "OrchestratedHealthCheckWorkflow",
     "orchestrated-release-prep": "OrchestratedReleasePrepWorkflow",
     # Backward compatibility aliases (point to orchestrated versions)
     "release-prep": "OrchestratedReleasePrepWorkflow",
-    "orchestrated-health-check-experimental": "OrchestratedHealthCheckWorkflow",
-    "orchestrated-release-prep-experimental": "OrchestratedReleasePrepWorkflow",
+    # Experimental aliases removed (use production versions instead)
+    # Research and synthesis workflows
+    "research-synthesis": "ResearchSynthesisWorkflow",
+    # Test management workflows
+    "test-coverage-boost": "TestCoverageBoostCrew",
+    "test-maintenance": "TestMaintenanceWorkflow",
+    "autonomous-test-gen": "AutonomousTestGenerator",
+    # Batch and progressive workflows
+    "batch-processing": "BatchProcessingWorkflow",
+    "progressive-test-gen": "ProgressiveTestGenWorkflow",
 }
 
 # Opt-in workflows - class names for lazy loading
@@ -440,6 +457,18 @@ def __getattr__(name: str) -> object:
         _load_cli_commands()
         return globals().get(name)
 
+    # Handle migration module exports
+    _MIGRATION_EXPORTS = {
+        "resolve_workflow_migration",
+        "MigrationConfig",
+        "WORKFLOW_ALIASES",
+        "show_migration_tip",
+        "list_migrations",
+    }
+    if name in _MIGRATION_EXPORTS:
+        from attune.workflows import migration
+        return getattr(migration, name)
+
     raise AttributeError(f"module 'attune.workflows' has no attribute '{name}'")
 
 
@@ -448,8 +477,10 @@ __all__ = [
     "PROVIDER_MODELS",
     # Registry and discovery
     "WORKFLOW_REGISTRY",
-    # Base classes
+    # Base classes and mixins
     "BaseWorkflow",
+    "CostTrackingMixin",
+    "ResponseParsingMixin",
     # Routing strategies
     "TierRoutingStrategy",
     "RoutingContext",
@@ -532,8 +563,20 @@ __all__ = [
     "OrchestratedReleasePrepWorkflow",
     "HealthCheckReport",
     "ReleaseReadinessReport",
+    # Additional workflows (added for completeness)
+    # Test5Workflow removed - test artifact
+    "TestMaintenanceWorkflow",
+    "BatchProcessingWorkflow",
+    "ProgressiveTestGenWorkflow",
+    "AutonomousTestGenerator",
     # XML-enhanced prompting
     "XMLAgent",
     "XMLTask",
     "parse_xml_response",
+    # Workflow migration system
+    "resolve_workflow_migration",
+    "MigrationConfig",
+    "WORKFLOW_ALIASES",
+    "show_migration_tip",
+    "list_migrations",
 ]