cribl-control-plane 0.0.39__py3-none-any.whl → 0.5.0b3__py3-none-any.whl

cribl_control_plane/models/inputmsk.py

@@ -1,9 +1,13 @@
 """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
 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
 
@@ -23,20 +27,32 @@ class InputMskConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputMskMode(str, Enum):
+class InputMskMode(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 = "smart"
+    # Always On
     ALWAYS = "always"
 
 
-class InputMskCompression(str, Enum):
+class InputMskCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
+    # None
     NONE = "none"
+    # Gzip
     GZIP = "gzip"
 
 
+class InputMskPqControlsTypedDict(TypedDict):
+    pass
+
+
+class InputMskPqControls(BaseModel):
+    pass
+
+
 class InputMskPqTypedDict(TypedDict):
     mode: NotRequired[InputMskMode]
     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."""
@@ -52,10 +68,13 @@ class InputMskPqTypedDict(TypedDict):
     r"""The location for the persistent queue files. To this field's value, the system will append: /<worker-id>/inputs/<input-id>"""
     compress: NotRequired[InputMskCompression]
     r"""Codec to use to compress the persisted data"""
+    pq_controls: NotRequired[InputMskPqControlsTypedDict]
 
 
 class InputMskPq(BaseModel):
-    mode: Optional[InputMskMode] = InputMskMode.ALWAYS
+    mode: Annotated[
+        Optional[InputMskMode], PlainValidator(validate_open_enum(False))
+    ] = InputMskMode.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[
@@ -79,9 +98,33 @@ class InputMskPq(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[InputMskCompression] = InputMskCompression.NONE
+    compress: Annotated[
+        Optional[InputMskCompression], PlainValidator(validate_open_enum(False))
+    ] = InputMskCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
+    pq_controls: Annotated[
+        Optional[InputMskPqControls], pydantic.Field(alias="pqControls")
+    ] = None
+
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputMskMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputMskCompression(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputMskMetadatumTypedDict(TypedDict):
     name: str
@@ -96,13 +139,6 @@ class InputMskMetadatum(BaseModel):
     r"""JavaScript expression to compute field's value, enclosed in quotes or backticks. (Can evaluate to a constant.)"""
 
 
-class InputMskSchemaType(str, Enum):
-    r"""The schema format used to encode and decode event data"""
-
-    AVRO = "avro"
-    JSON = "json"
-
-
 class InputMskAuthTypedDict(TypedDict):
     r"""Credentials to use when authenticating with the schema registry using basic HTTP authentication"""
 
@@ -122,14 +158,18 @@ class InputMskAuth(BaseModel):
     r"""Select or create a secret that references your credentials"""
 
 
-class InputMskKafkaSchemaRegistryMinimumTLSVersion(str, Enum):
+class InputMskKafkaSchemaRegistryMinimumTLSVersion(
+    str, Enum, metaclass=utils.OpenEnumMeta
+):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
     TL_SV1_3 = "TLSv1.3"
 
 
-class InputMskKafkaSchemaRegistryMaximumTLSVersion(str, Enum):
+class InputMskKafkaSchemaRegistryMaximumTLSVersion(
+    str, Enum, metaclass=utils.OpenEnumMeta
+):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
@@ -189,22 +229,44 @@ class InputMskKafkaSchemaRegistryTLSSettingsClientSide(BaseModel):
     r"""Passphrase to use to decrypt private key"""
 
     min_version: Annotated[
-        Optional[InputMskKafkaSchemaRegistryMinimumTLSVersion],
+        Annotated[
+            Optional[InputMskKafkaSchemaRegistryMinimumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="minVersion"),
     ] = None
 
     max_version: Annotated[
-        Optional[InputMskKafkaSchemaRegistryMaximumTLSVersion],
+        Annotated[
+            Optional[InputMskKafkaSchemaRegistryMaximumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="maxVersion"),
     ] = None
 
+    @field_serializer("min_version")
+    def serialize_min_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputMskKafkaSchemaRegistryMinimumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("max_version")
+    def serialize_max_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputMskKafkaSchemaRegistryMaximumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputMskKafkaSchemaRegistryAuthenticationTypedDict(TypedDict):
     disabled: NotRequired[bool]
     schema_registry_url: NotRequired[str]
     r"""URL for accessing the Confluent Schema Registry. Example: http://localhost:8081. To connect over TLS, use https instead of http."""
-    schema_type: NotRequired[InputMskSchemaType]
-    r"""The schema format used to encode and decode event data"""
     connection_timeout: NotRequired[float]
     r"""Maximum time to wait for a Schema Registry connection to complete successfully"""
     request_timeout: NotRequired[float]
@@ -224,11 +286,6 @@ class InputMskKafkaSchemaRegistryAuthentication(BaseModel):
     ] = "http://localhost:8081"
     r"""URL for accessing the Confluent Schema Registry. Example: http://localhost:8081. To connect over TLS, use https instead of http."""
 
-    schema_type: Annotated[
-        Optional[InputMskSchemaType], pydantic.Field(alias="schemaType")
-    ] = InputMskSchemaType.AVRO
-    r"""The schema format used to encode and decode event data"""
-
     connection_timeout: Annotated[
         Optional[float], pydantic.Field(alias="connectionTimeout")
     ] = 30000
@@ -248,29 +305,32 @@ class InputMskKafkaSchemaRegistryAuthentication(BaseModel):
     tls: Optional[InputMskKafkaSchemaRegistryTLSSettingsClientSide] = None
 
 
-class InputMskAuthenticationMethod(str, Enum):
+class InputMskAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""AWS authentication method. Choose Auto to use IAM roles."""
 
+    # Auto
     AUTO = "auto"
+    # Manual
     MANUAL = "manual"
+    # Secret Key pair
     SECRET = "secret"
 
 
-class InputMskSignatureVersion(str, Enum):
+class InputMskSignatureVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Signature version to use for signing MSK cluster requests"""
 
     V2 = "v2"
     V4 = "v4"
 
 
-class InputMskMinimumTLSVersion(str, Enum):
+class InputMskMinimumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
     TL_SV1_3 = "TLSv1.3"
 
 
-class InputMskMaximumTLSVersion(str, Enum):
+class InputMskMaximumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
@@ -330,13 +390,39 @@ class InputMskTLSSettingsClientSide(BaseModel):
     r"""Passphrase to use to decrypt private key"""
 
     min_version: Annotated[
-        Optional[InputMskMinimumTLSVersion], pydantic.Field(alias="minVersion")
+        Annotated[
+            Optional[InputMskMinimumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="minVersion"),
     ] = None
 
     max_version: Annotated[
-        Optional[InputMskMaximumTLSVersion], pydantic.Field(alias="maxVersion")
+        Annotated[
+            Optional[InputMskMaximumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="maxVersion"),
     ] = None
 
+    @field_serializer("min_version")
+    def serialize_min_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputMskMinimumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("max_version")
+    def serialize_max_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputMskMaximumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputMskTypedDict(TypedDict):
     type: InputMskType
@@ -367,7 +453,8 @@ class InputMskTypedDict(TypedDict):
     from_beginning: NotRequired[bool]
     r"""Leave enabled if you want the Source, upon first subscribing to a topic, to read starting with the earliest available message"""
     session_timeout: NotRequired[float]
-    r"""Timeout used to detect client failures when using Kafka's group-management facilities.
+    r"""
+    Timeout used to detect client failures when using Kafka's group-management facilities.
     If the client sends no heartbeats to the broker before the timeout expires,
     the broker will remove the client from the group and initiate a rebalance.
     Value must be between the broker's configured group.min.session.timeout.ms and group.max.session.timeout.ms.
@@ -490,7 +577,8 @@ class InputMsk(BaseModel):
     session_timeout: Annotated[
         Optional[float], pydantic.Field(alias="sessionTimeout")
     ] = 30000
-    r"""Timeout used to detect client failures when using Kafka's group-management facilities.
+    r"""
+    Timeout used to detect client failures when using Kafka's group-management facilities.
     If the client sends no heartbeats to the broker before the timeout expires,
     the broker will remove the client from the group and initiate a rebalance.
     Value must be between the broker's configured group.min.session.timeout.ms and group.max.session.timeout.ms.
@@ -556,7 +644,10 @@ class InputMsk(BaseModel):
     r"""Specifies a time window during which @{product} can reauthenticate if needed. Creates the window measuring backward from the moment when credentials are set to expire."""
 
     aws_authentication_method: Annotated[
-        Optional[InputMskAuthenticationMethod],
+        Annotated[
+            Optional[InputMskAuthenticationMethod],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="awsAuthenticationMethod"),
     ] = InputMskAuthenticationMethod.AUTO
     r"""AWS authentication method. Choose Auto to use IAM roles."""
@@ -569,7 +660,11 @@ class InputMsk(BaseModel):
     r"""MSK cluster service endpoint. If empty, defaults to the AWS Region-specific endpoint. Otherwise, it must point to MSK cluster-compatible endpoint."""
 
     signature_version: Annotated[
-        Optional[InputMskSignatureVersion], pydantic.Field(alias="signatureVersion")
+        Annotated[
+            Optional[InputMskSignatureVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="signatureVersion"),
     ] = InputMskSignatureVersion.V4
     r"""Signature version to use for signing MSK cluster requests"""
 
@@ -634,3 +729,21 @@ class InputMsk(BaseModel):
 
     aws_secret: Annotated[Optional[str], pydantic.Field(alias="awsSecret")] = None
     r"""Select or create a stored secret that references your access key and secret key"""
+
+    @field_serializer("aws_authentication_method")
+    def serialize_aws_authentication_method(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputMskAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("signature_version")
+    def serialize_signature_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputMskSignatureVersion(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/inputnetflow.py

@@ -1,9 +1,13 @@
 """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
 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
 
@@ -23,20 +27,32 @@ class InputNetflowConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputNetflowMode(str, Enum):
+class InputNetflowMode(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 = "smart"
+    # Always On
     ALWAYS = "always"
 
 
-class InputNetflowCompression(str, Enum):
+class InputNetflowCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
+    # None
     NONE = "none"
+    # Gzip
     GZIP = "gzip"
 
 
+class InputNetflowPqControlsTypedDict(TypedDict):
+    pass
+
+
+class InputNetflowPqControls(BaseModel):
+    pass
+
+
 class InputNetflowPqTypedDict(TypedDict):
     mode: NotRequired[InputNetflowMode]
     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."""
@@ -52,10 +68,13 @@ class InputNetflowPqTypedDict(TypedDict):
     r"""The location for the persistent queue files. To this field's value, the system will append: /<worker-id>/inputs/<input-id>"""
     compress: NotRequired[InputNetflowCompression]
     r"""Codec to use to compress the persisted data"""
+    pq_controls: NotRequired[InputNetflowPqControlsTypedDict]
 
 
 class InputNetflowPq(BaseModel):
-    mode: Optional[InputNetflowMode] = InputNetflowMode.ALWAYS
+    mode: Annotated[
+        Optional[InputNetflowMode], PlainValidator(validate_open_enum(False))
+    ] = InputNetflowMode.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[
@@ -79,9 +98,33 @@ class InputNetflowPq(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[InputNetflowCompression] = InputNetflowCompression.NONE
+    compress: Annotated[
+        Optional[InputNetflowCompression], PlainValidator(validate_open_enum(False))
+    ] = InputNetflowCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
+    pq_controls: Annotated[
+        Optional[InputNetflowPqControls], pydantic.Field(alias="pqControls")
+    ] = None
+
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputNetflowMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputNetflowCompression(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputNetflowMetadatumTypedDict(TypedDict):
     name: str

cribl_control_plane/models/inputoffice365mgmt.py

@@ -1,9 +1,13 @@
 """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
 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
 
@@ -23,20 +27,32 @@ class InputOffice365MgmtConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputOffice365MgmtMode(str, Enum):
+class InputOffice365MgmtMode(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 = "smart"
+    # Always On
     ALWAYS = "always"
 
 
-class InputOffice365MgmtCompression(str, Enum):
+class InputOffice365MgmtCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
+    # None
     NONE = "none"
+    # Gzip
     GZIP = "gzip"
 
 
+class InputOffice365MgmtPqControlsTypedDict(TypedDict):
+    pass
+
+
+class InputOffice365MgmtPqControls(BaseModel):
+    pass
+
+
 class InputOffice365MgmtPqTypedDict(TypedDict):
     mode: NotRequired[InputOffice365MgmtMode]
     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."""
@@ -52,10 +68,13 @@ class InputOffice365MgmtPqTypedDict(TypedDict):
     r"""The location for the persistent queue files. To this field's value, the system will append: /<worker-id>/inputs/<input-id>"""
     compress: NotRequired[InputOffice365MgmtCompression]
     r"""Codec to use to compress the persisted data"""
+    pq_controls: NotRequired[InputOffice365MgmtPqControlsTypedDict]
 
 
 class InputOffice365MgmtPq(BaseModel):
-    mode: Optional[InputOffice365MgmtMode] = InputOffice365MgmtMode.ALWAYS
+    mode: Annotated[
+        Optional[InputOffice365MgmtMode], PlainValidator(validate_open_enum(False))
+    ] = InputOffice365MgmtMode.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[
@@ -79,18 +98,45 @@ class InputOffice365MgmtPq(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[InputOffice365MgmtCompression] = (
-        InputOffice365MgmtCompression.NONE
-    )
+    compress: Annotated[
+        Optional[InputOffice365MgmtCompression],
+        PlainValidator(validate_open_enum(False)),
+    ] = InputOffice365MgmtCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
+    pq_controls: Annotated[
+        Optional[InputOffice365MgmtPqControls], pydantic.Field(alias="pqControls")
+    ] = None
 
-class InputOffice365MgmtSubscriptionPlan(str, Enum):
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputOffice365MgmtMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputOffice365MgmtCompression(value)
+            except ValueError:
+                return value
+        return value
+
+
+class InputOffice365MgmtSubscriptionPlan(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Office 365 subscription plan for your organization, typically Office 365 Enterprise"""
 
+    # Office 365 Enterprise
     ENTERPRISE_GCC = "enterprise_gcc"
+    # Office 365 GCC
     GCC = "gcc"
+    # Office 365 GCC High
     GCC_HIGH = "gcc_high"
+    # Office 365 DoD
     DOD = "dod"
 
 
@@ -107,7 +153,7 @@ class InputOffice365MgmtMetadatum(BaseModel):
     r"""JavaScript expression to compute field's value, enclosed in quotes or backticks. (Can evaluate to a constant.)"""
 
 
-class InputOffice365MgmtLogLevel(str, Enum):
+class InputOffice365MgmtLogLevel(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Collector runtime Log Level"""
 
     ERROR = "error"
@@ -137,18 +183,34 @@ class InputOffice365MgmtContentConfig(BaseModel):
     interval: Optional[float] = None
 
     log_level: Annotated[
-        Optional[InputOffice365MgmtLogLevel], pydantic.Field(alias="logLevel")
+        Annotated[
+            Optional[InputOffice365MgmtLogLevel],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="logLevel"),
     ] = None
     r"""Collector runtime Log Level"""
 
     enabled: Optional[bool] = None
 
+    @field_serializer("log_level")
+    def serialize_log_level(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputOffice365MgmtLogLevel(value)
+            except ValueError:
+                return value
+        return value
+
 
-class InputOffice365MgmtRetryType(str, Enum):
+class InputOffice365MgmtRetryType(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""The algorithm to use when performing HTTP retries"""
 
+    # Disabled
     NONE = "none"
+    # Backoff
     BACKOFF = "backoff"
+    # Static
     STATIC = "static"
 
 
@@ -172,7 +234,9 @@ class InputOffice365MgmtRetryRulesTypedDict(TypedDict):
 
 
 class InputOffice365MgmtRetryRules(BaseModel):
-    type: Optional[InputOffice365MgmtRetryType] = InputOffice365MgmtRetryType.BACKOFF
+    type: Annotated[
+        Optional[InputOffice365MgmtRetryType], PlainValidator(validate_open_enum(False))
+    ] = InputOffice365MgmtRetryType.BACKOFF
     r"""The algorithm to use when performing HTTP retries"""
 
     interval: Optional[float] = 1000
@@ -202,8 +266,17 @@ class InputOffice365MgmtRetryRules(BaseModel):
     ] = False
     r"""Retry request when a connection reset (ECONNRESET) error occurs"""
 
+    @field_serializer("type")
+    def serialize_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputOffice365MgmtRetryType(value)
+            except ValueError:
+                return value
+        return value
 
-class InputOffice365MgmtAuthenticationMethod(str, Enum):
+
+class InputOffice365MgmtAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Enter client secret directly, or select a stored secret"""
 
     MANUAL = "manual"
@@ -301,7 +374,11 @@ class InputOffice365Mgmt(BaseModel):
     pq: Optional[InputOffice365MgmtPq] = None
 
     plan_type: Annotated[
-        Optional[InputOffice365MgmtSubscriptionPlan], pydantic.Field(alias="planType")
+        Annotated[
+            Optional[InputOffice365MgmtSubscriptionPlan],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="planType"),
     ] = InputOffice365MgmtSubscriptionPlan.ENTERPRISE_GCC
     r"""Office 365 subscription plan for your organization, typically Office 365 Enterprise"""
 
@@ -351,7 +428,10 @@ class InputOffice365Mgmt(BaseModel):
     ] = None
 
     auth_type: Annotated[
-        Optional[InputOffice365MgmtAuthenticationMethod],
+        Annotated[
+            Optional[InputOffice365MgmtAuthenticationMethod],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="authType"),
     ] = InputOffice365MgmtAuthenticationMethod.MANUAL
     r"""Enter client secret directly, or select a stored secret"""
@@ -363,3 +443,21 @@ class InputOffice365Mgmt(BaseModel):
 
     text_secret: Annotated[Optional[str], pydantic.Field(alias="textSecret")] = None
     r"""Select or create a stored text secret"""
+
+    @field_serializer("plan_type")
+    def serialize_plan_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputOffice365MgmtSubscriptionPlan(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("auth_type")
+    def serialize_auth_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputOffice365MgmtAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value