cribl-control-plane 0.2.1rc4__py3-none-any.whl → 0.2.1rc11__py3-none-any.whl

cribl_control_plane/models/lakedatasetmetrics.py

@@ -0,0 +1,17 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from cribl_control_plane.types import BaseModel
+import pydantic
+from typing_extensions import Annotated, TypedDict
+
+
+class LakeDatasetMetricsTypedDict(TypedDict):
+    current_size_bytes: float
+    metrics_date: str
+
+
+class LakeDatasetMetrics(BaseModel):
+    current_size_bytes: Annotated[float, pydantic.Field(alias="currentSizeBytes")]
+
+    metrics_date: Annotated[str, pydantic.Field(alias="metricsDate")]

cribl_control_plane/models/listconfiggroupbyproductop.py

@@ -1,8 +1,8 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .configgroup import ConfigGroup, ConfigGroupTypedDict
 from .productscore import ProductsCore
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import (
     FieldMetadata,
@@ -10,8 +10,9 @@ from cribl_control_plane.utils import (
     QueryParamMetadata,
     validate_open_enum,
 )
+from pydantic import field_serializer
 from pydantic.functional_validators import PlainValidator
-from typing import List, Optional
+from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -35,19 +36,11 @@ class ListConfigGroupByProductRequest(BaseModel):
     ] = None
     r"""Comma-separated list of additional properties to include in the response. Available values are <code>git.commit</code>, <code>git.localChanges</code>, and <code>git.log</code>."""
 
