cribl-control-plane 0.0.39__py3-none-any.whl → 0.5.0b3__py3-none-any.whl

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[
@@ -36,7 +39,7 @@ class LakeDatasets(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.CreateCriblLakeDatasetByLakeIDResponse:
+    ) -> models.CountedCriblLakeDataset:
         r"""Create a Lake Dataset
 
         Create a new Lake Dataset in the specified Lake.
@@ -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]
@@ -111,16 +118,21 @@ class LakeDatasets(BaseSDK):
                 "json",
                 models.CriblLakeDataset,
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
         if retries == UNSET:
             if self.sdk_configuration.retry_config is not UNSET:
                 retries = self.sdk_configuration.retry_config
+            else:
+                retries = utils.RetryConfig(
+                    "backoff", utils.BackoffStrategy(500, 60000, 1.5, 3600000), True
+                )
 
         retry_config = None
         if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
+            retry_config = (retries, ["429"])
 
         http_res = self.do_request(
             hook_ctx=HookContext(
@@ -139,9 +151,7 @@ class LakeDatasets(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.CreateCriblLakeDatasetByLakeIDResponse, http_res
-            )
+            return unmarshal_json_response(models.CountedCriblLakeDataset, 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)
@@ -168,6 +178,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[
@@ -180,7 +193,7 @@ class LakeDatasets(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.CreateCriblLakeDatasetByLakeIDResponse:
+    ) -> models.CountedCriblLakeDataset:
         r"""Create a Lake Dataset
 
         Create a new Lake Dataset in the specified Lake.
@@ -194,6 +207,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 +240,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]
@@ -255,16 +272,21 @@ class LakeDatasets(BaseSDK):
                 "json",
                 models.CriblLakeDataset,
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
         if retries == UNSET:
             if self.sdk_configuration.retry_config is not UNSET:
                 retries = self.sdk_configuration.retry_config
+            else:
+                retries = utils.RetryConfig(
+                    "backoff", utils.BackoffStrategy(500, 60000, 1.5, 3600000), True
+                )
 
         retry_config = None
         if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
+            retry_config = (retries, ["429"])
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
@@ -283,9 +305,7 @@ class LakeDatasets(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.CreateCriblLakeDatasetByLakeIDResponse, http_res
-            )
+            return unmarshal_json_response(models.CountedCriblLakeDataset, 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)
@@ -302,16 +322,28 @@ class LakeDatasets(BaseSDK):
         self,
         *,
         lake_id: str,
+        storage_location_id: Optional[str] = None,
+        format_: Optional[str] = None,
+        exclude_ddss: Optional[bool] = None,
+        exclude_deleted: Optional[bool] = None,
+        exclude_internal: Optional[bool] = None,
+        exclude_byos: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.GetCriblLakeDatasetByLakeIDResponse:
+    ) -> models.CountedCriblLakeDataset:
         r"""List all Lake Datasets
 
         Get a list of all Lake Datasets in the specified Lake.
 
         :param lake_id: The <code>id</code> of the Lake that contains the Lake Datasets to list.
+        :param storage_location_id: Filter datasets by storage location ID. Use <code>default</code> for default storage location.
+        :param format_: Filter datasets by format. Set to <code>ddss</code> to return only DDSS datasets.
+        :param exclude_ddss: Exclude DDSS format datasets from the response.
+        :param exclude_deleted: Exclude deleted datasets from the response.
+        :param exclude_internal: Exclude internal datasets (those with IDs starting with <code>cribl_</code>) from the response.
+        :param exclude_byos: Exclude BYOS (Bring Your Own Storage) datasets from the response.
         :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
@@ -329,6 +361,12 @@ class LakeDatasets(BaseSDK):
 
         request = models.GetCriblLakeDatasetByLakeIDRequest(
             lake_id=lake_id,
+            storage_location_id=storage_location_id,
+            format_=format_,
+            exclude_ddss=exclude_ddss,
+            exclude_deleted=exclude_deleted,
+            exclude_internal=exclude_internal,
+            exclude_byos=exclude_byos,
         )
 
         req = self._build_request(
@@ -344,16 +382,21 @@ class LakeDatasets(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
         if retries == UNSET:
             if self.sdk_configuration.retry_config is not UNSET:
                 retries = self.sdk_configuration.retry_config
+            else:
+                retries = utils.RetryConfig(
+                    "backoff", utils.BackoffStrategy(500, 60000, 1.5, 3600000), True
+                )
 
         retry_config = None
         if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
+            retry_config = (retries, ["429"])
 
         http_res = self.do_request(
             hook_ctx=HookContext(
@@ -372,9 +415,7 @@ class LakeDatasets(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.GetCriblLakeDatasetByLakeIDResponse, http_res
-            )
+            return unmarshal_json_response(models.CountedCriblLakeDataset, 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)
@@ -391,16 +432,28 @@ class LakeDatasets(BaseSDK):
         self,
         *,
         lake_id: str,
+        storage_location_id: Optional[str] = None,
+        format_: Optional[str] = None,
+        exclude_ddss: Optional[bool] = None,
+        exclude_deleted: Optional[bool] = None,
+        exclude_internal: Optional[bool] = None,
+        exclude_byos: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.GetCriblLakeDatasetByLakeIDResponse:
+    ) -> models.CountedCriblLakeDataset:
         r"""List all Lake Datasets
 
         Get a list of all Lake Datasets in the specified Lake.
 
         :param lake_id: The <code>id</code> of the Lake that contains the Lake Datasets to list.
