cribl-control-plane 0.0.44__py3-none-any.whl → 0.0.44a2__py3-none-any.whl

cribl_control_plane/models/inputzscalerhec.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 InputZscalerHecConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputZscalerHecMode(str, Enum):
+class InputZscalerHecMode(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 InputZscalerHecCompression(str, Enum):
+class InputZscalerHecCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
     NONE = "none"
@@ -64,7 +67,9 @@ class InputZscalerHecPqTypedDict(TypedDict):
 
 
 class InputZscalerHecPq(BaseModel):
-    mode: Optional[InputZscalerHecMode] = InputZscalerHecMode.ALWAYS
+    mode: Annotated[
+        Optional[InputZscalerHecMode], PlainValidator(validate_open_enum(False))
+    ] = InputZscalerHecMode.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 InputZscalerHecPq(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[InputZscalerHecCompression] = InputZscalerHecCompression.NONE
+    compress: Annotated[
+        Optional[InputZscalerHecCompression], PlainValidator(validate_open_enum(False))
+    ] = InputZscalerHecCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
     pq_controls: Annotated[
@@ -96,7 +103,7 @@ class InputZscalerHecPq(BaseModel):
     ] = None
 
 
-class InputZscalerHecAuthenticationMethod(str, Enum):
+class InputZscalerHecAuthenticationMethod(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"
@@ -133,7 +140,11 @@ class InputZscalerHecAuthToken(BaseModel):
     token: Any
 
     auth_type: Annotated[
-        Optional[InputZscalerHecAuthenticationMethod], pydantic.Field(alias="authType")
+        Annotated[
+            Optional[InputZscalerHecAuthenticationMethod],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="authType"),
     ] = InputZscalerHecAuthenticationMethod.MANUAL
     r"""Select Manual to enter an auth token directly, or select Secret to use a text secret to authenticate"""
 
@@ -152,14 +163,14 @@ class InputZscalerHecAuthToken(BaseModel):
     r"""Fields to add to events referencing this token"""
 
 
-class InputZscalerHecMinimumTLSVersion(str, Enum):
+class InputZscalerHecMinimumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
     TL_SV1_3 = "TLSv1.3"
 
 
-class InputZscalerHecMaximumTLSVersion(str, Enum):
+class InputZscalerHecMaximumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
@@ -218,11 +229,19 @@ class InputZscalerHecTLSSettingsServerSide(BaseModel):
     ] = None
 
     min_version: Annotated[
-        Optional[InputZscalerHecMinimumTLSVersion], pydantic.Field(alias="minVersion")
+        Annotated[
+            Optional[InputZscalerHecMinimumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="minVersion"),
     ] = None
 
     max_version: Annotated[
-        Optional[InputZscalerHecMaximumTLSVersion], pydantic.Field(alias="maxVersion")
+        Annotated[
+            Optional[InputZscalerHecMaximumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="maxVersion"),
     ] = None
 
 

cribl_control_plane/models/lakehouseconnectiontype.py

@@ -1,9 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from enum import Enum
 
 
-class LakehouseConnectionType(str, Enum):
+class LakehouseConnectionType(str, Enum, metaclass=utils.OpenEnumMeta):
     CACHE = "cache"
     ZERO_POINT = "zeroPoint"

cribl_control_plane/models/listconfiggroupbyproductop.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
 
@@ -22,7 +24,7 @@ class ListConfigGroupByProductRequestTypedDict(TypedDict):
 
 class ListConfigGroupByProductRequest(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/masterworkerentry.py

@@ -4,9 +4,12 @@ from __future__ import annotations
 from .heartbeatmetadata import HeartbeatMetadata, HeartbeatMetadataTypedDict
 from .nodeprovidedinfo import NodeProvidedInfo, NodeProvidedInfoTypedDict
 from .nodeupgradestatus import NodeUpgradeStatus, NodeUpgradeStatusTypedDict
+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
 
@@ -19,7 +22,7 @@ class LastMetrics(BaseModel):
     pass
 
 
-class MasterWorkerEntryType(str, Enum):
+class MasterWorkerEntryType(str, Enum, metaclass=utils.OpenEnumMeta):
     INFO = "info"
     REQ = "req"
     RESP = "resp"
@@ -79,6 +82,8 @@ class MasterWorkerEntry(BaseModel):
 
     status: Optional[str] = None
 
-    type: Optional[MasterWorkerEntryType] = None
+    type: Annotated[
+        Optional[MasterWorkerEntryType], PlainValidator(validate_open_enum(False))
+    ] = None
 
     workers: Optional[MasterWorkerEntryWorkers] = None

cribl_control_plane/models/nodeactiveupgradestatus.py

@@ -1,10 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from enum import Enum
 
 
-class NodeActiveUpgradeStatus(int, Enum):
+class NodeActiveUpgradeStatus(int, Enum, metaclass=utils.OpenEnumMeta):
     ZERO = 0
     ONE = 1
     TWO = 2

cribl_control_plane/models/nodefailedupgradestatus.py

@@ -1,9 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from enum import Enum
 
 
-class NodeFailedUpgradeStatus(int, Enum):
+class NodeFailedUpgradeStatus(int, Enum, metaclass=utils.OpenEnumMeta):
     ZERO = 0
     ONE = 1

cribl_control_plane/models/nodeskippedupgradestatus.py

@@ -1,10 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from enum import Enum
 
 
-class NodeSkippedUpgradeStatus(int, Enum):
+class NodeSkippedUpgradeStatus(int, Enum, metaclass=utils.OpenEnumMeta):
     ZERO = 0
     ONE = 1
     TWO = 2

cribl_control_plane/models/nodeupgradestate.py

@@ -1,10 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from enum import Enum
 
 
-class NodeUpgradeState(int, Enum):
+class NodeUpgradeState(int, Enum, metaclass=utils.OpenEnumMeta):
     ZERO = 0
     ONE = 1
     TWO = 2

cribl_control_plane/models/nodeupgradestatus.py

@@ -6,8 +6,10 @@ from .nodefailedupgradestatus import NodeFailedUpgradeStatus
 from .nodeskippedupgradestatus import NodeSkippedUpgradeStatus
 from .nodeupgradestate import NodeUpgradeState
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
+from pydantic.functional_validators import PlainValidator
 from typing import Optional
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class NodeUpgradeStatusTypedDict(TypedDict):
@@ -19,12 +21,18 @@ class NodeUpgradeStatusTypedDict(TypedDict):
 
 
 class NodeUpgradeStatus(BaseModel):
-    state: NodeUpgradeState
+    state: Annotated[NodeUpgradeState, PlainValidator(validate_open_enum(True))]
 
     timestamp: float
 
-    active: Optional[NodeActiveUpgradeStatus] = None
+    active: Annotated[
+        Optional[NodeActiveUpgradeStatus], PlainValidator(validate_open_enum(True))
+    ] = None
 
-    failed: Optional[NodeFailedUpgradeStatus] = None
+    failed: Annotated[
+        Optional[NodeFailedUpgradeStatus], PlainValidator(validate_open_enum(True))
+    ] = None
 
-    skipped: Optional[NodeSkippedUpgradeStatus] = None
+    skipped: Annotated[
+        Optional[NodeSkippedUpgradeStatus], PlainValidator(validate_open_enum(True))
+    ] = None

cribl_control_plane/models/outputazureblob.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
 
@@ -12,7 +15,7 @@ class OutputAzureBlobType(str, Enum):
     AZURE_BLOB = "azure_blob"
 
 
-class OutputAzureBlobDataFormat(str, Enum):
+class OutputAzureBlobDataFormat(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Format of the output data"""
 
     JSON = "json"
@@ -20,28 +23,28 @@ class OutputAzureBlobDataFormat(str, Enum):
     PARQUET = "parquet"
 
 
-class OutputAzureBlobBackpressureBehavior(str, Enum):
+class OutputAzureBlobBackpressureBehavior(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""How to handle events when all receivers are exerting backpressure"""
 
     BLOCK = "block"
     DROP = "drop"
 
 
-class OutputAzureBlobDiskSpaceProtection(str, Enum):
+class OutputAzureBlobDiskSpaceProtection(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""How to handle events when disk space is below the global 'Min free disk space' limit"""
 
     BLOCK = "block"
     DROP = "drop"
 
 
-class OutputAzureBlobAuthenticationMethod(str, Enum):
+class OutputAzureBlobAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMeta):
     MANUAL = "manual"
     SECRET = "secret"
     CLIENT_SECRET = "clientSecret"
     CLIENT_CERT = "clientCert"
 
 
-class BlobAccessTier(str, Enum):
+class BlobAccessTier(str, Enum, metaclass=utils.OpenEnumMeta):
     INFERRED = "Inferred"
     HOT = "Hot"
     COOL = "Cool"
@@ -49,14 +52,14 @@ class BlobAccessTier(str, Enum):
     ARCHIVE = "Archive"
 
 
-class OutputAzureBlobCompression(str, Enum):
+class OutputAzureBlobCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Data compression format to apply to HTTP content before it is delivered"""
 
     NONE = "none"
     GZIP = "gzip"
 
 
-class OutputAzureBlobCompressionLevel(str, Enum):
+class OutputAzureBlobCompressionLevel(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Compression level to apply before moving files to final destination"""
 
     BEST_SPEED = "best_speed"
@@ -64,7 +67,7 @@ class OutputAzureBlobCompressionLevel(str, Enum):
     BEST_COMPRESSION = "best_compression"
 
 
-class OutputAzureBlobParquetVersion(str, Enum):
+class OutputAzureBlobParquetVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Determines which data types are supported and how they are represented"""
 
     PARQUET_1_0 = "PARQUET_1_0"
@@ -72,7 +75,7 @@ class OutputAzureBlobParquetVersion(str, Enum):
     PARQUET_2_6 = "PARQUET_2_6"
 
 
-class OutputAzureBlobDataPageVersion(str, Enum):
+class OutputAzureBlobDataPageVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Serialization format of data pages. Note that some reader implementations use Data page V2's attributes to work more efficiently, while others ignore it."""
 
     DATA_PAGE_V1 = "DATA_PAGE_V1"
@@ -261,7 +264,11 @@ class OutputAzureBlob(BaseModel):
     r"""JavaScript expression defining how files are partitioned and organized. Default is date-based. If blank, Stream will fall back to the event's __partition field value – if present – otherwise to each location's root directory."""
 
     format_: Annotated[
-        Optional[OutputAzureBlobDataFormat], pydantic.Field(alias="format")
+        Annotated[
+            Optional[OutputAzureBlobDataFormat],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="format"),
     ] = OutputAzureBlobDataFormat.JSON
     r"""Format of the output data"""
 
@@ -304,7 +311,10 @@ class OutputAzureBlob(BaseModel):
     r"""Buffer size used to write to a file"""
 
     on_backpressure: Annotated[
-        Optional[OutputAzureBlobBackpressureBehavior],
+        Annotated[
+            Optional[OutputAzureBlobBackpressureBehavior],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="onBackpressure"),
     ] = OutputAzureBlobBackpressureBehavior.BLOCK
     r"""How to handle events when all receivers are exerting backpressure"""
@@ -315,26 +325,39 @@ class OutputAzureBlob(BaseModel):
     r"""If a file fails to move to its final destination after the maximum number of retries, move it to a designated directory to prevent further errors"""
 
     on_disk_full_backpressure: Annotated[
-        Optional[OutputAzureBlobDiskSpaceProtection],
+        Annotated[
+            Optional[OutputAzureBlobDiskSpaceProtection],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="onDiskFullBackpressure"),
     ] = OutputAzureBlobDiskSpaceProtection.BLOCK
     r"""How to handle events when disk space is below the global 'Min free disk space' limit"""
 
     auth_type: Annotated[
-        Optional[OutputAzureBlobAuthenticationMethod], pydantic.Field(alias="authType")
+        Annotated[
+            Optional[OutputAzureBlobAuthenticationMethod],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="authType"),
     ] = OutputAzureBlobAuthenticationMethod.MANUAL
 
     storage_class: Annotated[
-        Optional[BlobAccessTier], pydantic.Field(alias="storageClass")
+        Annotated[Optional[BlobAccessTier], PlainValidator(validate_open_enum(False))],
+        pydantic.Field(alias="storageClass"),
     ] = BlobAccessTier.INFERRED
 
     description: Optional[str] = None
 
-    compress: Optional[OutputAzureBlobCompression] = OutputAzureBlobCompression.GZIP
+    compress: Annotated[
+        Optional[OutputAzureBlobCompression], PlainValidator(validate_open_enum(False))
+    ] = OutputAzureBlobCompression.GZIP
     r"""Data compression format to apply to HTTP content before it is delivered"""
 
     compression_level: Annotated[
-        Optional[OutputAzureBlobCompressionLevel],
+        Annotated[
+            Optional[OutputAzureBlobCompressionLevel],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="compressionLevel"),
     ] = OutputAzureBlobCompressionLevel.BEST_SPEED
     r"""Compression level to apply before moving files to final destination"""
@@ -345,12 +368,19 @@ class OutputAzureBlob(BaseModel):
     r"""Automatically calculate the schema based on the events of each Parquet file generated"""
 
     parquet_version: Annotated[
-        Optional[OutputAzureBlobParquetVersion], pydantic.Field(alias="parquetVersion")
+        Annotated[
+            Optional[OutputAzureBlobParquetVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="parquetVersion"),
     ] = OutputAzureBlobParquetVersion.PARQUET_2_6
     r"""Determines which data types are supported and how they are represented"""
 
     parquet_data_page_version: Annotated[
-        Optional[OutputAzureBlobDataPageVersion],
+        Annotated[
+            Optional[OutputAzureBlobDataPageVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="parquetDataPageVersion"),
     ] = OutputAzureBlobDataPageVersion.DATA_PAGE_V2
     r"""Serialization format of data pages. Note that some reader implementations use Data page V2's attributes to work more efficiently, while others ignore it."""