tapps-agents 3.5.40__py3-none-any.whl → 3.6.0__py3-none-any.whl

tapps_agents/health/registry.py

@@ -1,166 +1,166 @@
-"""
-Health Check Registry.
-
-Manages registration and discovery of health checks.
-"""
-
-from __future__ import annotations
-
-import logging
-from typing import Any
-
-from .base import HealthCheck, HealthCheckResult
-
-logger = logging.getLogger(__name__)
-
-
-class HealthCheckRegistry:
-    """Registry for health checks."""
-
-    def __init__(self):
-        """Initialize health check registry."""
-        self._checks: dict[str, HealthCheck] = {}
-
-    def register(self, check: HealthCheck) -> None:
-        """
-        Register a health check.
-
-        Args:
-            check: Health check instance to register
-
-        Raises:
-            ValueError: If check with same name already registered
-        """
-        if check.name in self._checks:
-            raise ValueError(f"Health check '{check.name}' is already registered")
-
-        self._checks[check.name] = check
-        logger.debug(f"Registered health check: {check.name}")
-
-    def get(self, name: str) -> HealthCheck | None:
-        """
-        Get a health check by name.
-
-        Args:
-            name: Name of the health check
-
-        Returns:
-            HealthCheck instance or None if not found
-        """
-        return self._checks.get(name)
-
-    def get_all(self) -> dict[str, HealthCheck]:
-        """
-        Get all registered health checks.
-
-        Returns:
-            Dictionary mapping check names to HealthCheck instances
-        """
-        return self._checks.copy()
-
-    def list_names(self) -> list[str]:
-        """
-        List all registered health check names.
-
-        Returns:
-            List of health check names
-        """
-        return list(self._checks.keys())
-
-    def run(self, name: str) -> HealthCheckResult | None:
-        """
-        Run a specific health check.
-
-        Args:
-            name: Name of the health check to run
-
-        Returns:
-            HealthCheckResult or None if check not found
-        """
-        check = self.get(name)
-        if not check:
-            logger.warning(f"Health check '{name}' not found")
-            return None
-
-        try:
-            return check.run()
-        except Exception as e:
-            logger.error(f"Error running health check '{name}': {e}", exc_info=True)
-            return HealthCheckResult(
-                name=name,
-                status="unhealthy",
-                score=0.0,
-                message=f"Health check failed with error: {e}",
-                details={"error": str(e)},
-            )
-
-    def run_all(self, check_names: list[str] | None = None) -> dict[str, HealthCheckResult]:
-        """
-        Run all registered health checks (or specified subset).
-
-        Args:
-            check_names: Optional list of check names to run. If None, runs all.
-
-        Returns:
-            Dictionary mapping check names to HealthCheckResult instances
-        """
-        if check_names is None:
-            check_names = self.list_names()
-
-        results: dict[str, HealthCheckResult] = {}
-
-        # Build dependency graph and determine execution order
-        execution_order = self._topological_sort(check_names)
-
-        for name in execution_order:
-            if name not in check_names:
-                continue
-            results[name] = self.run(name)
-
-        return results
-
-    def _topological_sort(self, check_names: list[str]) -> list[str]:
-        """
-        Topologically sort health checks based on dependencies.
-
-        Args:
-            check_names: List of check names to sort
-
-        Returns:
-            Sorted list of check names respecting dependencies
-        """
-        # Build dependency graph
-        graph: dict[str, set[str]] = {}
-        for name in check_names:
-            check = self.get(name)
-            if check:
-                graph[name] = set(check.get_dependencies())
-
-        # Topological sort using Kahn's algorithm
-        in_degree: dict[str, int] = {name: 0 for name in check_names}
-        for name in check_names:
-            for dep in graph.get(name, set()):
-                if dep in in_degree:
-                    in_degree[name] += 1
-
-        queue = [name for name, degree in in_degree.items() if degree == 0]
-        result = []
-
-        while queue:
-            node = queue.pop(0)
-            result.append(node)
-
-            # Reduce in-degree for dependent nodes
-            for name in check_names:
-                if node in graph.get(name, set()):
-                    in_degree[name] -= 1
-                    if in_degree[name] == 0:
-                        queue.append(name)
-
-        # Add any remaining nodes (shouldn't happen if no cycles)
-        for name in check_names:
-            if name not in result:
-                result.append(name)
-
-        return result
-
+"""
+Health Check Registry.
+
+Manages registration and discovery of health checks.
+"""
+
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from .base import HealthCheck, HealthCheckResult
+
+logger = logging.getLogger(__name__)
+
+
+class HealthCheckRegistry:
+    """Registry for health checks."""
+
+    def __init__(self):
+        """Initialize health check registry."""
+        self._checks: dict[str, HealthCheck] = {}
+
+    def register(self, check: HealthCheck) -> None:
+        """
+        Register a health check.
+
+        Args:
+            check: Health check instance to register
+
+        Raises:
+            ValueError: If check with same name already registered
+        """
+        if check.name in self._checks:
+            raise ValueError(f"Health check '{check.name}' is already registered")
+
+        self._checks[check.name] = check
+        logger.debug(f"Registered health check: {check.name}")
+
+    def get(self, name: str) -> HealthCheck | None:
+        """
+        Get a health check by name.
+
+        Args:
+            name: Name of the health check
+
+        Returns:
+            HealthCheck instance or None if not found
+        """
+        return self._checks.get(name)
+
+    def get_all(self) -> dict[str, HealthCheck]:
+        """
+        Get all registered health checks.
+
+        Returns:
+            Dictionary mapping check names to HealthCheck instances
+        """
+        return self._checks.copy()
+
+    def list_names(self) -> list[str]:
+        """
+        List all registered health check names.
+
+        Returns:
+            List of health check names
+        """
+        return list(self._checks.keys())
+
+    def run(self, name: str) -> HealthCheckResult | None:
+        """
+        Run a specific health check.
+
+        Args:
+            name: Name of the health check to run
+
+        Returns:
+            HealthCheckResult or None if check not found
+        """
+        check = self.get(name)
+        if not check:
+            logger.warning(f"Health check '{name}' not found")
+            return None
+
+        try:
+            return check.run()
+        except Exception as e:
+            logger.error(f"Error running health check '{name}': {e}", exc_info=True)
+            return HealthCheckResult(
+                name=name,
+                status="unhealthy",
+                score=0.0,
+                message=f"Health check failed with error: {e}",
+                details={"error": str(e)},
+            )
+
+    def run_all(self, check_names: list[str] | None = None) -> dict[str, HealthCheckResult]:
+        """
+        Run all registered health checks (or specified subset).
+
+        Args:
+            check_names: Optional list of check names to run. If None, runs all.
+
+        Returns:
+            Dictionary mapping check names to HealthCheckResult instances
+        """
+        if check_names is None:
+            check_names = self.list_names()
+
+        results: dict[str, HealthCheckResult] = {}
+
+        # Build dependency graph and determine execution order
+        execution_order = self._topological_sort(check_names)
+
+        for name in execution_order:
+            if name not in check_names:
+                continue
+            results[name] = self.run(name)
+
+        return results
+
+    def _topological_sort(self, check_names: list[str]) -> list[str]:
+        """
+        Topologically sort health checks based on dependencies.
+
+        Args:
+            check_names: List of check names to sort
+
+        Returns:
+            Sorted list of check names respecting dependencies
+        """
+        # Build dependency graph
+        graph: dict[str, set[str]] = {}
+        for name in check_names:
+            check = self.get(name)
+            if check:
+                graph[name] = set(check.get_dependencies())
+
+        # Topological sort using Kahn's algorithm
+        in_degree: dict[str, int] = {name: 0 for name in check_names}
+        for name in check_names:
+            for dep in graph.get(name, set()):
+                if dep in in_degree:
+                    in_degree[name] += 1
+
+        queue = [name for name, degree in in_degree.items() if degree == 0]
+        result = []
+
+        while queue:
+            node = queue.pop(0)
+            result.append(node)
+
+            # Reduce in-degree for dependent nodes
+            for name in check_names:
+                if node in graph.get(name, set()):
+                    in_degree[name] -= 1
+                    if in_degree[name] == 0:
+                        queue.append(name)
+
+        # Add any remaining nodes (shouldn't happen if no cycles)
+        for name in check_names:
+            if name not in result:
+                result.append(name)
+
+        return result
+

