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

truthound_dashboard/api/cross_alerts.py

@@ -0,0 +1,352 @@
+"""Cross-alert correlation API endpoints.
+
+This module provides REST API endpoints for cross-feature integration
+between Anomaly Detection and Drift Monitoring alerts.
+"""
+
+from __future__ import annotations
+
+from typing import Annotated
+
+from fastapi import APIRouter, Depends, HTTPException, Query
+
+from truthound_dashboard.core.cross_alerts import CrossAlertService
+from truthound_dashboard.schemas.cross_alerts import (
+    CrossAlertCorrelation,
+    CrossAlertCorrelationListResponse,
+    AutoTriggerConfig,
+    AutoTriggerConfigCreate,
+    AutoTriggerConfigUpdate,
+    AutoTriggerEvent,
+    AutoTriggerEventListResponse,
+    CrossAlertSummary,
+)
+from .deps import SessionDep
+
+router = APIRouter()
+
+
+# Dependency
+async def get_cross_alert_service(session: SessionDep) -> CrossAlertService:
+    """Get cross-alert service dependency."""
+    return CrossAlertService(session)
+
+
+CrossAlertServiceDep = Annotated[CrossAlertService, Depends(get_cross_alert_service)]
+
+
+# =============================================================================
+# Correlation Endpoints
+# =============================================================================
+
+
+@router.get(
+    "/cross-alerts/correlations",
+    response_model=CrossAlertCorrelationListResponse,
+    summary="Get correlated alerts",
+    description="Get correlations between anomaly and drift alerts.",
+)
+async def get_correlations(
+    service: CrossAlertServiceDep,
+    source_id: str | None = Query(None, description="Filter by source ID"),
+    limit: int = Query(50, ge=1, le=100, description="Maximum items to return"),
+    offset: int = Query(0, ge=0, description="Number of items to skip"),
+) -> CrossAlertCorrelationListResponse:
+    """Get correlated alerts.
+
+    Args:
+        service: Injected cross-alert service.
+        source_id: Optional source ID filter.
+        limit: Maximum items to return.
+        offset: Number of items to skip.
+
+    Returns:
+        Paginated list of correlations.
+    """
+    correlations, total = await service.get_correlations(
+        source_id=source_id,
+        limit=limit,
+        offset=offset,
+    )
+
+    return CrossAlertCorrelationListResponse(
+        success=True,
+        data=[CrossAlertCorrelation(**c) for c in correlations],
+        total=total,
+        offset=offset,
+        limit=limit,
+    )
+
+
+@router.get(
+    "/cross-alerts/correlations/{source_id}",
+    response_model=dict,
+    summary="Find correlations for source",
+    description="Find correlated anomaly and drift alerts for a specific source.",
+)
+async def find_correlations_for_source(
+    source_id: str,
+    service: CrossAlertServiceDep,
+    time_window_hours: int = Query(24, ge=1, le=168, description="Time window in hours"),
+    limit: int = Query(50, ge=1, le=100, description="Maximum items to return"),
+) -> dict:
+    """Find correlations for a specific source.
+
+    This actively searches for correlations between recent anomaly
+    detections and drift alerts for the given source.
+
+    Args:
+        source_id: Source ID to analyze.
+        service: Injected cross-alert service.
+        time_window_hours: Time window to search (1-168 hours).
+        limit: Maximum correlations to return.
+
+    Returns:
+        Dictionary with correlations found.
+    """
+    correlations = await service.correlate_anomaly_drift(
+        source_id=source_id,
+        time_window_hours=time_window_hours,
+        limit=limit,
+    )
+
+    return {
+        "success": True,
+        "data": correlations,
+        "total": len(correlations),
+    }
+
+
+# =============================================================================
+# Auto-Trigger Configuration Endpoints
+# =============================================================================
+
+
+@router.get(
+    "/cross-alerts/config",
+    response_model=dict,
+    summary="Get auto-trigger config",
+    description="Get auto-trigger configuration (global or source-specific).",
+)
+async def get_config(
+    service: CrossAlertServiceDep,
+    source_id: str | None = Query(None, description="Source ID for source-specific config"),
+) -> dict:
+    """Get auto-trigger configuration.
+
+    Args:
+        service: Injected cross-alert service.
+        source_id: Optional source ID for source-specific config.
+
+    Returns:
+        Configuration dictionary.
+    """
+    config = service.get_config(source_id)
+    return {
+        "success": True,
+        "data": config,
+    }
+
+
+@router.post(
+    "/cross-alerts/config",
+    response_model=dict,
+    status_code=201,
+    summary="Create/update auto-trigger config",
+    description="Create or update auto-trigger configuration.",
+)
+async def update_config(
+    request: AutoTriggerConfigCreate,
+    service: CrossAlertServiceDep,
+) -> dict:
+    """Create or update auto-trigger configuration.
+
+    Args:
+        request: Configuration data.
+        service: Injected cross-alert service.
+
+    Returns:
+        Updated configuration.
+    """
+    update_data = request.model_dump(exclude_unset=True)
+    source_id = update_data.pop("source_id", None)
+
+    config = service.update_config(source_id, **update_data)
+
+    return {
+        "success": True,
+        "data": config,
+    }
+
+
+@router.put(
+    "/cross-alerts/config",
+    response_model=dict,
+    summary="Update auto-trigger config",
+    description="Update existing auto-trigger configuration.",
+)
+async def patch_config(
+    request: AutoTriggerConfigUpdate,
+    service: CrossAlertServiceDep,
+    source_id: str | None = Query(None, description="Source ID for source-specific config"),
+) -> dict:
+    """Update auto-trigger configuration.
+
+    Args:
+        request: Configuration update data.
+        service: Injected cross-alert service.
+        source_id: Optional source ID for source-specific config.
+
+    Returns:
+        Updated configuration.
+    """
+    update_data = request.model_dump(exclude_unset=True)
+    config = service.update_config(source_id, **update_data)
+
+    return {
+        "success": True,
+        "data": config,
+    }
+
+
+# =============================================================================
+# Auto-Trigger Event Endpoints
+# =============================================================================
+
+
+@router.get(
+    "/cross-alerts/events",
+    response_model=AutoTriggerEventListResponse,
+    summary="List auto-trigger events",
+    description="Get history of auto-triggered checks.",
+)
+async def list_events(
+    service: CrossAlertServiceDep,
+    source_id: str | None = Query(None, description="Filter by source ID"),
+    limit: int = Query(50, ge=1, le=100, description="Maximum items to return"),
+    offset: int = Query(0, ge=0, description="Number of items to skip"),
+) -> AutoTriggerEventListResponse:
+    """List auto-trigger events.
+
+    Args:
+        service: Injected cross-alert service.
+        source_id: Optional source ID filter.
+        limit: Maximum items to return.
+        offset: Number of items to skip.
+
+    Returns:
+        Paginated list of events.
+    """
+    events, total = await service.get_auto_trigger_events(
+        source_id=source_id,
+        limit=limit,
+        offset=offset,
+    )
+
+    return AutoTriggerEventListResponse(
+        success=True,
+        data=[AutoTriggerEvent(**e) for e in events],
+        total=total,
+        offset=offset,
+        limit=limit,
+    )
+
+
+# =============================================================================
+# Manual Trigger Endpoints
+# =============================================================================
+
+
+@router.post(
+    "/cross-alerts/trigger/drift-on-anomaly/{detection_id}",
+    response_model=dict,
+    summary="Trigger drift check on anomaly",
+    description="Manually trigger a drift check based on an anomaly detection result.",
+)
+async def trigger_drift_on_anomaly(
+    detection_id: str,
+    service: CrossAlertServiceDep,
+) -> dict:
+    """Manually trigger drift check after anomaly detection.
+
+    Args:
+        detection_id: Anomaly detection ID.
+        service: Injected cross-alert service.
+
+    Returns:
+        Trigger event result.
+
+    Raises:
+        HTTPException: 404 if detection not found.
+    """
+    event = await service.auto_trigger_drift_on_anomaly(detection_id)
+
+    if not event:
+        raise HTTPException(status_code=404, detail="Detection not found")
+
+    return {
+        "success": True,
+        "data": event,
+    }
+
+
+@router.post(
+    "/cross-alerts/trigger/anomaly-on-drift/{monitor_id}",
+    response_model=dict,
+    summary="Trigger anomaly check on drift",
+    description="Manually trigger an anomaly check based on drift detection.",
+)
+async def trigger_anomaly_on_drift(
+    monitor_id: str,
+    service: CrossAlertServiceDep,
+) -> dict:
+    """Manually trigger anomaly check after drift detection.
+
+    Args:
+        monitor_id: Drift monitor ID.
+        service: Injected cross-alert service.
+
+    Returns:
+        Trigger event result.
+
+    Raises:
+        HTTPException: 404 if monitor not found.
+    """
+    event = await service.auto_trigger_anomaly_on_drift(monitor_id)
+
+    if not event:
+        raise HTTPException(status_code=404, detail="Monitor not found")
+
+    return {
+        "success": True,
+        "data": event,
+    }
+
+
+# =============================================================================
+# Summary Endpoint
+# =============================================================================
+
+
+@router.get(
+    "/cross-alerts/summary",
+    response_model=dict,
+    summary="Get cross-alert summary",
+    description="Get summary statistics for cross-alert correlations.",
+)
+async def get_summary(
+    service: CrossAlertServiceDep,
+) -> dict:
+    """Get cross-alert summary statistics.
+
+    Args:
+        service: Injected cross-alert service.
+
+    Returns:
+        Summary statistics dictionary.
+    """
+    summary = await service.get_summary()
+    return {
+        "success": True,
+        "data": summary,
+    }

