empathy-framework 4.7.0__py3-none-any.whl → 4.8.0__py3-none-any.whl

empathy_os/vscode_bridge 2.py

@@ -0,0 +1,173 @@
+"""VS Code Extension Bridge
+
+Provides functions to write data that the VS Code extension can pick up.
+Enables Claude Code CLI output to appear in VS Code webview panels.
+
+Copyright 2026 Smart-AI-Memory
+Licensed under Fair Source License 0.9
+"""
+
+import json
+from dataclasses import asdict, dataclass
+from datetime import datetime
+from pathlib import Path
+from typing import Any
+
+
+@dataclass
+class ReviewFinding:
+    """A code review finding."""
+
+    id: str
+    file: str
+    line: int
+    severity: str  # 'critical' | 'high' | 'medium' | 'low' | 'info'
+    category: str  # 'security' | 'performance' | 'maintainability' | 'style' | 'correctness'
+    message: str
+    column: int = 1
+    details: str | None = None
+    recommendation: str | None = None
+
+
+@dataclass
+class CodeReviewResult:
+    """Code review results for VS Code bridge."""
+
+    findings: list[dict[str, Any]]
+    summary: dict[str, Any]
+    verdict: str  # 'approve' | 'approve_with_suggestions' | 'request_changes' | 'reject'
+    security_score: int
+    formatted_report: str
+    model_tier_used: str
+    timestamp: str
+
+
+def get_empathy_dir() -> Path:
+    """Get the .empathy directory, creating if needed."""
+    empathy_dir = Path(".empathy")
+    empathy_dir.mkdir(exist_ok=True)
+    return empathy_dir
+
+
+def write_code_review_results(
+    findings: list[dict[str, Any]] | None = None,
+    summary: dict[str, Any] | None = None,
+    verdict: str = "approve_with_suggestions",
+    security_score: int = 85,
+    formatted_report: str = "",
+    model_tier_used: str = "capable",
+) -> Path:
+    """Write code review results for VS Code extension to pick up.
+
+    Args:
+        findings: List of finding dicts with keys: id, file, line, severity, category, message
+        summary: Summary dict with keys: total_findings, by_severity, by_category, files_affected
+        verdict: One of 'approve', 'approve_with_suggestions', 'request_changes', 'reject'
+        security_score: 0-100 score
+        formatted_report: Markdown formatted report
+        model_tier_used: 'cheap', 'capable', or 'premium'
+
+    Returns:
+        Path to the written file
+    """
+    findings = findings or []
+
+    # Build summary if not provided
+    if summary is None:
+        by_severity: dict[str, int] = {}
+        by_category: dict[str, int] = {}
+        files_affected: set[str] = set()
+
+        for f in findings:
+            sev = f.get("severity", "info")
+            cat = f.get("category", "correctness")
+            by_severity[sev] = by_severity.get(sev, 0) + 1
+            by_category[cat] = by_category.get(cat, 0) + 1
+            if f.get("file"):
+                files_affected.add(f["file"])
+
+        summary = {
+            "total_findings": len(findings),
+            "by_severity": by_severity,
+            "by_category": by_category,
+            "files_affected": list(files_affected),
+        }
+
+    result = CodeReviewResult(
+        findings=findings,
+        summary=summary,
+        verdict=verdict,
+        security_score=security_score,
+        formatted_report=formatted_report,
+        model_tier_used=model_tier_used,
+        timestamp=datetime.now().isoformat(),
+    )
+
+    output_path = get_empathy_dir() / "code-review-results.json"
+
+    with open(output_path, "w") as f:
+        json.dump(asdict(result), f, indent=2)
+
+    return output_path
+
+
+def write_pr_review_results(
+    pr_number: int | str,
+    title: str,
+    findings: list[dict[str, Any]],
+    verdict: str = "approve_with_suggestions",
+    summary_text: str = "",
+) -> Path:
+    """Write PR review results for VS Code extension.
+
+    Convenience wrapper for PR reviews from GitHub.
+
+    Args:
+        pr_number: The PR number
+        title: PR title
+        findings: List of review findings
+        verdict: Review verdict
+        summary_text: Summary of the review
+
+    Returns:
+        Path to the written file
+    """
+    formatted_report = f"""## PR #{pr_number}: {title}
+
+{summary_text}
+
+### Findings ({len(findings)})
+
+"""
+    for f in findings:
+        formatted_report += f"- **{f.get('severity', 'info').upper()}** [{f.get('file', 'unknown')}:{f.get('line', 0)}]: {f.get('message', '')}\n"
+
+    return write_code_review_results(
+        findings=findings,
+        verdict=verdict,
+        formatted_report=formatted_report,
+        model_tier_used="capable",
+    )
+
+
+# Quick helper for Claude Code to call
+def send_to_vscode(
+    message: str,
+    findings: list[dict[str, Any]] | None = None,
+    verdict: str = "approve_with_suggestions",
+) -> str:
+    """Quick helper to send review results to VS Code.
+
+    Usage in Claude Code:
+        from empathy_os.vscode_bridge import send_to_vscode
+        send_to_vscode("Review complete", findings=[...])
+
+    Returns:
+        Confirmation message
+    """
+    path = write_code_review_results(
+        findings=findings or [],
+        formatted_report=message,
+        verdict=verdict,
+    )
+    return f"Results written to {path} - VS Code will update automatically"

