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

cribl_control_plane/models/inputprometheusrw.py

@@ -1,10 +1,14 @@
 """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 typing import Any, List, Optional
+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 InputPrometheusRwConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputPrometheusRwMode(str, Enum):
+class InputPrometheusRwMode(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 InputPrometheusRwCompression(str, Enum):
+class InputPrometheusRwCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
+    # None
     NONE = "none"
+    # Gzip
     GZIP = "gzip"
 
 
+class InputPrometheusRwPqControlsTypedDict(TypedDict):
+    pass
+
+
+class InputPrometheusRwPqControls(BaseModel):
+    pass
+
+
 class InputPrometheusRwPqTypedDict(TypedDict):
     mode: NotRequired[InputPrometheusRwMode]
     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 InputPrometheusRwPqTypedDict(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[InputPrometheusRwCompression]
     r"""Codec to use to compress the persisted data"""
+    pq_controls: NotRequired[InputPrometheusRwPqControlsTypedDict]
 
 
 class InputPrometheusRwPq(BaseModel):
-    mode: Optional[InputPrometheusRwMode] = InputPrometheusRwMode.ALWAYS
+    mode: Annotated[
+        Optional[InputPrometheusRwMode], PlainValidator(validate_open_enum(False))
+    ] = InputPrometheusRwMode.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,43 @@ class InputPrometheusRwPq(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[InputPrometheusRwCompression] = InputPrometheusRwCompression.NONE
+    compress: Annotated[
+        Optional[InputPrometheusRwCompression],
+        PlainValidator(validate_open_enum(False)),
+    ] = InputPrometheusRwCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
+    pq_controls: Annotated[
+        Optional[InputPrometheusRwPqControls], pydantic.Field(alias="pqControls")
+    ] = None
 
-class InputPrometheusRwMinimumTLSVersion(str, Enum):
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputPrometheusRwMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputPrometheusRwCompression(value)
+            except ValueError:
+                return value
+        return value
+
+
+class InputPrometheusRwMinimumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
     TL_SV1_3 = "TLSv1.3"
 
 
-class InputPrometheusRwMaximumTLSVersion(str, Enum):
+class InputPrometheusRwMaximumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
@@ -99,6 +143,12 @@ class InputPrometheusRwMaximumTLSVersion(str, Enum):
 
 class InputPrometheusRwTLSSettingsServerSideTypedDict(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]
@@ -109,10 +159,6 @@ class InputPrometheusRwTLSSettingsServerSideTypedDict(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[InputPrometheusRwMinimumTLSVersion]
     max_version: NotRequired[InputPrometheusRwMaximumTLSVersion]
 
@@ -120,6 +166,19 @@ class InputPrometheusRwTLSSettingsServerSideTypedDict(TypedDict):
 class InputPrometheusRwTLSSettingsServerSide(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
@@ -137,27 +196,42 @@ class InputPrometheusRwTLSSettingsServerSide(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[
-        Optional[InputPrometheusRwMinimumTLSVersion], pydantic.Field(alias="minVersion")
+        Annotated[
+            Optional[InputPrometheusRwMinimumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="minVersion"),
     ] = None
 
     max_version: Annotated[
-        Optional[InputPrometheusRwMaximumTLSVersion], pydantic.Field(alias="maxVersion")
+        Annotated[
+            Optional[InputPrometheusRwMaximumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="maxVersion"),
     ] = None
 
-
-class InputPrometheusRwAuthenticationType(str, Enum):
+    @field_serializer("min_version")
+    def serialize_min_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputPrometheusRwMinimumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("max_version")
+    def serialize_max_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputPrometheusRwMaximumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
+
+class InputPrometheusRwAuthenticationType(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Remote Write authentication type"""
 
     NONE = "none"
@@ -388,7 +462,11 @@ class InputPrometheusRw(BaseModel):
     r"""Absolute path on which to listen for Prometheus requests. Defaults to /write, which will expand as: http://<your‑upstream‑URL>:<your‑port>/write."""
 
     auth_type: Annotated[
-        Optional[InputPrometheusRwAuthenticationType], pydantic.Field(alias="authType")
+        Annotated[
+            Optional[InputPrometheusRwAuthenticationType],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="authType"),
     ] = InputPrometheusRwAuthenticationType.NONE
     r"""Remote Write authentication type"""
 
@@ -448,3 +526,12 @@ class InputPrometheusRw(BaseModel):
         pydantic.Field(alias="oauthHeaders"),
     ] = None
     r"""Additional headers to send in the OAuth login request. @{product} will automatically add the content-type header 'application/x-www-form-urlencoded' when sending this request."""
