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

cribl_control_plane/models/cribllakedataset.py

@@ -6,11 +6,12 @@ from .lakedatasetsearchconfig import (
     LakeDatasetSearchConfig,
     LakeDatasetSearchConfigTypedDict,
 )
-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
@@ -78,3 +79,12 @@ class CriblLakeDataset(BaseModel):
     ] = None
 
     view_name: Annotated[Optional[str], pydantic.Field(alias="viewName")] = None
+
+    @field_serializer("format_")
+    def serialize_format_(self, value):
+        if isinstance(value, str):
+            try:
+                return models.CriblLakeDatasetFormat(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/cribllakedatasetupdate.py

@@ -6,11 +6,12 @@ from .lakedatasetsearchconfig import (
     LakeDatasetSearchConfig,
     LakeDatasetSearchConfigTypedDict,
 )
-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
@@ -79,3 +80,12 @@ class CriblLakeDatasetUpdate(BaseModel):
     ] = None
 
     view_name: Annotated[Optional[str], pydantic.Field(alias="viewName")] = None
+
+    @field_serializer("format_")
+    def serialize_format_(self, value):
+        if isinstance(value, str):
+            try:
+                return models.CriblLakeDatasetUpdateFormat(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/datasetmetadata.py

@@ -5,11 +5,12 @@ from .datasetmetadataruninfo import (
     DatasetMetadataRunInfo,
     DatasetMetadataRunInfoTypedDict,
 )
-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
@@ -43,3 +44,12 @@ class DatasetMetadata(BaseModel):
     latest_run_info: Annotated[
         Optional[DatasetMetadataRunInfo], pydantic.Field(alias="latestRunInfo")
     ] = None
+
+    @field_serializer("scan_mode")
+    def serialize_scan_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ScanMode(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/deleteconfiggroupbyproductandidop.py

@@ -3,12 +3,14 @@
 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,
     PathParamMetadata,
     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
@@ -33,6 +35,15 @@ class DeleteConfigGroupByProductAndIDRequest(BaseModel):
     ]
     r"""The <code>id</code> of the Worker Group or Edge Fleet to delete."""
 
+    @field_serializer("product")
+    def serialize_product(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ProductsCore(value)
+            except ValueError:
+                return value
+        return value
+
 
 class DeleteConfigGroupByProductAndIDResponseTypedDict(TypedDict):
     r"""a list of ConfigGroup objects"""

cribl_control_plane/models/getconfiggroupaclbyproductandidop.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 from .productscore import ProductsCore
 from .rbacresource import RbacResource
 from .useraccesscontrollist import UserAccessControlList, UserAccessControlListTypedDict
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import (
     FieldMetadata,
@@ -11,6 +12,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
@@ -43,6 +45,24 @@ class GetConfigGroupACLByProductAndIDRequest(BaseModel):
     ] = None
     r"""Filter for limiting the response to ACL entries for the specified RBAC resource type."""
 
+    @field_serializer("product")
+    def serialize_product(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ProductsCore(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("type")
+    def serialize_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.RbacResource(value)
+            except ValueError:
+                return value
+        return value
+
 
 class GetConfigGroupACLByProductAndIDResponseTypedDict(TypedDict):
     r"""a list of UserAccessControlList objects"""

cribl_control_plane/models/getconfiggroupaclteamsbyproductandidop.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 from .productscore import ProductsCore
 from .rbacresource import RbacResource
 from .teamaccesscontrollist import TeamAccessControlList, TeamAccessControlListTypedDict
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import (
     FieldMetadata,
@@ -11,6 +12,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
@@ -43,6 +45,24 @@ class GetConfigGroupACLTeamsByProductAndIDRequest(BaseModel):
     ] = None
     r"""Filter for limiting the response to ACL entries for the specified RBAC resource type."""
 
+    @field_serializer("product")
+    def serialize_product(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ProductsCore(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("type")
+    def serialize_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.RbacResource(value)
+            except ValueError:
+                return value
+        return value
+
 
 class GetConfigGroupACLTeamsByProductAndIDResponseTypedDict(TypedDict):
     r"""a list of TeamAccessControlList objects"""

cribl_control_plane/models/getconfiggroupbyproductandidop.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
@@ -42,6 +44,15 @@ class GetConfigGroupByProductAndIDRequest(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 GetConfigGroupByProductAndIDResponseTypedDict(TypedDict):
     r"""a list of ConfigGroup objects"""

cribl_control_plane/models/getconfiggroupconfigversionbyproductandidop.py

@@ -2,12 +2,14 @@
 
 from __future__ import annotations
 from .productscore import ProductsCore
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import (
     FieldMetadata,
     PathParamMetadata,
     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
@@ -32,6 +34,15 @@ class GetConfigGroupConfigVersionByProductAndIDRequest(BaseModel):
     ]
     r"""The <code>id</code> of the Worker Group or Edge Fleet to get the configuration version for."""
 
+    @field_serializer("product")
+    def serialize_product(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ProductsCore(value)
+            except ValueError:
+                return value
+        return value
+
 
 class GetConfigGroupConfigVersionByProductAndIDResponseTypedDict(TypedDict):
     r"""a list of string objects"""

cribl_control_plane/models/getsummaryop.py

@@ -3,12 +3,14 @@
 from __future__ import annotations
 from .distributedsummary import DistributedSummary, DistributedSummaryTypedDict
 from .workertypes import WorkerTypes
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import (
     FieldMetadata,
     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
@@ -26,6 +28,15 @@ class GetSummaryRequest(BaseModel):
     ] = None
     r"""Filter for limiting the response by Cribl product: Cribl Stream (<code>worker</code>) or Cribl Edge (<code>managed-edge</code>)."""
 
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.WorkerTypes(value)
+            except ValueError:
+                return value
+        return value
+
 
 class GetSummaryResponseTypedDict(TypedDict):
     r"""a list of DistributedSummary objects"""

cribl_control_plane/models/groupcreaterequest.py

@@ -4,11 +4,12 @@ from __future__ import annotations
 from .commit import Commit, CommitTypedDict
 from .configgroupcloud import ConfigGroupCloud, ConfigGroupCloudTypedDict
 from .configgrouplookups import ConfigGroupLookups, ConfigGroupLookupsTypedDict
-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
@@ -150,3 +151,21 @@ class GroupCreateRequest(BaseModel):
     worker_remote_access: Annotated[
         Optional[bool], pydantic.Field(alias="workerRemoteAccess")
     ] = None
+
+    @field_serializer("estimated_ingest_rate")
+    def serialize_estimated_ingest_rate(self, value):
+        if isinstance(value, str):
+            try:
+                return models.GroupCreateRequestEstimatedIngestRate(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("type")
+    def serialize_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.GroupCreateRequestType(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/hbcriblinfo.py

@@ -3,11 +3,12 @@
 from __future__ import annotations
 from .hbleaderinfo import HBLeaderInfo, HBLeaderInfoTypedDict
 from .lookupversions import LookupVersions, LookupVersionsTypedDict
-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
@@ -104,3 +105,12 @@ class HBCriblInfo(BaseModel):
     )
 
     version: Optional[str] = None
+
+    @field_serializer("dist_mode")
+    def serialize_dist_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.DistMode(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/healthserverstatus.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 Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
@@ -34,3 +35,21 @@ class HealthServerStatus(BaseModel):
     status: Annotated[Status, PlainValidator(validate_open_enum(False))]
 
     role: Annotated[Optional[Role], PlainValidator(validate_open_enum(False))] = None
+
+    @field_serializer("role")
+    def serialize_role(self, value):
+        if isinstance(value, str):
+            try:
+                return models.Role(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("status")
+    def serialize_status(self, value):
+        if isinstance(value, str):
+            try:
+                return models.Status(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/input.py

@@ -75,59 +75,59 @@ from typing_extensions import Annotated, TypeAliasType
 InputTypedDict = TypeAliasType(
     "InputTypedDict",
     Union[
-        InputDatagenTypedDict,
-        InputKubeEventsTypedDict,
         InputCriblTypedDict,
+        InputKubeEventsTypedDict,
+        InputDatagenTypedDict,
         InputCriblmetricsTypedDict,
         InputKubeMetricsTypedDict,
-        InputCollectionTypedDict,
         InputSystemStateTypedDict,
+        InputCollectionTypedDict,
         InputModelDrivenTelemetryTypedDict,
         InputSystemMetricsTypedDict,
         InputWindowsMetricsTypedDict,
         InputJournalFilesTypedDict,
-        InputWinEventLogsTypedDict,
         InputRawUDPTypedDict,
-        InputExecTypedDict,
         InputKubeLogsTypedDict,
-        InputMetricsTypedDict,
+        InputExecTypedDict,
         InputSnmpTypedDict,
+        InputMetricsTypedDict,
+        InputWinEventLogsTypedDict,
         InputCriblTCPTypedDict,
         InputNetflowTypedDict,
         InputTcpjsonTypedDict,
         InputGooglePubsubTypedDict,
         InputOffice365ServiceTypedDict,
         InputWizTypedDict,
-        InputTCPTypedDict,
         InputFirehoseTypedDict,
         InputCriblHTTPTypedDict,
         InputDatadogAgentTypedDict,
         InputOffice365MgmtTypedDict,
-        InputFileTypedDict,
+        InputTCPTypedDict,
         InputSplunkTypedDict,
-        InputWefTypedDict,
+        InputFileTypedDict,
         InputAppscopeTypedDict,
+        InputWefTypedDict,
         InputHTTPRawTypedDict,
         InputWizWebhookTypedDict,
-        InputHTTPTypedDict,
         InputCriblLakeHTTPTypedDict,
-        InputAzureBlobTypedDict,
-        InputSqsTypedDict,
+        InputHTTPTypedDict,
         InputZscalerHecTypedDict,
+        InputSqsTypedDict,
+        InputAzureBlobTypedDict,
         InputKinesisTypedDict,
         InputConfluentCloudTypedDict,
         InputEventhubTypedDict,
         InputKafkaTypedDict,
         InputElasticTypedDict,
-        InputOffice365MsgTraceTypedDict,
         InputSplunkHecTypedDict,
+        InputOffice365MsgTraceTypedDict,
         InputLokiTypedDict,
         InputPrometheusRwTypedDict,
-        InputPrometheusTypedDict,
         InputCrowdstrikeTypedDict,
-        InputEdgePrometheusTypedDict,
+        InputPrometheusTypedDict,
         InputOpenTelemetryTypedDict,
         InputS3TypedDict,
+        InputEdgePrometheusTypedDict,
         InputSecurityLakeTypedDict,
         InputMskTypedDict,
         InputS3InventoryTypedDict,

cribl_control_plane/models/inputappscope.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 InputAppscopePq(BaseModel):
         Optional[InputAppscopePqControls], pydantic.Field(alias="pqControls")
     ] = None
 
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputAppscopeMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputAppscopeCompression(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputAppscopeMetadatumTypedDict(TypedDict):
     name: str
@@ -197,6 +216,15 @@ class InputAppscopePersistence(BaseModel):
     )
     r"""Path to use to write metrics. Defaults to $CRIBL_HOME/state/appscope"""
 
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputAppscopeDataCompressionFormat(value)
+            except ValueError:
+                return value
+        return value
+
 
 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"""
@@ -221,6 +249,12 @@ class InputAppscopeMaximumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
 
 class InputAppscopeTLSSettingsServerSideTypedDict(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]
@@ -231,10 +265,6 @@ class InputAppscopeTLSSettingsServerSideTypedDict(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[InputAppscopeMinimumTLSVersion]
     max_version: NotRequired[InputAppscopeMaximumTLSVersion]
 
@@ -242,6 +272,19 @@ class InputAppscopeTLSSettingsServerSideTypedDict(TypedDict):
 class InputAppscopeTLSSettingsServerSide(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
@@ -259,17 +302,6 @@ class InputAppscopeTLSSettingsServerSide(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[InputAppscopeMinimumTLSVersion],
@@ -286,6 +318,24 @@ class InputAppscopeTLSSettingsServerSide(BaseModel):
         pydantic.Field(alias="maxVersion"),
     ] = None
 
+    @field_serializer("min_version")
+    def serialize_min_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputAppscopeMinimumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("max_version")
+    def serialize_max_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputAppscopeMaximumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputAppscopeTypedDict(TypedDict):
     type: InputAppscopeType
@@ -463,3 +513,12 @@ class InputAppscope(BaseModel):
 
     text_secret: Annotated[Optional[str], pydantic.Field(alias="textSecret")] = None
     r"""Select or create a stored text secret"""
+
+    @field_serializer("auth_type")
+    def serialize_auth_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputAppscopeAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/inputazureblob.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
@@ -106,6 +107,24 @@ class InputAzureBlobPq(BaseModel):
         Optional[InputAzureBlobPqControls], pydantic.Field(alias="pqControls")
     ] = None
 
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputAzureBlobMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputAzureBlobCompression(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputAzureBlobMetadatumTypedDict(TypedDict):
     name: str
@@ -323,3 +342,12 @@ class InputAzureBlob(BaseModel):
     r"""Select or create a stored text secret"""
 
     certificate: Optional[InputAzureBlobCertificate] = None
+
+    @field_serializer("auth_type")
+    def serialize_auth_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputAzureBlobAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value