cribl-control-plane 0.1.0b1__py3-none-any.whl → 0.1.1rc1__py3-none-any.whl

cribl_control_plane/models/__init__.py

@@ -22,6 +22,7 @@ if TYPE_CHECKING:
         ConfigGroup,
         ConfigGroupType,
         ConfigGroupTypedDict,
+        EstimatedIngestRate,
         Git,
         GitTypedDict,
     )
@@ -97,6 +98,11 @@ if TYPE_CHECKING:
         CriblLakeDatasetFormat,
         CriblLakeDatasetTypedDict,
     )
+    from .cribllakedatasetupdate import (
+        CriblLakeDatasetUpdate,
+        CriblLakeDatasetUpdateFormat,
+        CriblLakeDatasetUpdateTypedDict,
+    )
     from .currentbranchresult import CurrentBranchResult, CurrentBranchResultTypedDict
     from .datasetmetadata import DatasetMetadata, DatasetMetadataTypedDict, ScanMode
     from .datasetmetadataruninfo import (
@@ -335,7 +341,7 @@ if TYPE_CHECKING:
     from .gitdiffresult import GitDiffResult, GitDiffResultTypedDict
     from .gitfile import GitFile, GitFileTypedDict
     from .gitfilesresponse import GitFilesResponse, GitFilesResponseTypedDict
-    from .gitinfo import GitInfo, GitInfoTypedDict, Remote, RemoteTypedDict
+    from .gitinfo import GitInfo, GitInfoTypedDict, Remote, RemoteEnum, RemoteTypedDict
     from .gitlogresult import GitLogResult, GitLogResultTypedDict
     from .gitrevertparams import GitRevertParams, GitRevertParamsTypedDict
     from .gitrevertresult import (
@@ -1920,6 +1926,27 @@ if TYPE_CHECKING:
         OutputAzureLogsType,
         OutputAzureLogsTypedDict,
     )
+    from .outputchronicle import (
+        OutputChronicle,
+        OutputChronicleAuthenticationMethod,
+        OutputChronicleBackpressureBehavior,
+        OutputChronicleCompression,
+        OutputChronicleCustomLabel,
+        OutputChronicleCustomLabelTypedDict,
+        OutputChronicleExtraHTTPHeader,
+        OutputChronicleExtraHTTPHeaderTypedDict,
+        OutputChronicleFailedRequestLoggingMode,
+        OutputChronicleMode,
+        OutputChroniclePqControls,
+        OutputChroniclePqControlsTypedDict,
+        OutputChronicleQueueFullBehavior,
+        OutputChronicleResponseRetrySetting,
+        OutputChronicleResponseRetrySettingTypedDict,
+        OutputChronicleTimeoutRetrySettings,
+        OutputChronicleTimeoutRetrySettingsTypedDict,
+        OutputChronicleType,
+        OutputChronicleTypedDict,
+    )
     from .outputclickhouse import (
         ColumnMapping,
         ColumnMappingTypedDict,
@@ -2073,14 +2100,8 @@ if TYPE_CHECKING:
         OutputDatabricks,
         OutputDatabricksAuthenticationMethod,
         OutputDatabricksBackpressureBehavior,
-        OutputDatabricksCompression,
-        OutputDatabricksCompressionLevel,
         OutputDatabricksDataFormat,
-        OutputDatabricksDataPageVersion,
         OutputDatabricksDiskSpaceProtection,
-        OutputDatabricksKeyValueMetadatum,
-        OutputDatabricksKeyValueMetadatumTypedDict,
-        OutputDatabricksParquetVersion,
         OutputDatabricksType,
         OutputDatabricksTypedDict,
     )
@@ -2277,8 +2298,6 @@ if TYPE_CHECKING:
         OutputFilesystemTypedDict,
     )
     from .outputgooglechronicle import (
-        CustomLabel,
-        CustomLabelTypedDict,
         ExtraLogType,
         ExtraLogTypeTypedDict,
         OutputGoogleChronicle,
@@ -2286,6 +2305,8 @@ if TYPE_CHECKING:
         OutputGoogleChronicleAuthenticationMethod,
         OutputGoogleChronicleBackpressureBehavior,
         OutputGoogleChronicleCompression,
+        OutputGoogleChronicleCustomLabel,
+        OutputGoogleChronicleCustomLabelTypedDict,
         OutputGoogleChronicleExtraHTTPHeader,
         OutputGoogleChronicleExtraHTTPHeaderTypedDict,
         OutputGoogleChronicleFailedRequestLoggingMode,
@@ -3274,6 +3295,7 @@ if TYPE_CHECKING:
         UpdatePacksByIDResponse,
         UpdatePacksByIDResponseTypedDict,
     )
