cribl-control-plane 0.2.1rc7__py3-none-any.whl → 0.3.0a1__py3-none-any.whl

cribl_control_plane/models/hbcriblinfo.py

@@ -1,14 +1,12 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .appmode import AppMode
 from .hbleaderinfo import HBLeaderInfo, HBLeaderInfoTypedDict
 from .lookupversions import LookupVersions, LookupVersionsTypedDict
-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
@@ -38,19 +36,9 @@ class Config(BaseModel):
     version: Optional[str] = None
 
 
-class DistMode(str, Enum, metaclass=utils.OpenEnumMeta):
-    EDGE = "edge"
-    WORKER = "worker"
-    SINGLE = "single"
-    MASTER = "master"
-    MANAGED_EDGE = "managed-edge"
-    OUTPOST = "outpost"
-    SEARCH_SUPERVISOR = "search-supervisor"
-
-
 class HBCriblInfoTypedDict(TypedDict):
     config: ConfigTypedDict
-    dist_mode: DistMode
+    dist_mode: AppMode
     group: str
     guid: str
     start_time: float
@@ -70,7 +58,7 @@ class HBCriblInfo(BaseModel):
     config: Config
 
     dist_mode: Annotated[
-        Annotated[DistMode, PlainValidator(validate_open_enum(False))],
+        Annotated[AppMode, PlainValidator(validate_open_enum(False))],
         pydantic.Field(alias="distMode"),
     ]
 
@@ -105,12 +93,3 @@ 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 → healthstatus.py}

