diff --git a/src/fides/api/schemas/connection_configuration/connection_config.py b/src/fides/api/schemas/connection_configuration/connection_config.py index 15ac22f330..44a872cca3 100644 --- a/src/fides/api/schemas/connection_configuration/connection_config.py +++ b/src/fides/api/schemas/connection_configuration/connection_config.py @@ -1,5 +1,5 @@ from datetime import datetime -from typing import Any, Dict, List, Optional, cast +from typing import TYPE_CHECKING, Any, Dict, List, Optional, cast from fideslang.models import Dataset from fideslang.validation import FidesKey @@ -14,6 +14,9 @@ from fides.api.schemas.saas.saas_config import SaaSConfigBase from fides.api.util.connection_type import get_connection_type_secret_schema +if TYPE_CHECKING: + from fides.api.models.detection_discovery import MonitorConfig + class CreateConnectionConfiguration(BaseModel): """ @@ -119,6 +122,12 @@ def mask_sensitive_values(self) -> "ConnectionConfigurationResponse": model_config = ConfigDict(from_attributes=True) +class ConnectionConfigurationWithMonitorConfigResponse(ConnectionConfigurationResponse): + """Schema for ConnectionConfiguration with MonitorConfigs associated with it.""" + + monitors: List["MonitorConfig"] = None + + class BulkPutConnectionConfiguration(BulkResponse): """Schema with mixed success/failure responses for Bulk Create/Update of ConnectionConfiguration responses."""