cribl-control-plane 0.0.50rc2__py3-none-any.whl → 0.0.52__py3-none-any.whl

cribl_control_plane/models/nodeprovidedinfo.py

@@ -19,7 +19,6 @@ class NodeProvidedInfoTags(BaseModel):
 
 class NodeProvidedInfoAwsTypedDict(TypedDict):
     enabled: bool
-    instance_id: str
     region: str
     type: str
     zone: str
@@ -29,8 +28,6 @@ class NodeProvidedInfoAwsTypedDict(TypedDict):
 class NodeProvidedInfoAws(BaseModel):
     enabled: bool
 
-    instance_id: Annotated[str, pydantic.Field(alias="instanceId")]
-
     region: str
 
     type: str

cribl_control_plane/models/nodeskippedupgradestatus.py

@@ -1,11 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from cribl_control_plane import utils
 from enum import Enum
 
 
-class NodeSkippedUpgradeStatus(int, Enum, metaclass=utils.OpenEnumMeta):
+class NodeSkippedUpgradeStatus(int, Enum):
     ZERO = 0
     ONE = 1
     TWO = 2

cribl_control_plane/models/nodeupgradestate.py

@@ -1,11 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from cribl_control_plane import utils
 from enum import Enum
 
 
-class NodeUpgradeState(int, Enum, metaclass=utils.OpenEnumMeta):
+class NodeUpgradeState(int, Enum):
     ZERO = 0
     ONE = 1
     TWO = 2

cribl_control_plane/models/nodeupgradestatus.py

@@ -6,10 +6,8 @@ from .nodefailedupgradestatus import NodeFailedUpgradeStatus
 from .nodeskippedupgradestatus import NodeSkippedUpgradeStatus
 from .nodeupgradestate import NodeUpgradeState
 from cribl_control_plane.types import BaseModel
-from cribl_control_plane.utils import validate_open_enum
-from pydantic.functional_validators import PlainValidator
 from typing import Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class NodeUpgradeStatusTypedDict(TypedDict):
@@ -21,18 +19,12 @@ class NodeUpgradeStatusTypedDict(TypedDict):
 
 
 class NodeUpgradeStatus(BaseModel):
-    state: Annotated[NodeUpgradeState, PlainValidator(validate_open_enum(True))]
+    state: NodeUpgradeState
 
     timestamp: float
 
-    active: Annotated[
-        Optional[NodeActiveUpgradeStatus], PlainValidator(validate_open_enum(True))
-    ] = None
+    active: Optional[NodeActiveUpgradeStatus] = None
 
-    failed: Annotated[
-        Optional[NodeFailedUpgradeStatus], PlainValidator(validate_open_enum(True))
-    ] = None
+    failed: Optional[NodeFailedUpgradeStatus] = None
 
-    skipped: Annotated[
-        Optional[NodeSkippedUpgradeStatus], PlainValidator(validate_open_enum(True))
-    ] = None
+    skipped: Optional[NodeSkippedUpgradeStatus] = None

cribl_control_plane/models/output.py

@@ -18,7 +18,6 @@ from .outputcrowdstrikenextgensiem import (
     OutputCrowdstrikeNextGenSiem,
     OutputCrowdstrikeNextGenSiemTypedDict,
 )
-from .outputdatabricks import OutputDatabricks, OutputDatabricksTypedDict
 from .outputdatadog import OutputDatadog, OutputDatadogTypedDict
 from .outputdataset import OutputDataset, OutputDatasetTypedDict
 from .outputdefault import OutputDefault, OutputDefaultTypedDict
@@ -81,10 +80,8 @@ from .outputtcpjson import OutputTcpjson, OutputTcpjsonTypedDict
 from .outputwavefront import OutputWavefront, OutputWavefrontTypedDict
 from .outputwebhook import OutputWebhook, OutputWebhookTypedDict
 from .outputxsiam import OutputXsiam, OutputXsiamTypedDict
