cribl-control-plane 0.3.0b3__py3-none-any.whl → 0.3.0b12__py3-none-any.whl

cribl_control_plane/models/inputwizwebhook.py

@@ -1,13 +1,14 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from cribl_control_plane import utils
+from cribl_control_plane import models, 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 import field_serializer
 from pydantic.functional_validators import PlainValidator
-from typing import Any, List, Optional
+from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -106,6 +107,24 @@ class InputWizWebhookPq(BaseModel):
         Optional[InputWizWebhookPqControls], pydantic.Field(alias="pqControls")
     ] = None
 
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWizWebhookMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWizWebhookCompression(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputWizWebhookMinimumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
@@ -123,6 +142,12 @@ class InputWizWebhookMaximumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
 
 class InputWizWebhookTLSSettingsServerSideTypedDict(TypedDict):
     disabled: NotRequired[bool]
+    request_cert: NotRequired[bool]
+    r"""Require clients to present their certificates. Used to perform client authentication using SSL certs."""
+    reject_unauthorized: NotRequired[bool]
+    r"""Reject certificates not authorized by a CA in the CA certificate path or by another trusted CA (such as the system's)"""
+    common_name_regex: NotRequired[str]
+    r"""Regex matching allowable common names in peer certificates' subject attribute"""
     certificate_name: NotRequired[str]
     r"""The name of the predefined certificate"""
     priv_key_path: NotRequired[str]
@@ -133,10 +158,6 @@ class InputWizWebhookTLSSettingsServerSideTypedDict(TypedDict):
     r"""Path on server containing certificates to use. PEM format. Can reference $ENV_VARS."""
     ca_path: NotRequired[str]
     r"""Path on server containing CA certificates to use. PEM format. Can reference $ENV_VARS."""
-    request_cert: NotRequired[bool]
-    r"""Require clients to present their certificates. Used to perform client authentication using SSL certs."""
-    reject_unauthorized: NotRequired[Any]
-    common_name_regex: NotRequired[Any]
     min_version: NotRequired[InputWizWebhookMinimumTLSVersion]
     max_version: NotRequired[InputWizWebhookMaximumTLSVersion]
 
@@ -144,6 +165,19 @@ class InputWizWebhookTLSSettingsServerSideTypedDict(TypedDict):
 class InputWizWebhookTLSSettingsServerSide(BaseModel):
     disabled: Optional[bool] = True
 
+    request_cert: Annotated[Optional[bool], pydantic.Field(alias="requestCert")] = False
+    r"""Require clients to present their certificates. Used to perform client authentication using SSL certs."""
+
+    reject_unauthorized: Annotated[
+        Optional[bool], pydantic.Field(alias="rejectUnauthorized")
+    ] = True
+    r"""Reject certificates not authorized by a CA in the CA certificate path or by another trusted CA (such as the system's)"""
+
+    common_name_regex: Annotated[
+        Optional[str], pydantic.Field(alias="commonNameRegex")
+    ] = "/.*/"
+    r"""Regex matching allowable common names in peer certificates' subject attribute"""
+
     certificate_name: Annotated[
         Optional[str], pydantic.Field(alias="certificateName")
     ] = None
@@ -161,17 +195,6 @@ class InputWizWebhookTLSSettingsServerSide(BaseModel):
     ca_path: Annotated[Optional[str], pydantic.Field(alias="caPath")] = None
     r"""Path on server containing CA certificates to use. PEM format. Can reference $ENV_VARS."""
 
-    request_cert: Annotated[Optional[bool], pydantic.Field(alias="requestCert")] = False
-    r"""Require clients to present their certificates. Used to perform client authentication using SSL certs."""
-
-    reject_unauthorized: Annotated[
-        Optional[Any], pydantic.Field(alias="rejectUnauthorized")
-    ] = None
-
-    common_name_regex: Annotated[
-        Optional[Any], pydantic.Field(alias="commonNameRegex")
-    ] = None
-
     min_version: Annotated[
         Annotated[
             Optional[InputWizWebhookMinimumTLSVersion],
@@ -188,6 +211,24 @@ class InputWizWebhookTLSSettingsServerSide(BaseModel):
         pydantic.Field(alias="maxVersion"),
     ] = None
 
+    @field_serializer("min_version")
+    def serialize_min_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWizWebhookMinimumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("max_version")
+    def serialize_max_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWizWebhookMaximumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputWizWebhookMetadatumTypedDict(TypedDict):
     name: str

cribl_control_plane/models/inputzscalerhec.py

@@ -1,11 +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 import models, 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 import field_serializer
 from pydantic.functional_validators import PlainValidator
 from typing import Any, List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
@@ -106,6 +107,24 @@ class InputZscalerHecPq(BaseModel):
         Optional[InputZscalerHecPqControls], pydantic.Field(alias="pqControls")
     ] = None
 
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputZscalerHecMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputZscalerHecCompression(value)
+            except ValueError:
+                return value
+        return value
+
 
 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"""
@@ -166,6 +185,15 @@ class InputZscalerHecAuthToken(BaseModel):
     metadata: Optional[List[InputZscalerHecAuthTokenMetadatum]] = None
     r"""Fields to add to events referencing this token"""
 
+    @field_serializer("auth_type")
+    def serialize_auth_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputZscalerHecAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputZscalerHecMinimumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
@@ -183,6 +211,12 @@ class InputZscalerHecMaximumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
 
 class InputZscalerHecTLSSettingsServerSideTypedDict(TypedDict):
     disabled: NotRequired[bool]
+    request_cert: NotRequired[bool]
+    r"""Require clients to present their certificates. Used to perform client authentication using SSL certs."""
+    reject_unauthorized: NotRequired[bool]
+    r"""Reject certificates not authorized by a CA in the CA certificate path or by another trusted CA (such as the system's)"""
+    common_name_regex: NotRequired[str]
+    r"""Regex matching allowable common names in peer certificates' subject attribute"""
     certificate_name: NotRequired[str]
     r"""The name of the predefined certificate"""
     priv_key_path: NotRequired[str]
@@ -193,10 +227,6 @@ class InputZscalerHecTLSSettingsServerSideTypedDict(TypedDict):
     r"""Path on server containing certificates to use. PEM format. Can reference $ENV_VARS."""
     ca_path: NotRequired[str]
     r"""Path on server containing CA certificates to use. PEM format. Can reference $ENV_VARS."""
-    request_cert: NotRequired[bool]
-    r"""Require clients to present their certificates. Used to perform client authentication using SSL certs."""
-    reject_unauthorized: NotRequired[Any]
-    common_name_regex: NotRequired[Any]
     min_version: NotRequired[InputZscalerHecMinimumTLSVersion]
     max_version: NotRequired[InputZscalerHecMaximumTLSVersion]
 
@@ -204,6 +234,19 @@ class InputZscalerHecTLSSettingsServerSideTypedDict(TypedDict):
 class InputZscalerHecTLSSettingsServerSide(BaseModel):
     disabled: Optional[bool] = True
 
+    request_cert: Annotated[Optional[bool], pydantic.Field(alias="requestCert")] = False
+    r"""Require clients to present their certificates. Used to perform client authentication using SSL certs."""
+
+    reject_unauthorized: Annotated[
+        Optional[bool], pydantic.Field(alias="rejectUnauthorized")
+    ] = True
+    r"""Reject certificates not authorized by a CA in the CA certificate path or by another trusted CA (such as the system's)"""
+
+    common_name_regex: Annotated[
+        Optional[str], pydantic.Field(alias="commonNameRegex")
+    ] = "/.*/"
+    r"""Regex matching allowable common names in peer certificates' subject attribute"""
+
     certificate_name: Annotated[
         Optional[str], pydantic.Field(alias="certificateName")
     ] = None
@@ -221,17 +264,6 @@ class InputZscalerHecTLSSettingsServerSide(BaseModel):
     ca_path: Annotated[Optional[str], pydantic.Field(alias="caPath")] = None
     r"""Path on server containing CA certificates to use. PEM format. Can reference $ENV_VARS."""
 
-    request_cert: Annotated[Optional[bool], pydantic.Field(alias="requestCert")] = False
-    r"""Require clients to present their certificates. Used to perform client authentication using SSL certs."""
-
-    reject_unauthorized: Annotated[
-        Optional[Any], pydantic.Field(alias="rejectUnauthorized")
-    ] = None
-
-    common_name_regex: Annotated[
-        Optional[Any], pydantic.Field(alias="commonNameRegex")
-    ] = None
-
     min_version: Annotated[
         Annotated[
             Optional[InputZscalerHecMinimumTLSVersion],
@@ -248,6 +280,24 @@ class InputZscalerHecTLSSettingsServerSide(BaseModel):
         pydantic.Field(alias="maxVersion"),
     ] = None
 
+    @field_serializer("min_version")
+    def serialize_min_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputZscalerHecMinimumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("max_version")
+    def serialize_max_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputZscalerHecMaximumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputZscalerHecMetadatumTypedDict(TypedDict):
     name: str

cribl_control_plane/models/jobinfo.py

@@ -4,14 +4,20 @@ from __future__ import annotations
 from .jobstatus import JobStatus, JobStatusTypedDict
 from .runnablejob import RunnableJob, RunnableJobTypedDict
 from cribl_control_plane.types import BaseModel
-from typing import Dict, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Dict, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
+
+
+StatsTypedDict = TypeAliasType("StatsTypedDict", Union[float, Dict[str, float]])
+
+
+Stats = TypeAliasType("Stats", Union[float, Dict[str, float]])
 
 
 class JobInfoTypedDict(TypedDict):
     args: RunnableJobTypedDict
     id: str
-    stats: Dict[str, float]
+    stats: Dict[str, StatsTypedDict]
     status: JobStatusTypedDict
     keep: NotRequired[bool]
 
@@ -21,7 +27,7 @@ class JobInfo(BaseModel):
 
     id: str
 
-    stats: Dict[str, float]
+    stats: Dict[str, Stats]
 
     status: JobStatus
 

cribl_control_plane/models/jobstatus.py

@@ -1,17 +1,48 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import models, utils
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
+from enum import Enum
+from pydantic import field_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import Any, Dict, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class State(int, Enum, metaclass=utils.OpenEnumMeta):
+    r"""State of the Job"""
+
+    INITIALIZING = 0
+    PENDING = 1
+    RUNNING = 2
+    PAUSED = 3
+    CANCELLED = 4
+    FINISHED = 5
+    FAILED = 6
+    ORPHANED = 7
+    UNKNOWN = 8
+    LENGTH = 9
 
 
 class JobStatusTypedDict(TypedDict):
-    state: Dict[str, Any]
+    state: State
+    r"""State of the Job"""
     reason: NotRequired[Dict[str, Any]]
 
 
 class JobStatus(BaseModel):
-    state: Dict[str, Any]
+    state: Annotated[State, PlainValidator(validate_open_enum(True))]
+    r"""State of the Job"""
 
     reason: Optional[Dict[str, Any]] = None
+
+    @field_serializer("state")
+    def serialize_state(self, value):
+        if isinstance(value, str):
+            try:
+                return models.State(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/lakedatasetmetrics.py

@@ -0,0 +1,17 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from cribl_control_plane.types import BaseModel
+import pydantic
+from typing_extensions import Annotated, TypedDict
+
+
+class LakeDatasetMetricsTypedDict(TypedDict):
+    current_size_bytes: float
+    metrics_date: str
+
+
+class LakeDatasetMetrics(BaseModel):
+    current_size_bytes: Annotated[float, pydantic.Field(alias="currentSizeBytes")]
+
+    metrics_date: Annotated[str, pydantic.Field(alias="metricsDate")]

cribl_control_plane/models/listconfiggroupbyproductop.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 from .configgroup import ConfigGroup, ConfigGroupTypedDict
 from .productscore import ProductsCore
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import (
     FieldMetadata,
@@ -10,6 +11,7 @@ from cribl_control_plane.utils import (
     QueryParamMetadata,
     validate_open_enum,
 )
+from pydantic import field_serializer
 from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
@@ -35,6 +37,15 @@ class ListConfigGroupByProductRequest(BaseModel):
     ] = None
     r"""Comma-separated list of additional properties to include in the response. Available values are <code>git.commit</code>, <code>git.localChanges</code>, and <code>git.log</code>."""
 
+    @field_serializer("product")
+    def serialize_product(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ProductsCore(value)
+            except ValueError:
+                return value
+        return value
+
 
 class ListConfigGroupByProductResponseTypedDict(TypedDict):
     r"""a list of ConfigGroup objects"""

cribl_control_plane/models/masterworkerentry.py

@@ -4,11 +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 import models, 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 import field_serializer
 from pydantic.functional_validators import PlainValidator
 from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
@@ -87,3 +88,12 @@ class MasterWorkerEntry(BaseModel):
     ] = None
 
     workers: Optional[MasterWorkerEntryWorkers] = None
+
+    @field_serializer("type")
+    def serialize_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.MasterWorkerEntryType(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/nodeupgradestatus.py

@@ -5,8 +5,10 @@ from .nodeactiveupgradestatus import NodeActiveUpgradeStatus
 from .nodefailedupgradestatus import NodeFailedUpgradeStatus
 from .nodeskippedupgradestatus import NodeSkippedUpgradeStatus
 from .nodeupgradestate import NodeUpgradeState
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import validate_open_enum
+from pydantic import field_serializer
 from pydantic.functional_validators import PlainValidator
 from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
@@ -36,3 +38,39 @@ class NodeUpgradeStatus(BaseModel):
     skipped: Annotated[
         Optional[NodeSkippedUpgradeStatus], PlainValidator(validate_open_enum(True))
     ] = None
+
+    @field_serializer("active")
+    def serialize_active(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeActiveUpgradeStatus(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("failed")
+    def serialize_failed(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeFailedUpgradeStatus(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("skipped")
+    def serialize_skipped(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeSkippedUpgradeStatus(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("state")
+    def serialize_state(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeUpgradeState(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/output.py

@@ -98,64 +98,64 @@ OutputTypedDict = TypeAliasType(
         OutputNetflowTypedDict,
         OutputDiskSpoolTypedDict,
         OutputRingTypedDict,
-        OutputStatsdExtTypedDict,
         OutputGraphiteTypedDict,
+        OutputStatsdExtTypedDict,
         OutputStatsdTypedDict,
         OutputGooglePubsubTypedDict,
-        OutputCriblTCPTypedDict,
+        OutputExabeamTypedDict,
         OutputSplunkTypedDict,
         OutputSnsTypedDict,
-        OutputCloudwatchTypedDict,
+        OutputCriblTCPTypedDict,
         OutputAzureEventhubTypedDict,
-        OutputWavefrontTypedDict,
+        OutputCloudwatchTypedDict,
         OutputSignalfxTypedDict,
         OutputHoneycombTypedDict,
+        OutputWavefrontTypedDict,
+        OutputHumioHecTypedDict,
         OutputSumoLogicTypedDict,
         OutputCrowdstrikeNextGenSiemTypedDict,
-        OutputHumioHecTypedDict,
         OutputTcpjsonTypedDict,
         OutputElasticCloudTypedDict,
-        OutputKinesisTypedDict,
+        OutputNewrelicEventsTypedDict,
+        OutputFilesystemTypedDict,
         OutputConfluentCloudTypedDict,
         OutputKafkaTypedDict,
-        OutputExabeamTypedDict,
-        OutputNewrelicEventsTypedDict,
         OutputAzureLogsTypedDict,
+        OutputKinesisTypedDict,
         OutputSplunkLbTypedDict,
         OutputSyslogTypedDict,
         OutputSqsTypedDict,
-        OutputNewrelicTypedDict,
         OutputCriblHTTPTypedDict,
+        OutputNewrelicTypedDict,
         OutputXsiamTypedDict,
-        OutputFilesystemTypedDict,
         OutputDatasetTypedDict,
         OutputLokiTypedDict,
-        OutputSplunkHecTypedDict,
+        OutputCriblLakeTypedDict,
         OutputDynatraceHTTPTypedDict,
         OutputServiceNowTypedDict,
+        OutputSplunkHecTypedDict,
         OutputChronicleTypedDict,
         OutputDynatraceOtlpTypedDict,
-        OutputGoogleChronicleTypedDict,
+        OutputDatabricksTypedDict,
         OutputElasticTypedDict,
         OutputDatadogTypedDict,
-        OutputCriblLakeTypedDict,
-        OutputDatabricksTypedDict,
+        OutputGoogleChronicleTypedDict,
         OutputPrometheusTypedDict,
-        OutputMskTypedDict,
         OutputSentinelOneAiSiemTypedDict,
-        OutputSentinelTypedDict,
-        OutputInfluxdbTypedDict,
+        OutputMskTypedDict,
         OutputGoogleCloudStorageTypedDict,
+        OutputSentinelTypedDict,
         OutputAzureBlobTypedDict,
-        OutputOpenTelemetryTypedDict,
+        OutputInfluxdbTypedDict,
         OutputMinioTypedDict,
-        OutputClickHouseTypedDict,
         OutputSecurityLakeTypedDict,
-        OutputS3TypedDict,
+        OutputOpenTelemetryTypedDict,
+        OutputClickHouseTypedDict,
         OutputDlS3TypedDict,
+        OutputS3TypedDict,
         OutputWebhookTypedDict,
-        OutputAzureDataExplorerTypedDict,
         OutputGoogleCloudLoggingTypedDict,
+        OutputAzureDataExplorerTypedDict,
         OutputGrafanaCloudTypedDict,
     ],
 )

cribl_control_plane/models/outputazureblob.py

@@ -1,11 +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 import models, 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 import field_serializer
 from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
@@ -184,6 +185,8 @@ class OutputAzureBlobTypedDict(TypedDict):
     r"""Compression level to apply before moving files to final destination"""
     automatic_schema: NotRequired[bool]
     r"""Automatically calculate the schema based on the events of each Parquet file generated"""
+    parquet_schema: NotRequired[str]
+    r"""To add a new schema, navigate to Processing > Knowledge > Parquet Schemas"""
     parquet_version: NotRequired[OutputAzureBlobParquetVersion]
     r"""Determines which data types are supported and how they are represented"""
     parquet_data_page_version: NotRequired[OutputAzureBlobDataPageVersion]
@@ -387,6 +390,11 @@ class OutputAzureBlob(BaseModel):
     ] = False
     r"""Automatically calculate the schema based on the events of each Parquet file generated"""
 
+    parquet_schema: Annotated[Optional[str], pydantic.Field(alias="parquetSchema")] = (
+        None
+    )
+    r"""To add a new schema, navigate to Processing > Knowledge > Parquet Schemas"""
+
     parquet_version: Annotated[
         Annotated[
             Optional[OutputAzureBlobParquetVersion],
@@ -487,3 +495,84 @@ class OutputAzureBlob(BaseModel):
     r"""Select or create a stored text secret"""
 
     certificate: Optional[OutputAzureBlobCertificate] = None
+
+    @field_serializer("format_")
+    def serialize_format_(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobDataFormat(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("on_backpressure")
+    def serialize_on_backpressure(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobBackpressureBehavior(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("on_disk_full_backpressure")
+    def serialize_on_disk_full_backpressure(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobDiskSpaceProtection(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("auth_type")
+    def serialize_auth_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("storage_class")
+    def serialize_storage_class(self, value):
+        if isinstance(value, str):
+            try:
+                return models.BlobAccessTier(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobCompression(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compression_level")
+    def serialize_compression_level(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobCompressionLevel(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("parquet_version")
+    def serialize_parquet_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobParquetVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("parquet_data_page_version")
+    def serialize_parquet_data_page_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobDataPageVersion(value)
+            except ValueError:
+                return value
+        return value