databricks-sdk 0.25.0__tar.gz → 0.26.0__tar.gz

{databricks-sdk-0.25.0/databricks_sdk.egg-info → databricks-sdk-0.26.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: databricks-sdk
-Version: 0.25.0
+Version: 0.26.0
 Summary: Databricks SDK for Python (Beta)
 Home-page: https://databricks-sdk-py.readthedocs.io
 Author: Serge Smertin

{databricks-sdk-0.25.0 → databricks-sdk-0.26.0}/databricks/sdk/mixins/compute.py

@@ -86,6 +86,7 @@ class ClustersExt(compute.ClustersAPI):
         :param beta: bool
         :param latest: bool
         :param ml: bool
+        :param genomics: bool
         :param gpu: bool
         :param scala: str
         :param spark_version: str
@@ -100,7 +101,7 @@ class ClustersExt(compute.ClustersAPI):
         for version in sv.versions:
             if "-scala" + scala not in version.key:
                 continue
-            matches = ((not "apache-spark-" in version.key) and (("-ml-" in version.key) == ml)
+            matches = (("apache-spark-" not in version.key) and (("-ml-" in version.key) == ml)
                        and (("-hls-" in version.key) == genomics) and (("-gpu-" in version.key) == gpu)
                        and (("-photon-" in version.key) == photon)
                        and (("-aarch64-" in version.key) == graviton) and (("Beta" in version.name) == beta))
@@ -137,7 +138,7 @@ class ClustersExt(compute.ClustersAPI):
             return False
         val = compute.CloudProviderNodeStatus
         for st in nt.node_info.status:
-            if st in (val.NotAvailableInRegion, val.NotEnabledOnSubscription):
+            if st in (val.NOT_AVAILABLE_IN_REGION, val.NOT_ENABLED_ON_SUBSCRIPTION):
                 return True
         return False
 
@@ -168,6 +169,8 @@ class ClustersExt(compute.ClustersAPI):
         :param photon_driver_capable: bool
         :param graviton: bool
         :param is_io_cache_enabled: bool
+        :param support_port_forwarding: bool
+        :param fleet: bool
 
         :returns: `node_type` compatible string
         """

{databricks-sdk-0.25.0 → databricks-sdk-0.26.0}/databricks/sdk/service/catalog.py

@@ -3850,6 +3850,7 @@ class PrimaryKeyConstraint:
 
 class Privilege(Enum):
 
+    ACCESS = 'ACCESS'
     ALL_PRIVILEGES = 'ALL_PRIVILEGES'
     APPLY_TAG = 'APPLY_TAG'
     CREATE = 'CREATE'
@@ -3866,6 +3867,7 @@ class Privilege(Enum):
     CREATE_PROVIDER = 'CREATE_PROVIDER'
     CREATE_RECIPIENT = 'CREATE_RECIPIENT'
     CREATE_SCHEMA = 'CREATE_SCHEMA'
+    CREATE_SERVICE_CREDENTIAL = 'CREATE_SERVICE_CREDENTIAL'
     CREATE_SHARE = 'CREATE_SHARE'
     CREATE_STORAGE_CREDENTIAL = 'CREATE_STORAGE_CREDENTIAL'
     CREATE_TABLE = 'CREATE_TABLE'
@@ -4114,6 +4116,9 @@ class SchemaInfo:
     properties: Optional[Dict[str, str]] = None
     """A map of key-value properties attached to the securable."""
 
+    schema_id: Optional[str] = None
+    """The unique identifier of the schema."""
+
     storage_location: Optional[str] = None
     """Storage location for managed tables within schema."""
 
@@ -4146,6 +4151,7 @@ class SchemaInfo:
         if self.name is not None: body['name'] = self.name
         if self.owner is not None: body['owner'] = self.owner
         if self.properties: body['properties'] = self.properties
+        if self.schema_id is not None: body['schema_id'] = self.schema_id
         if self.storage_location is not None: body['storage_location'] = self.storage_location
         if self.storage_root is not None: body['storage_root'] = self.storage_root
         if self.updated_at is not None: body['updated_at'] = self.updated_at
@@ -4170,6 +4176,7 @@ class SchemaInfo:
                    name=d.get('name', None),
                    owner=d.get('owner', None),
                    properties=d.get('properties', None),
+                   schema_id=d.get('schema_id', None),
                    storage_location=d.get('storage_location', None),
                    storage_root=d.get('storage_root', None),
                    updated_at=d.get('updated_at', None),
@@ -4555,7 +4562,7 @@ class TableInfo:
     """List of table constraints. Note: this field is not set in the output of the __listTables__ API."""
 
     table_id: Optional[str] = None
-    """Name of table, relative to parent schema."""
+    """The unique identifier of the table."""
 
     table_type: Optional[TableType] = None
 
@@ -5507,63 +5514,32 @@ class ValidateStorageCredentialResponse:
 
 @dataclass
 class ValidationResult:
-    aws_operation: Optional[ValidationResultAwsOperation] = None
-    """The operation tested."""
-
-    azure_operation: Optional[ValidationResultAzureOperation] = None
-    """The operation tested."""
-
-    gcp_operation: Optional[ValidationResultGcpOperation] = None
-    """The operation tested."""
-
     message: Optional[str] = None
     """Error message would exist when the result does not equal to **PASS**."""
 
+    operation: Optional[ValidationResultOperation] = None
+    """The operation tested."""
+
     result: Optional[ValidationResultResult] = None
     """The results of the tested operation."""
 
     def as_dict(self) -> dict:
         """Serializes the ValidationResult into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.aws_operation is not None: body['aws_operation'] = self.aws_operation.value
-        if self.azure_operation is not None: body['azure_operation'] = self.azure_operation.value
-        if self.gcp_operation is not None: body['gcp_operation'] = self.gcp_operation.value
         if self.message is not None: body['message'] = self.message
+        if self.operation is not None: body['operation'] = self.operation.value
         if self.result is not None: body['result'] = self.result.value
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, any]) -> ValidationResult:
         """Deserializes the ValidationResult from a dictionary."""
