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

cribl_control_plane/models/nodeprovidedinfo.py

@@ -19,6 +19,7 @@ class NodeProvidedInfoTags(BaseModel):
 
 class NodeProvidedInfoAwsTypedDict(TypedDict):
     enabled: bool
+    instance_id: str
     region: str
     type: str
     zone: str
@@ -28,6 +29,8 @@ 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,10 +1,11 @@
 """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):
+class NodeSkippedUpgradeStatus(int, Enum, metaclass=utils.OpenEnumMeta):
     ZERO = 0
     ONE = 1
     TWO = 2

cribl_control_plane/models/nodeupgradestate.py

@@ -1,10 +1,11 @@
 """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):
+class NodeUpgradeState(int, Enum, metaclass=utils.OpenEnumMeta):
     ZERO = 0
     ONE = 1
     TWO = 2

cribl_control_plane/models/nodeupgradestatus.py

@@ -6,8 +6,10 @@ 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 NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class NodeUpgradeStatusTypedDict(TypedDict):
@@ -19,12 +21,18 @@ class NodeUpgradeStatusTypedDict(TypedDict):
 
 
 class NodeUpgradeStatus(BaseModel):
-    state: NodeUpgradeState
+    state: Annotated[NodeUpgradeState, PlainValidator(validate_open_enum(True))]
 
     timestamp: float
 
-    active: Optional[NodeActiveUpgradeStatus] = None
+    active: Annotated[
+        Optional[NodeActiveUpgradeStatus], PlainValidator(validate_open_enum(True))
+    ] = None
 
-    failed: Optional[NodeFailedUpgradeStatus] = None
+    failed: Annotated[
+        Optional[NodeFailedUpgradeStatus], PlainValidator(validate_open_enum(True))
+    ] = None
 
-    skipped: Optional[NodeSkippedUpgradeStatus] = None
+    skipped: Annotated[
+        Optional[NodeSkippedUpgradeStatus], PlainValidator(validate_open_enum(True))
+    ] = None

cribl_control_plane/models/output.py