+
+    @field_serializer("auth_type")
+    def serialize_auth_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputPrometheusRwAuthenticationType(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/inputrawudp.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 InputRawUDPConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputRawUDPMode(str, Enum):
+class InputRawUDPMode(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 InputRawUDPCompression(str, Enum):
+class InputRawUDPCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
+    # None
     NONE = "none"
+    # Gzip
     GZIP = "gzip"
 
 
+class InputRawUDPPqControlsTypedDict(TypedDict):
+    pass
+
+
+class InputRawUDPPqControls(BaseModel):
+    pass
+
+
 class InputRawUDPPqTypedDict(TypedDict):
     mode: NotRequired[InputRawUDPMode]
     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 InputRawUDPPqTypedDict(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[InputRawUDPCompression]
     r"""Codec to use to compress the persisted data"""
+    pq_controls: NotRequired[InputRawUDPPqControlsTypedDict]
 
 
 class InputRawUDPPq(BaseModel):
-    mode: Optional[InputRawUDPMode] = InputRawUDPMode.ALWAYS
+    mode: Annotated[
+        Optional[InputRawUDPMode], PlainValidator(validate_open_enum(False))
+    ] = InputRawUDPMode.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 InputRawUDPPq(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[InputRawUDPCompression] = InputRawUDPCompression.NONE
+    compress: Annotated[
+        Optional[InputRawUDPCompression], PlainValidator(validate_open_enum(False))
+    ] = InputRawUDPCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
+    pq_controls: Annotated[
+        Optional[InputRawUDPPqControls], pydantic.Field(alias="pqControls")
+    ] = None
+
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputRawUDPMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputRawUDPCompression(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputRawUDPMetadatumTypedDict(TypedDict):
     name: str

cribl_control_plane/models/inputs3.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 InputS3Connection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputS3Mode(str, Enum):
+class InputS3Mode(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 InputS3Compression(str, Enum):
+class InputS3Compression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
+    # None
     NONE = "none"
+    # Gzip
     GZIP = "gzip"
 
 
+class InputS3PqControlsTypedDict(TypedDict):
+    pass
+
+
+class InputS3PqControls(BaseModel):
+    pass
+
+
 class InputS3PqTypedDict(TypedDict):
     mode: NotRequired[InputS3Mode]
     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 InputS3PqTypedDict(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[InputS3Compression]
     r"""Codec to use to compress the persisted data"""
+    pq_controls: NotRequired[InputS3PqControlsTypedDict]
 
 
 class InputS3Pq(BaseModel):
-    mode: Optional[InputS3Mode] = InputS3Mode.ALWAYS
+    mode: Annotated[
+        Optional[InputS3Mode], PlainValidator(validate_open_enum(False))
+    ] = InputS3Mode.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,19 +98,46 @@ class InputS3Pq(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[InputS3Compression] = InputS3Compression.NONE
+    compress: Annotated[
+        Optional[InputS3Compression], PlainValidator(validate_open_enum(False))
+    ] = InputS3Compression.NONE
     r"""Codec to use to compress the persisted data"""
 
+    pq_controls: Annotated[
+        Optional[InputS3PqControls], pydantic.Field(alias="pqControls")
+    ] = None
 
-class InputS3AuthenticationMethod(str, Enum):
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputS3Mode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputS3Compression(value)
+            except ValueError:
+                return value
+        return value
+
+
+class InputS3AuthenticationMethod(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 InputS3SignatureVersion(str, Enum):
+class InputS3SignatureVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Signature version to use for signing S3 requests"""
 
     V2 = "v2"
@@ -273,7 +319,10 @@ class InputS3(BaseModel):
     r"""SQS queue owner's AWS account ID. Leave empty if SQS queue is in same AWS account."""
 
     aws_authentication_method: Annotated[
-        Optional[InputS3AuthenticationMethod],
+        Annotated[
+            Optional[InputS3AuthenticationMethod],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="awsAuthenticationMethod"),
     ] = InputS3AuthenticationMethod.AUTO
     r"""AWS authentication method. Choose Auto to use IAM roles."""
@@ -289,7 +338,10 @@ class InputS3(BaseModel):
     r"""S3 service endpoint. If empty, defaults to the AWS Region-specific endpoint. Otherwise, it must point to S3-compatible endpoint."""
 
     signature_version: Annotated[
-        Optional[InputS3SignatureVersion], pydantic.Field(alias="signatureVersion")
+        Annotated[
+            Optional[InputS3SignatureVersion], PlainValidator(validate_open_enum(False))
+        ],
+        pydantic.Field(alias="signatureVersion"),
     ] = InputS3SignatureVersion.V4
     r"""Signature version to use for signing S3 requests"""
 
@@ -408,3 +460,21 @@ class InputS3(BaseModel):
         Optional[str], pydantic.Field(alias="processedTagValue")
     ] = None
     r"""The value for the S3 object tag applied after processing. This field accepts an expression for dynamic generation."""
+
+    @field_serializer("aws_authentication_method")
+    def serialize_aws_authentication_method(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputS3AuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("signature_version")
+    def serialize_signature_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputS3SignatureVersion(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/inputs3inventory.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 InputS3InventoryConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputS3InventoryMode(str, Enum):
+class InputS3InventoryMode(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 InputS3InventoryCompression(str, Enum):
+class InputS3InventoryCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
+    # None
     NONE = "none"
+    # Gzip
     GZIP = "gzip"
 
 
+class InputS3InventoryPqControlsTypedDict(TypedDict):
+    pass
+
+
+class InputS3InventoryPqControls(BaseModel):
+    pass
+
+
 class InputS3InventoryPqTypedDict(TypedDict):
     mode: NotRequired[InputS3InventoryMode]
     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 InputS3InventoryPqTypedDict(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[InputS3InventoryCompression]
     r"""Codec to use to compress the persisted data"""
+    pq_controls: NotRequired[InputS3InventoryPqControlsTypedDict]
 
 
 class InputS3InventoryPq(BaseModel):
-    mode: Optional[InputS3InventoryMode] = InputS3InventoryMode.ALWAYS
+    mode: Annotated[
+        Optional[InputS3InventoryMode], PlainValidator(validate_open_enum(False))
+    ] = InputS3InventoryMode.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,19 +98,46 @@ class InputS3InventoryPq(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[InputS3InventoryCompression] = InputS3InventoryCompression.NONE
+    compress: Annotated[
+        Optional[InputS3InventoryCompression], PlainValidator(validate_open_enum(False))
+    ] = InputS3InventoryCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
+    pq_controls: Annotated[
+        Optional[InputS3InventoryPqControls], pydantic.Field(alias="pqControls")
+    ] = None
 
-class InputS3InventoryAuthenticationMethod(str, Enum):
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputS3InventoryMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputS3InventoryCompression(value)
+            except ValueError:
+                return value
+        return value
+
+
+class InputS3InventoryAuthenticationMethod(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 InputS3InventorySignatureVersion(str, Enum):
+class InputS3InventorySignatureVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Signature version to use for signing S3 requests"""
 
     V2 = "v2"
@@ -144,7 +190,7 @@ class InputS3InventoryCheckpointing(BaseModel):
     r"""The number of times to retry processing when a processing error occurs. If Skip file on error is enabled, this setting is ignored."""
 
 
-class InputS3InventoryTagAfterProcessing(str, Enum):
+class InputS3InventoryTagAfterProcessing(str, Enum, metaclass=utils.OpenEnumMeta):
     FALSE = "false"
     TRUE = "true"
 
@@ -281,7 +327,10 @@ class InputS3Inventory(BaseModel):
     r"""SQS queue owner's AWS account ID. Leave empty if SQS queue is in same AWS account."""
 
     aws_authentication_method: Annotated[
-        Optional[InputS3InventoryAuthenticationMethod],
+        Annotated[
+            Optional[InputS3InventoryAuthenticationMethod],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="awsAuthenticationMethod"),
     ] = InputS3InventoryAuthenticationMethod.AUTO
     r"""AWS authentication method. Choose Auto to use IAM roles."""
@@ -297,7 +346,10 @@ class InputS3Inventory(BaseModel):
     r"""S3 service endpoint. If empty, defaults to the AWS Region-specific endpoint. Otherwise, it must point to S3-compatible endpoint."""
 
     signature_version: Annotated[
-        Optional[InputS3InventorySignatureVersion],
+        Annotated[
+            Optional[InputS3InventorySignatureVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="signatureVersion"),
     ] = InputS3InventorySignatureVersion.V4
     r"""Signature version to use for signing S3 requests"""
@@ -416,7 +468,10 @@ class InputS3Inventory(BaseModel):
     r"""Select or create a stored secret that references your access key and secret key"""
 
     tag_after_processing: Annotated[
-        Optional[InputS3InventoryTagAfterProcessing],
+        Annotated[
+            Optional[InputS3InventoryTagAfterProcessing],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="tagAfterProcessing"),
     ] = None
 
@@ -429,3 +484,30 @@ class InputS3Inventory(BaseModel):
         Optional[str], pydantic.Field(alias="processedTagValue")
     ] = None
     r"""The value for the S3 object tag applied after processing. This field accepts an expression for dynamic generation."""
+
+    @field_serializer("aws_authentication_method")
+    def serialize_aws_authentication_method(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputS3InventoryAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("signature_version")
+    def serialize_signature_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputS3InventorySignatureVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("tag_after_processing")
+    def serialize_tag_after_processing(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputS3InventoryTagAfterProcessing(value)
+            except ValueError:
+                return value
+        return value