@@ -1,55 +1,36 @@
 """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 import 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
 
 
 class Role(str, Enum, metaclass=utils.OpenEnumMeta):
-    STANDBY = "standby"
     PRIMARY = "primary"
+    STANDBY = "standby"
 
 
 class Status(str, Enum, metaclass=utils.OpenEnumMeta):
-    SHUTTING_DOWN = "shutting down"
     HEALTHY = "healthy"
+    SHUTTING_DOWN = "shutting down"
     STANDBY = "standby"
 
 
-class HealthServerStatusTypedDict(TypedDict):
-    start_time: float
+class HealthStatusTypedDict(TypedDict):
     status: Status
+    start_time: float
     role: NotRequired[Role]
 
 
-class HealthServerStatus(BaseModel):
-    start_time: Annotated[float, pydantic.Field(alias="startTime")]
-
+class HealthStatus(BaseModel):
     status: Annotated[Status, PlainValidator(validate_open_enum(False))]
 
-    role: Annotated[Optional[Role], PlainValidator(validate_open_enum(False))] = None
+    start_time: Annotated[float, pydantic.Field(alias="startTime")]
 
-    @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
+    role: Annotated[Optional[Role], PlainValidator(validate_open_enum(False))] = None

cribl_control_plane/models/heartbeatmetadata.py

@@ -17,7 +17,6 @@ class HeartbeatMetadataTags(BaseModel):
 
 class HeartbeatMetadataAwsTypedDict(TypedDict):
     enabled: bool
-    instance_id: str
     region: str
     type: str
     zone: str
@@ -27,8 +26,6 @@ class HeartbeatMetadataAwsTypedDict(TypedDict):
 class HeartbeatMetadataAws(BaseModel):
     enabled: bool
 
-    instance_id: Annotated[str, pydantic.Field(alias="instanceId")]
-
     region: str
 
     type: str

cribl_control_plane/models/input.py

@@ -66,68 +66,66 @@ from .inputwineventlogs import InputWinEventLogs, InputWinEventLogsTypedDict
 from .inputwiz import InputWiz, InputWizTypedDict
 from .inputwizwebhook import InputWizWebhook, InputWizWebhookTypedDict
 from .inputzscalerhec import InputZscalerHec, InputZscalerHecTypedDict
-from cribl_control_plane.utils import get_discriminator
-from pydantic import Discriminator, Tag
 from typing import Union
-from typing_extensions import Annotated, TypeAliasType
+from typing_extensions import TypeAliasType
 
 
 InputTypedDict = TypeAliasType(
     "InputTypedDict",
     Union[
-        InputCriblTypedDict,
-        InputKubeEventsTypedDict,
         InputDatagenTypedDict,
+        InputKubeEventsTypedDict,
+        InputCriblTypedDict,
         InputCriblmetricsTypedDict,
         InputKubeMetricsTypedDict,
-        InputSystemStateTypedDict,
         InputCollectionTypedDict,
+        InputSystemStateTypedDict,
         InputModelDrivenTelemetryTypedDict,
         InputSystemMetricsTypedDict,
         InputWindowsMetricsTypedDict,
         InputJournalFilesTypedDict,
+        InputWinEventLogsTypedDict,
         InputRawUDPTypedDict,
-        InputKubeLogsTypedDict,
         InputExecTypedDict,
-        InputSnmpTypedDict,
+        InputKubeLogsTypedDict,
         InputMetricsTypedDict,
-        InputWinEventLogsTypedDict,
+        InputSnmpTypedDict,
         InputCriblTCPTypedDict,
         InputNetflowTypedDict,
         InputTcpjsonTypedDict,
         InputGooglePubsubTypedDict,
         InputOffice365ServiceTypedDict,
         InputWizTypedDict,
+        InputTCPTypedDict,
         InputFirehoseTypedDict,
         InputCriblHTTPTypedDict,
         InputDatadogAgentTypedDict,
         InputOffice365MgmtTypedDict,
-        InputTCPTypedDict,
-        InputSplunkTypedDict,
         InputFileTypedDict,
-        InputAppscopeTypedDict,
+        InputSplunkTypedDict,
         InputWefTypedDict,
+        InputAppscopeTypedDict,
         InputHTTPRawTypedDict,
         InputWizWebhookTypedDict,
-        InputCriblLakeHTTPTypedDict,
         InputHTTPTypedDict,
-        InputZscalerHecTypedDict,
-        InputSqsTypedDict,
+        InputCriblLakeHTTPTypedDict,
         InputAzureBlobTypedDict,
+        InputSqsTypedDict,
+        InputZscalerHecTypedDict,
         InputKinesisTypedDict,
         InputConfluentCloudTypedDict,
         InputEventhubTypedDict,
         InputKafkaTypedDict,
         InputElasticTypedDict,
-        InputSplunkHecTypedDict,
         InputOffice365MsgTraceTypedDict,
+        InputSplunkHecTypedDict,
         InputLokiTypedDict,
         InputPrometheusRwTypedDict,
-        InputCrowdstrikeTypedDict,
         InputPrometheusTypedDict,
+        InputCrowdstrikeTypedDict,
+        InputEdgePrometheusTypedDict,
         InputOpenTelemetryTypedDict,
         InputS3TypedDict,
-        InputEdgePrometheusTypedDict,
         InputSecurityLakeTypedDict,
         InputMskTypedDict,
         InputS3InventoryTypedDict,
@@ -138,67 +136,67 @@ InputTypedDict = TypeAliasType(
 )
 
 
-Input = Annotated[
+Input = TypeAliasType(
+    "Input",
     Union[
-        Annotated[InputCollection, Tag("collection")],
-        Annotated[InputKafka, Tag("kafka")],
-        Annotated[InputMsk, Tag("msk")],
-        Annotated[InputHTTP, Tag("http")],
-        Annotated[InputSplunk, Tag("splunk")],
-        Annotated[InputSplunkSearch, Tag("splunk_search")],
-        Annotated[InputSplunkHec, Tag("splunk_hec")],
-        Annotated[InputAzureBlob, Tag("azure_blob")],
-        Annotated[InputElastic, Tag("elastic")],
-        Annotated[InputConfluentCloud, Tag("confluent_cloud")],
-        Annotated[InputGrafana, Tag("grafana")],
-        Annotated[InputLoki, Tag("loki")],
-        Annotated[InputPrometheusRw, Tag("prometheus_rw")],
-        Annotated[InputPrometheus, Tag("prometheus")],
-        Annotated[InputEdgePrometheus, Tag("edge_prometheus")],
-        Annotated[InputOffice365Mgmt, Tag("office365_mgmt")],
-        Annotated[InputOffice365Service, Tag("office365_service")],
-        Annotated[InputOffice365MsgTrace, Tag("office365_msg_trace")],
-        Annotated[InputEventhub, Tag("eventhub")],
-        Annotated[InputExec, Tag("exec")],
-        Annotated[InputFirehose, Tag("firehose")],
-        Annotated[InputGooglePubsub, Tag("google_pubsub")],
-        Annotated[InputCribl, Tag("cribl")],
-        Annotated[InputCriblTCP, Tag("cribl_tcp")],
-        Annotated[InputCriblHTTP, Tag("cribl_http")],
-        Annotated[InputCriblLakeHTTP, Tag("cribl_lake_http")],
-        Annotated[InputTcpjson, Tag("tcpjson")],
-        Annotated[InputSystemMetrics, Tag("system_metrics")],
-        Annotated[InputSystemState, Tag("system_state")],
-        Annotated[InputKubeMetrics, Tag("kube_metrics")],
-        Annotated[InputKubeLogs, Tag("kube_logs")],
-        Annotated[InputKubeEvents, Tag("kube_events")],
-        Annotated[InputWindowsMetrics, Tag("windows_metrics")],
-        Annotated[InputCrowdstrike, Tag("crowdstrike")],
-        Annotated[InputDatadogAgent, Tag("datadog_agent")],
-        Annotated[InputDatagen, Tag("datagen")],
-        Annotated[InputHTTPRaw, Tag("http_raw")],
-        Annotated[InputKinesis, Tag("kinesis")],
-        Annotated[InputCriblmetrics, Tag("criblmetrics")],
-        Annotated[InputMetrics, Tag("metrics")],
-        Annotated[InputS3, Tag("s3")],
-        Annotated[InputS3Inventory, Tag("s3_inventory")],
-        Annotated[InputSnmp, Tag("snmp")],
-        Annotated[InputOpenTelemetry, Tag("open_telemetry")],
-        Annotated[InputModelDrivenTelemetry, Tag("model_driven_telemetry")],
-        Annotated[InputSqs, Tag("sqs")],
-        Annotated[InputSyslog, Tag("syslog")],
-        Annotated[InputFile, Tag("file")],
-        Annotated[InputTCP, Tag("tcp")],
-        Annotated[InputAppscope, Tag("appscope")],
-        Annotated[InputWef, Tag("wef")],
-        Annotated[InputWinEventLogs, Tag("win_event_logs")],
-        Annotated[InputRawUDP, Tag("raw_udp")],
-        Annotated[InputJournalFiles, Tag("journal_files")],
-        Annotated[InputWiz, Tag("wiz")],
-        Annotated[InputWizWebhook, Tag("wiz_webhook")],
-        Annotated[InputNetflow, Tag("netflow")],
-        Annotated[InputSecurityLake, Tag("security_lake")],
-        Annotated[InputZscalerHec, Tag("zscaler_hec")],
+        InputDatagen,
+        InputKubeEvents,
+        InputCribl,
+        InputCriblmetrics,
+        InputKubeMetrics,
+        InputCollection,
+        InputSystemState,
+        InputModelDrivenTelemetry,
+        InputSystemMetrics,
+        InputWindowsMetrics,
+        InputJournalFiles,
+        InputWinEventLogs,
+        InputRawUDP,
+        InputExec,
+        InputKubeLogs,
+        InputMetrics,
+        InputSnmp,
+        InputCriblTCP,
+        InputNetflow,
+        InputTcpjson,
+        InputGooglePubsub,
+        InputOffice365Service,
+        InputWiz,
+        InputTCP,
+        InputFirehose,
+        InputCriblHTTP,
+        InputDatadogAgent,
+        InputOffice365Mgmt,
+        InputFile,
+        InputSplunk,
+        InputWef,
+        InputAppscope,
+        InputHTTPRaw,
+        InputWizWebhook,
+        InputHTTP,
+        InputCriblLakeHTTP,
+        InputAzureBlob,
+        InputSqs,
+        InputZscalerHec,
+        InputKinesis,
+        InputConfluentCloud,
+        InputEventhub,
+        InputKafka,
+        InputElastic,
+        InputOffice365MsgTrace,
+        InputSplunkHec,
+        InputLoki,
+        InputPrometheusRw,
+        InputPrometheus,
+        InputCrowdstrike,
+        InputEdgePrometheus,
+        InputOpenTelemetry,
+        InputS3,
+        InputSecurityLake,
+        InputMsk,
+        InputS3Inventory,
+        InputSplunkSearch,
+        InputSyslog,
+        InputGrafana,
     ],
-    Discriminator(lambda m: get_discriminator(m, "type", "type")),
-]
+)

cribl_control_plane/models/inputappscope.py

@@ -1,14 +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 import 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 import Any, List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -30,18 +29,14 @@ class InputAppscopeConnection(BaseModel):
 class InputAppscopeMode(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 InputAppscopeCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
-    # None
     NONE = "none"
-    # Gzip
     GZIP = "gzip"
 
 
@@ -107,24 +102,6 @@ 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
@@ -216,15 +193,6 @@ 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"""
@@ -249,12 +217,6 @@ 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]
@@ -265,6 +227,10 @@ 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]
 