empathy_os/workflows/__init__.py

@@ -80,8 +80,20 @@ from .base import (
     get_workflow_stats,
 )
 
+# Builder pattern for workflow construction
+from .builder import WorkflowBuilder, workflow_builder
+
 # Config is small and frequently needed
 from .config import DEFAULT_MODELS, ModelConfig, WorkflowConfig, create_example_config, get_model
+
+# Routing strategies (small, frequently needed for builder pattern)
+from .routing import (
+    BalancedRouting,
+    CostOptimizedRouting,
+    PerformanceOptimizedRouting,
+    RoutingContext,
+    TierRoutingStrategy,
+)
 from .step_config import WorkflowStepConfig, steps_from_tier_map, validate_step_config
 
 # Lazy import mapping for workflow classes
@@ -99,9 +111,15 @@ _LAZY_WORKFLOW_IMPORTS: dict[str, tuple[str, str]] = {
     "KeyboardShortcutWorkflow": (".keyboard_shortcuts", "KeyboardShortcutWorkflow"),
     "ManageDocumentationCrew": (".manage_documentation", "ManageDocumentationCrew"),
     "ManageDocumentationCrewResult": (".manage_documentation", "ManageDocumentationCrewResult"),
-    "OrchestratedHealthCheckWorkflow": (".orchestrated_health_check", "OrchestratedHealthCheckWorkflow"),
+    "OrchestratedHealthCheckWorkflow": (
+        ".orchestrated_health_check",
+        "OrchestratedHealthCheckWorkflow",
+    ),
     "HealthCheckReport": (".orchestrated_health_check", "HealthCheckReport"),
-    "OrchestratedReleasePrepWorkflow": (".orchestrated_release_prep", "OrchestratedReleasePrepWorkflow"),
+    "OrchestratedReleasePrepWorkflow": (
+        ".orchestrated_release_prep",
+        "OrchestratedReleasePrepWorkflow",
+    ),
     "ReleaseReadinessReport": (".orchestrated_release_prep", "ReleaseReadinessReport"),
     "PerformanceAuditWorkflow": (".perf_audit", "PerformanceAuditWorkflow"),
     "PRReviewWorkflow": (".pr_review", "PRReviewWorkflow"),
@@ -141,6 +159,7 @@ def _lazy_import_workflow(name: str) -> object:
 
     # Import the module and get the attribute
     import importlib
+
     module = importlib.import_module(module_path, package="empathy_os.workflows")
     attr = getattr(module, attr_name)
 
@@ -148,6 +167,7 @@ def _lazy_import_workflow(name: str) -> object:
     _loaded_workflow_modules[cache_key] = attr
     return attr
 
+
 # Re-export CLI commands from workflow_commands.py (lazy loaded)
 _parent_dir = os.path.dirname(os.path.dirname(__file__))
 _workflows_module_path = os.path.join(_parent_dir, "workflow_commands.py")
@@ -420,6 +440,15 @@ __all__ = [
     "WORKFLOW_REGISTRY",
     # Base classes
     "BaseWorkflow",
+    # Routing strategies
+    "TierRoutingStrategy",
+    "RoutingContext",
+    "CostOptimizedRouting",
+    "PerformanceOptimizedRouting",
+    "BalancedRouting",
+    # Builder pattern
+    "WorkflowBuilder",
+    "workflow_builder",
     # New high-value workflows
     "BugPredictionWorkflow",
     # Code review crew integration (v3.1)