empathy-framework 3.9.3__py3-none-any.whl → 3.10.1__py3-none-any.whl

empathy_os/workflows/health_check.py

@@ -87,6 +87,7 @@ class HealthCheckWorkflow(BaseWorkflow):
         check_tests: bool = True,
         check_deps: bool = True,
         xml_prompts: bool = True,
+        health_score_threshold: int = 95,
         **kwargs: Any,
     ):
         """Initialize health check workflow.
@@ -98,6 +99,8 @@ class HealthCheckWorkflow(BaseWorkflow):
             check_tests: Run test checks
             check_deps: Run dependency checks
             xml_prompts: Use XML-enhanced prompts
+            health_score_threshold: Minimum health score required (0-100, default: 95)
+                                   100 = perfect health, 95 = very strict (default), 80 = moderate
             **kwargs: Additional arguments passed to BaseWorkflow
 
         """
@@ -108,6 +111,7 @@ class HealthCheckWorkflow(BaseWorkflow):
         self.check_tests = check_tests
         self.check_deps = check_deps
         self.xml_prompts = xml_prompts
+        self.health_score_threshold = health_score_threshold
         self._crew: Any = None
         self._crew_available = False
 
@@ -153,6 +157,39 @@ class HealthCheckWorkflow(BaseWorkflow):
             return await self._fix(input_data, tier)
         raise ValueError(f"Unknown stage: {stage_name}")
 
+    def validate_output(self, stage_output: dict) -> tuple[bool, str | None]:
+        """Validate health check output quality.
+
+        For health-check workflow, we validate that:
+        1. Diagnosis data is present
+        2. Health score meets the configured threshold (default: 95 = very strict quality)
+        3. No critical execution errors occurred
+
+        Args:
+            stage_output: Output from diagnose or fix stage
+
+        Returns:
+            Tuple of (is_valid, failure_reason)
+
+        """
+        # First run parent validation (checks for empty output, errors)
+        is_valid, reason = super().validate_output(stage_output)
+        if not is_valid:
+            return False, reason
+
+        # Check diagnosis data exists
+        diagnosis = stage_output.get("diagnosis", {})
+        if not diagnosis:
+            return False, "diagnosis_missing"
+
+        # Check health score meets configured threshold
+        health_score = diagnosis.get("health_score", 0)
+        if health_score < self.health_score_threshold:
+            return False, "health_score_low"
+
+        # All validation passed
+        return True, None
+
     async def _diagnose(self, input_data: dict, tier: ModelTier) -> tuple[dict, int, int]:
         """Run health diagnosis using HealthCheckCrew.
 

empathy_os/workflows/new_sample_workflow1.py

@@ -80,7 +80,7 @@ class NewSampleWorkflow1Workflow(BaseWorkflow):
 
         """
         # TODO: Implement analyze logic
-        prompt = f"{stage} stage: {input_data}"
+        prompt = f"analyze stage: {input_data}"
 
         if self.executor:
             result = await self.executor.execute(
@@ -107,7 +107,7 @@ class NewSampleWorkflow1Workflow(BaseWorkflow):
 
         """
         # TODO: Implement process logic
-        prompt = f"{stage} stage: {input_data}"
+        prompt = f"process stage: {input_data}"
 
         if self.executor:
             result = await self.executor.execute(
@@ -134,7 +134,7 @@ class NewSampleWorkflow1Workflow(BaseWorkflow):
 
         """
         # TODO: Implement report logic
-        prompt = f"{stage} stage: {input_data}"
+        prompt = f"report stage: {input_data}"
 
         if self.executor:
             result = await self.executor.execute(