-        return cls(aws_operation=_enum(d, 'aws_operation', ValidationResultAwsOperation),
-                   azure_operation=_enum(d, 'azure_operation', ValidationResultAzureOperation),
-                   gcp_operation=_enum(d, 'gcp_operation', ValidationResultGcpOperation),
-                   message=d.get('message', None),
+        return cls(message=d.get('message', None),
+                   operation=_enum(d, 'operation', ValidationResultOperation),
                    result=_enum(d, 'result', ValidationResultResult))
 
 
-class ValidationResultAwsOperation(Enum):
-    """The operation tested."""
-
-    DELETE = 'DELETE'
-    LIST = 'LIST'
-    PATH_EXISTS = 'PATH_EXISTS'
-    READ = 'READ'
-    WRITE = 'WRITE'
-
-
-class ValidationResultAzureOperation(Enum):
-    """The operation tested."""
-
-    DELETE = 'DELETE'
-    HIERARCHICAL_NAMESPACE_ENABLED = 'HIERARCHICAL_NAMESPACE_ENABLED'
-    LIST = 'LIST'
-    PATH_EXISTS = 'PATH_EXISTS'
-    READ = 'READ'
-    WRITE = 'WRITE'
-
-
-class ValidationResultGcpOperation(Enum):
+class ValidationResultOperation(Enum):
     """The operation tested."""
 
     DELETE = 'DELETE'

{databricks-sdk-0.25.0 → databricks-sdk-0.26.0}/databricks/sdk/service/compute.py

@@ -1637,6 +1637,28 @@ class ClusterSpec:
                    workload_type=_from_dict(d, 'workload_type', WorkloadType))
 
 
+@dataclass
+class ClusterStatusResponse:
+    cluster_id: Optional[str] = None
+    """Unique identifier for the cluster."""
+
+    library_statuses: Optional[List[LibraryFullStatus]] = None
+    """Status of all libraries on the cluster."""
+
+    def as_dict(self) -> dict:
+        """Serializes the ClusterStatusResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.cluster_id is not None: body['cluster_id'] = self.cluster_id
+        if self.library_statuses: body['library_statuses'] = [v.as_dict() for v in self.library_statuses]
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> ClusterStatusResponse:
+        """Deserializes the ClusterStatusResponse from a dictionary."""
+        return cls(cluster_id=d.get('cluster_id', None),
+                   library_statuses=_repeated_dict(d, 'library_statuses', LibraryFullStatus))
+
+
 @dataclass
 class Command:
     cluster_id: Optional[str] = None
@@ -2988,8 +3010,9 @@ class Environment:
     supported. Next ID: 5"""
 
     client: str
