empathy-framework 5.0.3__py3-none-any.whl → 5.1.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

@@ -66,6 +66,8 @@ if TYPE_CHECKING:
     from .test5 import Test5Workflow
     from .test_coverage_boost_crew import TestCoverageBoostCrew, TestCoverageBoostCrewResult
     from .test_gen import TestGenerationWorkflow
+    from .test_gen_behavioral import BehavioralTestGenerationWorkflow
+    from .test_gen_parallel import ParallelTestGenerationWorkflow
     from .xml_enhanced_crew import XMLAgent, XMLTask
 
 # Only import base module eagerly (small, needed for type checks)
@@ -136,6 +138,8 @@ _LAZY_WORKFLOW_IMPORTS: dict[str, tuple[str, str]] = {
     "TestCoverageBoostCrew": (".test_coverage_boost_crew", "TestCoverageBoostCrew"),
     "TestCoverageBoostCrewResult": (".test_coverage_boost_crew", "TestCoverageBoostCrewResult"),
     "TestGenerationWorkflow": (".test_gen", "TestGenerationWorkflow"),
+    "BehavioralTestGenerationWorkflow": (".test_gen_behavioral", "BehavioralTestGenerationWorkflow"),
+    "ParallelTestGenerationWorkflow": (".test_gen_parallel", "ParallelTestGenerationWorkflow"),
     "XMLAgent": (".xml_enhanced_crew", "XMLAgent"),
     "XMLTask": (".xml_enhanced_crew", "XMLTask"),
     "parse_xml_response": (".xml_enhanced_crew", "parse_xml_response"),
@@ -213,6 +217,8 @@ _DEFAULT_WORKFLOW_NAMES: dict[str, str] = {
     "perf-audit": "PerformanceAuditWorkflow",
     # Generation workflows
     "test-gen": "TestGenerationWorkflow",
+    "test-gen-behavioral": "BehavioralTestGenerationWorkflow",
+    "test-gen-parallel": "ParallelTestGenerationWorkflow",
     "refactor-plan": "RefactorPlanWorkflow",
     # Operational workflows
     "dependency-check": "DependencyCheckWorkflow",
@@ -484,6 +490,8 @@ __all__ = [
     "SecureReleaseResult",
     "SecurityAuditWorkflow",
     "TestGenerationWorkflow",
+    "BehavioralTestGenerationWorkflow",
+    "ParallelTestGenerationWorkflow",
     # Configuration
     "WorkflowConfig",
     "WorkflowResult",