cribl-control-plane 0.0.21__py3-none-any.whl → 0.4.0b23__py3-none-any.whl

cribl_control_plane/{lake.py → lakedatasets.py}

@@ -9,10 +9,8 @@ from cribl_control_plane.utils.unmarshal_json_response import unmarshal_json_res
 from typing import Any, List, Mapping, Optional, Union
 
 
-class Lake(BaseSDK):
-    r"""Actions related to Lake"""
-
-    def create_cribl_lake_dataset_by_lake_id(
+class LakeDatasets(BaseSDK):
+    def create(
         self,
         *,
         lake_id: str,
@@ -26,6 +24,9 @@ class Lake(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[
@@ -38,12 +39,12 @@ class Lake(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.CreateCriblLakeDatasetByLakeIDResponse:
-        r"""Create a Dataset in the specified Lake
+    ) -> models.CountedCriblLakeDataset:
+        r"""Create a Lake Dataset
 
-        Create a Dataset in the specified Lake
+        Create a new Lake Dataset in the specified Lake.
 
-        :param lake_id: lake id that contains the Datasets
+        :param lake_id: The <code>id</code> of the Lake to create the Lake Dataset in.
         :param id:
         :param accelerated_fields:
         :param bucket_name:
@@ -52,6 +53,7 @@ class Lake(BaseSDK):
         :param description:
         :param format_:
         :param http_da_used:
+        :param metrics:
         :param retention_period_in_days:
         :param search_config:
         :param storage_location_id:
@@ -84,6 +86,9 @@ class Lake(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]
@@ -113,16 +118,21 @@ class Lake(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(
@@ -141,9 +151,7 @@ class Lake(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)
@@ -156,7 +164,7 @@ class Lake(BaseSDK):
 
         raise errors.APIError("Unexpected response received", http_res)
 
-    async def create_cribl_lake_dataset_by_lake_id_async(
+    async def create_async(
         self,
         *,
         lake_id: str,
@@ -170,6 +178,9 @@ class Lake(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[
@@ -182,12 +193,12 @@ class Lake(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.CreateCriblLakeDatasetByLakeIDResponse:
-        r"""Create a Dataset in the specified Lake
+    ) -> models.CountedCriblLakeDataset:
+        r"""Create a Lake Dataset
 
-        Create a Dataset in the specified Lake
+        Create a new Lake Dataset in the specified Lake.
 
-        :param lake_id: lake id that contains the Datasets
+        :param lake_id: The <code>id</code> of the Lake to create the Lake Dataset in.
         :param id:
         :param accelerated_fields:
         :param bucket_name:
@@ -196,6 +207,7 @@ class Lake(BaseSDK):
         :param description:
         :param format_:
         :param http_da_used:
+        :param metrics:
         :param retention_period_in_days:
         :param search_config:
         :param storage_location_id:
@@ -228,6 +240,9 @@ class Lake(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]
@@ -257,16 +272,21 @@ class Lake(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(
@@ -285,9 +305,7 @@ class Lake(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)
@@ -300,20 +318,32 @@ class Lake(BaseSDK):
 
         raise errors.APIError("Unexpected response received", http_res)
 
-    def get_cribl_lake_dataset_by_lake_id(
+    def list(
         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:
-        r"""Get the list of Dataset contained in the specified Lake
-
-        Get the list of Dataset contained in the specified Lake
-
-        :param lake_id: lake id that contains the Datasets
+    ) -> 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
@@ -331,6 +361,12 @@ class Lake(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(
@@ -346,16 +382,21 @@ class Lake(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(
@@ -374,9 +415,7 @@ class Lake(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)
@@ -389,20 +428,32 @@ class Lake(BaseSDK):
 
         raise errors.APIError("Unexpected response received", http_res)
 
-    async def get_cribl_lake_dataset_by_lake_id_async(
+    async def list_async(
         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:
-        r"""Get the list of Dataset contained in the specified Lake
-
-        Get the list of Dataset contained in the specified Lake
-
-        :param lake_id: lake id that contains the Datasets
+    ) -> 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
@@ -420,6 +471,12 @@ class Lake(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(
@@ -435,16 +492,21 @@ class Lake(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(
@@ -463,9 +525,7 @@ class Lake(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)
@@ -478,7 +538,7 @@ class Lake(BaseSDK):
 
         raise errors.APIError("Unexpected response received", http_res)
 
-    def delete_cribl_lake_dataset_by_lake_id_and_id(
+    def delete(
         self,
         *,
         lake_id: str,
@@ -487,13 +547,13 @@ class Lake(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.DeleteCriblLakeDatasetByLakeIDAndIDResponse:
-        r"""Delete a Dataset in the specified Lake
+    ) -> models.CountedCriblLakeDataset:
+        r"""Delete a Lake Dataset
 
-        Delete a Dataset in the specified Lake
+        Delete the specified Lake Dataset in the specified Lake
 
-        :param lake_id: lake id that contains the Datasets
-        :param id: dataset id to delete
+        :param lake_id: The <code>id</code> of the Lake that contains the Lake Dataset to delete.
+        :param id: The <code>id</code> of the Lake Dataset to delete.
         :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
@@ -527,16 +587,21 @@ class Lake(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(
@@ -555,9 +620,7 @@ class Lake(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)
@@ -570,7 +633,7 @@ class Lake(BaseSDK):
 
         raise errors.APIError("Unexpected response received", http_res)
 
-    async def delete_cribl_lake_dataset_by_lake_id_and_id_async(
+    async def delete_async(
         self,
         *,
         lake_id: str,
@@ -579,13 +642,13 @@ class Lake(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.DeleteCriblLakeDatasetByLakeIDAndIDResponse:
-        r"""Delete a Dataset in the specified Lake
+    ) -> models.CountedCriblLakeDataset:
+        r"""Delete a Lake Dataset
 
-        Delete a Dataset in the specified Lake
+        Delete the specified Lake Dataset in the specified Lake
 
-        :param lake_id: lake id that contains the Datasets
-        :param id: dataset id to delete
+        :param lake_id: The <code>id</code> of the Lake that contains the Lake Dataset to delete.
+        :param id: The <code>id</code> of the Lake Dataset to delete.
         :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
@@ -619,16 +682,21 @@ class Lake(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(
@@ -647,9 +715,7 @@ class Lake(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)
@@ -662,7 +728,7 @@ class Lake(BaseSDK):
 
         raise errors.APIError("Unexpected response received", http_res)
 
-    def get_cribl_lake_dataset_by_lake_id_and_id(
+    def get(
         self,
         *,
         lake_id: str,
@@ -671,13 +737,13 @@ class Lake(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.GetCriblLakeDatasetByLakeIDAndIDResponse:
-        r"""Get a Dataset in the specified Lake
+    ) -> models.CountedCriblLakeDataset:
+        r"""Get a Lake Dataset
 
-        Get a Dataset in the specified Lake
+        Get the specified Lake Dataset in the specified Lake.
 
-        :param lake_id: lake id that contains the Datasets
-        :param id: dataset id to get
+        :param lake_id: The <code>id</code> of the Lake that contains the Lake Dataset to get.
+        :param id: The <code>id</code> of the Lake Dataset to get.
         :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
@@ -711,16 +777,21 @@ class Lake(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(
@@ -739,9 +810,7 @@ class Lake(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)
@@ -754,7 +823,7 @@ class Lake(BaseSDK):
 
         raise errors.APIError("Unexpected response received", http_res)
 
-    async def get_cribl_lake_dataset_by_lake_id_and_id_async(
+    async def get_async(
         self,
         *,
         lake_id: str,
@@ -763,13 +832,13 @@ class Lake(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.GetCriblLakeDatasetByLakeIDAndIDResponse:
-        r"""Get a Dataset in the specified Lake
+    ) -> models.CountedCriblLakeDataset:
+        r"""Get a Lake Dataset
 
-        Get a Dataset in the specified Lake
+        Get the specified Lake Dataset in the specified Lake.
 
-        :param lake_id: lake id that contains the Datasets
-        :param id: dataset id to get
+        :param lake_id: The <code>id</code> of the Lake that contains the Lake Dataset to get.
+        :param id: The <code>id</code> of the Lake Dataset to get.
         :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
@@ -803,16 +872,21 @@ class Lake(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(
@@ -831,9 +905,7 @@ class Lake(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)
@@ -846,12 +918,11 @@ class Lake(BaseSDK):
 
         raise errors.APIError("Unexpected response received", http_res)
 
-    def update_cribl_lake_dataset_by_lake_id_and_id(
+    def update(
         self,
         *,
         lake_id: str,
         id_param: str,
-        id: str,
         accelerated_fields: Optional[List[str]] = None,
         bucket_name: Optional[str] = None,
         cache_connection: Optional[
@@ -859,8 +930,12 @@ class Lake(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[
@@ -873,14 +948,13 @@ class Lake(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.UpdateCriblLakeDatasetByLakeIDAndIDResponse:
-        r"""Update a Dataset in the specified Lake
+    ) -> models.CountedCriblLakeDataset:
+        r"""Update a Lake Dataset
 
-        Update a Dataset in the specified Lake
+        Update the specified Lake Dataset in the specified Lake.
 
-        :param lake_id: lake id that contains the Datasets
-        :param id_param: dataset id to update
-        :param id:
+        :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 accelerated_fields:
         :param bucket_name:
         :param cache_connection:
@@ -888,6 +962,8 @@ class Lake(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:
@@ -910,7 +986,7 @@ class Lake(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(
@@ -921,6 +997,9 @@ class Lake(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]
@@ -944,22 +1023,27 @@ class Lake(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(
@@ -978,9 +1062,7 @@ class Lake(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)
@@ -993,12 +1075,11 @@ class Lake(BaseSDK):
 
         raise errors.APIError("Unexpected response received", http_res)
 
-    async def update_cribl_lake_dataset_by_lake_id_and_id_async(
+    async def update_async(
         self,
         *,
         lake_id: str,
         id_param: str,
-        id: str,
         accelerated_fields: Optional[List[str]] = None,
         bucket_name: Optional[str] = None,
         cache_connection: Optional[
@@ -1006,8 +1087,12 @@ class Lake(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[
@@ -1020,14 +1105,13 @@ class Lake(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.UpdateCriblLakeDatasetByLakeIDAndIDResponse:
-        r"""Update a Dataset in the specified Lake
+    ) -> models.CountedCriblLakeDataset:
+        r"""Update a Lake Dataset
 
-        Update a Dataset in the specified Lake
+        Update the specified Lake Dataset in the specified Lake.
 
-        :param lake_id: lake id that contains the Datasets
-        :param id_param: dataset id to update
-        :param id:
+        :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 accelerated_fields:
         :param bucket_name:
         :param cache_connection:
@@ -1035,6 +1119,8 @@ class Lake(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:
@@ -1057,7 +1143,7 @@ class Lake(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(
@@ -1068,6 +1154,9 @@ class Lake(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]
@@ -1091,22 +1180,27 @@ class Lake(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(
@@ -1125,9 +1219,7 @@ class Lake(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)