cribl-control-plane 0.3.0b3__py3-none-any.whl → 0.3.0b5__py3-none-any.whl

cribl_control_plane/models/nodeupgradestatus.py

@@ -5,8 +5,10 @@ from .nodeactiveupgradestatus import NodeActiveUpgradeStatus
 from .nodefailedupgradestatus import NodeFailedUpgradeStatus
 from .nodeskippedupgradestatus import NodeSkippedUpgradeStatus
 from .nodeupgradestate import NodeUpgradeState
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import validate_open_enum
+from pydantic import field_serializer
 from pydantic.functional_validators import PlainValidator
 from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
@@ -36,3 +38,39 @@ class NodeUpgradeStatus(BaseModel):
     skipped: Annotated[
         Optional[NodeSkippedUpgradeStatus], PlainValidator(validate_open_enum(True))
     ] = None
+
+    @field_serializer("active")
+    def serialize_active(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeActiveUpgradeStatus(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("failed")
+    def serialize_failed(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeFailedUpgradeStatus(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("skipped")
+    def serialize_skipped(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeSkippedUpgradeStatus(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("state")
+    def serialize_state(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeUpgradeState(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/output.py

@@ -98,64 +98,64 @@ OutputTypedDict = TypeAliasType(
         OutputNetflowTypedDict,
         OutputDiskSpoolTypedDict,
         OutputRingTypedDict,
-        OutputStatsdExtTypedDict,
         OutputGraphiteTypedDict,
+        OutputStatsdExtTypedDict,
         OutputStatsdTypedDict,
         OutputGooglePubsubTypedDict,
-        OutputCriblTCPTypedDict,
+        OutputExabeamTypedDict,
         OutputSplunkTypedDict,
         OutputSnsTypedDict,
-        OutputCloudwatchTypedDict,
+        OutputCriblTCPTypedDict,
         OutputAzureEventhubTypedDict,
-        OutputWavefrontTypedDict,
+        OutputCloudwatchTypedDict,
         OutputSignalfxTypedDict,
         OutputHoneycombTypedDict,
+        OutputWavefrontTypedDict,
+        OutputHumioHecTypedDict,
         OutputSumoLogicTypedDict,
         OutputCrowdstrikeNextGenSiemTypedDict,
-        OutputHumioHecTypedDict,
         OutputTcpjsonTypedDict,
         OutputElasticCloudTypedDict,
-        OutputKinesisTypedDict,
+        OutputNewrelicEventsTypedDict,
+        OutputFilesystemTypedDict,
         OutputConfluentCloudTypedDict,
         OutputKafkaTypedDict,
-        OutputExabeamTypedDict,
-        OutputNewrelicEventsTypedDict,
         OutputAzureLogsTypedDict,
+        OutputKinesisTypedDict,
         OutputSplunkLbTypedDict,
         OutputSyslogTypedDict,
         OutputSqsTypedDict,
-        OutputNewrelicTypedDict,
         OutputCriblHTTPTypedDict,
+        OutputNewrelicTypedDict,
         OutputXsiamTypedDict,
-        OutputFilesystemTypedDict,
         OutputDatasetTypedDict,
         OutputLokiTypedDict,
-        OutputSplunkHecTypedDict,
+        OutputCriblLakeTypedDict,
         OutputDynatraceHTTPTypedDict,
         OutputServiceNowTypedDict,
+        OutputSplunkHecTypedDict,
         OutputChronicleTypedDict,
         OutputDynatraceOtlpTypedDict,
-        OutputGoogleChronicleTypedDict,
+        OutputDatabricksTypedDict,
         OutputElasticTypedDict,
         OutputDatadogTypedDict,
-        OutputCriblLakeTypedDict,
-        OutputDatabricksTypedDict,
+        OutputGoogleChronicleTypedDict,
         OutputPrometheusTypedDict,
-        OutputMskTypedDict,
         OutputSentinelOneAiSiemTypedDict,
-        OutputSentinelTypedDict,
-        OutputInfluxdbTypedDict,
+        OutputMskTypedDict,
         OutputGoogleCloudStorageTypedDict,
+        OutputSentinelTypedDict,
         OutputAzureBlobTypedDict,
-        OutputOpenTelemetryTypedDict,
+        OutputInfluxdbTypedDict,
         OutputMinioTypedDict,
-        OutputClickHouseTypedDict,
         OutputSecurityLakeTypedDict,
-        OutputS3TypedDict,
+        OutputOpenTelemetryTypedDict,
+        OutputClickHouseTypedDict,
         OutputDlS3TypedDict,
+        OutputS3TypedDict,
         OutputWebhookTypedDict,
-        OutputAzureDataExplorerTypedDict,
         OutputGoogleCloudLoggingTypedDict,
+        OutputAzureDataExplorerTypedDict,
         OutputGrafanaCloudTypedDict,
     ],
 )

cribl_control_plane/models/outputazureblob.py

@@ -1,11 +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 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
@@ -184,6 +185,8 @@ class OutputAzureBlobTypedDict(TypedDict):
     r"""Compression level to apply before moving files to final destination"""
     automatic_schema: NotRequired[bool]
     r"""Automatically calculate the schema based on the events of each Parquet file generated"""
+    parquet_schema: NotRequired[str]
+    r"""To add a new schema, navigate to Processing > Knowledge > Parquet Schemas"""
     parquet_version: NotRequired[OutputAzureBlobParquetVersion]
     r"""Determines which data types are supported and how they are represented"""
     parquet_data_page_version: NotRequired[OutputAzureBlobDataPageVersion]
@@ -387,6 +390,11 @@ class OutputAzureBlob(BaseModel):
     ] = False
     r"""Automatically calculate the schema based on the events of each Parquet file generated"""
 
+    parquet_schema: Annotated[Optional[str], pydantic.Field(alias="parquetSchema")] = (
+        None
+    )
+    r"""To add a new schema, navigate to Processing > Knowledge > Parquet Schemas"""
+
     parquet_version: Annotated[
         Annotated[
             Optional[OutputAzureBlobParquetVersion],
@@ -487,3 +495,84 @@ class OutputAzureBlob(BaseModel):
     r"""Select or create a stored text secret"""
 
     certificate: Optional[OutputAzureBlobCertificate] = None
+
+    @field_serializer("format_")
+    def serialize_format_(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobDataFormat(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("on_backpressure")
+    def serialize_on_backpressure(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobBackpressureBehavior(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("on_disk_full_backpressure")
+    def serialize_on_disk_full_backpressure(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobDiskSpaceProtection(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("auth_type")
+    def serialize_auth_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("storage_class")
+    def serialize_storage_class(self, value):
+        if isinstance(value, str):
+            try:
+                return models.BlobAccessTier(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobCompression(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compression_level")
+    def serialize_compression_level(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobCompressionLevel(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("parquet_version")
+    def serialize_parquet_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobParquetVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("parquet_data_page_version")
+    def serialize_parquet_data_page_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureBlobDataPageVersion(value)
+            except ValueError:
+                return value
+        return value