tapps_agents/hooks/__init__.py

@@ -1,33 +1,33 @@
-"""
-Hook system for TappsCodingAgents.
-
-Provides event-driven automation: UserPromptSubmit, PostToolUse, SessionStart,
-SessionEnd, WorkflowComplete. Configuration via .tapps-agents/hooks.yaml.
-"""
-
-from .config import HookDefinition, HooksConfig, load_hooks_config
-from .executor import HookResult, run_hook
-from .events import (
-    HookEventType,
-    PostToolUseEvent,
-    SessionEndEvent,
-    SessionStartEvent,
-    UserPromptSubmitEvent,
-    WorkflowCompleteEvent,
-)
-from .manager import HookManager
-
-__all__ = [
-    "HookDefinition",
-    "HookManager",
-    "HooksConfig",
-    "HookResult",
-    "load_hooks_config",
-    "run_hook",
-    "HookEventType",
-    "PostToolUseEvent",
-    "SessionEndEvent",
-    "SessionStartEvent",
-    "UserPromptSubmitEvent",
-    "WorkflowCompleteEvent",
-]
+"""
+Hook system for TappsCodingAgents.
+
+Provides event-driven automation: UserPromptSubmit, PostToolUse, SessionStart,
+SessionEnd, WorkflowComplete. Configuration via .tapps-agents/hooks.yaml.
+"""
+
+from .config import HookDefinition, HooksConfig, load_hooks_config
+from .executor import HookResult, run_hook
+from .events import (
+    HookEventType,
+    PostToolUseEvent,
+    SessionEndEvent,
+    SessionStartEvent,
+    UserPromptSubmitEvent,
+    WorkflowCompleteEvent,
+)
+from .manager import HookManager
+
+__all__ = [
+    "HookDefinition",
+    "HookManager",
+    "HooksConfig",
+    "HookResult",
+    "load_hooks_config",
+    "run_hook",
+    "HookEventType",
+    "PostToolUseEvent",
+    "SessionEndEvent",
+    "SessionStartEvent",
+    "UserPromptSubmitEvent",
+    "WorkflowCompleteEvent",
+]

