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

cribl_control_plane/models/inputwef.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 Any, List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -23,20 +27,32 @@ class InputWefConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputWefMode(str, Enum):
+class InputWefMode(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 InputWefCompression(str, Enum):
+class InputWefCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
+    # None
     NONE = "none"
+    # Gzip
     GZIP = "gzip"
 
 
+class InputWefPqControlsTypedDict(TypedDict):
+    pass
+
+
+class InputWefPqControls(BaseModel):
+    pass
+
+
 class InputWefPqTypedDict(TypedDict):
     mode: NotRequired[InputWefMode]
     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 InputWefPqTypedDict(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[InputWefCompression]
     r"""Codec to use to compress the persisted data"""
+    pq_controls: NotRequired[InputWefPqControlsTypedDict]
 
 
 class InputWefPq(BaseModel):
-    mode: Optional[InputWefMode] = InputWefMode.ALWAYS
+    mode: Annotated[
+        Optional[InputWefMode], PlainValidator(validate_open_enum(False))
+    ] = InputWefMode.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,25 +98,51 @@ class InputWefPq(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[InputWefCompression] = InputWefCompression.NONE
+    compress: Annotated[
+        Optional[InputWefCompression], PlainValidator(validate_open_enum(False))
+    ] = InputWefCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
+    pq_controls: Annotated[
+        Optional[InputWefPqControls], pydantic.Field(alias="pqControls")
+    ] = None
 
-class InputWefAuthenticationMethod(str, Enum):
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWefMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWefCompression(value)
+            except ValueError:
+                return value
+        return value
+
+
+class InputWefAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""How to authenticate incoming client connections"""
 
+    # Client certificate
     CLIENT_CERT = "clientCert"
+    # Kerberos
     KERBEROS = "kerberos"
 
 
-class InputWefMinimumTLSVersion(str, Enum):
+class InputWefMinimumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
     TL_SV1_3 = "TLSv1.3"
 
 
-class InputWefMaximumTLSVersion(str, Enum):
+class InputWefMaximumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
     TL_SV1_1 = "TLSv1.1"
     TL_SV1_2 = "TLSv1.2"
@@ -168,11 +213,19 @@ class MTLSSettings(BaseModel):
     r"""Regex matching allowable common names in peer certificates' subject attribute"""
 
     min_version: Annotated[
-        Optional[InputWefMinimumTLSVersion], pydantic.Field(alias="minVersion")
+        Annotated[
+            Optional[InputWefMinimumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="minVersion"),
     ] = None
 
     max_version: Annotated[
-        Optional[InputWefMaximumTLSVersion], pydantic.Field(alias="maxVersion")
+        Annotated[
+            Optional[InputWefMaximumTLSVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="maxVersion"),
     ] = None
 
     ocsp_check: Annotated[Optional[bool], pydantic.Field(alias="ocspCheck")] = False
@@ -187,15 +240,33 @@ class MTLSSettings(BaseModel):
     ] = False
     r"""If enabled, checks will fail on any OCSP error. Otherwise, checks will fail only when a certificate is revoked, ignoring other errors."""
 
-
-class InputWefFormat(str, Enum):
+    @field_serializer("min_version")
+    def serialize_min_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWefMinimumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("max_version")
+    def serialize_max_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWefMaximumTLSVersion(value)
+            except ValueError:
+                return value
+        return value
+
+
+class InputWefFormat(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Content format in which the endpoint should deliver events"""
 
     RAW = "Raw"
     RENDERED_TEXT = "RenderedText"
 
 
-class QueryBuilderMode(str, Enum):
+class QueryBuilderMode(str, Enum, metaclass=utils.OpenEnumMeta):
     SIMPLE = "simple"
     XML = "xml"
 
@@ -213,6 +284,21 @@ class SubscriptionMetadatum(BaseModel):
     r"""JavaScript expression to compute field's value, enclosed in quotes or backticks. (Can evaluate to a constant.)"""
 
 
+class QueryTypedDict(TypedDict):
+    path: str
+    r"""The Path attribute from the relevant XML Select element"""
+    query_expression: str
+    r"""The XPath query inside the relevant XML Select element"""
+
+
+class Query(BaseModel):
+    path: str
+    r"""The Path attribute from the relevant XML Select element"""
+
+    query_expression: Annotated[str, pydantic.Field(alias="queryExpression")]
+    r"""The XPath query inside the relevant XML Select element"""
+
+
 class SubscriptionTypedDict(TypedDict):
     subscription_name: str
     targets: List[str]
@@ -236,6 +322,9 @@ class SubscriptionTypedDict(TypedDict):
     query_selector: NotRequired[QueryBuilderMode]
     metadata: NotRequired[List[SubscriptionMetadatumTypedDict]]
     r"""Fields to add to events ingested under this subscription"""
+    queries: NotRequired[List[QueryTypedDict]]
+    xml_query: NotRequired[str]
+    r"""The XPath query to use for selecting events"""
 
 
 class Subscription(BaseModel):
@@ -248,7 +337,8 @@ class Subscription(BaseModel):
     r"""Version UUID for this subscription. If any subscription parameters are modified, this value will change."""
 
     content_format: Annotated[
-        Optional[InputWefFormat], pydantic.Field(alias="contentFormat")
+        Annotated[Optional[InputWefFormat], PlainValidator(validate_open_enum(False))],
+        pydantic.Field(alias="contentFormat"),
     ] = InputWefFormat.RAW
     r"""Content format in which the endpoint should deliver events"""
 
@@ -277,12 +367,38 @@ class Subscription(BaseModel):
     r"""The RFC-3066 locale the Windows clients should use when sending events. Defaults to \"en-US\"."""
 
     query_selector: Annotated[
-        Optional[QueryBuilderMode], pydantic.Field(alias="querySelector")
+        Annotated[
+            Optional[QueryBuilderMode], PlainValidator(validate_open_enum(False))
+        ],
+        pydantic.Field(alias="querySelector"),
     ] = QueryBuilderMode.SIMPLE
 
     metadata: Optional[List[SubscriptionMetadatum]] = None
     r"""Fields to add to events ingested under this subscription"""
 
+    queries: Optional[List[Query]] = None
+
+    xml_query: Annotated[Optional[str], pydantic.Field(alias="xmlQuery")] = None
+    r"""The XPath query to use for selecting events"""
+
+    @field_serializer("content_format")
+    def serialize_content_format(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWefFormat(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("query_selector")
+    def serialize_query_selector(self, value):
+        if isinstance(value, str):
+            try:
+                return models.QueryBuilderMode(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputWefMetadatumTypedDict(TypedDict):
     name: str
@@ -397,7 +513,11 @@ class InputWef(BaseModel):
     r"""Port to listen on"""
 
     auth_method: Annotated[
-        Optional[InputWefAuthenticationMethod], pydantic.Field(alias="authMethod")
+        Annotated[
+            Optional[InputWefAuthenticationMethod],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="authMethod"),
     ] = InputWefAuthenticationMethod.CLIENT_CERT
     r"""How to authenticate incoming client connections"""
 
@@ -473,3 +593,12 @@ class InputWef(BaseModel):
         Optional[bool], pydantic.Field(alias="logFingerprintMismatch")
     ] = False
     r"""Log a warning if the client certificate authority (CA) fingerprint does not match the expected value. A mismatch prevents Cribl from receiving events from the Windows Event Forwarder."""
+
+    @field_serializer("auth_method")
+    def serialize_auth_method(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWefAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/inputwindowsmetrics.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 InputWindowsMetricsConnection(BaseModel):
     pipeline: Optional[str] = None
 
 
-class InputWindowsMetricsPqMode(str, Enum):
+class InputWindowsMetricsPqMode(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 InputWindowsMetricsCompression(str, Enum):
+class InputWindowsMetricsCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
+    # None
     NONE = "none"
+    # Gzip
     GZIP = "gzip"
 
 
+class InputWindowsMetricsPqControlsTypedDict(TypedDict):
+    pass
+
+
+class InputWindowsMetricsPqControls(BaseModel):
+    pass
+
+
 class InputWindowsMetricsPqTypedDict(TypedDict):
     mode: NotRequired[InputWindowsMetricsPqMode]
     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 InputWindowsMetricsPqTypedDict(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[InputWindowsMetricsCompression]
     r"""Codec to use to compress the persisted data"""
+    pq_controls: NotRequired[InputWindowsMetricsPqControlsTypedDict]
 
 
 class InputWindowsMetricsPq(BaseModel):
-    mode: Optional[InputWindowsMetricsPqMode] = InputWindowsMetricsPqMode.ALWAYS
+    mode: Annotated[
+        Optional[InputWindowsMetricsPqMode], PlainValidator(validate_open_enum(False))
+    ] = InputWindowsMetricsPqMode.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,27 +98,58 @@ class InputWindowsMetricsPq(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[InputWindowsMetricsCompression] = (
-        InputWindowsMetricsCompression.NONE
-    )
+    compress: Annotated[
+        Optional[InputWindowsMetricsCompression],
+        PlainValidator(validate_open_enum(False)),
+    ] = InputWindowsMetricsCompression.NONE
     r"""Codec to use to compress the persisted data"""
 
-
-class InputWindowsMetricsHostMode(str, Enum):
+    pq_controls: Annotated[
+        Optional[InputWindowsMetricsPqControls], pydantic.Field(alias="pqControls")
+    ] = None
+
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWindowsMetricsPqMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWindowsMetricsCompression(value)
+            except ValueError:
+                return value
+        return value
+
+
+class InputWindowsMetricsHostMode(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Select level of detail for host metrics"""
 
+    # Basic
     BASIC = "basic"
+    # All
     ALL = "all"
+    # Custom
     CUSTOM = "custom"
+    # Disabled
     DISABLED = "disabled"
 
 
-class InputWindowsMetricsSystemMode(str, Enum):
+class InputWindowsMetricsSystemMode(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Select the level of details for system metrics"""
 
+    # Basic
     BASIC = "basic"
+    # All
     ALL = "all"
+    # Custom
     CUSTOM = "custom"
+    # Disabled
     DISABLED = "disabled"
 
 
@@ -111,19 +161,35 @@ class InputWindowsMetricsSystemTypedDict(TypedDict):
 
 
 class InputWindowsMetricsSystem(BaseModel):
-    mode: Optional[InputWindowsMetricsSystemMode] = InputWindowsMetricsSystemMode.BASIC
+    mode: Annotated[
+        Optional[InputWindowsMetricsSystemMode],
+        PlainValidator(validate_open_enum(False)),
+    ] = InputWindowsMetricsSystemMode.BASIC
     r"""Select the level of details for system metrics"""
 
     detail: Optional[bool] = False
     r"""Generate metrics for all system information"""
 
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWindowsMetricsSystemMode(value)
+            except ValueError:
+                return value
+        return value
+
 
-class InputWindowsMetricsCPUMode(str, Enum):
+class InputWindowsMetricsCPUMode(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Select the level of details for CPU metrics"""
 
+    # Basic
     BASIC = "basic"
+    # All
     ALL = "all"
+    # Custom
     CUSTOM = "custom"
+    # Disabled
     DISABLED = "disabled"
 
 
@@ -139,7 +205,9 @@ class InputWindowsMetricsCPUTypedDict(TypedDict):
 
 
 class InputWindowsMetricsCPU(BaseModel):
-    mode: Optional[InputWindowsMetricsCPUMode] = InputWindowsMetricsCPUMode.BASIC
+    mode: Annotated[
+        Optional[InputWindowsMetricsCPUMode], PlainValidator(validate_open_enum(False))
+    ] = InputWindowsMetricsCPUMode.BASIC
     r"""Select the level of details for CPU metrics"""
 
     per_cpu: Annotated[Optional[bool], pydantic.Field(alias="perCpu")] = False
@@ -151,13 +219,26 @@ class InputWindowsMetricsCPU(BaseModel):
     time: Optional[bool] = False
     r"""Generate raw, monotonic CPU time counters"""
 
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWindowsMetricsCPUMode(value)
+            except ValueError:
+                return value
+        return value
 
-class InputWindowsMetricsMemoryMode(str, Enum):
+
+class InputWindowsMetricsMemoryMode(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Select the level of details for memory metrics"""
 
+    # Basic
     BASIC = "basic"
+    # All
     ALL = "all"
+    # Custom
     CUSTOM = "custom"
+    # Disabled
     DISABLED = "disabled"
 
 
@@ -169,39 +250,64 @@ class InputWindowsMetricsMemoryTypedDict(TypedDict):
 
 
 class InputWindowsMetricsMemory(BaseModel):
-    mode: Optional[InputWindowsMetricsMemoryMode] = InputWindowsMetricsMemoryMode.BASIC
+    mode: Annotated[
+        Optional[InputWindowsMetricsMemoryMode],
+        PlainValidator(validate_open_enum(False)),
+    ] = InputWindowsMetricsMemoryMode.BASIC
     r"""Select the level of details for memory metrics"""
 
     detail: Optional[bool] = False
     r"""Generate metrics for all memory states"""
 
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWindowsMetricsMemoryMode(value)
+            except ValueError:
+                return value
+        return value
+
 
-class InputWindowsMetricsNetworkMode(str, Enum):
+class InputWindowsMetricsNetworkMode(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Select the level of details for network metrics"""
 
+    # Basic
     BASIC = "basic"
+    # All
     ALL = "all"
+    # Custom
     CUSTOM = "custom"
+    # Disabled
     DISABLED = "disabled"
 
 
 class InputWindowsMetricsNetworkTypedDict(TypedDict):
     mode: NotRequired[InputWindowsMetricsNetworkMode]
     r"""Select the level of details for network metrics"""
+    detail: NotRequired[bool]
+    r"""Generate full network metrics"""
+    protocols: NotRequired[bool]
+    r"""Generate protocol metrics for ICMP, ICMPMsg, IP, TCP, UDP and UDPLite"""
     devices: NotRequired[List[str]]
     r"""Network interfaces to include/exclude. All interfaces are included if this list is empty."""
     per_interface: NotRequired[bool]
     r"""Generate separate metrics for each interface"""
-    detail: NotRequired[bool]
-    r"""Generate full network metrics"""
 
 
 class InputWindowsMetricsNetwork(BaseModel):
-    mode: Optional[InputWindowsMetricsNetworkMode] = (
-        InputWindowsMetricsNetworkMode.BASIC
-    )
+    mode: Annotated[
+        Optional[InputWindowsMetricsNetworkMode],
+        PlainValidator(validate_open_enum(False)),
+    ] = InputWindowsMetricsNetworkMode.BASIC
     r"""Select the level of details for network metrics"""
 
+    detail: Optional[bool] = False
+    r"""Generate full network metrics"""
+
+    protocols: Optional[bool] = False
+    r"""Generate protocol metrics for ICMP, ICMPMsg, IP, TCP, UDP and UDPLite"""
+
     devices: Optional[List[str]] = None
     r"""Network interfaces to include/exclude. All interfaces are included if this list is empty."""
 
@@ -210,37 +316,63 @@ class InputWindowsMetricsNetwork(BaseModel):
     )
     r"""Generate separate metrics for each interface"""
 
-    detail: Optional[bool] = False
-    r"""Generate full network metrics"""
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWindowsMetricsNetworkMode(value)
+            except ValueError:
+                return value
+        return value
 
 
-class InputWindowsMetricsDiskMode(str, Enum):
+class InputWindowsMetricsDiskMode(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Select the level of details for disk metrics"""
 
+    # Basic
     BASIC = "basic"
+    # All
     ALL = "all"
+    # Custom
     CUSTOM = "custom"
+    # Disabled
     DISABLED = "disabled"
 
 
 class InputWindowsMetricsDiskTypedDict(TypedDict):
     mode: NotRequired[InputWindowsMetricsDiskMode]
     r"""Select the level of details for disk metrics"""
-    volumes: NotRequired[List[str]]
-    r"""Windows volumes to include/exclude. E.g.: C:, !E:, etc. Wildcards and ! (not) operators are supported. All volumes are included if this list is empty."""
     per_volume: NotRequired[bool]
     r"""Generate separate metrics for each volume"""
+    detail: NotRequired[bool]
+    r"""Generate full disk metrics"""
+    volumes: NotRequired[List[str]]
+    r"""Windows volumes to include/exclude. E.g.: C:, !E:, etc. Wildcards and ! (not) operators are supported. All volumes are included if this list is empty."""
 
 
 class InputWindowsMetricsDisk(BaseModel):
-    mode: Optional[InputWindowsMetricsDiskMode] = InputWindowsMetricsDiskMode.BASIC
+    mode: Annotated[
+        Optional[InputWindowsMetricsDiskMode], PlainValidator(validate_open_enum(False))
+    ] = InputWindowsMetricsDiskMode.BASIC
     r"""Select the level of details for disk metrics"""
 
+    per_volume: Annotated[Optional[bool], pydantic.Field(alias="perVolume")] = False
+    r"""Generate separate metrics for each volume"""
+
+    detail: Optional[bool] = False
+    r"""Generate full disk metrics"""
+
     volumes: Optional[List[str]] = None
     r"""Windows volumes to include/exclude. E.g.: C:, !E:, etc. Wildcards and ! (not) operators are supported. All volumes are included if this list is empty."""
 
-    per_volume: Annotated[Optional[bool], pydantic.Field(alias="perVolume")] = False
-    r"""Generate separate metrics for each volume"""
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWindowsMetricsDiskMode(value)
+            except ValueError:
+                return value
+        return value
 
 
 class InputWindowsMetricsCustomTypedDict(TypedDict):
@@ -270,11 +402,22 @@ class InputWindowsMetricsHostTypedDict(TypedDict):
 
 
 class InputWindowsMetricsHost(BaseModel):
-    mode: Optional[InputWindowsMetricsHostMode] = InputWindowsMetricsHostMode.BASIC
+    mode: Annotated[
+        Optional[InputWindowsMetricsHostMode], PlainValidator(validate_open_enum(False))
+    ] = InputWindowsMetricsHostMode.BASIC
     r"""Select level of detail for host metrics"""
 
     custom: Optional[InputWindowsMetricsCustom] = None
 
+    @field_serializer("mode")
+    def serialize_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWindowsMetricsHostMode(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputWindowsMetricsSetTypedDict(TypedDict):
     name: str
@@ -315,7 +458,7 @@ class InputWindowsMetricsMetadatum(BaseModel):
     r"""JavaScript expression to compute field's value, enclosed in quotes or backticks. (Can evaluate to a constant.)"""
 
 
-class InputWindowsMetricsDataCompressionFormat(str, Enum):
+class InputWindowsMetricsDataCompressionFormat(str, Enum, metaclass=utils.OpenEnumMeta):
     NONE = "none"
     GZIP = "gzip"
 
@@ -347,15 +490,25 @@ class InputWindowsMetricsPersistence(BaseModel):
     max_data_time: Annotated[Optional[str], pydantic.Field(alias="maxDataTime")] = "24h"
     r"""Maximum amount of time to retain data (examples: 2h, 4d). When limit is reached, older data will be deleted."""
 
-    compress: Optional[InputWindowsMetricsDataCompressionFormat] = (
-        InputWindowsMetricsDataCompressionFormat.GZIP
-    )
+    compress: Annotated[
+        Optional[InputWindowsMetricsDataCompressionFormat],
+        PlainValidator(validate_open_enum(False)),
+    ] = InputWindowsMetricsDataCompressionFormat.GZIP
 
     dest_path: Annotated[Optional[str], pydantic.Field(alias="destPath")] = (
         "$CRIBL_HOME/state/windows_metrics"
     )
     r"""Path to use to write metrics. Defaults to $CRIBL_HOME/state/windows_metrics"""
 
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.InputWindowsMetricsDataCompressionFormat(value)
+            except ValueError:
+                return value
+        return value
+
 
 class InputWindowsMetricsTypedDict(TypedDict):
     type: InputWindowsMetricsType