databricks-sdk 0.66.0__py3-none-any.whl → 0.68.0__py3-none-any.whl

databricks/sdk/service/files.py

@@ -6,7 +6,8 @@ import logging
 from dataclasses import dataclass
 from typing import Any, BinaryIO, Dict, Iterator, List, Optional
 
-from ._internal import _escape_multi_segment_path_parameter, _repeated_dict
+from databricks.sdk.service._internal import (
+    _escape_multi_segment_path_parameter, _repeated_dict)
 
 _LOG = logging.getLogger("databricks.sdk")
 

databricks/sdk/service/iam.py

@@ -7,7 +7,8 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional
 
-from ._internal import _enum, _from_dict, _repeated_dict, _repeated_enum
+from databricks.sdk.service._internal import (_enum, _from_dict,
+                                              _repeated_dict, _repeated_enum)
 
 _LOG = logging.getLogger("databricks.sdk")
 
@@ -3583,7 +3584,7 @@ class PermissionsAPI:
 
         :param request_object_type: str
           The type of the request object. Can be one of the following: alerts, alertsv2, authorization,
-          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files,
+          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, genie,
           instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, serving-endpoints, or
           warehouses.
         :param request_object_id: str
@@ -3604,7 +3605,7 @@ class PermissionsAPI:
 
         :param request_object_type: str
           The type of the request object. Can be one of the following: alerts, alertsv2, authorization,
-          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files,
+          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, genie,
           instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, serving-endpoints, or
           warehouses.
         :param request_object_id: str
@@ -3634,7 +3635,7 @@ class PermissionsAPI:
 
         :param request_object_type: str
           The type of the request object. Can be one of the following: alerts, alertsv2, authorization,
-          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files,
+          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, genie,
           instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, serving-endpoints, or
           warehouses.
         :param request_object_id: str
@@ -3668,7 +3669,7 @@ class PermissionsAPI:
 
         :param request_object_type: str
           The type of the request object. Can be one of the following: alerts, alertsv2, authorization,
-          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files,
+          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, genie,
           instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, serving-endpoints, or
           warehouses.
         :param request_object_id: str

databricks/sdk/service/iamv2.py

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Any, Dict, List, Optional
 
-from ._internal import _enum, _from_dict, _repeated_enum
+from databricks.sdk.service._internal import _enum, _from_dict, _repeated_enum
 
 _LOG = logging.getLogger("databricks.sdk")
 

databricks/sdk/service/jobs.py

@@ -10,14 +10,15 @@ from datetime import timedelta
 from enum import Enum
 from typing import Any, Callable, Dict, Iterator, List, Optional
 
+from databricks.sdk.service import compute
+from databricks.sdk.service._internal import (Wait, _enum, _from_dict,
+                                              _repeated_dict)
+
 from ..errors import OperationFailed
-from ._internal import Wait, _enum, _from_dict, _repeated_dict
 
 _LOG = logging.getLogger("databricks.sdk")
 
 
-from databricks.sdk.service import compute
-
 # all definitions in this file are in alphabetical order
 
 
@@ -7318,7 +7319,7 @@ class TableUpdateTriggerConfiguration:
     last time the trigger fired. The minimum allowed value is 60 seconds."""
 
     table_names: Optional[List[str]] = None
-    """A list of Delta tables to monitor for changes. The table name must be in the format
+    """A list of tables to monitor for changes. The table name must be in the format
     `catalog_name.schema_name.table_name`."""
 
     wait_after_last_change_seconds: Optional[int] = None
@@ -8428,11 +8429,7 @@ class JobsAPI:
         }
 
         op_response = self._api.do("POST", "/api/2.2/jobs/runs/cancel", body=body, headers=headers)
-        return Wait(
-            self.wait_get_run_job_terminated_or_skipped,
-            response=CancelRunResponse.from_dict(op_response),
-            run_id=run_id,
-        )
+        return Wait(self.wait_get_run_job_terminated_or_skipped, run_id=run_id)
 
     def cancel_run_and_wait(self, run_id: int, timeout=timedelta(minutes=20)) -> Run:
         return self.cancel_run(run_id=run_id).result(timeout=timeout)

databricks/sdk/service/marketplace.py

@@ -7,7 +7,8 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional
 
-from ._internal import _enum, _from_dict, _repeated_dict, _repeated_enum
+from databricks.sdk.service._internal import (_enum, _from_dict,
+                                              _repeated_dict, _repeated_enum)
 
 _LOG = logging.getLogger("databricks.sdk")
 
@@ -44,6 +45,7 @@ class AssetType(Enum):
     ASSET_TYPE_APP = "ASSET_TYPE_APP"
     ASSET_TYPE_DATA_TABLE = "ASSET_TYPE_DATA_TABLE"
     ASSET_TYPE_GIT_REPO = "ASSET_TYPE_GIT_REPO"
+    ASSET_TYPE_MCP = "ASSET_TYPE_MCP"
     ASSET_TYPE_MEDIA = "ASSET_TYPE_MEDIA"
     ASSET_TYPE_MODEL = "ASSET_TYPE_MODEL"
     ASSET_TYPE_NOTEBOOK = "ASSET_TYPE_NOTEBOOK"

databricks/sdk/service/ml.py

@@ -10,8 +10,10 @@ from datetime import timedelta
 from enum import Enum
 from typing import Any, Callable, Dict, Iterator, List, Optional
 
+from databricks.sdk.service._internal import (Wait, _enum, _from_dict,
+                                              _repeated_dict, _repeated_enum)
+
 from ..errors import OperationFailed
-from ._internal import Wait, _enum, _from_dict, _repeated_dict, _repeated_enum
 
 _LOG = logging.getLogger("databricks.sdk")
 

databricks/sdk/service/oauth2.py

@@ -6,7 +6,7 @@ import logging
 from dataclasses import dataclass
 from typing import Any, Dict, Iterator, List, Optional
 
-from ._internal import _from_dict, _repeated_dict
+from databricks.sdk.service._internal import _from_dict, _repeated_dict
 
 _LOG = logging.getLogger("databricks.sdk")
 

databricks/sdk/service/pipelines.py

@@ -10,14 +10,15 @@ from datetime import timedelta
 from enum import Enum
 from typing import Any, Callable, Dict, Iterator, List, Optional
 
+from databricks.sdk.service import compute
+from databricks.sdk.service._internal import (Wait, _enum, _from_dict,
+                                              _repeated_dict, _repeated_enum)
+
 from ..errors import OperationFailed
-from ._internal import Wait, _enum, _from_dict, _repeated_dict, _repeated_enum
 
 _LOG = logging.getLogger("databricks.sdk")
 
 
-from databricks.sdk.service import compute
-
 # all definitions in this file are in alphabetical order
 
 
@@ -3775,9 +3776,7 @@ class PipelinesAPI:
         }
 
         op_response = self._api.do("POST", f"/api/2.0/pipelines/{pipeline_id}/stop", headers=headers)
-        return Wait(
-            self.wait_get_pipeline_idle, response=StopPipelineResponse.from_dict(op_response), pipeline_id=pipeline_id
-        )
+        return Wait(self.wait_get_pipeline_idle, pipeline_id=pipeline_id)
 
     def stop_and_wait(self, pipeline_id: str, timeout=timedelta(minutes=20)) -> GetPipelineResponse:
         return self.stop(pipeline_id=pipeline_id).result(timeout=timeout)