tapps_agents/hooks/config.py

@@ -1,140 +1,140 @@
-"""
-Hook configuration schema and loader.
-
-Loads and validates .tapps-agents/hooks.yaml with event definitions,
-matchers, and commands. Supported events: UserPromptSubmit, PostToolUse,
-SessionStart, SessionEnd, WorkflowComplete.
-"""
-
-from __future__ import annotations
-
-from pathlib import Path
-
-import yaml
-from pydantic import BaseModel, Field, model_validator
-
-# Supported hook event types
-HOOK_EVENT_TYPES = frozenset({
-    "UserPromptSubmit",
-    "PostToolUse",
-    "SessionStart",
-    "SessionEnd",
-    "WorkflowComplete",
-})
-
-
-class HookDefinition(BaseModel):
-    """Single hook definition within an event."""
-
-    name: str = Field(..., min_length=1, description="Hook display name")
-    command: str = Field(..., min_length=1, description="Shell command to execute")
-    enabled: bool = Field(default=True, description="Whether hook is enabled")
-    matcher: str | None = Field(
-        default=None,
-        description="Tool name matcher for PostToolUse (e.g. 'Write|Edit')",
-    )
-    file_patterns: list[str] | None = Field(
-        default=None,
-        description="Glob patterns for file filtering (e.g. ['*.py'])",
-    )
-    fail_on_error: bool = Field(
-        default=False,
-        description="If true, non-zero exit fails workflow",
-    )
-
-    model_config = {"extra": "forbid"}
-
-
-class HooksConfig(BaseModel):
-    """
-    Root schema for hooks.yaml.
-
-    Top-level key is 'hooks'. Each event maps to a list of HookDefinition.
-    """
-
-    hooks: dict[str, list[HookDefinition]] = Field(
-        default_factory=dict,
-        description="Event name -> list of hook definitions",
-    )
-
-    model_config = {"extra": "forbid"}
-
-    @model_validator(mode="after")
-    def validate_event_names(self) -> "HooksConfig":
-        """Ensure only supported event names are used."""
-        for event_name in self.hooks:
-            if event_name not in HOOK_EVENT_TYPES:
-                raise ValueError(
-                    f"Unsupported hook event '{event_name}'. "
-                    f"Supported: {', '.join(sorted(HOOK_EVENT_TYPES))}"
-                )
-        return self
-
-
-def load_hooks_config(
-    config_path: Path | str | None = None,
-    project_root: Path | None = None,
-) -> HooksConfig:
-    """
-    Load and validate hooks.yaml.
-
-    Args:
-        config_path: Explicit path to hooks.yaml. If None, uses
-            project_root/.tapps-agents/hooks.yaml.
-        project_root: Project root (default: cwd). Used when config_path is None.
-
-    Returns:
-        Validated HooksConfig. Returns empty config (all events empty) when
-        file is missing (safe defaults).
-
-    Raises:
-        FileNotFoundError: If config_path is given and file doesn't exist.
-        yaml.YAMLError: On YAML parse errors (includes path/line in message).
-        ValueError: On schema validation failures.
-    """
-    project_root = project_root or Path.cwd()
-    if config_path is None:
-        config_path = project_root / ".tapps-agents" / "hooks.yaml"
-    else:
-        config_path = Path(config_path)
-        if not config_path.is_absolute():
-            config_path = project_root / config_path
-
-    if not config_path.exists():
-        return HooksConfig(hooks={})
-
-    try:
-        content = config_path.read_text(encoding="utf-8")
-    except OSError as e:
-        raise FileNotFoundError(
-            f"Cannot read hooks config from {config_path}: {e}"
-        ) from e
-
-    try:
-        raw = yaml.safe_load(content)
-    except yaml.YAMLError as e:
-        path_hint = str(config_path)
-        if hasattr(e, "problem_mark") and e.problem_mark:
-            line = e.problem_mark.line + 1
-            col = e.problem_mark.column + 1
-            path_hint = f"{config_path}:{line}:{col}"
-        raise yaml.YAMLError(f"Invalid YAML in {path_hint}: {e}") from e
-
-    if raw is None:
-        return HooksConfig(hooks={})
-
-    if not isinstance(raw, dict):
-        raise ValueError(
-            f"hooks.yaml must be a YAML object (dict), got {type(raw).__name__} "
-            f"at {config_path}"
-        )
-
-    if "hooks" not in raw:
-        return HooksConfig(hooks={})
-
-    try:
-        return HooksConfig.model_validate(raw)
-    except Exception as e:
-        raise ValueError(
-            f"Hooks config validation failed at {config_path}: {e}"
-        ) from e
+"""
+Hook configuration schema and loader.
+
+Loads and validates .tapps-agents/hooks.yaml with event definitions,
+matchers, and commands. Supported events: UserPromptSubmit, PostToolUse,
+SessionStart, SessionEnd, WorkflowComplete.
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+import yaml
+from pydantic import BaseModel, Field, model_validator
+
+# Supported hook event types
+HOOK_EVENT_TYPES = frozenset({
+    "UserPromptSubmit",
+    "PostToolUse",
+    "SessionStart",
+    "SessionEnd",
+    "WorkflowComplete",
+})
+
+
+class HookDefinition(BaseModel):
+    """Single hook definition within an event."""
+
+    name: str = Field(..., min_length=1, description="Hook display name")
+    command: str = Field(..., min_length=1, description="Shell command to execute")
+    enabled: bool = Field(default=True, description="Whether hook is enabled")
+    matcher: str | None = Field(
+        default=None,
+        description="Tool name matcher for PostToolUse (e.g. 'Write|Edit')",
+    )
+    file_patterns: list[str] | None = Field(
+        default=None,
+        description="Glob patterns for file filtering (e.g. ['*.py'])",
+    )
+    fail_on_error: bool = Field(
+        default=False,
+        description="If true, non-zero exit fails workflow",
+    )
+
+    model_config = {"extra": "forbid"}
+
+
+class HooksConfig(BaseModel):
+    """
+    Root schema for hooks.yaml.
+
+    Top-level key is 'hooks'. Each event maps to a list of HookDefinition.
+    """
+
+    hooks: dict[str, list[HookDefinition]] = Field(
+        default_factory=dict,
+        description="Event name -> list of hook definitions",
+    )
+
+    model_config = {"extra": "forbid"}
+
+    @model_validator(mode="after")
+    def validate_event_names(self) -> "HooksConfig":
+        """Ensure only supported event names are used."""
+        for event_name in self.hooks:
+            if event_name not in HOOK_EVENT_TYPES:
+                raise ValueError(
+                    f"Unsupported hook event '{event_name}'. "
+                    f"Supported: {', '.join(sorted(HOOK_EVENT_TYPES))}"
+                )
+        return self
+
+
+def load_hooks_config(
+    config_path: Path | str | None = None,
+    project_root: Path | None = None,
+) -> HooksConfig:
+    """
+    Load and validate hooks.yaml.
+
+    Args:
+        config_path: Explicit path to hooks.yaml. If None, uses
+            project_root/.tapps-agents/hooks.yaml.
+        project_root: Project root (default: cwd). Used when config_path is None.
+
+    Returns:
+        Validated HooksConfig. Returns empty config (all events empty) when
+        file is missing (safe defaults).
+
+    Raises:
+        FileNotFoundError: If config_path is given and file doesn't exist.
+        yaml.YAMLError: On YAML parse errors (includes path/line in message).
+        ValueError: On schema validation failures.
+    """
+    project_root = project_root or Path.cwd()
+    if config_path is None:
+        config_path = project_root / ".tapps-agents" / "hooks.yaml"
+    else:
+        config_path = Path(config_path)
+        if not config_path.is_absolute():
+            config_path = project_root / config_path
+
+    if not config_path.exists():
+        return HooksConfig(hooks={})
+
+    try:
+        content = config_path.read_text(encoding="utf-8")
+    except OSError as e:
+        raise FileNotFoundError(
+            f"Cannot read hooks config from {config_path}: {e}"
+        ) from e
+
+    try:
+        raw = yaml.safe_load(content)
+    except yaml.YAMLError as e:
+        path_hint = str(config_path)
+        if hasattr(e, "problem_mark") and e.problem_mark:
+            line = e.problem_mark.line + 1
+            col = e.problem_mark.column + 1
+            path_hint = f"{config_path}:{line}:{col}"
+        raise yaml.YAMLError(f"Invalid YAML in {path_hint}: {e}") from e
+
+    if raw is None:
+        return HooksConfig(hooks={})
+
+    if not isinstance(raw, dict):
+        raise ValueError(
+            f"hooks.yaml must be a YAML object (dict), got {type(raw).__name__} "
+            f"at {config_path}"
+        )
+
+    if "hooks" not in raw:
+        return HooksConfig(hooks={})
+
+    try:
+        return HooksConfig.model_validate(raw)
+    except Exception as e:
+        raise ValueError(
+            f"Hooks config validation failed at {config_path}: {e}"
+        ) from e