truthound-dashboard 1.3.1__py3-none-any.whl → 1.4.1__py3-none-any.whl

truthound_dashboard/schemas/unified_alerts.py

@@ -0,0 +1,223 @@
+"""Pydantic schemas for Unified Alerts.
+
+Provides schemas for aggregating alerts from all sources:
+- Model monitoring alerts
+- Drift monitoring alerts
+- Anomaly detection alerts
+- Validation failures
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+from enum import Enum
+from typing import Any, Literal
+
+from pydantic import BaseModel, Field
+
+from .base import BaseSchema, IDMixin, ListResponseWrapper, TimestampMixin
+
+
+# =============================================================================
+# Enums
+# =============================================================================
+
+
+class AlertSource(str, Enum):
+    """Sources of alerts in the unified system."""
+
+    MODEL = "model"
+    DRIFT = "drift"
+    ANOMALY = "anomaly"
+    VALIDATION = "validation"
+
+
+class AlertSeverity(str, Enum):
+    """Alert severity levels."""
+
+    CRITICAL = "critical"
+    HIGH = "high"
+    MEDIUM = "medium"
+    LOW = "low"
+    INFO = "info"
+
+
+class AlertStatus(str, Enum):
+    """Alert status values."""
+
+    OPEN = "open"
+    ACKNOWLEDGED = "acknowledged"
+    RESOLVED = "resolved"
+    IGNORED = "ignored"
+
+
+# =============================================================================
+# Unified Alert Schemas
+# =============================================================================
+
+
+class UnifiedAlertBase(BaseSchema):
+    """Base unified alert schema."""
+
+    source: AlertSource = Field(..., description="Alert source type")
+    source_id: str = Field(..., description="Source-specific alert ID")
+    source_name: str = Field(..., description="Name of the source (model, source, etc.)")
+    severity: AlertSeverity = Field(..., description="Alert severity")
+    status: AlertStatus = Field(default=AlertStatus.OPEN, description="Alert status")
+    title: str = Field(..., description="Alert title", min_length=1, max_length=500)
+    message: str = Field(..., description="Alert message/description")
+    details: dict[str, Any] = Field(default_factory=dict, description="Source-specific details")
+
+
+class UnifiedAlertResponse(UnifiedAlertBase, IDMixin, TimestampMixin):
+    """Schema for unified alert response."""
+
+    acknowledged_at: datetime | None = Field(None, description="When acknowledged")
+    acknowledged_by: str | None = Field(None, description="Who acknowledged")
+    resolved_at: datetime | None = Field(None, description="When resolved")
+    resolved_by: str | None = Field(None, description="Who resolved")
+    related_alert_ids: list[str] = Field(default_factory=list, description="IDs of related alerts")
+
+
+class UnifiedAlertListResponse(ListResponseWrapper):
+    """List response for unified alerts."""
+
+    items: list[UnifiedAlertResponse]
+
+
+# =============================================================================
+# Alert Summary Schemas
+# =============================================================================
+
+
+class AlertCountBySeverity(BaseModel):
+    """Alert counts by severity."""
+
+    critical: int = Field(default=0, description="Critical alerts")
+    high: int = Field(default=0, description="High severity alerts")
+    medium: int = Field(default=0, description="Medium severity alerts")
+    low: int = Field(default=0, description="Low severity alerts")
+    info: int = Field(default=0, description="Info alerts")
+
+
+class AlertCountBySource(BaseModel):
+    """Alert counts by source."""
+
+    model: int = Field(default=0, description="Model monitoring alerts")
+    drift: int = Field(default=0, description="Drift monitoring alerts")
+    anomaly: int = Field(default=0, description="Anomaly detection alerts")
+    validation: int = Field(default=0, description="Validation failure alerts")
+
+
+class AlertCountByStatus(BaseModel):
+    """Alert counts by status."""
+
+    open: int = Field(default=0, description="Open alerts")
+    acknowledged: int = Field(default=0, description="Acknowledged alerts")
+    resolved: int = Field(default=0, description="Resolved alerts")
+    ignored: int = Field(default=0, description="Ignored alerts")
+
+
+class AlertTrendPoint(BaseModel):
+    """Single point in alert trend data."""
+
+    timestamp: datetime = Field(..., description="Time point")
+    count: int = Field(default=0, description="Alert count at this time")
+
+
+class AlertSummary(BaseSchema):
+    """Summary statistics for alerts."""
+
+    total_alerts: int = Field(default=0, description="Total active alerts")
+    active_alerts: int = Field(default=0, description="Non-resolved alerts")
+    by_severity: AlertCountBySeverity = Field(
+        default_factory=AlertCountBySeverity,
+        description="Counts by severity",
+    )
+    by_source: AlertCountBySource = Field(
+        default_factory=AlertCountBySource,
+        description="Counts by source",
+    )
+    by_status: AlertCountByStatus = Field(
+        default_factory=AlertCountByStatus,
+        description="Counts by status",
+    )
+    trend_24h: list[AlertTrendPoint] = Field(
+        default_factory=list,
+        description="Alert trend over last 24 hours",
+    )
+    top_sources: list[dict[str, Any]] = Field(
+        default_factory=list,
+        description="Top sources with most alerts",
+    )
+
+
+# =============================================================================
+# Alert Correlation Schemas
+# =============================================================================
+
+
+class AlertCorrelation(BaseSchema):
+    """Correlation between alerts."""
+
+    alert_id: str = Field(..., description="Primary alert ID")
+    related_alerts: list[UnifiedAlertResponse] = Field(
+        default_factory=list,
+        description="Related alerts",
+    )
+    correlation_type: str = Field(..., description="Type of correlation")
+    correlation_score: float = Field(
+        default=0.0,
+        ge=0.0,
+        le=1.0,
+        description="Correlation confidence score",
+    )
+    common_factors: list[str] = Field(
+        default_factory=list,
+        description="Common factors between alerts",
+    )
+
+
+class AlertCorrelationResponse(BaseSchema):
+    """Response for alert correlation query."""
+
+    correlations: list[AlertCorrelation] = Field(
+        default_factory=list,
+        description="List of correlations",
+    )
+    total_correlated: int = Field(default=0, description="Total correlated alerts")
+
+
+# =============================================================================
+# Request/Action Schemas
+# =============================================================================
+
+
+class AcknowledgeAlertRequest(BaseModel):
+    """Request to acknowledge an alert."""
+
+    actor: str = Field(..., description="Who is acknowledging", min_length=1)
+    message: str = Field(default="", description="Optional acknowledgement message")
+
+
+class ResolveAlertRequest(BaseModel):
+    """Request to resolve an alert."""
+
+    actor: str = Field(..., description="Who is resolving", min_length=1)
+    message: str = Field(default="", description="Resolution message")
+
+
+class BulkAlertActionRequest(BaseModel):
+    """Request for bulk alert actions."""
+
+    alert_ids: list[str] = Field(..., description="Alert IDs to act on", min_length=1)
+    actor: str = Field(..., description="Who is performing the action")
+    message: str = Field(default="", description="Optional message")
+
+
+class BulkAlertActionResponse(BaseModel):
+    """Response for bulk alert actions."""
+
+    success_count: int = Field(default=0, description="Number of successful updates")
+    failed_count: int = Field(default=0, description="Number of failed updates")
+    failed_ids: list[str] = Field(default_factory=list, description="IDs that failed")

truthound_dashboard/schemas/validation.py

@@ -42,15 +42,30 @@ class ValidationIssue(BaseSchema):
     )
 
 
+class CustomValidatorConfig(BaseSchema):
+    """Configuration for running a custom validator."""
+
+    validator_id: str = Field(
+        ..., description="ID of the custom validator to run"
+    )
+    column: str = Field(
+        ..., description="Column to validate"
+    )
+    params: dict[str, Any] | None = Field(
+        default=None, description="Parameter values for the validator"
+    )
+
+
 class ValidationRunRequest(BaseSchema):
     """Request to run validation on a source.
 
     This schema maps to truthound's th.check() parameters for maximum flexibility.
     All optional parameters default to None/False to use truthound's defaults.
 