truthound_dashboard/api/deps.py

@@ -31,6 +31,14 @@ from truthound_dashboard.core import (
     SourceService,
     ValidationService,
 )
+from truthound_dashboard.core.schema_evolution import SchemaEvolutionService
+from truthound_dashboard.core.rule_generator import RuleGeneratorService
+from truthound_dashboard.core.profile_comparison import ProfileComparisonService
+from truthound_dashboard.core.lineage import LineageService
+from truthound_dashboard.core.anomaly import AnomalyDetectionService
+from truthound_dashboard.core.anomaly_explainer import AnomalyExplainerService
+from truthound_dashboard.core.openlineage import OpenLineageEmitterService, OpenLineageWebhookService
+from truthound_dashboard.core.report_history import ReportHistoryService
 from truthound_dashboard.db import get_db_session
 
 
@@ -168,6 +176,102 @@ async def get_mask_service(session: SessionDep) -> MaskService:
     return MaskService(session)
 
 
+async def get_schema_evolution_service(session: SessionDep) -> SchemaEvolutionService:
+    """Get schema evolution service dependency.
+
+    Args:
+        session: Database session.
+
+    Returns:
+        SchemaEvolutionService instance.
+    """
+    return SchemaEvolutionService(session)
+
+
+async def get_rule_generator_service(session: SessionDep) -> RuleGeneratorService:
+    """Get rule generator service dependency.
+
+    Args:
+        session: Database session.
+
+    Returns:
+        RuleGeneratorService instance.
+    """
+    return RuleGeneratorService(session)
+
+
+async def get_profile_comparison_service(session: SessionDep) -> ProfileComparisonService:
+    """Get profile comparison service dependency.
+
+    Args:
+        session: Database session.
+
+    Returns:
+        ProfileComparisonService instance.
+    """
+    return ProfileComparisonService(session)
+
+
+async def get_lineage_service(session: SessionDep) -> LineageService:
+    """Get lineage service dependency.
+
+    Args:
+        session: Database session.
+
+    Returns:
+        LineageService instance.
+    """
+    return LineageService(session)
+
+
+async def get_anomaly_detection_service(session: SessionDep) -> AnomalyDetectionService:
+    """Get anomaly detection service dependency.
+
+    Args:
+        session: Database session.
+
+    Returns:
+        AnomalyDetectionService instance.
+    """
+    return AnomalyDetectionService(session)
+
+
+async def get_anomaly_explainer_service(session: SessionDep) -> AnomalyExplainerService:
+    """Get anomaly explainer service dependency.
+
+    Args:
+        session: Database session.
+
+    Returns:
+        AnomalyExplainerService instance for SHAP/LIME explanations.
+    """
+    return AnomalyExplainerService(session)
+
+
+async def get_openlineage_emitter_service(session: SessionDep) -> OpenLineageEmitterService:
+    """Get OpenLineage emitter service dependency.
+
+    Args:
+        session: Database session.
+
+    Returns:
+        OpenLineageEmitterService instance.
+    """
+    return OpenLineageEmitterService(session)
+
+
+async def get_openlineage_webhook_service(session: SessionDep) -> OpenLineageWebhookService:
+    """Get OpenLineage webhook service dependency.
+
+    Args:
+        session: Database session.
+
+    Returns:
+        OpenLineageWebhookService instance.
+    """
+    return OpenLineageWebhookService(session)
+
+
 # Type aliases for service dependencies
 SourceServiceDep = Annotated[SourceService, Depends(get_source_service)]
 ValidationServiceDep = Annotated[ValidationService, Depends(get_validation_service)]