@@ -18,6 +18,7 @@ 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
@@ -80,8 +81,10 @@ 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 TypeAliasType
+from typing_extensions import Annotated, TypeAliasType
 
 
 OutputTypedDict = TypeAliasType(
@@ -90,31 +93,32 @@ OutputTypedDict = TypeAliasType(
         OutputDevnullTypedDict,
         OutputDefaultTypedDict,
         OutputRouterTypedDict,
-        OutputSnmpTypedDict,
         OutputNetflowTypedDict,
+        OutputSnmpTypedDict,
         OutputDiskSpoolTypedDict,
         OutputRingTypedDict,
-        OutputStatsdExtTypedDict,
         OutputGraphiteTypedDict,
+        OutputStatsdExtTypedDict,
         OutputStatsdTypedDict,
         OutputGooglePubsubTypedDict,
-        OutputCriblTCPTypedDict,
         OutputSplunkTypedDict,
+        OutputCriblTCPTypedDict,
         OutputSnsTypedDict,
         OutputCloudwatchTypedDict,
         OutputAzureEventhubTypedDict,
-        OutputWavefrontTypedDict,
         OutputSignalfxTypedDict,
+        OutputWavefrontTypedDict,
         OutputHoneycombTypedDict,
-        OutputSumoLogicTypedDict,
-        OutputCrowdstrikeNextGenSiemTypedDict,
         OutputHumioHecTypedDict,
         OutputTcpjsonTypedDict,
+        OutputSumoLogicTypedDict,
+        OutputCrowdstrikeNextGenSiemTypedDict,
         OutputElasticCloudTypedDict,
-        OutputKinesisTypedDict,
-        OutputConfluentCloudTypedDict,
-        OutputKafkaTypedDict,
         OutputExabeamTypedDict,
+        OutputKafkaTypedDict,
+        OutputConfluentCloudTypedDict,
+        OutputKinesisTypedDict,
+        OutputDatabricksTypedDict,
         OutputNewrelicEventsTypedDict,
         OutputAzureLogsTypedDict,
         OutputSplunkLbTypedDict,
@@ -130,10 +134,10 @@ OutputTypedDict = TypeAliasType(
         OutputDynatraceHTTPTypedDict,
         OutputServiceNowTypedDict,
         OutputDynatraceOtlpTypedDict,
-        OutputElasticTypedDict,
         OutputGoogleChronicleTypedDict,
-        OutputCriblLakeTypedDict,
+        OutputElasticTypedDict,
         OutputDatadogTypedDict,
+        OutputCriblLakeTypedDict,
         OutputPrometheusTypedDict,
         OutputMskTypedDict,
         OutputSentinelOneAiSiemTypedDict,
@@ -155,72 +159,73 @@ OutputTypedDict = TypeAliasType(
 )
 
 
-Output = TypeAliasType(
-    "Output",
+Output = Annotated[
     Union[
-        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,
+        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")],
     ],
-)
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]

cribl_control_plane/models/outputazureblob.py

@@ -1,9 +1,12 @@
 """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
 
@@ -12,7 +15,7 @@ class OutputAzureBlobType(str, Enum):
     AZURE_BLOB = "azure_blob"
 
 
-class OutputAzureBlobDataFormat(str, Enum):
+class OutputAzureBlobDataFormat(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Format of the output data"""
 
     JSON = "json"
@@ -20,28 +23,28 @@ class OutputAzureBlobDataFormat(str, Enum):
     PARQUET = "parquet"
 
 
-class OutputAzureBlobBackpressureBehavior(str, Enum):
+class OutputAzureBlobBackpressureBehavior(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""How to handle events when all receivers are exerting backpressure"""
 
     BLOCK = "block"
     DROP = "drop"
 
 
-class OutputAzureBlobDiskSpaceProtection(str, Enum):
+class OutputAzureBlobDiskSpaceProtection(str, Enum, metaclass=utils.OpenEnumMeta):
     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):
+class OutputAzureBlobAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMeta):
     MANUAL = "manual"
     SECRET = "secret"
     CLIENT_SECRET = "clientSecret"
     CLIENT_CERT = "clientCert"
 
 
-class BlobAccessTier(str, Enum):
+class BlobAccessTier(str, Enum, metaclass=utils.OpenEnumMeta):
     INFERRED = "Inferred"
     HOT = "Hot"
     COOL = "Cool"
@@ -49,14 +52,14 @@ class BlobAccessTier(str, Enum):
     ARCHIVE = "Archive"
 
 
-class OutputAzureBlobCompression(str, Enum):
+class OutputAzureBlobCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Data compression format to apply to HTTP content before it is delivered"""
 
     NONE = "none"
     GZIP = "gzip"
 
 
-class OutputAzureBlobCompressionLevel(str, Enum):
+class OutputAzureBlobCompressionLevel(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Compression level to apply before moving files to final destination"""
 
     BEST_SPEED = "best_speed"
@@ -64,7 +67,7 @@ class OutputAzureBlobCompressionLevel(str, Enum):
     BEST_COMPRESSION = "best_compression"
 
 
-class OutputAzureBlobParquetVersion(str, Enum):
+class OutputAzureBlobParquetVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Determines which data types are supported and how they are represented"""
 
     PARQUET_1_0 = "PARQUET_1_0"
@@ -72,7 +75,7 @@ class OutputAzureBlobParquetVersion(str, Enum):
     PARQUET_2_6 = "PARQUET_2_6"
 
 
-class OutputAzureBlobDataPageVersion(str, Enum):
+class OutputAzureBlobDataPageVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     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"
@@ -261,7 +264,11 @@ 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[
-        Optional[OutputAzureBlobDataFormat], pydantic.Field(alias="format")
+        Annotated[
+            Optional[OutputAzureBlobDataFormat],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="format"),
     ] = OutputAzureBlobDataFormat.JSON
     r"""Format of the output data"""
 
@@ -304,7 +311,10 @@ class OutputAzureBlob(BaseModel):
     r"""Buffer size used to write to a file"""
 
     on_backpressure: Annotated[
-        Optional[OutputAzureBlobBackpressureBehavior],
+        Annotated[
+            Optional[OutputAzureBlobBackpressureBehavior],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="onBackpressure"),
     ] = OutputAzureBlobBackpressureBehavior.BLOCK
     r"""How to handle events when all receivers are exerting backpressure"""
@@ -315,26 +325,39 @@ 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[
-        Optional[OutputAzureBlobDiskSpaceProtection],
+        Annotated[
+            Optional[OutputAzureBlobDiskSpaceProtection],
+            PlainValidator(validate_open_enum(False)),
+        ],
         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[
-        Optional[OutputAzureBlobAuthenticationMethod], pydantic.Field(alias="authType")
+        Annotated[
+            Optional[OutputAzureBlobAuthenticationMethod],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="authType"),
     ] = OutputAzureBlobAuthenticationMethod.MANUAL
 
     storage_class: Annotated[
-        Optional[BlobAccessTier], pydantic.Field(alias="storageClass")
+        Annotated[Optional[BlobAccessTier], PlainValidator(validate_open_enum(False))],
+        pydantic.Field(alias="storageClass"),
     ] = BlobAccessTier.INFERRED
 
     description: Optional[str] = None
 
-    compress: Optional[OutputAzureBlobCompression] = OutputAzureBlobCompression.GZIP
+    compress: Annotated[
+        Optional[OutputAzureBlobCompression], PlainValidator(validate_open_enum(False))
+    ] = OutputAzureBlobCompression.GZIP
     r"""Data compression format to apply to HTTP content before it is delivered"""
 
     compression_level: Annotated[
-        Optional[OutputAzureBlobCompressionLevel],
+        Annotated[
+            Optional[OutputAzureBlobCompressionLevel],
+            PlainValidator(validate_open_enum(False)),
+        ],
         pydantic.Field(alias="compressionLevel"),
     ] = OutputAzureBlobCompressionLevel.BEST_SPEED
     r"""Compression level to apply before moving files to final destination"""
@@ -345,12 +368,19 @@ class OutputAzureBlob(BaseModel):
     r"""Automatically calculate the schema based on the events of each Parquet file generated"""
 
     parquet_version: Annotated[
-        Optional[OutputAzureBlobParquetVersion], pydantic.Field(alias="parquetVersion")
+        Annotated[
+            Optional[OutputAzureBlobParquetVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        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[
-        Optional[OutputAzureBlobDataPageVersion],
+        Annotated[
+            Optional[OutputAzureBlobDataPageVersion],
+            PlainValidator(validate_open_enum(False)),
+        ],
         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."""