cribl-control-plane 0.0.49__py3-none-any.whl → 0.1.0b1__py3-none-any.whl

cribl_control_plane/models/createversionpushop.py

@@ -2,22 +2,22 @@
 
 from __future__ import annotations
 from cribl_control_plane.types import BaseModel
-from typing import Any, Dict, List, Optional
+from typing import List, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
 class CreateVersionPushResponseTypedDict(TypedDict):
-    r"""a list of any objects"""
+    r"""a list of string objects"""
 
     count: NotRequired[int]
     r"""number of items present in the items array"""
-    items: NotRequired[List[Dict[str, Any]]]
+    items: NotRequired[List[str]]
 
 
 class CreateVersionPushResponse(BaseModel):
-    r"""a list of any objects"""
+    r"""a list of string objects"""
 
     count: Optional[int] = None
     r"""number of items present in the items array"""
 
-    items: Optional[List[Dict[str, Any]]] = None
+    items: Optional[List[str]] = None

cribl_control_plane/models/cribllakedataset.py

@@ -6,14 +6,17 @@ from .lakedatasetsearchconfig import (
     LakeDatasetSearchConfig,
     LakeDatasetSearchConfigTypedDict,
 )
+from cribl_control_plane import utils
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
 from enum import Enum
 import pydantic
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class CriblLakeDatasetFormat(str, Enum):
+class CriblLakeDatasetFormat(str, Enum, metaclass=utils.OpenEnumMeta):
     JSON = "json"
     DDSS = "ddss"
     PARQUET = "parquet"
@@ -54,7 +57,10 @@ class CriblLakeDataset(BaseModel):
     description: Optional[str] = None
 
     format_: Annotated[
-        Optional[CriblLakeDatasetFormat], pydantic.Field(alias="format")
+        Annotated[
+            Optional[CriblLakeDatasetFormat], PlainValidator(validate_open_enum(False))
+        ],
+        pydantic.Field(alias="format"),
     ] = None
 
     http_da_used: Annotated[Optional[bool], pydantic.Field(alias="httpDAUsed")] = None

cribl_control_plane/models/datasetmetadata.py

@@ -5,14 +5,17 @@ from .datasetmetadataruninfo import (
     DatasetMetadataRunInfo,
     DatasetMetadataRunInfoTypedDict,
 )
+from cribl_control_plane import utils
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
 from enum import Enum
 import pydantic
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class ScanMode(str, Enum):
+class ScanMode(str, Enum, metaclass=utils.OpenEnumMeta):
     DETAILED = "detailed"
     QUICK = "quick"
 
@@ -32,7 +35,10 @@ class DatasetMetadata(BaseModel):
 
     field_list: Annotated[List[str], pydantic.Field(alias="fieldList")]
 