@@ -179,3 +283,42 @@ DriftServiceDep = Annotated[DriftService, Depends(get_drift_service)]
 ScheduleServiceDep = Annotated[ScheduleService, Depends(get_schedule_service)]
 PIIScanServiceDep = Annotated[PIIScanService, Depends(get_pii_scan_service)]
 MaskServiceDep = Annotated[MaskService, Depends(get_mask_service)]
+SchemaEvolutionServiceDep = Annotated[
+    SchemaEvolutionService, Depends(get_schema_evolution_service)
+]
+RuleGeneratorServiceDep = Annotated[
+    RuleGeneratorService, Depends(get_rule_generator_service)
+]
+ProfileComparisonServiceDep = Annotated[
+    ProfileComparisonService, Depends(get_profile_comparison_service)
+]
+LineageServiceDep = Annotated[LineageService, Depends(get_lineage_service)]
+AnomalyDetectionServiceDep = Annotated[
+    AnomalyDetectionService, Depends(get_anomaly_detection_service)
+]
+AnomalyExplainerServiceDep = Annotated[
+    AnomalyExplainerService, Depends(get_anomaly_explainer_service)
+]
+OpenLineageEmitterServiceDep = Annotated[
+    OpenLineageEmitterService, Depends(get_openlineage_emitter_service)
+]
+OpenLineageWebhookServiceDep = Annotated[
+    OpenLineageWebhookService, Depends(get_openlineage_webhook_service)
+]
+
+
+async def get_report_history_service(session: SessionDep) -> ReportHistoryService:
+    """Get report history service dependency.
+
+    Args:
+        session: Database session.
+
+    Returns:
+        ReportHistoryService instance.
+    """
+    return ReportHistoryService(session)
+
+
+ReportHistoryServiceDep = Annotated[
+    ReportHistoryService, Depends(get_report_history_service)
+]