-    Supports two modes:
+    Supports three modes:
     1. Simple mode: Use `validators` list with validator names (backward compatible)
     2. Advanced mode: Use `validator_configs` for per-validator parameter configuration
+    3. Custom validators: Use `custom_validators` to include user-defined validators
     """
 
     # Core validation options - Simple mode (backward compatible)
@@ -69,6 +84,15 @@ class ValidationRunRequest(BaseSchema):
         ),
     )
 
+    # Custom validators - User-defined validators
+    custom_validators: list[CustomValidatorConfig] | None = Field(
+        default=None,
+        description=(
+            "Custom validators to run alongside built-in validators. "
+            "Each config specifies the validator_id, target column, and parameters."
+        ),
+    )
+
     schema_path: str | None = Field(
         default=None,
         description="Path to schema YAML file for schema validation",

truthound_dashboard/schemas/validators/__init__.py

@@ -18,12 +18,17 @@ Usage:
 from __future__ import annotations
 
 from .base import (
+    CustomValidatorExecuteRequest,
+    CustomValidatorExecuteResponse,
     ParameterDefinition,
     ParameterType,
+    UnifiedValidatorDefinition,
+    UnifiedValidatorListResponse,
     ValidatorCategory,
     ValidatorConfig,
     ValidatorConfigList,
     ValidatorDefinition,
+    ValidatorSource,
     configs_to_truthound_format,
     has_custom_params,
     merge_severity_overrides,
@@ -45,6 +50,12 @@ __all__ = [
     "ValidatorConfig",
     "ValidatorConfigList",
     "ValidatorDefinition",
+    # Unified validator types (built-in + custom)
+    "ValidatorSource",
+    "UnifiedValidatorDefinition",
+    "UnifiedValidatorListResponse",
+    "CustomValidatorExecuteRequest",
+    "CustomValidatorExecuteResponse",
     # Utility functions
     "configs_to_truthound_format",
     "has_custom_params",

truthound_dashboard/schemas/validators/base.py

@@ -261,3 +261,154 @@ def merge_severity_overrides(
         for config in configs
         if config.enabled and config.severity_override is not None
     }
+
+
+# ============================================================================
+# Unified Validator Types (Built-in + Custom)
+# ============================================================================
+
+
+class ValidatorSource(str, Enum):
+    """Source of a validator (built-in or custom)."""
+
+    BUILTIN = "builtin"
+    CUSTOM = "custom"
+
+
+class UnifiedValidatorDefinition(BaseModel):
+    """Unified validator definition that can represent both built-in and custom validators.
+
+    This is used for the unified validator list endpoint that combines
+    both built-in truthound validators and user-defined custom validators.
+    """
+
+    id: str | None = Field(
+        default=None, description="Unique ID (for custom validators only)"
+    )
+    name: str = Field(..., description="Validator class name")
+    display_name: str = Field(..., description="Human-readable name")
+    category: str = Field(..., description="Validator category")
+    description: str = Field(..., description="What this validator checks")
+    parameters: list[ParameterDefinition] = Field(
+        default_factory=list, description="Configurable parameters"
+    )
+    tags: list[str] = Field(default_factory=list, description="Searchable tags")
+    severity_default: Literal["low", "medium", "high", "critical"] = Field(
+        default="medium", description="Default issue severity"
+    )
+    source: ValidatorSource = Field(
+        default=ValidatorSource.BUILTIN, description="Whether built-in or custom"
+    )
+    is_enabled: bool = Field(default=True, description="Whether validator is enabled")
+    requires_extra: str | None = Field(
+        default=None, description="Extra dependency required"
+    )
+    experimental: bool = Field(default=False, description="Whether experimental")
+    deprecated: bool = Field(default=False, description="Whether deprecated")
+    usage_count: int = Field(
+        default=0, description="Number of times used (custom only)"
+    )
+    is_verified: bool = Field(
+        default=False, description="Whether verified (custom only)"
+    )
+
+    @classmethod
+    def from_builtin(cls, validator: ValidatorDefinition) -> "UnifiedValidatorDefinition":
+        """Create from a built-in ValidatorDefinition."""
+        return cls(
+            id=None,
+            name=validator.name,
+            display_name=validator.display_name,
+            category=validator.category.value,
+            description=validator.description,
+            parameters=validator.parameters,
+            tags=validator.tags,
+            severity_default=validator.severity_default,
+            source=ValidatorSource.BUILTIN,
+            is_enabled=True,
+            requires_extra=validator.requires_extra,
+            experimental=validator.experimental,
+            deprecated=validator.deprecated,
+            usage_count=0,
+            is_verified=True,
+        )
+
+    @classmethod
+    def from_custom(cls, custom_validator: Any) -> "UnifiedValidatorDefinition":
+        """Create from a CustomValidator database model."""
+        # Convert custom validator parameters to ParameterDefinition format
+        parameters = []
+        for param in custom_validator.parameters or []:
+            param_def = ParameterDefinition(
+                name=param.get("name", ""),
+                label=param.get("name", "").replace("_", " ").title(),
+                type=ParameterType(param.get("type", "string")),
+                description=param.get("description", ""),
+                required=param.get("required", False),
+                default=param.get("default"),
+                options=param.get("options"),
+                min_value=param.get("min_value"),
+                max_value=param.get("max_value"),
+            )
+            parameters.append(param_def)
+
+        return cls(
+            id=str(custom_validator.id),
+            name=f"custom:{custom_validator.name}",  # Prefix to distinguish
+            display_name=custom_validator.display_name,
+            category=custom_validator.category,
+            description=custom_validator.description,
+            parameters=parameters,
+            tags=custom_validator.tags or [],
+            severity_default=custom_validator.severity or "medium",
+            source=ValidatorSource.CUSTOM,
+            is_enabled=custom_validator.is_enabled,
+            requires_extra=None,
+            experimental=False,
+            deprecated=False,
+            usage_count=custom_validator.usage_count,
+            is_verified=custom_validator.is_verified,
+        )
+
+
+class UnifiedValidatorListResponse(BaseModel):
+    """Response for unified validator list."""
+
+    data: list[UnifiedValidatorDefinition] = Field(
+        default_factory=list, description="List of validators"
+    )
+    total: int = Field(..., description="Total number of validators")
+    builtin_count: int = Field(..., description="Number of built-in validators")
+    custom_count: int = Field(..., description="Number of custom validators")
+    categories: list[dict[str, Any]] = Field(
+        default_factory=list, description="Category summary with counts"
+    )
+
+
+class CustomValidatorExecuteRequest(BaseModel):
+    """Request to execute a custom validator on a data source."""
+
+    source_id: str = Field(..., description="Data source ID to validate")
+    column_name: str = Field(..., description="Column to validate")
+    param_values: dict[str, Any] = Field(
+        default_factory=dict, description="Parameter values"
+    )
+    sample_size: int | None = Field(
+        default=None, ge=1, le=100000, description="Sample size (optional)"
+    )
+
+
+class CustomValidatorExecuteResponse(BaseModel):
+    """Response from custom validator execution."""
+
+    success: bool = Field(..., description="Whether execution succeeded")
+    passed: bool | None = Field(default=None, description="Whether validation passed")
+    execution_time_ms: float = Field(default=0, description="Execution time in ms")
+    issues: list[dict[str, Any]] = Field(
+        default_factory=list, description="Validation issues found"
+    )
+    message: str = Field(default="", description="Summary message")
+    details: dict[str, Any] = Field(
+        default_factory=dict, description="Additional details"
+    )
+    error: str | None = Field(default=None, description="Error message if failed")

truthound_dashboard/schemas/versioning.py

@@ -0,0 +1,152 @@
+"""Versioning schemas for validation result versioning.
+
+Defines request/response models for the versioning API endpoints.
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+from typing import Any, Literal
+
+from pydantic import Field
+
+from .base import BaseSchema
+
+
+# Versioning strategy types
+VersioningStrategyType = Literal["incremental", "semantic", "timestamp", "gitlike"]
+
+
+class VersionInfoResponse(BaseSchema):
+    """Version information response."""
+
+    version_id: str = Field(..., description="Unique version identifier")
+    version_number: str = Field(..., description="Human-readable version number")
+    validation_id: str = Field(..., description="Associated validation ID")
+    source_id: str = Field(..., description="Associated source ID")
+    strategy: VersioningStrategyType = Field(..., description="Versioning strategy used")
+    created_at: datetime = Field(..., description="Version creation timestamp")
+    parent_version_id: str | None = Field(None, description="Parent version ID")
+    metadata: dict[str, Any] = Field(default_factory=dict, description="Additional metadata")
+    content_hash: str | None = Field(None, description="Hash of validation content")
+
+
+class VersionListResponse(BaseSchema):
+    """Response for listing versions."""
+
+    success: bool = Field(default=True)
+    data: list[VersionInfoResponse] = Field(..., description="List of versions")
+    total: int = Field(..., description="Total number of versions")
+    source_id: str = Field(..., description="Source ID for these versions")
+
+
+class IssueChange(BaseSchema):
+    """Represents a changed issue between versions."""
+
+    key: str = Field(..., description="Issue key (column:issue_type)")
+    from_issue: dict[str, Any] = Field(..., alias="from", description="Original issue")
+    to_issue: dict[str, Any] = Field(..., alias="to", description="New issue")
+
+    class Config:
+        populate_by_name = True
+
+
+class VersionDiffResponse(BaseSchema):
+    """Version comparison/diff response."""
+
+    from_version: VersionInfoResponse = Field(..., description="Source version")
+    to_version: VersionInfoResponse = Field(..., description="Target version")
+    issues_added: list[dict[str, Any]] = Field(
+        default_factory=list, description="Issues added in target version"
+    )
+    issues_removed: list[dict[str, Any]] = Field(
+        default_factory=list, description="Issues removed in target version"
+    )
+    issues_changed: list[dict[str, Any]] = Field(
+        default_factory=list, description="Issues that changed between versions"
+    )
+    summary_changes: dict[str, Any] = Field(
+        default_factory=dict, description="Summary of changes"
+    )
+    has_changes: bool = Field(..., description="Whether there are any changes")
+
+
+class VersionCompareRequest(BaseSchema):
+    """Request to compare two versions."""
+
+    from_version_id: str = Field(..., description="Source version ID to compare from")
+    to_version_id: str = Field(..., description="Target version ID to compare to")
+
+
+class VersionHistoryResponse(BaseSchema):
+    """Response for version history chain."""
+
+    success: bool = Field(default=True)
+    data: list[VersionInfoResponse] = Field(..., description="Version history chain")
+    depth: int = Field(..., description="Depth of history returned")
+
+
+class CreateVersionRequest(BaseSchema):
+    """Request to create a new version for a validation."""
+
+    validation_id: str = Field(..., description="Validation ID to version")
+    strategy: VersioningStrategyType | None = Field(
+        None, description="Versioning strategy (defaults to incremental)"
+    )
+    metadata: dict[str, Any] | None = Field(
+        None, description="Additional version metadata"
+    )
+
+
+class CreateVersionResponse(BaseSchema):
+    """Response after creating a version."""
+
+    success: bool = Field(default=True)
+    data: VersionInfoResponse = Field(..., description="Created version info")
+    message: str = Field(..., description="Success message")
+
+
+# Rollback schemas
+class RollbackRequest(BaseSchema):
+    """Request to rollback to a previous version."""
+
+    target_version_id: str = Field(..., description="Version ID to rollback to")
+    create_new_validation: bool = Field(
+        default=True,
+        description="Whether to create a new validation record for the rollback"
+    )
+
+
+class RollbackResponse(BaseSchema):
+    """Response after a rollback operation."""
+
+    success: bool = Field(default=True)
+    source_id: str = Field(..., description="Source ID that was rolled back")
+    from_version: VersionInfoResponse | None = Field(
+        None, description="Version rolled back from"
+    )
+    to_version: VersionInfoResponse | None = Field(
+        None, description="Version rolled back to"
+    )
+    new_validation_id: str | None = Field(
+        None, description="ID of the new validation created (if any)"
+    )
+    message: str = Field(..., description="Status message")
+    rolled_back_at: datetime = Field(..., description="When rollback was performed")
+
+
+class RollbackAvailabilityResponse(BaseSchema):
+    """Response for checking rollback availability."""
+
+    success: bool = Field(default=True)
+    can_rollback: bool = Field(..., description="Whether rollback is available")
+    current_version_id: str | None = Field(
+        None, description="Current active version ID"
+    )
+    available_versions: int = Field(
+        ..., description="Number of available versions"
+    )
+    rollback_targets: list[dict[str, Any]] = Field(
+        default_factory=list,
+        description="List of versions available for rollback"
+    )

truthound_dashboard/static/index.html

@@ -6,8 +6,8 @@
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <meta name="description" content="Truthound Dashboard - Open-source data quality monitoring" />
     <title>Truthound Dashboard</title>
-    <script type="module" crossorigin src="/assets/index-BZG20KuF.js"></script>
-    <link rel="stylesheet" crossorigin href="/assets/index-D_HyZ3pb.css">
+    <script type="module" crossorigin src="/assets/index-DkU82VsU.js"></script>
+    <link rel="stylesheet" crossorigin href="/assets/index-C6JSrkHo.css">
   </head>
   <body>
     <div id="root"></div>