-    scan_mode: Annotated[ScanMode, pydantic.Field(alias="scanMode")]
+    scan_mode: Annotated[
+        Annotated[ScanMode, PlainValidator(validate_open_enum(False))],
+        pydantic.Field(alias="scanMode"),
+    ]
 
     latest_run_info: Annotated[
         Optional[DatasetMetadataRunInfo], pydantic.Field(alias="latestRunInfo")

cribl_control_plane/models/deleteconfiggroupbyproductandidop.py

@@ -4,7 +4,12 @@ from __future__ import annotations
 from .configgroup import ConfigGroup, ConfigGroupTypedDict
 from .productscore import ProductsCore
 from cribl_control_plane.types import BaseModel
-from cribl_control_plane.utils import FieldMetadata, PathParamMetadata
+from cribl_control_plane.utils import (
+    FieldMetadata,
+    PathParamMetadata,
+    validate_open_enum,
+)
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -18,7 +23,7 @@ class DeleteConfigGroupByProductAndIDRequestTypedDict(TypedDict):
 
 class DeleteConfigGroupByProductAndIDRequest(BaseModel):
     product: Annotated[
-        ProductsCore,
+        Annotated[ProductsCore, PlainValidator(validate_open_enum(False))],
         FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
     ]
     r"""Name of the Cribl product to get the Worker Groups or Edge Fleets for."""

cribl_control_plane/models/error.py

@@ -0,0 +1,16 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from cribl_control_plane.types import BaseModel
+from typing import Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+class ErrorTypedDict(TypedDict):
+    message: NotRequired[str]
+    r"""Error message"""
+
+
+class Error(BaseModel):
+    message: Optional[str] = None
+    r"""Error message"""

cribl_control_plane/models/getconfiggroupaclbyproductandidop.py

@@ -9,7 +9,9 @@ from cribl_control_plane.utils import (
     FieldMetadata,
     PathParamMetadata,
     QueryParamMetadata,
+    validate_open_enum,
 )
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -25,7 +27,7 @@ class GetConfigGroupACLByProductAndIDRequestTypedDict(TypedDict):
 
 class GetConfigGroupACLByProductAndIDRequest(BaseModel):
     product: Annotated[
-        ProductsCore,
+        Annotated[ProductsCore, PlainValidator(validate_open_enum(False))],
         FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
     ]
     r"""Name of the Cribl product to get the Worker Groups or Edge Fleets for."""
@@ -36,7 +38,7 @@ class GetConfigGroupACLByProductAndIDRequest(BaseModel):
     r"""The <code>id</code> of the Worker Group or Edge Fleet to get the ACL for."""
 
     type: Annotated[
-        Optional[RbacResource],
+        Annotated[Optional[RbacResource], PlainValidator(validate_open_enum(False))],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = None
     r"""Filter for limiting the response to ACL entries for the specified RBAC resource type."""

cribl_control_plane/models/getconfiggroupaclteamsbyproductandidop.py

@@ -9,7 +9,9 @@ from cribl_control_plane.utils import (
     FieldMetadata,
     PathParamMetadata,
     QueryParamMetadata,
+    validate_open_enum,
 )
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -25,7 +27,7 @@ class GetConfigGroupACLTeamsByProductAndIDRequestTypedDict(TypedDict):
 
 class GetConfigGroupACLTeamsByProductAndIDRequest(BaseModel):
     product: Annotated[
-        ProductsCore,
+        Annotated[ProductsCore, PlainValidator(validate_open_enum(False))],
         FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
     ]
     r"""Name of the Cribl product that contains the Worker Group or Edge Fleet."""
@@ -36,7 +38,7 @@ class GetConfigGroupACLTeamsByProductAndIDRequest(BaseModel):
     r"""The <code>id</code> of the Worker Group or Edge Fleet to get the team ACL for."""
 
     type: Annotated[
-        Optional[RbacResource],
+        Annotated[Optional[RbacResource], PlainValidator(validate_open_enum(False))],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = None
     r"""Filter for limiting the response to ACL entries for the specified RBAC resource type."""

cribl_control_plane/models/getconfiggroupbyproductandidop.py

@@ -8,7 +8,9 @@ from cribl_control_plane.utils import (
     FieldMetadata,
     PathParamMetadata,
     QueryParamMetadata,
+    validate_open_enum,
 )
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -24,7 +26,7 @@ class GetConfigGroupByProductAndIDRequestTypedDict(TypedDict):
 
 class GetConfigGroupByProductAndIDRequest(BaseModel):
     product: Annotated[
-        ProductsCore,
+        Annotated[ProductsCore, PlainValidator(validate_open_enum(False))],
         FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
     ]
     r"""Name of the Cribl product to get the Worker Groups or Edge Fleets for."""

cribl_control_plane/models/getconfiggroupconfigversionbyproductandidop.py

@@ -3,7 +3,12 @@
 from __future__ import annotations
 from .productscore import ProductsCore
 from cribl_control_plane.types import BaseModel
-from cribl_control_plane.utils import FieldMetadata, PathParamMetadata
+from cribl_control_plane.utils import (
+    FieldMetadata,
+    PathParamMetadata,
+    validate_open_enum,
+)
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -17,7 +22,7 @@ class GetConfigGroupConfigVersionByProductAndIDRequestTypedDict(TypedDict):
 
 class GetConfigGroupConfigVersionByProductAndIDRequest(BaseModel):
     product: Annotated[
-        ProductsCore,
+        Annotated[ProductsCore, PlainValidator(validate_open_enum(False))],
         FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
     ]
     r"""Name of the Cribl product to get the Worker Groups or Edge Fleets for."""

cribl_control_plane/models/gethealthinfoop.py

@@ -0,0 +1,17 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .error import Error, ErrorTypedDict
+from .healthstatus import HealthStatus, HealthStatusTypedDict
+from typing import Union
+from typing_extensions import TypeAliasType
+
+
+GetHealthInfoResponseTypedDict = TypeAliasType(
+    "GetHealthInfoResponseTypedDict", Union[ErrorTypedDict, HealthStatusTypedDict]
+)
+
+
+GetHealthInfoResponse = TypeAliasType(
+    "GetHealthInfoResponse", Union[Error, HealthStatus]
+)

cribl_control_plane/models/getsummaryop.py

@@ -4,7 +4,12 @@ from __future__ import annotations
 from .distributedsummary import DistributedSummary, DistributedSummaryTypedDict
 from .workertypes import WorkerTypes
 from cribl_control_plane.types import BaseModel
-from cribl_control_plane.utils import FieldMetadata, QueryParamMetadata
+from cribl_control_plane.utils import (
+    FieldMetadata,
+    QueryParamMetadata,
+    validate_open_enum,
+)
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -16,7 +21,7 @@ class GetSummaryRequestTypedDict(TypedDict):
 
 class GetSummaryRequest(BaseModel):
     mode: Annotated[
-        Optional[WorkerTypes],
+        Annotated[Optional[WorkerTypes], PlainValidator(validate_open_enum(False))],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = None
     r"""Filter for limiting the response by Cribl product: Cribl Stream (<code>worker</code>) or Cribl Edge (<code>managed-edge</code>)."""

cribl_control_plane/models/getversionshowop.py

@@ -1,10 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .gitshowresult import GitShowResult, GitShowResultTypedDict
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import FieldMetadata, QueryParamMetadata
 import pydantic
-from typing import Any, Dict, List, Optional
+from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -48,17 +49,17 @@ class GetVersionShowRequest(BaseModel):
 
 
 class GetVersionShowResponseTypedDict(TypedDict):
-    r"""a list of any objects"""
+    r"""a list of GitShowResult objects"""
 
     count: NotRequired[int]
     r"""number of items present in the items array"""
-    items: NotRequired[List[Dict[str, Any]]]
+    items: NotRequired[List[GitShowResultTypedDict]]
 
 
 class GetVersionShowResponse(BaseModel):
-    r"""a list of any objects"""
+    r"""a list of GitShowResult objects"""
 
     count: Optional[int] = None
     r"""number of items present in the items array"""
 
-    items: Optional[List[Dict[str, Any]]] = None
+    items: Optional[List[GitShowResult]] = None

cribl_control_plane/models/gitshowresult.py

@@ -0,0 +1,19 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .difffiles import DiffFiles, DiffFilesTypedDict
+from cribl_control_plane.types import BaseModel
+import pydantic
+from typing import List
+from typing_extensions import Annotated, TypedDict
+
+
+class GitShowResultTypedDict(TypedDict):
+    commit_message: str
+    diff_json: List[DiffFilesTypedDict]
+
+
+class GitShowResult(BaseModel):
+    commit_message: Annotated[str, pydantic.Field(alias="commitMessage")]
+
+    diff_json: Annotated[List[DiffFiles], pydantic.Field(alias="diffJson")]

cribl_control_plane/models/hbcriblinfo.py

@@ -1,11 +1,14 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .appmode import AppMode
 from .hbleaderinfo import HBLeaderInfo, HBLeaderInfoTypedDict
 from .lookupversions import LookupVersions, LookupVersionsTypedDict
+from cribl_control_plane import utils
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
+from enum import Enum
 import pydantic
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -34,9 +37,19 @@ class Config(BaseModel):
     version: Optional[str] = None
 
 
+class DistMode(str, Enum, metaclass=utils.OpenEnumMeta):
+    EDGE = "edge"
+    WORKER = "worker"
+    SINGLE = "single"
+    MASTER = "master"
+    MANAGED_EDGE = "managed-edge"
+    OUTPOST = "outpost"
+    SEARCH_SUPERVISOR = "search-supervisor"
+
+
 class HBCriblInfoTypedDict(TypedDict):
     config: ConfigTypedDict
-    dist_mode: AppMode
+    dist_mode: DistMode
     group: str
     guid: str
     start_time: float
@@ -48,13 +61,17 @@ class HBCriblInfoTypedDict(TypedDict):
     lookup_versions: NotRequired[LookupVersionsTypedDict]
     master: NotRequired[HBLeaderInfoTypedDict]
     pid: NotRequired[float]
+    socks_enabled: NotRequired[bool]
     version: NotRequired[str]
 
 
 class HBCriblInfo(BaseModel):
     config: Config
 
-    dist_mode: Annotated[AppMode, pydantic.Field(alias="distMode")]
+    dist_mode: Annotated[
+        Annotated[DistMode, PlainValidator(validate_open_enum(False))],
+        pydantic.Field(alias="distMode"),
+    ]
 
     group: str
 
@@ -82,4 +99,8 @@ class HBCriblInfo(BaseModel):
 
     pid: Optional[float] = None
 
+    socks_enabled: Annotated[Optional[bool], pydantic.Field(alias="socksEnabled")] = (
+        None
+    )
+
     version: Optional[str] = None

cribl_control_plane/models/healthstatus.py

@@ -1,19 +1,22 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
 from enum import Enum
 import pydantic
+from pydantic.functional_validators import PlainValidator
 from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class Role(str, Enum):
+class Role(str, Enum, metaclass=utils.OpenEnumMeta):
     PRIMARY = "primary"
     STANDBY = "standby"
 
 
-class Status(str, Enum):
+class Status(str, Enum, metaclass=utils.OpenEnumMeta):
     HEALTHY = "healthy"
     SHUTTING_DOWN = "shutting down"
     STANDBY = "standby"
@@ -26,8 +29,8 @@ class HealthStatusTypedDict(TypedDict):
 
 
 class HealthStatus(BaseModel):
-    status: Status
+    status: Annotated[Status, PlainValidator(validate_open_enum(False))]
 
     start_time: Annotated[float, pydantic.Field(alias="startTime")]
 
-    role: Optional[Role] = None
+    role: Annotated[Optional[Role], PlainValidator(validate_open_enum(False))] = None

cribl_control_plane/models/heartbeatmetadata.py

@@ -17,6 +17,7 @@ class HeartbeatMetadataTags(BaseModel):
 
 class HeartbeatMetadataAwsTypedDict(TypedDict):
     enabled: bool
+    instance_id: str
     region: str
     type: str
     zone: str
@@ -26,6 +27,8 @@ class HeartbeatMetadataAwsTypedDict(TypedDict):
 class HeartbeatMetadataAws(BaseModel):
     enabled: bool
 
+    instance_id: Annotated[str, pydantic.Field(alias="instanceId")]
+
     region: str
 
     type: str

cribl_control_plane/models/inputappscope.py

@@ -1,9 +1,12 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
 from enum import Enum
 import pydantic
+from pydantic.functional_validators import PlainValidator
 from typing import Any, List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -23,14 +26,14 @@ class InputAppscopeConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputAppscopeMode(str, Enum):
+class InputAppscopeMode(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""With Smart mode, PQ will write events to the filesystem only when it detects backpressure from the processing engine. With Always On mode, PQ will always write events directly to the queue before forwarding them to the processing engine."""
 
     SMART = "smart"
     ALWAYS = "always"
 
 
-class InputAppscopeCompression(str, Enum):
+class InputAppscopeCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
     NONE = "none"
@@ -64,7 +67,9 @@ class InputAppscopePqTypedDict(TypedDict):
 
 
 class InputAppscopePq(BaseModel):
-    mode: Optional[InputAppscopeMode] = InputAppscopeMode.ALWAYS
+    mode: Annotated[
+        Optional[InputAppscopeMode], PlainValidator(validate_open_enum(False))
+    ] = InputAppscopeMode.ALWAYS
     r"""With Smart mode, PQ will write events to the filesystem only when it detects backpressure from the processing engine. With Always On mode, PQ will always write events directly to the queue before forwarding them to the processing engine."""
 
     max_buffer_size: Annotated[
@@ -88,7 +93,9 @@ class InputAppscopePq(BaseModel):
     path: Optional[str] = "$CRIBL_HOME/state/queues"
     r"""The location for the persistent queue files. To this field's value, the system will append: /<worker-id>/inputs/<input-id>"""
 
-    compress: Optional[InputAppscopeCompression] = InputAppscopeCompression.NONE
+    compress: Annotated[
+        Optional[InputAppscopeCompression], PlainValidator(validate_open_enum(False))
+    ] = InputAppscopeCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
     pq_controls: Annotated[
@@ -144,7 +151,7 @@ class InputAppscopeFilter(BaseModel):
     r"""To override the UNIX domain socket or address/port specified in General Settings (while leaving Authentication settings as is), enter a URL."""
 
 
-class InputAppscopeDataCompressionFormat(str, Enum):
+class InputAppscopeDataCompressionFormat(str, Enum, metaclass=utils.OpenEnumMeta):
     NONE = "none"
     GZIP = "gzip"
 
@@ -176,9 +183,10 @@ class InputAppscopePersistence(BaseModel):
     max_data_time: Annotated[Optional[str], pydantic.Field(alias="maxDataTime")] = "24h"
     r"""Maximum amount of time to retain data (examples: 2h, 4d). When limit is reached, older data will be deleted."""
 
-    compress: Optional[InputAppscopeDataCompressionFormat] = (
-        InputAppscopeDataCompressionFormat.GZIP
-    )
+    compress: Annotated[
+        Optional[InputAppscopeDataCompressionFormat],
+        PlainValidator(validate_open_enum(False)),
+    ] = InputAppscopeDataCompressionFormat.GZIP
 
     dest_path: Annotated[Optional[str], pydantic.Field(alias="destPath")] = (
         "$CRIBL_HOME/state/appscope"
@@ -186,21 +194,21 @@ class InputAppscopePersistence(BaseModel):
     r"""Path to use to write metrics. Defaults to $CRIBL_HOME/state/appscope"""
 
 
-class InputAppscopeAuthenticationMethod(str, Enum):
+class InputAppscopeAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Select Manual to enter an auth token directly, or select Secret to use a text secret to authenticate"""
 
     MANUAL = "manual"
     SECRET = "secret"
 
 
-class InputAppscopeMinimumTLSVersion(str, Enum):
+class InputAppscopeMinimumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
     TL_SV1_3 = "TLSv1.3"
 
 
-class InputAppscopeMaximumTLSVersion(str, Enum):
+class InputAppscopeMaximumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
@@ -259,11 +267,19 @@ class InputAppscopeTLSSettingsServerSide(BaseModel):
     ] = None
 
     min_version: Annotated[
-        Optional[InputAppscopeMinimumTLSVersion], pydantic.Field(alias="minVersion")
+        Annotated[
+            Optional[InputAppscopeMinimumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="minVersion"),
     ] = None
 
     max_version: Annotated[
-        Optional[InputAppscopeMaximumTLSVersion], pydantic.Field(alias="maxVersion")
+        Annotated[
+            Optional[InputAppscopeMaximumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="maxVersion"),
     ] = None
 
 
@@ -410,7 +426,11 @@ class InputAppscope(BaseModel):
     persistence: Optional[InputAppscopePersistence] = None
 
     auth_type: Annotated[
-        Optional[InputAppscopeAuthenticationMethod], pydantic.Field(alias="authType")
+        Annotated[
+            Optional[InputAppscopeAuthenticationMethod],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="authType"),
     ] = InputAppscopeAuthenticationMethod.MANUAL
     r"""Select Manual to enter an auth token directly, or select Secret to use a text secret to authenticate"""
 

cribl_control_plane/models/inputazureblob.py

@@ -1,9 +1,12 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
 from enum import Enum
 import pydantic
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -23,14 +26,14 @@ class InputAzureBlobConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputAzureBlobMode(str, Enum):
+class InputAzureBlobMode(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""With Smart mode, PQ will write events to the filesystem only when it detects backpressure from the processing engine. With Always On mode, PQ will always write events directly to the queue before forwarding them to the processing engine."""
 
     SMART = "smart"
     ALWAYS = "always"
 
 
-class InputAzureBlobCompression(str, Enum):
+class InputAzureBlobCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
     NONE = "none"
@@ -64,7 +67,9 @@ class InputAzureBlobPqTypedDict(TypedDict):
 
 
 class InputAzureBlobPq(BaseModel):
-    mode: Optional[InputAzureBlobMode] = InputAzureBlobMode.ALWAYS
+    mode: Annotated[
+        Optional[InputAzureBlobMode], PlainValidator(validate_open_enum(False))
+    ] = InputAzureBlobMode.ALWAYS
     r"""With Smart mode, PQ will write events to the filesystem only when it detects backpressure from the processing engine. With Always On mode, PQ will always write events directly to the queue before forwarding them to the processing engine."""
 
     max_buffer_size: Annotated[
@@ -88,7 +93,9 @@ class InputAzureBlobPq(BaseModel):
     path: Optional[str] = "$CRIBL_HOME/state/queues"
     r"""The location for the persistent queue files. To this field's value, the system will append: /<worker-id>/inputs/<input-id>"""
 
-    compress: Optional[InputAzureBlobCompression] = InputAzureBlobCompression.NONE
+    compress: Annotated[
+        Optional[InputAzureBlobCompression], PlainValidator(validate_open_enum(False))
+    ] = InputAzureBlobCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
     pq_controls: Annotated[
@@ -109,7 +116,7 @@ class InputAzureBlobMetadatum(BaseModel):
     r"""JavaScript expression to compute field's value, enclosed in quotes or backticks. (Can evaluate to a constant.)"""
 
 
-class InputAzureBlobAuthenticationMethod(str, Enum):
+class InputAzureBlobAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMeta):
     MANUAL = "manual"
     SECRET = "secret"
     CLIENT_SECRET = "clientSecret"
@@ -270,7 +277,11 @@ class InputAzureBlob(BaseModel):
     r"""The maximum time allowed for downloading a Parquet chunk. Processing will stop if a chunk cannot be downloaded within the time specified."""
 
     auth_type: Annotated[
-        Optional[InputAzureBlobAuthenticationMethod], pydantic.Field(alias="authType")
+        Annotated[
+            Optional[InputAzureBlobAuthenticationMethod],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="authType"),
     ] = InputAzureBlobAuthenticationMethod.MANUAL
 
     description: Optional[str] = None