@@ -272,19 +238,6 @@ 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
@@ -302,6 +255,17 @@ 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],
@@ -318,24 +282,6 @@ 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
@@ -513,12 +459,3 @@ 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,12 +1,11 @@
 """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 import 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
@@ -30,18 +29,14 @@ class InputAzureBlobConnection(BaseModel):
 class InputAzureBlobMode(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 InputAzureBlobCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
-    # None
     NONE = "none"
-    # Gzip
     GZIP = "gzip"
 
 
@@ -107,24 +102,6 @@ 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
@@ -342,12 +319,3 @@ 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

cribl_control_plane/models/inputcollection.py

@@ -1,12 +1,11 @@
 """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 import 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
@@ -30,18 +29,14 @@ class InputCollectionConnection(BaseModel):
 class InputCollectionMode(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 InputCollectionCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
-    # None
     NONE = "none"
-    # Gzip
     GZIP = "gzip"
 
 
@@ -107,24 +102,6 @@ class InputCollectionPq(BaseModel):
         Optional[InputCollectionPqControls], pydantic.Field(alias="pqControls")
     ] = None
 
-    @field_serializer("mode")
-    def serialize_mode(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputCollectionMode(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("compress")
-    def serialize_compress(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputCollectionCompression(value)
-            except ValueError:
-                return value
-        return value
-
 
 class InputCollectionPreprocessTypedDict(TypedDict):
     disabled: NotRequired[bool]