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

cribl_control_plane/models/inputkafka.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 InputKafkaConnection(BaseModel):
 class InputKafkaMode(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 InputKafkaCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
-    # None
     NONE = "none"
-    # Gzip
     GZIP = "gzip"
 
 
@@ -107,23 +102,12 @@ class InputKafkaPq(BaseModel):
         Optional[InputKafkaPqControls], pydantic.Field(alias="pqControls")
     ] = None
 
-    @field_serializer("mode")
-    def serialize_mode(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKafkaMode(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("compress")
-    def serialize_compress(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKafkaCompression(value)
-            except ValueError:
-                return value
-        return value
+
+class InputKafkaSchemaType(str, Enum, metaclass=utils.OpenEnumMeta):
+    r"""The schema format used to encode and decode event data"""
+
+    AVRO = "avro"
+    JSON = "json"
 
 
 class InputKafkaAuthTypedDict(TypedDict):
@@ -231,29 +215,13 @@ class InputKafkaKafkaSchemaRegistryTLSSettingsClientSide(BaseModel):
         pydantic.Field(alias="maxVersion"),
     ] = None
 
-    @field_serializer("min_version")
-    def serialize_min_version(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKafkaKafkaSchemaRegistryMinimumTLSVersion(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("max_version")
-    def serialize_max_version(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKafkaKafkaSchemaRegistryMaximumTLSVersion(value)
-            except ValueError:
-                return value
-        return value
-
 
 class InputKafkaKafkaSchemaRegistryAuthenticationTypedDict(TypedDict):
     disabled: NotRequired[bool]
     schema_registry_url: NotRequired[str]
     r"""URL for accessing the Confluent Schema Registry. Example: http://localhost:8081. To connect over TLS, use https instead of http."""
+    schema_type: NotRequired[InputKafkaSchemaType]
+    r"""The schema format used to encode and decode event data"""
     connection_timeout: NotRequired[float]
     r"""Maximum time to wait for a Schema Registry connection to complete successfully"""
     request_timeout: NotRequired[float]
@@ -273,6 +241,14 @@ class InputKafkaKafkaSchemaRegistryAuthentication(BaseModel):
     ] = "http://localhost:8081"
     r"""URL for accessing the Confluent Schema Registry. Example: http://localhost:8081. To connect over TLS, use https instead of http."""
 
+    schema_type: Annotated[
+        Annotated[
+            Optional[InputKafkaSchemaType], PlainValidator(validate_open_enum(False))
+        ],
+        pydantic.Field(alias="schemaType"),
+    ] = InputKafkaSchemaType.AVRO
+    r"""The schema format used to encode and decode event data"""
+
     connection_timeout: Annotated[
         Optional[float], pydantic.Field(alias="connectionTimeout")
     ] = 30000
@@ -292,76 +268,20 @@ class InputKafkaKafkaSchemaRegistryAuthentication(BaseModel):
     tls: Optional[InputKafkaKafkaSchemaRegistryTLSSettingsClientSide] = None
 
 
-class InputKafkaAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMeta):
-    r"""Enter credentials directly, or select a stored secret"""
-
-    MANUAL = "manual"
-    SECRET = "secret"
-
-
 class InputKafkaSASLMechanism(str, Enum, metaclass=utils.OpenEnumMeta):
-    # PLAIN
     PLAIN = "plain"
-    # SCRAM-SHA-256
     SCRAM_SHA_256 = "scram-sha-256"
-    # SCRAM-SHA-512
     SCRAM_SHA_512 = "scram-sha-512"
-    # GSSAPI/Kerberos
     KERBEROS = "kerberos"
 
 
-class InputKafkaOauthParamTypedDict(TypedDict):
-    name: str
-    value: str
-
-
-class InputKafkaOauthParam(BaseModel):
-    name: str
-
-    value: str
-
-
-class InputKafkaSaslExtensionTypedDict(TypedDict):
-    name: str
-    value: str
-
-
-class InputKafkaSaslExtension(BaseModel):
-    name: str
-
-    value: str
-
-
 class InputKafkaAuthenticationTypedDict(TypedDict):
     r"""Authentication parameters to use when connecting to brokers. Using TLS is highly recommended."""
 
     disabled: NotRequired[bool]
-    username: NotRequired[str]
-    password: NotRequired[str]
-    auth_type: NotRequired[InputKafkaAuthenticationMethod]
-    r"""Enter credentials directly, or select a stored secret"""
-    credentials_secret: NotRequired[str]
-    r"""Select or create a secret that references your credentials"""
     mechanism: NotRequired[InputKafkaSASLMechanism]
-    keytab_location: NotRequired[str]
-    r"""Location of keytab file for authentication principal"""
-    principal: NotRequired[str]
-    r"""Authentication principal, such as `kafka_user@example.com`"""
-    broker_service_class: NotRequired[str]
-    r"""Kerberos service class for Kafka brokers, such as `kafka`"""
     oauth_enabled: NotRequired[bool]
     r"""Enable OAuth authentication"""
-    token_url: NotRequired[str]
-    r"""URL of the token endpoint to use for OAuth authentication"""
-    client_id: NotRequired[str]
-    r"""Client ID to use for OAuth authentication"""
-    oauth_secret_type: NotRequired[str]
-    client_text_secret: NotRequired[str]
-    r"""Select or create a stored text secret"""
-    oauth_params: NotRequired[List[InputKafkaOauthParamTypedDict]]
-    r"""Additional fields to send to the token endpoint, such as scope or audience"""
-    sasl_extensions: NotRequired[List[InputKafkaSaslExtensionTypedDict]]
-    r"""Additional SASL extension fields, such as Confluent's logicalCluster or identityPoolId"""
 
 
 class InputKafkaAuthentication(BaseModel):
@@ -369,89 +289,15 @@ class InputKafkaAuthentication(BaseModel):
 
     disabled: Optional[bool] = True
 
-    username: Optional[str] = None
-
-    password: Optional[str] = None
-
-    auth_type: Annotated[
-        Annotated[
-            Optional[InputKafkaAuthenticationMethod],
-            PlainValidator(validate_open_enum(False)),
-        ],
-        pydantic.Field(alias="authType"),
-    ] = InputKafkaAuthenticationMethod.MANUAL
-    r"""Enter credentials directly, or select a stored secret"""
-
-    credentials_secret: Annotated[
-        Optional[str], pydantic.Field(alias="credentialsSecret")
-    ] = None
-    r"""Select or create a secret that references your credentials"""
-
     mechanism: Annotated[
         Optional[InputKafkaSASLMechanism], PlainValidator(validate_open_enum(False))
     ] = InputKafkaSASLMechanism.PLAIN
 
-    keytab_location: Annotated[
-        Optional[str], pydantic.Field(alias="keytabLocation")
-    ] = None
-    r"""Location of keytab file for authentication principal"""
-
-    principal: Optional[str] = None
-    r"""Authentication principal, such as `kafka_user@example.com`"""
-
-    broker_service_class: Annotated[
-        Optional[str], pydantic.Field(alias="brokerServiceClass")
-    ] = None
-    r"""Kerberos service class for Kafka brokers, such as `kafka`"""
-
     oauth_enabled: Annotated[Optional[bool], pydantic.Field(alias="oauthEnabled")] = (
         False
     )
     r"""Enable OAuth authentication"""
 
-    token_url: Annotated[Optional[str], pydantic.Field(alias="tokenUrl")] = None
-    r"""URL of the token endpoint to use for OAuth authentication"""
-
-    client_id: Annotated[Optional[str], pydantic.Field(alias="clientId")] = None
-    r"""Client ID to use for OAuth authentication"""
-
-    oauth_secret_type: Annotated[
-        Optional[str], pydantic.Field(alias="oauthSecretType")
-    ] = "secret"
-
-    client_text_secret: Annotated[
-        Optional[str], pydantic.Field(alias="clientTextSecret")
-    ] = None
-    r"""Select or create a stored text secret"""
-
-    oauth_params: Annotated[
-        Optional[List[InputKafkaOauthParam]], pydantic.Field(alias="oauthParams")
-    ] = None
-    r"""Additional fields to send to the token endpoint, such as scope or audience"""
-
-    sasl_extensions: Annotated[
-        Optional[List[InputKafkaSaslExtension]], pydantic.Field(alias="saslExtensions")
-    ] = None
-    r"""Additional SASL extension fields, such as Confluent's logicalCluster or identityPoolId"""
-
-    @field_serializer("auth_type")
-    def serialize_auth_type(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKafkaAuthenticationMethod(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("mechanism")
-    def serialize_mechanism(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKafkaSASLMechanism(value)
-            except ValueError:
-                return value
-        return value
-
 
 class InputKafkaMinimumTLSVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     TL_SV1 = "TLSv1"
@@ -535,24 +381,6 @@ class InputKafkaTLSSettingsClientSide(BaseModel):
         pydantic.Field(alias="maxVersion"),
     ] = None
 
-    @field_serializer("min_version")
-    def serialize_min_version(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKafkaMinimumTLSVersion(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("max_version")
-    def serialize_max_version(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKafkaMaximumTLSVersion(value)
-            except ValueError:
-                return value
-        return value
-
 
 class InputKafkaMetadatumTypedDict(TypedDict):
     name: str

cribl_control_plane/models/inputkinesis.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 InputKinesisConnection(BaseModel):
 class InputKinesisMode(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 InputKinesisCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
-    # None
     NONE = "none"
-    # Gzip
     GZIP = "gzip"
 
 
@@ -107,64 +102,35 @@ class InputKinesisPq(BaseModel):
         Optional[InputKinesisPqControls], pydantic.Field(alias="pqControls")
     ] = None
 
-    @field_serializer("mode")
-    def serialize_mode(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKinesisMode(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("compress")
-    def serialize_compress(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKinesisCompression(value)
-            except ValueError:
-                return value
-        return value
-
 
 class ShardIteratorStart(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Location at which to start reading a shard for the first time"""
 
-    # Earliest record
     TRIM_HORIZON = "TRIM_HORIZON"
-    # Latest record
     LATEST = "LATEST"
 
 
 class InputKinesisRecordDataFormat(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Format of data inside the Kinesis Stream records. Gzip compression is automatically detected."""
 
-    # Cribl
     CRIBL = "cribl"
-    # Newline JSON
     NDJSON = "ndjson"
-    # Cloudwatch Logs
     CLOUDWATCH = "cloudwatch"
-    # Event per line
     LINE = "line"
 
 
 class ShardLoadBalancing(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""The load-balancing algorithm to use for spreading out shards across Workers and Worker Processes"""
 
-    # Consistent Hashing
     CONSISTENT_HASHING = "ConsistentHashing"
-    # Round Robin
     ROUND_ROBIN = "RoundRobin"
 
 
 class InputKinesisAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""AWS authentication method. Choose Auto to use IAM roles."""
 
-    # Auto
     AUTO = "auto"
-    # Manual
     MANUAL = "manual"
-    # Secret Key pair
     SECRET = "secret"
 
 
@@ -408,48 +374,3 @@ class InputKinesis(BaseModel):
 
     aws_secret: Annotated[Optional[str], pydantic.Field(alias="awsSecret")] = None
     r"""Select or create a stored secret that references your access key and secret key"""
-
-    @field_serializer("shard_iterator_type")
-    def serialize_shard_iterator_type(self, value):
-        if isinstance(value, str):
-            try:
-                return models.ShardIteratorStart(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("payload_format")
-    def serialize_payload_format(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKinesisRecordDataFormat(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("load_balancing_algorithm")
-    def serialize_load_balancing_algorithm(self, value):
-        if isinstance(value, str):
-            try:
-                return models.ShardLoadBalancing(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("aws_authentication_method")
-    def serialize_aws_authentication_method(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKinesisAuthenticationMethod(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("signature_version")
-    def serialize_signature_version(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKinesisSignatureVersion(value)
-            except ValueError:
-                return value
-        return value

cribl_control_plane/models/inputkubeevents.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 InputKubeEventsConnection(BaseModel):
 class InputKubeEventsMode(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 InputKubeEventsCompression(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 InputKubeEventsPq(BaseModel):
         Optional[InputKubeEventsPqControls], pydantic.Field(alias="pqControls")
     ] = None
 
-    @field_serializer("mode")
-    def serialize_mode(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKubeEventsMode(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("compress")
-    def serialize_compress(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKubeEventsCompression(value)
-            except ValueError:
-                return value
-        return value
-
 
 class InputKubeEventsRuleTypedDict(TypedDict):
     filter_: str

cribl_control_plane/models/inputkubelogs.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 InputKubeLogsConnection(BaseModel):
 class InputKubeLogsMode(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 InputKubeLogsPqCompression(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 InputKubeLogsPq(BaseModel):
         Optional[InputKubeLogsPqControls], pydantic.Field(alias="pqControls")
     ] = None
 
-    @field_serializer("mode")
-    def serialize_mode(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKubeLogsMode(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("compress")
-    def serialize_compress(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKubeLogsPqCompression(value)
-            except ValueError:
-                return value
-        return value
-
 
 class InputKubeLogsRuleTypedDict(TypedDict):
     filter_: str
@@ -193,15 +170,6 @@ class InputKubeLogsDiskSpooling(BaseModel):
     ] = InputKubeLogsPersistenceCompression.GZIP
     r"""Data compression format. Default is gzip."""
 
-    @field_serializer("compress")
-    def serialize_compress(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKubeLogsPersistenceCompression(value)
-            except ValueError:
-                return value
-        return value
-
 
 class InputKubeLogsTypedDict(TypedDict):
     type: InputKubeLogsType

cribl_control_plane/models/inputkubemetrics.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 InputKubeMetricsConnection(BaseModel):
 class InputKubeMetricsMode(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 InputKubeMetricsCompression(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 InputKubeMetricsPq(BaseModel):
         Optional[InputKubeMetricsPqControls], pydantic.Field(alias="pqControls")
     ] = None
 
-    @field_serializer("mode")
-    def serialize_mode(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKubeMetricsMode(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("compress")
-    def serialize_compress(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKubeMetricsCompression(value)
-            except ValueError:
-                return value
-        return value
-
 
 class InputKubeMetricsRuleTypedDict(TypedDict):
     filter_: str
@@ -196,15 +173,6 @@ class InputKubeMetricsPersistence(BaseModel):
     )
     r"""Path to use to write metrics. Defaults to $CRIBL_HOME/state/<id>"""
 
-    @field_serializer("compress")
-    def serialize_compress(self, value):
-        if isinstance(value, str):
-            try:
-                return models.InputKubeMetricsDataCompressionFormat(value)
-            except ValueError:
-                return value
-        return value
-
 
 class InputKubeMetricsTypedDict(TypedDict):
     type: InputKubeMetricsType