-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
 
 
 OutputTypedDict = TypeAliasType(
@@ -93,32 +90,31 @@ OutputTypedDict = TypeAliasType(
         OutputDevnullTypedDict,
         OutputDefaultTypedDict,
         OutputRouterTypedDict,
-        OutputNetflowTypedDict,
         OutputSnmpTypedDict,
+        OutputNetflowTypedDict,
         OutputDiskSpoolTypedDict,
         OutputRingTypedDict,
-        OutputGraphiteTypedDict,
         OutputStatsdExtTypedDict,
+        OutputGraphiteTypedDict,
         OutputStatsdTypedDict,
         OutputGooglePubsubTypedDict,
-        OutputSplunkTypedDict,
         OutputCriblTCPTypedDict,
+        OutputSplunkTypedDict,
         OutputSnsTypedDict,
         OutputCloudwatchTypedDict,
         OutputAzureEventhubTypedDict,
-        OutputSignalfxTypedDict,
         OutputWavefrontTypedDict,
+        OutputSignalfxTypedDict,
         OutputHoneycombTypedDict,
-        OutputHumioHecTypedDict,
-        OutputTcpjsonTypedDict,
         OutputSumoLogicTypedDict,
         OutputCrowdstrikeNextGenSiemTypedDict,
+        OutputHumioHecTypedDict,
+        OutputTcpjsonTypedDict,
         OutputElasticCloudTypedDict,
-        OutputExabeamTypedDict,
-        OutputKafkaTypedDict,
-        OutputConfluentCloudTypedDict,
         OutputKinesisTypedDict,
-        OutputDatabricksTypedDict,
+        OutputConfluentCloudTypedDict,
+        OutputKafkaTypedDict,
+        OutputExabeamTypedDict,
         OutputNewrelicEventsTypedDict,
         OutputAzureLogsTypedDict,
         OutputSplunkLbTypedDict,
@@ -134,10 +130,10 @@ OutputTypedDict = TypeAliasType(
         OutputDynatraceHTTPTypedDict,
         OutputServiceNowTypedDict,
         OutputDynatraceOtlpTypedDict,
-        OutputGoogleChronicleTypedDict,
         OutputElasticTypedDict,
-        OutputDatadogTypedDict,
+        OutputGoogleChronicleTypedDict,
         OutputCriblLakeTypedDict,
+        OutputDatadogTypedDict,
         OutputPrometheusTypedDict,
         OutputMskTypedDict,
         OutputSentinelOneAiSiemTypedDict,
@@ -159,73 +155,72 @@ OutputTypedDict = TypeAliasType(
 )
 
 
-Output = Annotated[
+Output = TypeAliasType(
+    "Output",
     Union[
-        Annotated[OutputDefault, Tag("default")],
-        Annotated[OutputWebhook, Tag("webhook")],
-        Annotated[OutputSentinel, Tag("sentinel")],
-        Annotated[OutputDevnull, Tag("devnull")],
-        Annotated[OutputSyslog, Tag("syslog")],
-        Annotated[OutputSplunk, Tag("splunk")],
-        Annotated[OutputSplunkLb, Tag("splunk_lb")],
-        Annotated[OutputSplunkHec, Tag("splunk_hec")],
-        Annotated[OutputTcpjson, Tag("tcpjson")],
-        Annotated[OutputWavefront, Tag("wavefront")],
-        Annotated[OutputSignalfx, Tag("signalfx")],
-        Annotated[OutputFilesystem, Tag("filesystem")],
-        Annotated[OutputS3, Tag("s3")],
-        Annotated[OutputAzureBlob, Tag("azure_blob")],
-        Annotated[OutputAzureDataExplorer, Tag("azure_data_explorer")],
-        Annotated[OutputAzureLogs, Tag("azure_logs")],
-        Annotated[OutputKinesis, Tag("kinesis")],
-        Annotated[OutputHoneycomb, Tag("honeycomb")],
-        Annotated[OutputAzureEventhub, Tag("azure_eventhub")],
-        Annotated[OutputGoogleChronicle, Tag("google_chronicle")],
-        Annotated[OutputGoogleCloudStorage, Tag("google_cloud_storage")],
-        Annotated[OutputGoogleCloudLogging, Tag("google_cloud_logging")],
-        Annotated[OutputGooglePubsub, Tag("google_pubsub")],
-        Annotated[OutputExabeam, Tag("exabeam")],
-        Annotated[OutputKafka, Tag("kafka")],
-        Annotated[OutputConfluentCloud, Tag("confluent_cloud")],
-        Annotated[OutputMsk, Tag("msk")],
-        Annotated[OutputElastic, Tag("elastic")],
-        Annotated[OutputElasticCloud, Tag("elastic_cloud")],
-        Annotated[OutputNewrelic, Tag("newrelic")],
-        Annotated[OutputNewrelicEvents, Tag("newrelic_events")],
-        Annotated[OutputInfluxdb, Tag("influxdb")],
-        Annotated[OutputCloudwatch, Tag("cloudwatch")],
-        Annotated[OutputMinio, Tag("minio")],
-        Annotated[OutputStatsd, Tag("statsd")],
-        Annotated[OutputStatsdExt, Tag("statsd_ext")],
-        Annotated[OutputGraphite, Tag("graphite")],
-        Annotated[OutputRouter, Tag("router")],
-        Annotated[OutputSns, Tag("sns")],
-        Annotated[OutputSqs, Tag("sqs")],
-        Annotated[OutputSnmp, Tag("snmp")],
-        Annotated[OutputSumoLogic, Tag("sumo_logic")],
-        Annotated[OutputDatadog, Tag("datadog")],
-        Annotated[OutputGrafanaCloud, Tag("grafana_cloud")],
-        Annotated[OutputLoki, Tag("loki")],
-        Annotated[OutputPrometheus, Tag("prometheus")],
-        Annotated[OutputRing, Tag("ring")],
-        Annotated[OutputOpenTelemetry, Tag("open_telemetry")],
-        Annotated[OutputServiceNow, Tag("service_now")],
-        Annotated[OutputDataset, Tag("dataset")],
-        Annotated[OutputCriblTCP, Tag("cribl_tcp")],
-        Annotated[OutputCriblHTTP, Tag("cribl_http")],
-        Annotated[OutputHumioHec, Tag("humio_hec")],
-        Annotated[OutputCrowdstrikeNextGenSiem, Tag("crowdstrike_next_gen_siem")],
-        Annotated[OutputDlS3, Tag("dl_s3")],
-        Annotated[OutputSecurityLake, Tag("security_lake")],
-        Annotated[OutputCriblLake, Tag("cribl_lake")],
-        Annotated[OutputDiskSpool, Tag("disk_spool")],
-        Annotated[OutputClickHouse, Tag("click_house")],
-        Annotated[OutputXsiam, Tag("xsiam")],
-        Annotated[OutputNetflow, Tag("netflow")],
-        Annotated[OutputDynatraceHTTP, Tag("dynatrace_http")],
-        Annotated[OutputDynatraceOtlp, Tag("dynatrace_otlp")],
-        Annotated[OutputSentinelOneAiSiem, Tag("sentinel_one_ai_siem")],
-        Annotated[OutputDatabricks, Tag("databricks")],
+        OutputDevnull,
+        OutputDefault,
+        OutputRouter,
+        OutputSnmp,
+        OutputNetflow,
+        OutputDiskSpool,
+        OutputRing,
+        OutputStatsdExt,
+        OutputGraphite,
+        OutputStatsd,
+        OutputGooglePubsub,
+        OutputCriblTCP,
+        OutputSplunk,
+        OutputSns,
+        OutputCloudwatch,
+        OutputAzureEventhub,
+        OutputWavefront,
+        OutputSignalfx,
+        OutputHoneycomb,
+        OutputSumoLogic,
+        OutputCrowdstrikeNextGenSiem,
+        OutputHumioHec,
+        OutputTcpjson,
+        OutputElasticCloud,
+        OutputKinesis,
+        OutputConfluentCloud,
+        OutputKafka,
+        OutputExabeam,
+        OutputNewrelicEvents,
+        OutputAzureLogs,
+        OutputSplunkLb,
+        OutputSyslog,
+        OutputSqs,
+        OutputNewrelic,
+        OutputCriblHTTP,
+        OutputXsiam,
+        OutputFilesystem,
+        OutputDataset,
+        OutputLoki,
+        OutputSplunkHec,
+        OutputDynatraceHTTP,
+        OutputServiceNow,
+        OutputDynatraceOtlp,
+        OutputElastic,
+        OutputGoogleChronicle,
+        OutputCriblLake,
+        OutputDatadog,
+        OutputPrometheus,
+        OutputMsk,
+        OutputSentinelOneAiSiem,
+        OutputSentinel,
+        OutputInfluxdb,
+        OutputGoogleCloudStorage,
+        OutputAzureBlob,
+        OutputOpenTelemetry,
+        OutputMinio,
+        OutputClickHouse,
+        OutputSecurityLake,
+        OutputDlS3,
+        OutputS3,
+        OutputWebhook,
+        OutputAzureDataExplorer,
+        OutputGoogleCloudLogging,
+        OutputGrafanaCloud,
     ],
-    Discriminator(lambda m: get_discriminator(m, "type", "type")),
-]
+)

cribl_control_plane/models/outputazureblob.py

@@ -1,12 +1,9 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-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.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -15,7 +12,7 @@ class OutputAzureBlobType(str, Enum):
     AZURE_BLOB = "azure_blob"
 
 
-class OutputAzureBlobDataFormat(str, Enum, metaclass=utils.OpenEnumMeta):
+class OutputAzureBlobDataFormat(str, Enum):
     r"""Format of the output data"""
 
     JSON = "json"
@@ -23,28 +20,28 @@ class OutputAzureBlobDataFormat(str, Enum, metaclass=utils.OpenEnumMeta):
     PARQUET = "parquet"
 
 
-class OutputAzureBlobBackpressureBehavior(str, Enum, metaclass=utils.OpenEnumMeta):
+class OutputAzureBlobBackpressureBehavior(str, Enum):
     r"""How to handle events when all receivers are exerting backpressure"""
 
     BLOCK = "block"
     DROP = "drop"
 
 
-class OutputAzureBlobDiskSpaceProtection(str, Enum, metaclass=utils.OpenEnumMeta):
+class OutputAzureBlobDiskSpaceProtection(str, Enum):
     r"""How to handle events when disk space is below the global 'Min free disk space' limit"""
 
     BLOCK = "block"
     DROP = "drop"
 
 
-class OutputAzureBlobAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMeta):
+class OutputAzureBlobAuthenticationMethod(str, Enum):
     MANUAL = "manual"
     SECRET = "secret"
     CLIENT_SECRET = "clientSecret"
     CLIENT_CERT = "clientCert"
 
 
-class BlobAccessTier(str, Enum, metaclass=utils.OpenEnumMeta):
+class BlobAccessTier(str, Enum):
     INFERRED = "Inferred"
     HOT = "Hot"
     COOL = "Cool"
@@ -52,14 +49,14 @@ class BlobAccessTier(str, Enum, metaclass=utils.OpenEnumMeta):
     ARCHIVE = "Archive"
 
 
-class OutputAzureBlobCompression(str, Enum, metaclass=utils.OpenEnumMeta):
+class OutputAzureBlobCompression(str, Enum):
     r"""Data compression format to apply to HTTP content before it is delivered"""
 
     NONE = "none"
     GZIP = "gzip"
 
 
-class OutputAzureBlobCompressionLevel(str, Enum, metaclass=utils.OpenEnumMeta):
+class OutputAzureBlobCompressionLevel(str, Enum):
     r"""Compression level to apply before moving files to final destination"""
 
     BEST_SPEED = "best_speed"
@@ -67,7 +64,7 @@ class OutputAzureBlobCompressionLevel(str, Enum, metaclass=utils.OpenEnumMeta):
     BEST_COMPRESSION = "best_compression"
 
 
-class OutputAzureBlobParquetVersion(str, Enum, metaclass=utils.OpenEnumMeta):
+class OutputAzureBlobParquetVersion(str, Enum):
     r"""Determines which data types are supported and how they are represented"""
 
     PARQUET_1_0 = "PARQUET_1_0"
@@ -75,7 +72,7 @@ class OutputAzureBlobParquetVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     PARQUET_2_6 = "PARQUET_2_6"
 
 
-class OutputAzureBlobDataPageVersion(str, Enum, metaclass=utils.OpenEnumMeta):
+class OutputAzureBlobDataPageVersion(str, Enum):
     r"""Serialization format of data pages. Note that some reader implementations use Data page V2's attributes to work more efficiently, while others ignore it."""
 
     DATA_PAGE_V1 = "DATA_PAGE_V1"
@@ -264,11 +261,7 @@ class OutputAzureBlob(BaseModel):
     r"""JavaScript expression defining how files are partitioned and organized. Default is date-based. If blank, Stream will fall back to the event's __partition field value – if present – otherwise to each location's root directory."""
 
     format_: Annotated[
-        Annotated[
-            Optional[OutputAzureBlobDataFormat],
-            PlainValidator(validate_open_enum(False)),
-        ],
-        pydantic.Field(alias="format"),
+        Optional[OutputAzureBlobDataFormat], pydantic.Field(alias="format")
     ] = OutputAzureBlobDataFormat.JSON
     r"""Format of the output data"""
 
@@ -311,10 +304,7 @@ class OutputAzureBlob(BaseModel):
     r"""Buffer size used to write to a file"""
 
     on_backpressure: Annotated[
-        Annotated[
-            Optional[OutputAzureBlobBackpressureBehavior],
-            PlainValidator(validate_open_enum(False)),
-        ],
+        Optional[OutputAzureBlobBackpressureBehavior],
         pydantic.Field(alias="onBackpressure"),
     ] = OutputAzureBlobBackpressureBehavior.BLOCK
     r"""How to handle events when all receivers are exerting backpressure"""
@@ -325,39 +315,26 @@ class OutputAzureBlob(BaseModel):
     r"""If a file fails to move to its final destination after the maximum number of retries, move it to a designated directory to prevent further errors"""
 
     on_disk_full_backpressure: Annotated[
-        Annotated[
-            Optional[OutputAzureBlobDiskSpaceProtection],
-            PlainValidator(validate_open_enum(False)),
-        ],
+        Optional[OutputAzureBlobDiskSpaceProtection],
         pydantic.Field(alias="onDiskFullBackpressure"),
     ] = OutputAzureBlobDiskSpaceProtection.BLOCK
     r"""How to handle events when disk space is below the global 'Min free disk space' limit"""
 
     auth_type: Annotated[
-        Annotated[
-            Optional[OutputAzureBlobAuthenticationMethod],
-            PlainValidator(validate_open_enum(False)),
-        ],
-        pydantic.Field(alias="authType"),
+        Optional[OutputAzureBlobAuthenticationMethod], pydantic.Field(alias="authType")
     ] = OutputAzureBlobAuthenticationMethod.MANUAL
 
     storage_class: Annotated[
-        Annotated[Optional[BlobAccessTier], PlainValidator(validate_open_enum(False))],
-        pydantic.Field(alias="storageClass"),
+        Optional[BlobAccessTier], pydantic.Field(alias="storageClass")
     ] = BlobAccessTier.INFERRED
 
     description: Optional[str] = None
 
-    compress: Annotated[
-        Optional[OutputAzureBlobCompression], PlainValidator(validate_open_enum(False))
-    ] = OutputAzureBlobCompression.GZIP
+    compress: Optional[OutputAzureBlobCompression] = OutputAzureBlobCompression.GZIP
     r"""Data compression format to apply to HTTP content before it is delivered"""
 
     compression_level: Annotated[
-        Annotated[
-            Optional[OutputAzureBlobCompressionLevel],
-            PlainValidator(validate_open_enum(False)),
-        ],
+        Optional[OutputAzureBlobCompressionLevel],
         pydantic.Field(alias="compressionLevel"),
     ] = OutputAzureBlobCompressionLevel.BEST_SPEED
     r"""Compression level to apply before moving files to final destination"""
@@ -368,19 +345,12 @@ class OutputAzureBlob(BaseModel):
     r"""Automatically calculate the schema based on the events of each Parquet file generated"""
 
     parquet_version: Annotated[
-        Annotated[
-            Optional[OutputAzureBlobParquetVersion],
-            PlainValidator(validate_open_enum(False)),
-        ],
-        pydantic.Field(alias="parquetVersion"),
+        Optional[OutputAzureBlobParquetVersion], pydantic.Field(alias="parquetVersion")
     ] = OutputAzureBlobParquetVersion.PARQUET_2_6
     r"""Determines which data types are supported and how they are represented"""
 
     parquet_data_page_version: Annotated[
-        Annotated[
-            Optional[OutputAzureBlobDataPageVersion],
-            PlainValidator(validate_open_enum(False)),
-        ],
+        Optional[OutputAzureBlobDataPageVersion],
         pydantic.Field(alias="parquetDataPageVersion"),
     ] = OutputAzureBlobDataPageVersion.DATA_PAGE_V2
     r"""Serialization format of data pages. Note that some reader implementations use Data page V2's attributes to work more efficiently, while others ignore it."""