cribl-control-plane 0.0.38__py3-none-any.whl → 0.4.0a6__py3-none-any.whl

cribl_control_plane/groups_sdk.py

@@ -18,14 +18,16 @@ class GroupsSDK(BaseSDK):
     configs: GroupsConfigs
     acl: ACL
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
-        BaseSDK.__init__(self, sdk_config)
+    def __init__(
+        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
+    ) -> None:
+        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
         self.sdk_configuration = sdk_config
         self._init_sdks()
 
     def _init_sdks(self):
-        self.configs = GroupsConfigs(self.sdk_configuration)
-        self.acl = ACL(self.sdk_configuration)
+        self.configs = GroupsConfigs(self.sdk_configuration, parent_ref=self.parent_ref)
+        self.acl = ACL(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def list(
         self,
@@ -215,15 +217,18 @@ class GroupsSDK(BaseSDK):
         self,
         *,
         product: models.ProductsCore,
-        config_version: str,
         id: str,
         cloud: Optional[
             Union[models.ConfigGroupCloud, models.ConfigGroupCloudTypedDict]
         ] = None,
         deploying_worker_count: Optional[float] = None,
         description: Optional[str] = None,
-        estimated_ingest_rate: Optional[float] = None,
-        git: Optional[Union[models.Git, models.GitTypedDict]] = None,
+        estimated_ingest_rate: Optional[
+            models.GroupCreateRequestEstimatedIngestRate
+        ] = None,
+        git: Optional[
+            Union[models.GroupCreateRequestGit, models.GroupCreateRequestGitTypedDict]
+        ] = None,
         incompatible_worker_count: Optional[float] = None,
         inherits: Optional[str] = None,
         is_fleet: Optional[bool] = None,
@@ -238,9 +243,10 @@ class GroupsSDK(BaseSDK):
         name: Optional[str] = None,
         on_prem: Optional[bool] = None,
         provisioned: Optional[bool] = None,
+        source_group_id: Optional[str] = None,
         streamtags: Optional[List[str]] = None,
         tags: Optional[str] = None,
-        type_: Optional[models.ConfigGroupType] = None,
+        type_: Optional[models.GroupCreateRequestType] = None,
         upgrade_version: Optional[str] = None,
         worker_count: Optional[float] = None,
         worker_remote_access: Optional[bool] = None,
@@ -253,13 +259,12 @@ class GroupsSDK(BaseSDK):
 
         Create a new Worker Group or Edge Fleet for the specified Cribl product.
 
-        :param product: Name of the Cribl product to add the Worker Group or Edge Fleet to.
-        :param config_version:
+        :param product: required Name of the Cribl product to add the Worker Group or Edge Fleet to.
         :param id:
         :param cloud:
         :param deploying_worker_count:
         :param description:
-        :param estimated_ingest_rate:
+        :param estimated_ingest_rate: Maximum expected volume of data ingested by the @{group}. (This setting is available only on @{group}s consisting of Cribl-managed Cribl.Cloud @{node}s.)
         :param git:
         :param incompatible_worker_count:
         :param inherits:
@@ -270,6 +275,7 @@ class GroupsSDK(BaseSDK):
         :param name:
         :param on_prem:
         :param provisioned:
+        :param source_group_id:
         :param streamtags:
         :param tags:
         :param type:
@@ -293,15 +299,16 @@ class GroupsSDK(BaseSDK):
 
         request = models.CreateConfigGroupByProductRequest(
             product=product,
-            config_group=models.ConfigGroup(
+            group_create_request=models.GroupCreateRequest(
                 cloud=utils.get_pydantic_model(
                     cloud, Optional[models.ConfigGroupCloud]
                 ),
-                config_version=config_version,
                 deploying_worker_count=deploying_worker_count,
                 description=description,
                 estimated_ingest_rate=estimated_ingest_rate,
-                git=utils.get_pydantic_model(git, Optional[models.Git]),
+                git=utils.get_pydantic_model(
+                    git, Optional[models.GroupCreateRequestGit]
+                ),
                 id=id,
                 incompatible_worker_count=incompatible_worker_count,
                 inherits=inherits,
@@ -314,6 +321,7 @@ class GroupsSDK(BaseSDK):
                 name=name,
                 on_prem=on_prem,
                 provisioned=provisioned,
+                source_group_id=source_group_id,
                 streamtags=streamtags,
                 tags=tags,
                 type=type_,
@@ -337,7 +345,11 @@ class GroupsSDK(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request.config_group, False, False, "json", models.ConfigGroup
+                request.group_create_request,
+                False,
+                False,
+                "json",
+                models.GroupCreateRequest,
             ),
             timeout_ms=timeout_ms,
         )
@@ -386,15 +398,18 @@ class GroupsSDK(BaseSDK):
         self,
         *,
         product: models.ProductsCore,
-        config_version: str,
         id: str,
         cloud: Optional[
             Union[models.ConfigGroupCloud, models.ConfigGroupCloudTypedDict]
         ] = None,
         deploying_worker_count: Optional[float] = None,
         description: Optional[str] = None,
-        estimated_ingest_rate: Optional[float] = None,
-        git: Optional[Union[models.Git, models.GitTypedDict]] = None,
+        estimated_ingest_rate: Optional[
+            models.GroupCreateRequestEstimatedIngestRate
+        ] = None,
+        git: Optional[
+            Union[models.GroupCreateRequestGit, models.GroupCreateRequestGitTypedDict]
+        ] = None,
         incompatible_worker_count: Optional[float] = None,
         inherits: Optional[str] = None,
         is_fleet: Optional[bool] = None,
@@ -409,9 +424,10 @@ class GroupsSDK(BaseSDK):
         name: Optional[str] = None,
         on_prem: Optional[bool] = None,
         provisioned: Optional[bool] = None,
+        source_group_id: Optional[str] = None,
         streamtags: Optional[List[str]] = None,
         tags: Optional[str] = None,
-        type_: Optional[models.ConfigGroupType] = None,
+        type_: Optional[models.GroupCreateRequestType] = None,
         upgrade_version: Optional[str] = None,
         worker_count: Optional[float] = None,
         worker_remote_access: Optional[bool] = None,
@@ -424,13 +440,12 @@ class GroupsSDK(BaseSDK):
 
         Create a new Worker Group or Edge Fleet for the specified Cribl product.
 
-        :param product: Name of the Cribl product to add the Worker Group or Edge Fleet to.
-        :param config_version:
+        :param product: required Name of the Cribl product to add the Worker Group or Edge Fleet to.
         :param id:
         :param cloud:
         :param deploying_worker_count:
         :param description:
-        :param estimated_ingest_rate:
+        :param estimated_ingest_rate: Maximum expected volume of data ingested by the @{group}. (This setting is available only on @{group}s consisting of Cribl-managed Cribl.Cloud @{node}s.)
         :param git:
         :param incompatible_worker_count:
         :param inherits:
@@ -441,6 +456,7 @@ class GroupsSDK(BaseSDK):
         :param name:
         :param on_prem:
         :param provisioned:
+        :param source_group_id:
         :param streamtags:
         :param tags:
         :param type:
@@ -464,15 +480,16 @@ class GroupsSDK(BaseSDK):
 
         request = models.CreateConfigGroupByProductRequest(
             product=product,
-            config_group=models.ConfigGroup(
+            group_create_request=models.GroupCreateRequest(
                 cloud=utils.get_pydantic_model(
                     cloud, Optional[models.ConfigGroupCloud]
                 ),
-                config_version=config_version,
                 deploying_worker_count=deploying_worker_count,
                 description=description,
                 estimated_ingest_rate=estimated_ingest_rate,
-                git=utils.get_pydantic_model(git, Optional[models.Git]),
+                git=utils.get_pydantic_model(
+                    git, Optional[models.GroupCreateRequestGit]
+                ),
                 id=id,
                 incompatible_worker_count=incompatible_worker_count,
                 inherits=inherits,
@@ -485,6 +502,7 @@ class GroupsSDK(BaseSDK):
                 name=name,
                 on_prem=on_prem,
                 provisioned=provisioned,
+                source_group_id=source_group_id,
                 streamtags=streamtags,
                 tags=tags,
                 type=type_,
@@ -508,7 +526,11 @@ class GroupsSDK(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request.config_group, False, False, "json", models.ConfigGroup
+                request.group_create_request,
+                False,
+                False,
+                "json",
+                models.GroupCreateRequest,
             ),
             timeout_ms=timeout_ms,
         )
@@ -748,15 +770,17 @@ class GroupsSDK(BaseSDK):
         *,
         product: models.ProductsCore,
         id_param: str,
-        config_version: str,
         id: str,
         cloud: Optional[
             Union[models.ConfigGroupCloud, models.ConfigGroupCloudTypedDict]
         ] = None,
+        config_version: Optional[str] = None,
         deploying_worker_count: Optional[float] = None,
         description: Optional[str] = None,
-        estimated_ingest_rate: Optional[float] = None,
-        git: Optional[Union[models.Git, models.GitTypedDict]] = None,
+        estimated_ingest_rate: Optional[models.ConfigGroupEstimatedIngestRate] = None,
+        git: Optional[
+            Union[models.ConfigGroupGit, models.ConfigGroupGitTypedDict]
+        ] = None,
         incompatible_worker_count: Optional[float] = None,
         inherits: Optional[str] = None,
         is_fleet: Optional[bool] = None,
@@ -788,12 +812,12 @@ class GroupsSDK(BaseSDK):
 
         :param product: Name of the Cribl product to get the Worker Groups or Edge Fleets for.
         :param id_param: The <code>id</code> of the Worker Group or Edge Fleet to update.
-        :param config_version:
         :param id:
         :param cloud:
+        :param config_version:
         :param deploying_worker_count:
         :param description:
-        :param estimated_ingest_rate:
+        :param estimated_ingest_rate: Maximum expected volume of data ingested by the @{group}. (This setting is available only on @{group}s consisting of Cribl-managed Cribl.Cloud @{node}s.)
         :param git:
         :param incompatible_worker_count:
         :param inherits:
@@ -836,7 +860,7 @@ class GroupsSDK(BaseSDK):
                 deploying_worker_count=deploying_worker_count,
                 description=description,
                 estimated_ingest_rate=estimated_ingest_rate,
-                git=utils.get_pydantic_model(git, Optional[models.Git]),
+                git=utils.get_pydantic_model(git, Optional[models.ConfigGroupGit]),
                 id=id,
                 incompatible_worker_count=incompatible_worker_count,
                 inherits=inherits,
@@ -922,15 +946,17 @@ class GroupsSDK(BaseSDK):
         *,
         product: models.ProductsCore,
         id_param: str,
-        config_version: str,
         id: str,
         cloud: Optional[
             Union[models.ConfigGroupCloud, models.ConfigGroupCloudTypedDict]
         ] = None,
+        config_version: Optional[str] = None,
         deploying_worker_count: Optional[float] = None,
         description: Optional[str] = None,
-        estimated_ingest_rate: Optional[float] = None,
-        git: Optional[Union[models.Git, models.GitTypedDict]] = None,
+        estimated_ingest_rate: Optional[models.ConfigGroupEstimatedIngestRate] = None,
+        git: Optional[
+            Union[models.ConfigGroupGit, models.ConfigGroupGitTypedDict]
+        ] = None,
         incompatible_worker_count: Optional[float] = None,
         inherits: Optional[str] = None,
         is_fleet: Optional[bool] = None,
@@ -962,12 +988,12 @@ class GroupsSDK(BaseSDK):
 
         :param product: Name of the Cribl product to get the Worker Groups or Edge Fleets for.
         :param id_param: The <code>id</code> of the Worker Group or Edge Fleet to update.
-        :param config_version:
         :param id:
         :param cloud:
+        :param config_version:
         :param deploying_worker_count:
         :param description:
-        :param estimated_ingest_rate:
+        :param estimated_ingest_rate: Maximum expected volume of data ingested by the @{group}. (This setting is available only on @{group}s consisting of Cribl-managed Cribl.Cloud @{node}s.)
         :param git:
         :param incompatible_worker_count:
         :param inherits:
@@ -1010,7 +1036,7 @@ class GroupsSDK(BaseSDK):
                 deploying_worker_count=deploying_worker_count,
                 description=description,
                 estimated_ingest_rate=estimated_ingest_rate,
-                git=utils.get_pydantic_model(git, Optional[models.Git]),
+                git=utils.get_pydantic_model(git, Optional[models.ConfigGroupGit]),
                 id=id,
                 incompatible_worker_count=incompatible_worker_count,
                 inherits=inherits,

cribl_control_plane/health.py

@@ -18,9 +18,11 @@ class Health(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.HealthStatus:
+    ) -> models.HealthServerStatus:
         r"""Retrieve health status of the server
 
+        Get the current health status of the server.
+
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -62,23 +64,26 @@ class Health(BaseSDK):
             hook_ctx=HookContext(
                 config=self.sdk_configuration,
                 base_url=base_url or "",
-                operation_id="getHealthInfo",
+                operation_id="getHealth",
                 oauth2_scopes=[],
                 security_source=None,
             ),
             request=req,
-            error_status_codes=["420", "4XX", "5XX"],
+            error_status_codes=["420", "4XX", "500", "5XX"],
             retry_config=retry_config,
         )
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(models.HealthStatus, http_res)
+            return unmarshal_json_response(models.HealthServerStatus, http_res)
         if utils.match_response(http_res, "420", "application/json"):
             response_data = unmarshal_json_response(
-                errors.HealthStatusErrorData, http_res
+                errors.HealthServerStatusErrorData, http_res
             )
-            raise errors.HealthStatusError(response_data, http_res)
+            raise errors.HealthServerStatusError(response_data, http_res)
+        if utils.match_response(http_res, "500", "application/json"):
+            response_data = unmarshal_json_response(errors.ErrorData, http_res)
+            raise errors.Error(response_data, http_res)
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise errors.APIError("API error occurred", http_res, http_res_text)
@@ -95,9 +100,11 @@ class Health(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.HealthStatus:
+    ) -> models.HealthServerStatus:
         r"""Retrieve health status of the server
 
+        Get the current health status of the server.
+
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -139,23 +146,26 @@ class Health(BaseSDK):
             hook_ctx=HookContext(
                 config=self.sdk_configuration,
                 base_url=base_url or "",
-                operation_id="getHealthInfo",
+                operation_id="getHealth",
                 oauth2_scopes=[],
                 security_source=None,
             ),
             request=req,
-            error_status_codes=["420", "4XX", "5XX"],
+            error_status_codes=["420", "4XX", "500", "5XX"],
             retry_config=retry_config,
         )
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(models.HealthStatus, http_res)
+            return unmarshal_json_response(models.HealthServerStatus, http_res)
         if utils.match_response(http_res, "420", "application/json"):
             response_data = unmarshal_json_response(
-                errors.HealthStatusErrorData, http_res
+                errors.HealthServerStatusErrorData, http_res
             )
-            raise errors.HealthStatusError(response_data, http_res)
+            raise errors.HealthServerStatusError(response_data, http_res)
+        if utils.match_response(http_res, "500", "application/json"):
+            response_data = unmarshal_json_response(errors.ErrorData, http_res)
+            raise errors.Error(response_data, http_res)
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise errors.APIError("API error occurred", http_res, http_res_text)

cribl_control_plane/httpclient.py

@@ -107,7 +107,6 @@ def close_clients(
     # to them from the owning SDK instance and they can be reaped.
     owner.client = None
     owner.async_client = None
-
     if sync_client is not None and not sync_client_supplied:
         try:
             sync_client.close()

cribl_control_plane/lakedatasets.py

@@ -24,6 +24,9 @@ class LakeDatasets(BaseSDK):
         description: Optional[str] = None,
         format_: Optional[models.CriblLakeDatasetFormat] = None,
         http_da_used: Optional[bool] = None,
+        metrics: Optional[
+            Union[models.LakeDatasetMetrics, models.LakeDatasetMetricsTypedDict]
+        ] = None,
         retention_period_in_days: Optional[float] = None,
         search_config: Optional[
             Union[
@@ -50,6 +53,7 @@ class LakeDatasets(BaseSDK):
         :param description:
         :param format_:
         :param http_da_used:
+        :param metrics:
         :param retention_period_in_days:
         :param search_config:
         :param storage_location_id:
@@ -82,6 +86,9 @@ class LakeDatasets(BaseSDK):
                 format_=format_,
                 http_da_used=http_da_used,
                 id=id,
+                metrics=utils.get_pydantic_model(
+                    metrics, Optional[models.LakeDatasetMetrics]
+                ),
                 retention_period_in_days=retention_period_in_days,
                 search_config=utils.get_pydantic_model(
                     search_config, Optional[models.LakeDatasetSearchConfig]
@@ -168,6 +175,9 @@ class LakeDatasets(BaseSDK):
         description: Optional[str] = None,
         format_: Optional[models.CriblLakeDatasetFormat] = None,
         http_da_used: Optional[bool] = None,
+        metrics: Optional[
+            Union[models.LakeDatasetMetrics, models.LakeDatasetMetricsTypedDict]
+        ] = None,
         retention_period_in_days: Optional[float] = None,
         search_config: Optional[
             Union[
@@ -194,6 +204,7 @@ class LakeDatasets(BaseSDK):
         :param description:
         :param format_:
         :param http_da_used:
+        :param metrics:
         :param retention_period_in_days:
         :param search_config:
         :param storage_location_id:
@@ -226,6 +237,9 @@ class LakeDatasets(BaseSDK):
                 format_=format_,
                 http_da_used=http_da_used,
                 id=id,
+                metrics=utils.get_pydantic_model(
+                    metrics, Optional[models.LakeDatasetMetrics]
+                ),
                 retention_period_in_days=retention_period_in_days,
                 search_config=utils.get_pydantic_model(
                     search_config, Optional[models.LakeDatasetSearchConfig]
@@ -849,7 +863,6 @@ class LakeDatasets(BaseSDK):
         *,
         lake_id: str,
         id_param: str,
-        id: str,
         accelerated_fields: Optional[List[str]] = None,
         bucket_name: Optional[str] = None,
         cache_connection: Optional[
@@ -857,8 +870,12 @@ class LakeDatasets(BaseSDK):
         ] = None,
         deletion_started_at: Optional[float] = None,
         description: Optional[str] = None,
-        format_: Optional[models.CriblLakeDatasetFormat] = None,
+        format_: Optional[models.CriblLakeDatasetUpdateFormat] = None,
         http_da_used: Optional[bool] = None,
+        id: Optional[str] = None,
+        metrics: Optional[
+            Union[models.LakeDatasetMetrics, models.LakeDatasetMetricsTypedDict]
+        ] = None,
         retention_period_in_days: Optional[float] = None,
         search_config: Optional[
             Union[
@@ -878,7 +895,6 @@ class LakeDatasets(BaseSDK):
 
         :param lake_id: The <code>id</code> of the Lake that contains the Lake Dataset to update.
         :param id_param: The <code>id</code> of the Lake Dataset to update.
-        :param id:
         :param accelerated_fields:
         :param bucket_name:
         :param cache_connection:
@@ -886,6 +902,8 @@ class LakeDatasets(BaseSDK):
         :param description:
         :param format_:
         :param http_da_used:
+        :param id:
+        :param metrics:
         :param retention_period_in_days:
         :param search_config:
         :param storage_location_id:
@@ -908,7 +926,7 @@ class LakeDatasets(BaseSDK):
         request = models.UpdateCriblLakeDatasetByLakeIDAndIDRequest(
             lake_id=lake_id,
             id_param=id_param,
-            cribl_lake_dataset=models.CriblLakeDataset(
+            cribl_lake_dataset_update=models.CriblLakeDatasetUpdate(
                 accelerated_fields=accelerated_fields,
                 bucket_name=bucket_name,
                 cache_connection=utils.get_pydantic_model(
@@ -919,6 +937,9 @@ class LakeDatasets(BaseSDK):
                 format_=format_,
                 http_da_used=http_da_used,
                 id=id,
+                metrics=utils.get_pydantic_model(
+                    metrics, Optional[models.LakeDatasetMetrics]
+                ),
                 retention_period_in_days=retention_period_in_days,
                 search_config=utils.get_pydantic_model(
                     search_config, Optional[models.LakeDatasetSearchConfig]
@@ -942,11 +963,11 @@ class LakeDatasets(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request.cribl_lake_dataset,
+                request.cribl_lake_dataset_update,
                 False,
                 False,
                 "json",
-                models.CriblLakeDataset,
+                models.CriblLakeDatasetUpdate,
             ),
             timeout_ms=timeout_ms,
         )
@@ -996,7 +1017,6 @@ class LakeDatasets(BaseSDK):
         *,
         lake_id: str,
         id_param: str,
-        id: str,
         accelerated_fields: Optional[List[str]] = None,
         bucket_name: Optional[str] = None,
         cache_connection: Optional[
@@ -1004,8 +1024,12 @@ class LakeDatasets(BaseSDK):
         ] = None,
         deletion_started_at: Optional[float] = None,
         description: Optional[str] = None,
-        format_: Optional[models.CriblLakeDatasetFormat] = None,
+        format_: Optional[models.CriblLakeDatasetUpdateFormat] = None,
         http_da_used: Optional[bool] = None,
+        id: Optional[str] = None,
+        metrics: Optional[
+            Union[models.LakeDatasetMetrics, models.LakeDatasetMetricsTypedDict]
+        ] = None,
         retention_period_in_days: Optional[float] = None,
         search_config: Optional[
             Union[
@@ -1025,7 +1049,6 @@ class LakeDatasets(BaseSDK):
 
         :param lake_id: The <code>id</code> of the Lake that contains the Lake Dataset to update.
         :param id_param: The <code>id</code> of the Lake Dataset to update.
-        :param id:
         :param accelerated_fields:
         :param bucket_name:
         :param cache_connection:
@@ -1033,6 +1056,8 @@ class LakeDatasets(BaseSDK):
         :param description:
         :param format_:
         :param http_da_used:
+        :param id:
+        :param metrics:
         :param retention_period_in_days:
         :param search_config:
         :param storage_location_id:
@@ -1055,7 +1080,7 @@ class LakeDatasets(BaseSDK):
         request = models.UpdateCriblLakeDatasetByLakeIDAndIDRequest(
             lake_id=lake_id,
             id_param=id_param,
-            cribl_lake_dataset=models.CriblLakeDataset(
+            cribl_lake_dataset_update=models.CriblLakeDatasetUpdate(
                 accelerated_fields=accelerated_fields,
                 bucket_name=bucket_name,
                 cache_connection=utils.get_pydantic_model(
@@ -1066,6 +1091,9 @@ class LakeDatasets(BaseSDK):
                 format_=format_,
                 http_da_used=http_da_used,
                 id=id,
+                metrics=utils.get_pydantic_model(
+                    metrics, Optional[models.LakeDatasetMetrics]
+                ),
                 retention_period_in_days=retention_period_in_days,
                 search_config=utils.get_pydantic_model(
                     search_config, Optional[models.LakeDatasetSearchConfig]
@@ -1089,11 +1117,11 @@ class LakeDatasets(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request.cribl_lake_dataset,
+                request.cribl_lake_dataset_update,
                 False,
                 False,
                 "json",
-                models.CriblLakeDataset,
+                models.CriblLakeDatasetUpdate,
             ),
             timeout_ms=timeout_ms,
         )