-    """* User-friendly name for the client version: “client”: “1” The version is a string,
-    consisting of the major client version"""
+    """Client version used by the environment The client is the user-facing environment of the runtime.
+    Each client comes with a specific set of pre-installed libraries. The version is a string,
+    consisting of the major client version."""
 
     dependencies: Optional[List[str]] = None
     """List of pip dependencies, as supported by the version of pip in this environment. Each
@@ -4223,7 +4246,14 @@ class InstancePoolGcpAttributes:
     be of a form like "us-west1-a". The provided availability zone must be in the same region as the
     Databricks workspace. For example, "us-west1-a" is not a valid zone id if the Databricks
     workspace resides in the "us-east1" region. This is an optional field at instance pool creation,
-    and if not specified, a default zone will be used."""
+    and if not specified, a default zone will be used.
+    
+    This field can be one of the following: - "HA" => High availability, spread nodes across
+    availability zones for a Databricks deployment region - A GCP availability zone => Pick One of
+    the available zones for (machine type + region) from
+    https://cloud.google.com/compute/docs/regions-zones (e.g. "us-west1-a").
+    
+    If empty, Databricks picks an availability zone to schedule the cluster on."""
 
     def as_dict(self) -> dict:
         """Serializes the InstancePoolGcpAttributes into a dictionary suitable for use as a JSON request body."""
@@ -4453,16 +4483,18 @@ class Library:
     """Specification of a CRAN library to be installed as part of the library"""
 
     egg: Optional[str] = None
-    """URI of the egg to be installed. Currently only DBFS and S3 URIs are supported. For example: `{
-    "egg": "dbfs:/my/egg" }` or `{ "egg": "s3://my-bucket/egg" }`. If S3 is used, please make sure
-    the cluster has read access on the library. You may need to launch the cluster with an IAM role
-    to access the S3 URI."""
+    """URI of the egg library to install. Supported URIs include Workspace paths, Unity Catalog Volumes
+    paths, and S3 URIs. For example: `{ "egg": "/Workspace/path/to/library.egg" }`, `{ "egg" :
+    "/Volumes/path/to/library.egg" }` or `{ "egg": "s3://my-bucket/library.egg" }`. If S3 is used,
+    please make sure the cluster has read access on the library. You may need to launch the cluster
+    with an IAM role to access the S3 URI."""
 
     jar: Optional[str] = None
-    """URI of the jar to be installed. Currently only DBFS and S3 URIs are supported. For example: `{
-    "jar": "dbfs:/mnt/databricks/library.jar" }` or `{ "jar": "s3://my-bucket/library.jar" }`. If S3
-    is used, please make sure the cluster has read access on the library. You may need to launch the
-    cluster with an IAM role to access the S3 URI."""
+    """URI of the JAR library to install. Supported URIs include Workspace paths, Unity Catalog Volumes
+    paths, and S3 URIs. For example: `{ "jar": "/Workspace/path/to/library.jar" }`, `{ "jar" :
+    "/Volumes/path/to/library.jar" }` or `{ "jar": "s3://my-bucket/library.jar" }`. If S3 is used,
+    please make sure the cluster has read access on the library. You may need to launch the cluster
+    with an IAM role to access the S3 URI."""
 
     maven: Optional[MavenLibrary] = None
     """Specification of a maven library to be installed. For example: `{ "coordinates":
@@ -4471,10 +4503,17 @@ class Library:
     pypi: Optional[PythonPyPiLibrary] = None
     """Specification of a PyPi library to be installed. For example: `{ "package": "simplejson" }`"""
 
+    requirements: Optional[str] = None
+    """URI of the requirements.txt file to install. Only Workspace paths and Unity Catalog Volumes
+    paths are supported. For example: `{ "requirements": "/Workspace/path/to/requirements.txt" }` or
+    `{ "requirements" : "/Volumes/path/to/requirements.txt" }`"""
+
     whl: Optional[str] = None
-    """URI of the wheel to be installed. For example: `{ "whl": "dbfs:/my/whl" }` or `{ "whl":
-    "s3://my-bucket/whl" }`. If S3 is used, please make sure the cluster has read access on the
-    library. You may need to launch the cluster with an IAM role to access the S3 URI."""
+    """URI of the wheel library to install. Supported URIs include Workspace paths, Unity Catalog
+    Volumes paths, and S3 URIs. For example: `{ "whl": "/Workspace/path/to/library.whl" }`, `{ "whl"
+    : "/Volumes/path/to/library.whl" }` or `{ "whl": "s3://my-bucket/library.whl" }`. If S3 is used,
+    please make sure the cluster has read access on the library. You may need to launch the cluster
+    with an IAM role to access the S3 URI."""
 
     def as_dict(self) -> dict:
         """Serializes the Library into a dictionary suitable for use as a JSON request body."""
@@ -4484,6 +4523,7 @@ class Library:
         if self.jar is not None: body['jar'] = self.jar
         if self.maven: body['maven'] = self.maven.as_dict()
         if self.pypi: body['pypi'] = self.pypi.as_dict()
+        if self.requirements is not None: body['requirements'] = self.requirements
         if self.whl is not None: body['whl'] = self.whl
         return body
 
@@ -4495,11 +4535,14 @@ class Library:
                    jar=d.get('jar', None),
                    maven=_from_dict(d, 'maven', MavenLibrary),
                    pypi=_from_dict(d, 'pypi', PythonPyPiLibrary),
+                   requirements=d.get('requirements', None),
                    whl=d.get('whl', None))
 
 
 @dataclass
 class LibraryFullStatus:
+    """The status of the library on a specific cluster."""
+
     is_library_for_all_clusters: Optional[bool] = None
     """Whether the library was set to be installed on all clusters via the libraries UI."""
 
@@ -4509,7 +4552,7 @@ class LibraryFullStatus:
     messages: Optional[List[str]] = None
     """All the info and warning messages that have occurred so far for this library."""
 
-    status: Optional[LibraryFullStatusStatus] = None
+    status: Optional[LibraryInstallStatus] = None
     """Status of installing the library on the cluster."""
 
     def as_dict(self) -> dict:
@@ -4528,17 +4571,18 @@ class LibraryFullStatus:
         return cls(is_library_for_all_clusters=d.get('is_library_for_all_clusters', None),
                    library=_from_dict(d, 'library', Library),
                    messages=d.get('messages', None),
-                   status=_enum(d, 'status', LibraryFullStatusStatus))
+                   status=_enum(d, 'status', LibraryInstallStatus))
 
 
-class LibraryFullStatusStatus(Enum):
-    """Status of installing the library on the cluster."""
+class LibraryInstallStatus(Enum):
+    """The status of a library on a specific cluster."""
 
     FAILED = 'FAILED'
     INSTALLED = 'INSTALLED'
     INSTALLING = 'INSTALLING'
     PENDING = 'PENDING'
     RESOLVING = 'RESOLVING'
+    RESTORED = 'RESTORED'
     SKIPPED = 'SKIPPED'
     UNINSTALL_ON_RESTART = 'UNINSTALL_ON_RESTART'
 
@@ -8070,16 +8114,13 @@ class LibrariesAPI:
     cluster.
     
     To make third-party or custom code available to notebooks and jobs running on your clusters, you can
-    install a library. Libraries can be written in Python, Java, Scala, and R. You can upload Java, Scala, and
-    Python libraries and point to external packages in PyPI, Maven, and CRAN repositories.
+    install a library. Libraries can be written in Python, Java, Scala, and R. You can upload Python, Java,
+    Scala and R libraries and point to external packages in PyPI, Maven, and CRAN repositories.
     
     Cluster libraries can be used by all notebooks running on a cluster. You can install a cluster library
     directly from a public repository such as PyPI or Maven, using a previously installed workspace library,
     or using an init script.
     
-    When you install a library on a cluster, a notebook already attached to that cluster will not immediately
-    see the new library. You must first detach and then reattach the notebook to the cluster.
-    
     When you uninstall a library from a cluster, the library is removed only when you restart the cluster.
     Until you restart the cluster, the status of the uninstalled library appears as Uninstall pending restart."""
 
@@ -8089,9 +8130,8 @@ class LibrariesAPI:
     def all_cluster_statuses(self) -> ListAllClusterLibraryStatusesResponse:
         """Get all statuses.
         
-        Get the status of all libraries on all clusters. A status will be available for all libraries
-        installed on this cluster via the API or the libraries UI as well as libraries set to be installed on
-        all clusters via the libraries UI.
+        Get the status of all libraries on all clusters. A status is returned for all libraries installed on
+        this cluster via the API or the libraries UI.
         
         :returns: :class:`ListAllClusterLibraryStatusesResponse`
         """
@@ -8104,18 +8144,11 @@ class LibrariesAPI:
     def cluster_status(self, cluster_id: str) -> Iterator[LibraryFullStatus]:
         """Get status.
         
-        Get the status of libraries on a cluster. A status will be available for all libraries installed on
-        this cluster via the API or the libraries UI as well as libraries set to be installed on all clusters
-        via the libraries UI. The order of returned libraries will be as follows.
-        
-        1. Libraries set to be installed on this cluster will be returned first. Within this group, the final
-        order will be order in which the libraries were added to the cluster.
-        
-        2. Libraries set to be installed on all clusters are returned next. Within this group there is no
-        order guarantee.
-        
-        3. Libraries that were previously requested on this cluster or on all clusters, but now marked for
-        removal. Within this group there is no order guarantee.
+        Get the status of libraries on a cluster. A status is returned for all libraries installed on this
+        cluster via the API or the libraries UI. The order of returned libraries is as follows: 1. Libraries
+        set to be installed on this cluster, in the order that the libraries were added to the cluster, are
+        returned first. 2. Libraries that were previously requested to be installed on this cluster or, but
+        are now marked for removal, in no particular order, are returned last.
         
         :param cluster_id: str
           Unique identifier of the cluster whose status should be retrieved.
@@ -8128,17 +8161,14 @@ class LibrariesAPI:
         headers = {'Accept': 'application/json', }
 
         json = self._api.do('GET', '/api/2.0/libraries/cluster-status', query=query, headers=headers)
-        parsed = ClusterLibraryStatuses.from_dict(json).library_statuses
+        parsed = ClusterStatusResponse.from_dict(json).library_statuses
         return parsed if parsed is not None else []
 
     def install(self, cluster_id: str, libraries: List[Library]):
         """Add a library.
         
-        Add libraries to be installed on a cluster. The installation is asynchronous; it happens in the
-        background after the completion of this request.
-        
-        **Note**: The actual set of libraries to be installed on a cluster is the union of the libraries
-        specified via this method and the libraries set to be installed on all clusters via the libraries UI.
+        Add libraries to install on a cluster. The installation is asynchronous; it happens in the background
+        after the completion of this request.
         
         :param cluster_id: str
           Unique identifier for the cluster on which to install these libraries.
@@ -8157,9 +8187,8 @@ class LibrariesAPI:
     def uninstall(self, cluster_id: str, libraries: List[Library]):
         """Uninstall libraries.
         
-        Set libraries to be uninstalled on a cluster. The libraries won't be uninstalled until the cluster is
-        restarted. Uninstalling libraries that are not installed on the cluster will have no impact but is not
-        an error.
+        Set libraries to uninstall from a cluster. The libraries won't be uninstalled until the cluster is
+        restarted. A request to uninstall a library that is not currently installed is ignored.
         
         :param cluster_id: str
           Unique identifier for the cluster on which to uninstall these libraries.

{databricks-sdk-0.25.0 → databricks-sdk-0.26.0}/databricks/sdk/service/jobs.py

@@ -1963,12 +1963,20 @@ class NotebookTask:
     `git_source` is defined and `WORKSPACE` otherwise. * `WORKSPACE`: Notebook is located in
     Databricks workspace. * `GIT`: Notebook is located in cloud Git provider."""
 
+    warehouse_id: Optional[str] = None
+    """Optional `warehouse_id` to run the notebook on a SQL warehouse. Classic SQL warehouses are NOT
+    supported, please use serverless or pro SQL warehouses.
+    
+    Note that SQL warehouses only support SQL cells; if the notebook contains non-SQL cells, the run
+    will fail."""
+
     def as_dict(self) -> dict:
         """Serializes the NotebookTask into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.base_parameters: body['base_parameters'] = self.base_parameters
         if self.notebook_path is not None: body['notebook_path'] = self.notebook_path
         if self.source is not None: body['source'] = self.source.value
+        if self.warehouse_id is not None: body['warehouse_id'] = self.warehouse_id
         return body
 
     @classmethod
@@ -1976,7 +1984,8 @@ class NotebookTask:
         """Deserializes the NotebookTask from a dictionary."""
         return cls(base_parameters=d.get('base_parameters', None),
                    notebook_path=d.get('notebook_path', None),
-                   source=_enum(d, 'source', Source))
+                   source=_enum(d, 'source', Source),
+                   warehouse_id=d.get('warehouse_id', None))
 
 
 class PauseStatus(Enum):
@@ -3953,9 +3962,7 @@ class SqlTask:
     """If dashboard, indicates that this job must refresh a SQL dashboard."""
 
     file: Optional[SqlTaskFile] = None
-    """If file, indicates that this job runs a SQL file in a remote Git repository. Only one SQL
-    statement is supported in a file. Multiple SQL statements separated by semicolons (;) are not
-    permitted."""
+    """If file, indicates that this job runs a SQL file in a remote Git repository."""
 
     parameters: Optional[Dict[str, str]] = None
     """Parameters to be used for each run of this job. The SQL alert task does not support custom
@@ -4175,6 +4182,10 @@ class SubmitRun:
     queue: Optional[QueueSettings] = None
     """The queue settings of the one-time run."""
 
+    run_as: Optional[JobRunAs] = None
+    """Specifies the user or service principal that the job runs as. If not specified, the job runs as
+    the user who submits the request."""
+
     run_job_task: Optional[RunJobTask] = None
     """If run_job_task, indicates that this task must execute another job."""
 
@@ -4231,6 +4242,7 @@ class SubmitRun:
         if self.pipeline_task: body['pipeline_task'] = self.pipeline_task.as_dict()
         if self.python_wheel_task: body['python_wheel_task'] = self.python_wheel_task.as_dict()
         if self.queue: body['queue'] = self.queue.as_dict()
+        if self.run_as: body['run_as'] = self.run_as.as_dict()
         if self.run_job_task: body['run_job_task'] = self.run_job_task.as_dict()
         if self.run_name is not None: body['run_name'] = self.run_name
         if self.spark_jar_task: body['spark_jar_task'] = self.spark_jar_task.as_dict()
@@ -4257,6 +4269,7 @@ class SubmitRun:
                    pipeline_task=_from_dict(d, 'pipeline_task', PipelineTask),
                    python_wheel_task=_from_dict(d, 'python_wheel_task', PythonWheelTask),
                    queue=_from_dict(d, 'queue', QueueSettings),
+                   run_as=_from_dict(d, 'run_as', JobRunAs),
                    run_job_task=_from_dict(d, 'run_job_task', RunJobTask),
                    run_name=d.get('run_name', None),
                    spark_jar_task=_from_dict(d, 'spark_jar_task', SparkJarTask),
@@ -5890,6 +5903,7 @@ class JobsAPI:
                pipeline_task: Optional[PipelineTask] = None,
                python_wheel_task: Optional[PythonWheelTask] = None,
                queue: Optional[QueueSettings] = None,
+               run_as: Optional[JobRunAs] = None,
                run_job_task: Optional[RunJobTask] = None,
                run_name: Optional[str] = None,
                spark_jar_task: Optional[SparkJarTask] = None,
@@ -5951,6 +5965,9 @@ class JobsAPI:
           If python_wheel_task, indicates that this job must execute a PythonWheel.
         :param queue: :class:`QueueSettings` (optional)
           The queue settings of the one-time run.
+        :param run_as: :class:`JobRunAs` (optional)
+          Specifies the user or service principal that the job runs as. If not specified, the job runs as the
+          user who submits the request.
         :param run_job_task: :class:`RunJobTask` (optional)
           If run_job_task, indicates that this task must execute another job.
         :param run_name: str (optional)
@@ -6001,6 +6018,7 @@ class JobsAPI:
         if pipeline_task is not None: body['pipeline_task'] = pipeline_task.as_dict()
         if python_wheel_task is not None: body['python_wheel_task'] = python_wheel_task.as_dict()
         if queue is not None: body['queue'] = queue.as_dict()
+        if run_as is not None: body['run_as'] = run_as.as_dict()
         if run_job_task is not None: body['run_job_task'] = run_job_task.as_dict()
         if run_name is not None: body['run_name'] = run_name
         if spark_jar_task is not None: body['spark_jar_task'] = spark_jar_task.as_dict()
@@ -6032,6 +6050,7 @@ class JobsAPI:
         pipeline_task: Optional[PipelineTask] = None,
         python_wheel_task: Optional[PythonWheelTask] = None,
         queue: Optional[QueueSettings] = None,
+        run_as: Optional[JobRunAs] = None,
         run_job_task: Optional[RunJobTask] = None,
         run_name: Optional[str] = None,
         spark_jar_task: Optional[SparkJarTask] = None,
@@ -6054,6 +6073,7 @@ class JobsAPI:
                            pipeline_task=pipeline_task,
                            python_wheel_task=python_wheel_task,
                            queue=queue,
+                           run_as=run_as,
                            run_job_task=run_job_task,
                            run_name=run_name,
                            spark_jar_task=spark_jar_task,

{databricks-sdk-0.25.0 → databricks-sdk-0.26.0}/databricks/sdk/service/pipelines.py

@@ -43,6 +43,9 @@ class CreatePipeline:
     continuous: Optional[bool] = None
     """Whether the pipeline is continuous or triggered. This replaces `trigger`."""
 
+    deployment: Optional[PipelineDeployment] = None
+    """Deployment type of this pipeline."""
+
     development: Optional[bool] = None
     """Whether the pipeline is in Development mode. Defaults to false."""
 
@@ -92,6 +95,7 @@ class CreatePipeline:
         if self.clusters: body['clusters'] = [v.as_dict() for v in self.clusters]
         if self.configuration: body['configuration'] = self.configuration
         if self.continuous is not None: body['continuous'] = self.continuous
+        if self.deployment: body['deployment'] = self.deployment.as_dict()
         if self.development is not None: body['development'] = self.development
         if self.dry_run is not None: body['dry_run'] = self.dry_run
         if self.edition is not None: body['edition'] = self.edition
@@ -116,6 +120,7 @@ class CreatePipeline:
                    clusters=_repeated_dict(d, 'clusters', PipelineCluster),
                    configuration=d.get('configuration', None),
                    continuous=d.get('continuous', None),
+                   deployment=_from_dict(d, 'deployment', PipelineDeployment),
                    development=d.get('development', None),
                    dry_run=d.get('dry_run', None),
                    edition=d.get('edition', None),
@@ -208,6 +213,13 @@ class DeletePipelineResponse:
         return cls()
 
 
+class DeploymentKind(Enum):
+    """The deployment method that manages the pipeline: - BUNDLE: The pipeline is managed by a
+    Databricks Asset Bundle."""
+
+    BUNDLE = 'BUNDLE'
+
+
 @dataclass
 class EditPipeline:
     allow_duplicate_names: Optional[bool] = None
@@ -231,6 +243,9 @@ class EditPipeline:
     continuous: Optional[bool] = None
     """Whether the pipeline is continuous or triggered. This replaces `trigger`."""
 
+    deployment: Optional[PipelineDeployment] = None
+    """Deployment type of this pipeline."""
+
     development: Optional[bool] = None
     """Whether the pipeline is in Development mode. Defaults to false."""
 
@@ -285,6 +300,7 @@ class EditPipeline:
         if self.clusters: body['clusters'] = [v.as_dict() for v in self.clusters]
         if self.configuration: body['configuration'] = self.configuration
         if self.continuous is not None: body['continuous'] = self.continuous
+        if self.deployment: body['deployment'] = self.deployment.as_dict()
         if self.development is not None: body['development'] = self.development
         if self.edition is not None: body['edition'] = self.edition
         if self.expected_last_modified is not None:
@@ -311,6 +327,7 @@ class EditPipeline:
                    clusters=_repeated_dict(d, 'clusters', PipelineCluster),
                    configuration=d.get('configuration', None),
                    continuous=d.get('continuous', None),
+                   deployment=_from_dict(d, 'deployment', PipelineDeployment),
                    development=d.get('development', None),
                    edition=d.get('edition', None),
                    expected_last_modified=d.get('expected_last_modified', None),
@@ -1017,6 +1034,28 @@ class PipelineClusterAutoscaleMode(Enum):
     LEGACY = 'LEGACY'
 
 
+@dataclass
+class PipelineDeployment:
+    kind: Optional[DeploymentKind] = None
+    """The deployment method that manages the pipeline."""
+
+    metadata_file_path: Optional[str] = None
+    """The path to the file containing metadata about the deployment."""
+
+    def as_dict(self) -> dict:
+        """Serializes the PipelineDeployment into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.kind is not None: body['kind'] = self.kind.value
+        if self.metadata_file_path is not None: body['metadata_file_path'] = self.metadata_file_path
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> PipelineDeployment:
+        """Deserializes the PipelineDeployment from a dictionary."""
+        return cls(kind=_enum(d, 'kind', DeploymentKind),
+                   metadata_file_path=d.get('metadata_file_path', None))
+
+
 @dataclass
 class PipelineEvent:
     error: Optional[ErrorDetail] = None
@@ -1229,6 +1268,9 @@ class PipelineSpec:
     continuous: Optional[bool] = None
     """Whether the pipeline is continuous or triggered. This replaces `trigger`."""
 
+    deployment: Optional[PipelineDeployment] = None
+    """Deployment type of this pipeline."""
+
     development: Optional[bool] = None
     """Whether the pipeline is in Development mode. Defaults to false."""
 
@@ -1275,6 +1317,7 @@ class PipelineSpec:
         if self.clusters: body['clusters'] = [v.as_dict() for v in self.clusters]
         if self.configuration: body['configuration'] = self.configuration
         if self.continuous is not None: body['continuous'] = self.continuous
+        if self.deployment: body['deployment'] = self.deployment.as_dict()
         if self.development is not None: body['development'] = self.development
         if self.edition is not None: body['edition'] = self.edition
         if self.filters: body['filters'] = self.filters.as_dict()
@@ -1297,6 +1340,7 @@ class PipelineSpec:
                    clusters=_repeated_dict(d, 'clusters', PipelineCluster),
                    configuration=d.get('configuration', None),
                    continuous=d.get('continuous', None),
+                   deployment=_from_dict(d, 'deployment', PipelineDeployment),
                    development=d.get('development', None),
                    edition=d.get('edition', None),
                    filters=_from_dict(d, 'filters', Filters),
@@ -1784,6 +1828,7 @@ class PipelinesAPI:
                clusters: Optional[List[PipelineCluster]] = None,
                configuration: Optional[Dict[str, str]] = None,
                continuous: Optional[bool] = None,
+               deployment: Optional[PipelineDeployment] = None,
                development: Optional[bool] = None,
                dry_run: Optional[bool] = None,
                edition: Optional[str] = None,
@@ -1816,6 +1861,8 @@ class PipelinesAPI:
           String-String configuration for this pipeline execution.
         :param continuous: bool (optional)
           Whether the pipeline is continuous or triggered. This replaces `trigger`.
+        :param deployment: :class:`PipelineDeployment` (optional)
+          Deployment type of this pipeline.
         :param development: bool (optional)
           Whether the pipeline is in Development mode. Defaults to false.
         :param dry_run: bool (optional)
@@ -1852,6 +1899,7 @@ class PipelinesAPI:
         if clusters is not None: body['clusters'] = [v.as_dict() for v in clusters]
         if configuration is not None: body['configuration'] = configuration
         if continuous is not None: body['continuous'] = continuous
+        if deployment is not None: body['deployment'] = deployment.as_dict()
         if development is not None: body['development'] = development
         if dry_run is not None: body['dry_run'] = dry_run
         if edition is not None: body['edition'] = edition
@@ -2179,6 +2227,7 @@ class PipelinesAPI:
                clusters: Optional[List[PipelineCluster]] = None,
                configuration: Optional[Dict[str, str]] = None,
                continuous: Optional[bool] = None,
+               deployment: Optional[PipelineDeployment] = None,
                development: Optional[bool] = None,
                edition: Optional[str] = None,
                expected_last_modified: Optional[int] = None,
@@ -2212,6 +2261,8 @@ class PipelinesAPI:
           String-String configuration for this pipeline execution.
         :param continuous: bool (optional)
           Whether the pipeline is continuous or triggered. This replaces `trigger`.
+        :param deployment: :class:`PipelineDeployment` (optional)
+          Deployment type of this pipeline.
         :param development: bool (optional)
           Whether the pipeline is in Development mode. Defaults to false.
         :param edition: str (optional)
@@ -2250,6 +2301,7 @@ class PipelinesAPI:
         if clusters is not None: body['clusters'] = [v.as_dict() for v in clusters]
         if configuration is not None: body['configuration'] = configuration
         if continuous is not None: body['continuous'] = continuous
+        if deployment is not None: body['deployment'] = deployment.as_dict()
         if development is not None: body['development'] = development
         if edition is not None: body['edition'] = edition
         if expected_last_modified is not None: body['expected_last_modified'] = expected_last_modified

{databricks-sdk-0.25.0 → databricks-sdk-0.26.0}/databricks/sdk/service/sharing.py

@@ -758,6 +758,7 @@ class PartitionValueOp(Enum):
 
 class Privilege(Enum):
 
+    ACCESS = 'ACCESS'
     ALL_PRIVILEGES = 'ALL_PRIVILEGES'
     APPLY_TAG = 'APPLY_TAG'
     CREATE = 'CREATE'
@@ -774,6 +775,7 @@ class Privilege(Enum):
     CREATE_PROVIDER = 'CREATE_PROVIDER'
     CREATE_RECIPIENT = 'CREATE_RECIPIENT'
     CREATE_SCHEMA = 'CREATE_SCHEMA'
+    CREATE_SERVICE_CREDENTIAL = 'CREATE_SERVICE_CREDENTIAL'
     CREATE_SHARE = 'CREATE_SHARE'
     CREATE_STORAGE_CREDENTIAL = 'CREATE_STORAGE_CREDENTIAL'
     CREATE_TABLE = 'CREATE_TABLE'

databricks-sdk-0.26.0/databricks/sdk/version.py

@@ -0,0 +1 @@
+__version__ = '0.26.0'

{databricks-sdk-0.25.0 → databricks-sdk-0.26.0/databricks_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: databricks-sdk
-Version: 0.25.0
+Version: 0.26.0
 Summary: Databricks SDK for Python (Beta)
 Home-page: https://databricks-sdk-py.readthedocs.io
 Author: Serge Smertin

databricks-sdk-0.25.0/databricks/sdk/version.py

@@ -1 +0,0 @@
-__version__ = '0.25.0'