+        :param storage_location_id: Filter datasets by storage location ID. Use <code>default</code> for default storage location.
+        :param format_: Filter datasets by format. Set to <code>ddss</code> to return only DDSS datasets.
+        :param exclude_ddss: Exclude DDSS format datasets from the response.
+        :param exclude_deleted: Exclude deleted datasets from the response.
+        :param exclude_internal: Exclude internal datasets (those with IDs starting with <code>cribl_</code>) from the response.
+        :param exclude_byos: Exclude BYOS (Bring Your Own Storage) datasets from the response.
         :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
@@ -418,6 +471,12 @@ class LakeDatasets(BaseSDK):
 
         request = models.GetCriblLakeDatasetByLakeIDRequest(
             lake_id=lake_id,
+            storage_location_id=storage_location_id,
+            format_=format_,
+            exclude_ddss=exclude_ddss,
+            exclude_deleted=exclude_deleted,
+            exclude_internal=exclude_internal,
+            exclude_byos=exclude_byos,
         )
 
         req = self._build_request_async(
@@ -433,16 +492,21 @@ class LakeDatasets(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
         if retries == UNSET:
             if self.sdk_configuration.retry_config is not UNSET:
                 retries = self.sdk_configuration.retry_config
+            else:
+                retries = utils.RetryConfig(
+                    "backoff", utils.BackoffStrategy(500, 60000, 1.5, 3600000), True
+                )
 
         retry_config = None
         if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
+            retry_config = (retries, ["429"])
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
@@ -461,9 +525,7 @@ class LakeDatasets(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.GetCriblLakeDatasetByLakeIDResponse, http_res
-            )
+            return unmarshal_json_response(models.CountedCriblLakeDataset, 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)
@@ -485,7 +547,7 @@ class LakeDatasets(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.DeleteCriblLakeDatasetByLakeIDAndIDResponse:
+    ) -> models.CountedCriblLakeDataset:
         r"""Delete a Lake Dataset
 
         Delete the specified Lake Dataset in the specified Lake
@@ -525,16 +587,21 @@ class LakeDatasets(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
         if retries == UNSET:
             if self.sdk_configuration.retry_config is not UNSET:
                 retries = self.sdk_configuration.retry_config
+            else:
+                retries = utils.RetryConfig(
+                    "backoff", utils.BackoffStrategy(500, 60000, 1.5, 3600000), True
+                )
 
         retry_config = None
         if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
+            retry_config = (retries, ["429"])
 
         http_res = self.do_request(
             hook_ctx=HookContext(
@@ -553,9 +620,7 @@ class LakeDatasets(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.DeleteCriblLakeDatasetByLakeIDAndIDResponse, http_res
-            )
+            return unmarshal_json_response(models.CountedCriblLakeDataset, 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)
@@ -577,7 +642,7 @@ class LakeDatasets(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.DeleteCriblLakeDatasetByLakeIDAndIDResponse:
+    ) -> models.CountedCriblLakeDataset:
         r"""Delete a Lake Dataset
 
         Delete the specified Lake Dataset in the specified Lake
@@ -617,16 +682,21 @@ class LakeDatasets(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
         if retries == UNSET:
             if self.sdk_configuration.retry_config is not UNSET:
                 retries = self.sdk_configuration.retry_config
+            else:
+                retries = utils.RetryConfig(
+                    "backoff", utils.BackoffStrategy(500, 60000, 1.5, 3600000), True
+                )
 
         retry_config = None
         if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
+            retry_config = (retries, ["429"])
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
@@ -645,9 +715,7 @@ class LakeDatasets(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.DeleteCriblLakeDatasetByLakeIDAndIDResponse, http_res
-            )
+            return unmarshal_json_response(models.CountedCriblLakeDataset, 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)
@@ -669,7 +737,7 @@ class LakeDatasets(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.GetCriblLakeDatasetByLakeIDAndIDResponse:
+    ) -> models.CountedCriblLakeDataset:
         r"""Get a Lake Dataset
 
         Get the specified Lake Dataset in the specified Lake.
@@ -709,16 +777,21 @@ class LakeDatasets(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
         if retries == UNSET:
             if self.sdk_configuration.retry_config is not UNSET:
                 retries = self.sdk_configuration.retry_config
+            else:
+                retries = utils.RetryConfig(
+                    "backoff", utils.BackoffStrategy(500, 60000, 1.5, 3600000), True
+                )
 
         retry_config = None
         if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
+            retry_config = (retries, ["429"])
 
         http_res = self.do_request(
             hook_ctx=HookContext(
@@ -737,9 +810,7 @@ class LakeDatasets(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.GetCriblLakeDatasetByLakeIDAndIDResponse, http_res
-            )
+            return unmarshal_json_response(models.CountedCriblLakeDataset, 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)
@@ -761,7 +832,7 @@ class LakeDatasets(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.GetCriblLakeDatasetByLakeIDAndIDResponse:
+    ) -> models.CountedCriblLakeDataset:
         r"""Get a Lake Dataset
 
         Get the specified Lake Dataset in the specified Lake.
@@ -801,16 +872,21 @@ class LakeDatasets(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
         if retries == UNSET:
             if self.sdk_configuration.retry_config is not UNSET:
                 retries = self.sdk_configuration.retry_config
+            else:
+                retries = utils.RetryConfig(
+                    "backoff", utils.BackoffStrategy(500, 60000, 1.5, 3600000), True
+                )
 
         retry_config = None
         if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
+            retry_config = (retries, ["429"])
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
@@ -829,9 +905,7 @@ class LakeDatasets(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.GetCriblLakeDatasetByLakeIDAndIDResponse, http_res
-            )
+            return unmarshal_json_response(models.CountedCriblLakeDataset, 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)
@@ -849,7 +923,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 +930,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[
@@ -871,14 +948,13 @@ class LakeDatasets(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.UpdateCriblLakeDatasetByLakeIDAndIDResponse:
+    ) -> models.CountedCriblLakeDataset:
         r"""Update a Lake Dataset
 
         Update the specified Lake Dataset in the specified Lake.
 
         :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 +962,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 +986,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 +997,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,22 +1023,27 @@ 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,
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
         if retries == UNSET:
             if self.sdk_configuration.retry_config is not UNSET:
                 retries = self.sdk_configuration.retry_config
+            else:
+                retries = utils.RetryConfig(
+                    "backoff", utils.BackoffStrategy(500, 60000, 1.5, 3600000), True
+                )
 
         retry_config = None
         if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
+            retry_config = (retries, ["429"])
 
         http_res = self.do_request(
             hook_ctx=HookContext(
@@ -976,9 +1062,7 @@ class LakeDatasets(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.UpdateCriblLakeDatasetByLakeIDAndIDResponse, http_res
-            )
+            return unmarshal_json_response(models.CountedCriblLakeDataset, 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)
@@ -996,7 +1080,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 +1087,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[
@@ -1018,14 +1105,13 @@ class LakeDatasets(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.UpdateCriblLakeDatasetByLakeIDAndIDResponse:
+    ) -> models.CountedCriblLakeDataset:
         r"""Update a Lake Dataset
 
         Update the specified Lake Dataset in the specified Lake.
 
         :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 +1119,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 +1143,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 +1154,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,22 +1180,27 @@ 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,
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
         if retries == UNSET:
             if self.sdk_configuration.retry_config is not UNSET:
                 retries = self.sdk_configuration.retry_config
+            else:
+                retries = utils.RetryConfig(
+                    "backoff", utils.BackoffStrategy(500, 60000, 1.5, 3600000), True
+                )
 
         retry_config = None
         if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
+            retry_config = (retries, ["429"])
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
@@ -1123,9 +1219,7 @@ class LakeDatasets(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.UpdateCriblLakeDatasetByLakeIDAndIDResponse, http_res
-            )
+            return unmarshal_json_response(models.CountedCriblLakeDataset, 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)