+    from .updatepacksop import UpdatePacksRequest, UpdatePacksRequestTypedDict
     from .updatepipelinebyidop import (
         UpdatePipelineByIDRequest,
         UpdatePipelineByIDRequestTypedDict,
@@ -3286,6 +3308,7 @@ if TYPE_CHECKING:
         UpdateRoutesByIDResponse,
         UpdateRoutesByIDResponseTypedDict,
     )
+    from .uploadpackresponse import UploadPackResponse, UploadPackResponseTypedDict
     from .useraccesscontrollist import (
         UserAccessControlList,
         UserAccessControlListTypedDict,
@@ -3401,10 +3424,11 @@ __all__ = [
     "CriblLakeDataset",
     "CriblLakeDatasetFormat",
     "CriblLakeDatasetTypedDict",
+    "CriblLakeDatasetUpdate",
+    "CriblLakeDatasetUpdateFormat",
+    "CriblLakeDatasetUpdateTypedDict",
     "CurrentBranchResult",
     "CurrentBranchResultTypedDict",
-    "CustomLabel",
-    "CustomLabelTypedDict",
     "DNS",
     "DNSTypedDict",
     "DataSetSite",
@@ -3470,6 +3494,7 @@ __all__ = [
     "EndpointType",
     "Error",
     "ErrorTypedDict",
+    "EstimatedIngestRate",
     "EventFormat",
     "Executor",
     "ExecutorSpecificSettings",
@@ -4997,6 +5022,25 @@ __all__ = [
     "OutputAzureLogsTimeoutRetrySettingsTypedDict",
     "OutputAzureLogsType",
     "OutputAzureLogsTypedDict",
+    "OutputChronicle",
+    "OutputChronicleAuthenticationMethod",
+    "OutputChronicleBackpressureBehavior",
+    "OutputChronicleCompression",
+    "OutputChronicleCustomLabel",
+    "OutputChronicleCustomLabelTypedDict",
+    "OutputChronicleExtraHTTPHeader",
+    "OutputChronicleExtraHTTPHeaderTypedDict",
+    "OutputChronicleFailedRequestLoggingMode",
+    "OutputChronicleMode",
+    "OutputChroniclePqControls",
+    "OutputChroniclePqControlsTypedDict",
+    "OutputChronicleQueueFullBehavior",
+    "OutputChronicleResponseRetrySetting",
+    "OutputChronicleResponseRetrySettingTypedDict",
+    "OutputChronicleTimeoutRetrySettings",
+    "OutputChronicleTimeoutRetrySettingsTypedDict",
+    "OutputChronicleType",
+    "OutputChronicleTypedDict",
     "OutputClickHouse",
     "OutputClickHouseAuthenticationType",
     "OutputClickHouseBackpressureBehavior",
@@ -5131,14 +5175,8 @@ __all__ = [
     "OutputDatabricks",
     "OutputDatabricksAuthenticationMethod",
     "OutputDatabricksBackpressureBehavior",
-    "OutputDatabricksCompression",
-    "OutputDatabricksCompressionLevel",
     "OutputDatabricksDataFormat",
-    "OutputDatabricksDataPageVersion",
     "OutputDatabricksDiskSpaceProtection",
-    "OutputDatabricksKeyValueMetadatum",
-    "OutputDatabricksKeyValueMetadatumTypedDict",
-    "OutputDatabricksParquetVersion",
     "OutputDatabricksType",
     "OutputDatabricksTypedDict",
     "OutputDatadog",
@@ -5314,6 +5352,8 @@ __all__ = [
     "OutputGoogleChronicleAuthenticationMethod",
     "OutputGoogleChronicleBackpressureBehavior",
     "OutputGoogleChronicleCompression",
+    "OutputGoogleChronicleCustomLabel",
+    "OutputGoogleChronicleCustomLabelTypedDict",
     "OutputGoogleChronicleExtraHTTPHeader",
     "OutputGoogleChronicleExtraHTTPHeaderTypedDict",
     "OutputGoogleChronicleFailedRequestLoggingMode",
@@ -6073,6 +6113,7 @@ __all__ = [
     "RbacResource",
     "ReadMode",
     "Remote",
+    "RemoteEnum",
     "RemoteTypedDict",
     "Renamed",
     "RenamedTypedDict",
@@ -6209,6 +6250,8 @@ __all__ = [
     "UpdatePacksByIDRequestTypedDict",
     "UpdatePacksByIDResponse",
     "UpdatePacksByIDResponseTypedDict",
+    "UpdatePacksRequest",
+    "UpdatePacksRequestTypedDict",
     "UpdatePipelineByIDRequest",
     "UpdatePipelineByIDRequestTypedDict",
     "UpdatePipelineByIDResponse",
@@ -6217,6 +6260,8 @@ __all__ = [
     "UpdateRoutesByIDRequestTypedDict",
     "UpdateRoutesByIDResponse",
     "UpdateRoutesByIDResponseTypedDict",
+    "UploadPackResponse",
+    "UploadPackResponseTypedDict",
     "UserAccessControlList",
     "UserAccessControlListTypedDict",
     "UsersAndGroups",
@@ -6246,6 +6291,7 @@ _dynamic_imports: dict[str, str] = {
     "ConfigGroup": ".configgroup",
     "ConfigGroupType": ".configgroup",
     "ConfigGroupTypedDict": ".configgroup",
+    "EstimatedIngestRate": ".configgroup",
     "Git": ".configgroup",
     "GitTypedDict": ".configgroup",
     "ConfigGroupCloud": ".configgroupcloud",
@@ -6301,6 +6347,9 @@ _dynamic_imports: dict[str, str] = {
     "CriblLakeDataset": ".cribllakedataset",
     "CriblLakeDatasetFormat": ".cribllakedataset",
     "CriblLakeDatasetTypedDict": ".cribllakedataset",
+    "CriblLakeDatasetUpdate": ".cribllakedatasetupdate",
+    "CriblLakeDatasetUpdateFormat": ".cribllakedatasetupdate",
+    "CriblLakeDatasetUpdateTypedDict": ".cribllakedatasetupdate",
     "CurrentBranchResult": ".currentbranchresult",
     "CurrentBranchResultTypedDict": ".currentbranchresult",
     "DatasetMetadata": ".datasetmetadata",
@@ -6481,6 +6530,7 @@ _dynamic_imports: dict[str, str] = {
     "GitInfo": ".gitinfo",
     "GitInfoTypedDict": ".gitinfo",
     "Remote": ".gitinfo",
+    "RemoteEnum": ".gitinfo",
     "RemoteTypedDict": ".gitinfo",
     "GitLogResult": ".gitlogresult",
     "GitLogResultTypedDict": ".gitlogresult",
@@ -7940,6 +7990,25 @@ _dynamic_imports: dict[str, str] = {
     "OutputAzureLogsTimeoutRetrySettingsTypedDict": ".outputazurelogs",
     "OutputAzureLogsType": ".outputazurelogs",
     "OutputAzureLogsTypedDict": ".outputazurelogs",
+    "OutputChronicle": ".outputchronicle",
+    "OutputChronicleAuthenticationMethod": ".outputchronicle",
+    "OutputChronicleBackpressureBehavior": ".outputchronicle",
+    "OutputChronicleCompression": ".outputchronicle",
+    "OutputChronicleCustomLabel": ".outputchronicle",
+    "OutputChronicleCustomLabelTypedDict": ".outputchronicle",
+    "OutputChronicleExtraHTTPHeader": ".outputchronicle",
+    "OutputChronicleExtraHTTPHeaderTypedDict": ".outputchronicle",
+    "OutputChronicleFailedRequestLoggingMode": ".outputchronicle",
+    "OutputChronicleMode": ".outputchronicle",
+    "OutputChroniclePqControls": ".outputchronicle",
+    "OutputChroniclePqControlsTypedDict": ".outputchronicle",
+    "OutputChronicleQueueFullBehavior": ".outputchronicle",
+    "OutputChronicleResponseRetrySetting": ".outputchronicle",
+    "OutputChronicleResponseRetrySettingTypedDict": ".outputchronicle",
+    "OutputChronicleTimeoutRetrySettings": ".outputchronicle",
+    "OutputChronicleTimeoutRetrySettingsTypedDict": ".outputchronicle",
+    "OutputChronicleType": ".outputchronicle",
+    "OutputChronicleTypedDict": ".outputchronicle",
     "ColumnMapping": ".outputclickhouse",
     "ColumnMappingTypedDict": ".outputclickhouse",
     "MappingType": ".outputclickhouse",
@@ -8078,14 +8147,8 @@ _dynamic_imports: dict[str, str] = {
     "OutputDatabricks": ".outputdatabricks",
     "OutputDatabricksAuthenticationMethod": ".outputdatabricks",
     "OutputDatabricksBackpressureBehavior": ".outputdatabricks",
-    "OutputDatabricksCompression": ".outputdatabricks",
-    "OutputDatabricksCompressionLevel": ".outputdatabricks",
     "OutputDatabricksDataFormat": ".outputdatabricks",
-    "OutputDatabricksDataPageVersion": ".outputdatabricks",
     "OutputDatabricksDiskSpaceProtection": ".outputdatabricks",
-    "OutputDatabricksKeyValueMetadatum": ".outputdatabricks",
-    "OutputDatabricksKeyValueMetadatumTypedDict": ".outputdatabricks",
-    "OutputDatabricksParquetVersion": ".outputdatabricks",
     "OutputDatabricksType": ".outputdatabricks",
     "OutputDatabricksTypedDict": ".outputdatabricks",
     "DatadogSite": ".outputdatadog",
@@ -8264,8 +8327,6 @@ _dynamic_imports: dict[str, str] = {
     "OutputFilesystemParquetVersion": ".outputfilesystem",
     "OutputFilesystemType": ".outputfilesystem",
     "OutputFilesystemTypedDict": ".outputfilesystem",
-    "CustomLabel": ".outputgooglechronicle",
-    "CustomLabelTypedDict": ".outputgooglechronicle",
     "ExtraLogType": ".outputgooglechronicle",
     "ExtraLogTypeTypedDict": ".outputgooglechronicle",
     "OutputGoogleChronicle": ".outputgooglechronicle",
@@ -8273,6 +8334,8 @@ _dynamic_imports: dict[str, str] = {
     "OutputGoogleChronicleAuthenticationMethod": ".outputgooglechronicle",
     "OutputGoogleChronicleBackpressureBehavior": ".outputgooglechronicle",
     "OutputGoogleChronicleCompression": ".outputgooglechronicle",
+    "OutputGoogleChronicleCustomLabel": ".outputgooglechronicle",
+    "OutputGoogleChronicleCustomLabelTypedDict": ".outputgooglechronicle",
     "OutputGoogleChronicleExtraHTTPHeader": ".outputgooglechronicle",
     "OutputGoogleChronicleExtraHTTPHeaderTypedDict": ".outputgooglechronicle",
     "OutputGoogleChronicleFailedRequestLoggingMode": ".outputgooglechronicle",
@@ -9151,6 +9214,8 @@ _dynamic_imports: dict[str, str] = {
     "UpdatePacksByIDRequestTypedDict": ".updatepacksbyidop",
     "UpdatePacksByIDResponse": ".updatepacksbyidop",
     "UpdatePacksByIDResponseTypedDict": ".updatepacksbyidop",
+    "UpdatePacksRequest": ".updatepacksop",
+    "UpdatePacksRequestTypedDict": ".updatepacksop",
     "UpdatePipelineByIDRequest": ".updatepipelinebyidop",
     "UpdatePipelineByIDRequestTypedDict": ".updatepipelinebyidop",
     "UpdatePipelineByIDResponse": ".updatepipelinebyidop",
@@ -9159,6 +9224,8 @@ _dynamic_imports: dict[str, str] = {
     "UpdateRoutesByIDRequestTypedDict": ".updateroutesbyidop",
     "UpdateRoutesByIDResponse": ".updateroutesbyidop",
     "UpdateRoutesByIDResponseTypedDict": ".updateroutesbyidop",
+    "UploadPackResponse": ".uploadpackresponse",
+    "UploadPackResponseTypedDict": ".uploadpackresponse",
     "UserAccessControlList": ".useraccesscontrollist",
     "UserAccessControlListTypedDict": ".useraccesscontrollist",
     "WorkerTypes": ".workertypes",

cribl_control_plane/models/configgroup.py

@@ -14,6 +14,18 @@ from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
+class EstimatedIngestRate(int, Enum, metaclass=utils.OpenEnumMeta):
+    ONE_THOUSAND_AND_TWENTY_FOUR = 1024
+    FOUR_THOUSAND_AND_NINETY_SIX = 4096
+    TEN_THOUSAND_TWO_HUNDRED_AND_FORTY = 10240
+    TWO_THOUSAND_AND_FORTY_EIGHT = 2048
+    THREE_THOUSAND_AND_SEVENTY_TWO = 3072
+    FIVE_THOUSAND_ONE_HUNDRED_AND_TWENTY = 5120
+    SEVEN_THOUSAND_ONE_HUNDRED_AND_SIXTY_EIGHT = 7168
+    THIRTEEN_THOUSAND_THREE_HUNDRED_AND_TWELVE = 13312
+    FIFTEEN_THOUSAND_THREE_HUNDRED_AND_SIXTY = 15360
+
+
 class GitTypedDict(TypedDict):
     commit: NotRequired[str]
     local_changes: NotRequired[float]
@@ -40,7 +52,7 @@ class ConfigGroupTypedDict(TypedDict):
     config_version: NotRequired[str]
     deploying_worker_count: NotRequired[float]
     description: NotRequired[str]
-    estimated_ingest_rate: NotRequired[float]
+    estimated_ingest_rate: NotRequired[EstimatedIngestRate]
     git: NotRequired[GitTypedDict]
     incompatible_worker_count: NotRequired[float]
     inherits: NotRequired[str]
@@ -75,7 +87,10 @@ class ConfigGroup(BaseModel):
     description: Optional[str] = None
 
     estimated_ingest_rate: Annotated[
-        Optional[float], pydantic.Field(alias="estimatedIngestRate")
+        Annotated[
+            Optional[EstimatedIngestRate], PlainValidator(validate_open_enum(True))
+        ],
+        pydantic.Field(alias="estimatedIngestRate"),
     ] = None
 
     git: Optional[Git] = None

cribl_control_plane/models/createversionundoop.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 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 Any, Dict, List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -27,7 +27,7 @@ class CreateVersionUndoResponseTypedDict(TypedDict):
 
     count: NotRequired[int]
     r"""number of items present in the items array"""
-    items: NotRequired[List[str]]
+    items: NotRequired[List[Dict[str, Any]]]
 
 
 class CreateVersionUndoResponse(BaseModel):
@@ -36,4 +36,4 @@ class CreateVersionUndoResponse(BaseModel):
     count: Optional[int] = None
     r"""number of items present in the items array"""
 
-    items: Optional[List[str]] = None
+    items: Optional[List[Dict[str, Any]]] = None

cribl_control_plane/models/cribllakedatasetupdate.py

@@ -0,0 +1,81 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .cacheconnection import CacheConnection, CacheConnectionTypedDict
+from .lakedatasetsearchconfig import (
+    LakeDatasetSearchConfig,
+    LakeDatasetSearchConfigTypedDict,
+)
+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
+
+
+class CriblLakeDatasetUpdateFormat(str, Enum, metaclass=utils.OpenEnumMeta):
+    JSON = "json"
+    DDSS = "ddss"
+    PARQUET = "parquet"
+
+
+class CriblLakeDatasetUpdateTypedDict(TypedDict):
+    accelerated_fields: NotRequired[List[str]]
+    bucket_name: NotRequired[str]
+    cache_connection: NotRequired[CacheConnectionTypedDict]
+    deletion_started_at: NotRequired[float]
+    description: NotRequired[str]
+    format_: NotRequired[CriblLakeDatasetUpdateFormat]
+    http_da_used: NotRequired[bool]
+    id: NotRequired[str]
+    retention_period_in_days: NotRequired[float]
+    search_config: NotRequired[LakeDatasetSearchConfigTypedDict]
+    storage_location_id: NotRequired[str]
+    view_name: NotRequired[str]
+
+
+class CriblLakeDatasetUpdate(BaseModel):
+    accelerated_fields: Annotated[
+        Optional[List[str]], pydantic.Field(alias="acceleratedFields")
+    ] = None
+
+    bucket_name: Annotated[Optional[str], pydantic.Field(alias="bucketName")] = None
+
+    cache_connection: Annotated[
+        Optional[CacheConnection], pydantic.Field(alias="cacheConnection")
+    ] = None
+
+    deletion_started_at: Annotated[
+        Optional[float], pydantic.Field(alias="deletionStartedAt")
+    ] = None
+
+    description: Optional[str] = None
+
+    format_: Annotated[
+        Annotated[
+            Optional[CriblLakeDatasetUpdateFormat],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="format"),
+    ] = None
+
+    http_da_used: Annotated[Optional[bool], pydantic.Field(alias="httpDAUsed")] = None
+
+    id: Optional[str] = None
+
+    retention_period_in_days: Annotated[
+        Optional[float], pydantic.Field(alias="retentionPeriodInDays")
+    ] = None
+
+    search_config: Annotated[
+        Optional[LakeDatasetSearchConfig], pydantic.Field(alias="searchConfig")
+    ] = None
+
+    storage_location_id: Annotated[
+        Optional[str], pydantic.Field(alias="storageLocationId")
+    ] = None
+
+    view_name: Annotated[Optional[str], pydantic.Field(alias="viewName")] = None

cribl_control_plane/models/distributedsummary.py

@@ -9,7 +9,9 @@ from typing_extensions import Annotated, TypedDict
 class DistributedSummaryGroupsTypedDict(TypedDict):
     count: float
     destinations: float
+    packs: float
     pipelines: float
+    quick_connects: float
     routes: float
     sources: float
 
@@ -19,8 +21,12 @@ class DistributedSummaryGroups(BaseModel):
 
     destinations: float
 
+    packs: float
+
     pipelines: float
 
+    quick_connects: Annotated[float, pydantic.Field(alias="quickConnects")]
+
     routes: float
 
     sources: float

cribl_control_plane/models/gitinfo.py

@@ -1,15 +1,26 @@
 """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
+from pydantic.functional_validators import PlainValidator
 from typing import Union
-from typing_extensions import TypeAliasType, TypedDict
+from typing_extensions import Annotated, TypeAliasType, TypedDict
 
 
-RemoteTypedDict = TypeAliasType("RemoteTypedDict", Union[str, bool])
+class RemoteEnum(str, Enum, metaclass=utils.OpenEnumMeta):
+    FALSE = "false"
 
 
-Remote = TypeAliasType("Remote", Union[str, bool])
+RemoteTypedDict = TypeAliasType("RemoteTypedDict", Union[str, RemoteEnum])
+
+
+Remote = TypeAliasType(
+    "Remote",
+    Union[str, Annotated[RemoteEnum, PlainValidator(validate_open_enum(False))]],
+)
 
 
 class GitInfoTypedDict(TypedDict):

cribl_control_plane/models/input.py

@@ -66,8 +66,10 @@ from .inputwineventlogs import InputWinEventLogs, InputWinEventLogsTypedDict
 from .inputwiz import InputWiz, InputWizTypedDict
 from .inputwizwebhook import InputWizWebhook, InputWizWebhookTypedDict
 from .inputzscalerhec import InputZscalerHec, InputZscalerHecTypedDict
+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
 
 
 InputTypedDict = TypeAliasType(
@@ -136,67 +138,67 @@ InputTypedDict = TypeAliasType(
 )
 
 
-Input = TypeAliasType(
-    "Input",
+Input = Annotated[
     Union[
-        InputDatagen,
-        InputKubeEvents,
-        InputCribl,
-        InputCriblmetrics,
-        InputKubeMetrics,
-        InputCollection,
-        InputSystemState,
-        InputModelDrivenTelemetry,
-        InputSystemMetrics,
-        InputWindowsMetrics,
-        InputJournalFiles,
-        InputWinEventLogs,
-        InputRawUDP,
-        InputExec,
-        InputKubeLogs,
-        InputMetrics,
-        InputSnmp,
-        InputCriblTCP,
-        InputNetflow,
-        InputTcpjson,
-        InputGooglePubsub,
-        InputOffice365Service,
-        InputWiz,
-        InputTCP,
-        InputFirehose,
-        InputCriblHTTP,
-        InputDatadogAgent,
-        InputOffice365Mgmt,
-        InputFile,
-        InputSplunk,
-        InputWef,
-        InputAppscope,
-        InputHTTPRaw,
-        InputWizWebhook,
-        InputHTTP,
-        InputCriblLakeHTTP,
-        InputAzureBlob,
-        InputSqs,
-        InputZscalerHec,
-        InputKinesis,
-        InputConfluentCloud,
-        InputEventhub,
-        InputKafka,
-        InputElastic,
-        InputOffice365MsgTrace,
-        InputSplunkHec,
-        InputLoki,
-        InputPrometheusRw,
-        InputPrometheus,
-        InputCrowdstrike,
-        InputEdgePrometheus,
-        InputOpenTelemetry,
-        InputS3,
-        InputSecurityLake,
-        InputMsk,
-        InputS3Inventory,
-        InputSplunkSearch,
-        InputSyslog,
-        InputGrafana,
+        Annotated[InputCollection, Tag("collection")],
+        Annotated[InputKafka, Tag("kafka")],
+        Annotated[InputMsk, Tag("msk")],
+        Annotated[InputHTTP, Tag("http")],
+        Annotated[InputSplunk, Tag("splunk")],
+        Annotated[InputSplunkSearch, Tag("splunk_search")],
+        Annotated[InputSplunkHec, Tag("splunk_hec")],
+        Annotated[InputAzureBlob, Tag("azure_blob")],
+        Annotated[InputElastic, Tag("elastic")],
+        Annotated[InputConfluentCloud, Tag("confluent_cloud")],
+        Annotated[InputGrafana, Tag("grafana")],
+        Annotated[InputLoki, Tag("loki")],
+        Annotated[InputPrometheusRw, Tag("prometheus_rw")],
+        Annotated[InputPrometheus, Tag("prometheus")],
+        Annotated[InputEdgePrometheus, Tag("edge_prometheus")],
+        Annotated[InputOffice365Mgmt, Tag("office365_mgmt")],
+        Annotated[InputOffice365Service, Tag("office365_service")],
+        Annotated[InputOffice365MsgTrace, Tag("office365_msg_trace")],
+        Annotated[InputEventhub, Tag("eventhub")],
+        Annotated[InputExec, Tag("exec")],
+        Annotated[InputFirehose, Tag("firehose")],
+        Annotated[InputGooglePubsub, Tag("google_pubsub")],
+        Annotated[InputCribl, Tag("cribl")],
+        Annotated[InputCriblTCP, Tag("cribl_tcp")],
+        Annotated[InputCriblHTTP, Tag("cribl_http")],
+        Annotated[InputCriblLakeHTTP, Tag("cribl_lake_http")],
+        Annotated[InputTcpjson, Tag("tcpjson")],
+        Annotated[InputSystemMetrics, Tag("system_metrics")],
+        Annotated[InputSystemState, Tag("system_state")],
+        Annotated[InputKubeMetrics, Tag("kube_metrics")],
+        Annotated[InputKubeLogs, Tag("kube_logs")],
+        Annotated[InputKubeEvents, Tag("kube_events")],
+        Annotated[InputWindowsMetrics, Tag("windows_metrics")],
+        Annotated[InputCrowdstrike, Tag("crowdstrike")],
+        Annotated[InputDatadogAgent, Tag("datadog_agent")],
+        Annotated[InputDatagen, Tag("datagen")],
+        Annotated[InputHTTPRaw, Tag("http_raw")],
+        Annotated[InputKinesis, Tag("kinesis")],
+        Annotated[InputCriblmetrics, Tag("criblmetrics")],
+        Annotated[InputMetrics, Tag("metrics")],
+        Annotated[InputS3, Tag("s3")],
+        Annotated[InputS3Inventory, Tag("s3_inventory")],
+        Annotated[InputSnmp, Tag("snmp")],
+        Annotated[InputOpenTelemetry, Tag("open_telemetry")],
+        Annotated[InputModelDrivenTelemetry, Tag("model_driven_telemetry")],
+        Annotated[InputSqs, Tag("sqs")],
+        Annotated[InputSyslog, Tag("syslog")],
+        Annotated[InputFile, Tag("file")],
+        Annotated[InputTCP, Tag("tcp")],
+        Annotated[InputAppscope, Tag("appscope")],
+        Annotated[InputWef, Tag("wef")],
+        Annotated[InputWinEventLogs, Tag("win_event_logs")],
+        Annotated[InputRawUDP, Tag("raw_udp")],
+        Annotated[InputJournalFiles, Tag("journal_files")],
+        Annotated[InputWiz, Tag("wiz")],
+        Annotated[InputWizWebhook, Tag("wiz_webhook")],
+        Annotated[InputNetflow, Tag("netflow")],
+        Annotated[InputSecurityLake, Tag("security_lake")],
+        Annotated[InputZscalerHec, Tag("zscaler_hec")],
     ],
-)
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]

cribl_control_plane/models/inputappscope.py

@@ -29,14 +29,18 @@ class InputAppscopeConnection(BaseModel):
 class InputAppscopeMode(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 InputAppscopeCompression(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Codec to use to compress the persisted data"""
 
+    # None
     NONE = "none"
+    # Gzip
     GZIP = "gzip"