-
-class ListConfigGroupByProductResponseTypedDict(TypedDict):
-    r"""a list of ConfigGroup objects"""
-
-    count: NotRequired[int]
-    r"""number of items present in the items array"""
-    items: NotRequired[List[ConfigGroupTypedDict]]
-
-
-class ListConfigGroupByProductResponse(BaseModel):
-    r"""a list of ConfigGroup objects"""
-
-    count: Optional[int] = None
-    r"""number of items present in the items array"""
-
-    items: Optional[List[ConfigGroup]] = None
+    @field_serializer("product")
+    def serialize_product(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ProductsCore(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/listmasterworkerentryop.py

@@ -1,11 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .masterworkerentry import MasterWorkerEntry, MasterWorkerEntryTypedDict
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import FieldMetadata, QueryParamMetadata
 import pydantic
-from typing import List, Optional
+from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -63,20 +62,3 @@ class ListMasterWorkerEntryRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = None
     r"""Starting point from which to retrieve results for this request. Use with <code>limit</code> to paginate the response into manageable batches."""
-
-
-class ListMasterWorkerEntryResponseTypedDict(TypedDict):
-    r"""a list of MasterWorkerEntry objects"""
-
-    count: NotRequired[int]
-    r"""number of items present in the items array"""
-    items: NotRequired[List[MasterWorkerEntryTypedDict]]
-
-
-class ListMasterWorkerEntryResponse(BaseModel):
-    r"""a list of MasterWorkerEntry objects"""
-
-    count: Optional[int] = None
-    r"""number of items present in the items array"""
-
-    items: Optional[List[MasterWorkerEntry]] = None

cribl_control_plane/models/masterworkerentry.py

@@ -4,11 +4,12 @@ from __future__ import annotations
 from .heartbeatmetadata import HeartbeatMetadata, HeartbeatMetadataTypedDict
 from .nodeprovidedinfo import NodeProvidedInfo, NodeProvidedInfoTypedDict
 from .nodeupgradestatus import NodeUpgradeStatus, NodeUpgradeStatusTypedDict
-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 Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
@@ -87,3 +88,12 @@ class MasterWorkerEntry(BaseModel):
     ] = None
 
     workers: Optional[MasterWorkerEntryWorkers] = None
+
+    @field_serializer("type")
+    def serialize_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.MasterWorkerEntryType(value)
+            except ValueError:
+                return value
+        return value

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,42 +98,42 @@ OutputTypedDict = TypeAliasType(
         OutputNetflowTypedDict,
         OutputDiskSpoolTypedDict,
         OutputRingTypedDict,
-        OutputStatsdTypedDict,
         OutputGraphiteTypedDict,
         OutputStatsdExtTypedDict,
+        OutputStatsdTypedDict,
         OutputGooglePubsubTypedDict,
+        OutputExabeamTypedDict,
         OutputSplunkTypedDict,
         OutputSnsTypedDict,
+        OutputAzureEventhubTypedDict,
         OutputCriblTCPTypedDict,
-        OutputExabeamTypedDict,
         OutputCloudwatchTypedDict,
-        OutputAzureEventhubTypedDict,
         OutputSignalfxTypedDict,
-        OutputWavefrontTypedDict,
         OutputHoneycombTypedDict,
+        OutputWavefrontTypedDict,
         OutputHumioHecTypedDict,
-        OutputElasticCloudTypedDict,
         OutputSumoLogicTypedDict,
         OutputCrowdstrikeNextGenSiemTypedDict,
         OutputTcpjsonTypedDict,
-        OutputAzureLogsTypedDict,
+        OutputElasticCloudTypedDict,
+        OutputNewrelicEventsTypedDict,
         OutputFilesystemTypedDict,
-        OutputKafkaTypedDict,
         OutputConfluentCloudTypedDict,
+        OutputKafkaTypedDict,
+        OutputAzureLogsTypedDict,
         OutputKinesisTypedDict,
-        OutputNewrelicEventsTypedDict,
         OutputSplunkLbTypedDict,
         OutputSyslogTypedDict,
         OutputSqsTypedDict,
-        OutputCriblHTTPTypedDict,
         OutputNewrelicTypedDict,
+        OutputCriblHTTPTypedDict,
         OutputXsiamTypedDict,
         OutputDatasetTypedDict,
+        OutputLokiTypedDict,
         OutputCriblLakeTypedDict,
-        OutputSplunkHecTypedDict,
         OutputDynatraceHTTPTypedDict,
-        OutputLokiTypedDict,
         OutputServiceNowTypedDict,
+        OutputSplunkHecTypedDict,
         OutputChronicleTypedDict,
         OutputDynatraceOtlpTypedDict,
         OutputDatabricksTypedDict,
@@ -144,15 +144,15 @@ OutputTypedDict = TypeAliasType(
         OutputSentinelOneAiSiemTypedDict,
         OutputMskTypedDict,
         OutputGoogleCloudStorageTypedDict,
-        OutputAzureBlobTypedDict,
         OutputSentinelTypedDict,
+        OutputAzureBlobTypedDict,
         OutputInfluxdbTypedDict,
         OutputMinioTypedDict,
         OutputSecurityLakeTypedDict,
         OutputOpenTelemetryTypedDict,
         OutputClickHouseTypedDict,
-        OutputS3TypedDict,
         OutputDlS3TypedDict,
+        OutputS3TypedDict,
         OutputWebhookTypedDict,
         OutputGoogleCloudLoggingTypedDict,
         OutputAzureDataExplorerTypedDict,

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
@@ -494,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

cribl_control_plane/models/outputazuredataexplorer.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
@@ -162,6 +163,15 @@ class ExtentTag(BaseModel):
         Optional[PrefixOptional], PlainValidator(validate_open_enum(False))
     ] = None
 
+    @field_serializer("prefix")
+    def serialize_prefix(self, value):
+        if isinstance(value, str):
+            try:
+                return models.PrefixOptional(value)
+            except ValueError:
+                return value
+        return value
+
 
 class IngestIfNotExistTypedDict(TypedDict):
     value: str
@@ -881,3 +891,140 @@ class OutputAzureDataExplorer(BaseModel):
     pq_controls: Annotated[
         Optional[OutputAzureDataExplorerPqControls], pydantic.Field(alias="pqControls")
     ] = None
+
+    @field_serializer("ingest_mode")
+    def serialize_ingest_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.IngestionMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("oauth_endpoint")
+    def serialize_oauth_endpoint(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureDataExplorerMicrosoftEntraIDAuthenticationEndpoint(
+                    value
+                )
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("oauth_type")
+    def serialize_oauth_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureDataExplorerAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("format_")
+    def serialize_format_(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureDataExplorerDataFormat(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compress")
+    def serialize_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureDataExplorerCompressCompression(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("compression_level")
+    def serialize_compression_level(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureDataExplorerCompressionLevel(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("parquet_version")
+    def serialize_parquet_version(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureDataExplorerParquetVersion(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.OutputAzureDataExplorerDataPageVersion(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("on_backpressure")
+    def serialize_on_backpressure(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureDataExplorerBackpressureBehavior(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.OutputAzureDataExplorerDiskSpaceProtection(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("report_level")
+    def serialize_report_level(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ReportLevel(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("report_method")
+    def serialize_report_method(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ReportMethod(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("pq_mode")
+    def serialize_pq_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureDataExplorerMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("pq_compress")
+    def serialize_pq_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureDataExplorerPqCompressCompression(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("pq_on_backpressure")
+    def serialize_pq_on_backpressure(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureDataExplorerQueueFullBehavior(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/outputazureeventhub.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
@@ -179,6 +180,48 @@ class OutputAzureEventhubAuthentication(BaseModel):
     scope: Optional[str] = None
     r"""Scope to pass in the OAuth request parameter"""
 
+    @field_serializer("auth_type")
+    def serialize_auth_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureEventhubAuthTypeAuthenticationMethod(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("mechanism")
+    def serialize_mechanism(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureEventhubSASLMechanism(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("client_secret_auth_type")
+    def serialize_client_secret_auth_type(self, value):
+        if isinstance(value, str):
+            try:
+                return (
+                    models.OutputAzureEventhubClientSecretAuthTypeAuthenticationMethod(
+                        value
+                    )
+                )
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("oauth_endpoint")
+    def serialize_oauth_endpoint(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureEventhubMicrosoftEntraIDAuthenticationEndpoint(
+                    value
+                )
+            except ValueError:
+                return value
+        return value
+
 
 class OutputAzureEventhubTLSSettingsClientSideTypedDict(TypedDict):
     disabled: NotRequired[bool]
@@ -482,3 +525,57 @@ class OutputAzureEventhub(BaseModel):
     pq_controls: Annotated[
         Optional[OutputAzureEventhubPqControls], pydantic.Field(alias="pqControls")
     ] = None
+
+    @field_serializer("ack")
+    def serialize_ack(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureEventhubAcknowledgments(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("format_")
+    def serialize_format_(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureEventhubRecordDataFormat(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("on_backpressure")
+    def serialize_on_backpressure(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureEventhubBackpressureBehavior(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("pq_mode")
+    def serialize_pq_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureEventhubMode(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("pq_compress")
+    def serialize_pq_compress(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureEventhubCompression(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("pq_on_backpressure")
+    def serialize_pq_on_backpressure(self, value):
+        if isinstance(value, str):
+            try:
+                return models.OutputAzureEventhubQueueFullBehavior(value)
+            except ValueError:
+                return value
+        return value