Update SDK API to 6700bbe74746844cbb713c810c90e9187056b894 (#1199)

This PR updates the SDK to the latest API changes.

NO_CHANGELOG=true

Co-authored-by: databricks-ci-ghec-1[bot] <184311507+databricks-ci-ghec-1[bot]@users.noreply.github.com>

Author: databricks-ci-ghec-1[bot]

.codegen/_openapi_sha

@@ -1 +1 @@
-e16060430295515f0c147e5252a3fee3c19fde9f
+6700bbe74746844cbb713c810c90e9187056b894

docs/account/iam/workspace_assignment.rst

@@ -43,9 +43,9 @@
             
             a = AccountClient()
             
-            workspace_id = os.environ["TEST_WORKSPACE_ID"]
+            workspace_id = os.environ["DUMMY_WORKSPACE_ID"]
             
-            all = a.workspace_assignment.list(list=workspace_id)
+            all = a.workspace_assignment.list(workspace_id=workspace_id)
 
         Get the permission assignments for the specified Databricks account and Databricks workspace.
 
@@ -74,9 +74,9 @@
             
             spn_id = spn.id
             
-            workspace_id = os.environ["DUMMY_WORKSPACE_ID"]
+            workspace_id = os.environ["TEST_WORKSPACE_ID"]
             
-            _ = a.workspace_assignment.update(
+            a.workspace_assignment.update(
                 workspace_id=workspace_id,
                 principal_id=spn_id,
                 permissions=[iam.WorkspacePermission.USER],

docs/workspace/catalog/catalogs.rst

@@ -24,10 +24,10 @@
             
             w = WorkspaceClient()
             
-            created = w.catalogs.create(name=f"sdk-{time.time_ns()}")
+            created_catalog = w.catalogs.create(name=f"sdk-{time.time_ns()}")
             
             # cleanup
-            w.catalogs.delete(name=created.name, force=True)
+            w.catalogs.delete(name=created_catalog.name, force=True)
 
         Creates a new catalog instance in the parent metastore if the caller is a metastore admin or has the
         **CREATE_CATALOG** privilege.

docs/workspace/catalog/external_locations.rst

@@ -30,20 +30,22 @@
             
             w = WorkspaceClient()
             
-            credential = w.storage_credentials.create(
+            storage_credential = w.storage_credentials.create(
                 name=f"sdk-{time.time_ns()}",
                 aws_iam_role=catalog.AwsIamRoleRequest(role_arn=os.environ["TEST_METASTORE_DATA_ACCESS_ARN"]),
+                comment="created via SDK",
             )
             
-            created = w.external_locations.create(
+            external_location = w.external_locations.create(
                 name=f"sdk-{time.time_ns()}",
-                credential_name=credential.name,
-                url="s3://%s/%s" % (os.environ["TEST_BUCKET"], f"sdk-{time.time_ns()}"),
+                credential_name=storage_credential.name,
+                comment="created via SDK",
+                url="s3://" + os.environ["TEST_BUCKET"] + "/" + f"sdk-{time.time_ns()}",
             )
             
             # cleanup
-            w.storage_credentials.delete(name=credential.name)
-            w.external_locations.delete(name=created.name)
+            w.storage_credentials.delete(name=storage_credential.name)
+            w.external_locations.delete(name=external_location.name)
 
         Creates a new external location entry in the metastore. The caller must be a metastore admin or have
         the **CREATE_EXTERNAL_LOCATION** privilege on both the metastore and the associated storage

docs/workspace/catalog/storage_credentials.rst

@@ -30,13 +30,14 @@
             
             w = WorkspaceClient()
             
-            credential = w.storage_credentials.create(
+            storage_credential = w.storage_credentials.create(
                 name=f"sdk-{time.time_ns()}",
                 aws_iam_role=catalog.AwsIamRoleRequest(role_arn=os.environ["TEST_METASTORE_DATA_ACCESS_ARN"]),
+                comment="created via SDK",
             )
             
             # cleanup
-            w.storage_credentials.delete(name=credential.name)
+            w.storage_credentials.delete(name=storage_credential.name)
 
         Creates a new storage credential.
 
@@ -98,13 +99,13 @@
             
             created = w.storage_credentials.create(
                 name=f"sdk-{time.time_ns()}",
-                aws_iam_role=catalog.AwsIamRole(role_arn=os.environ["TEST_METASTORE_DATA_ACCESS_ARN"]),
+                aws_iam_role=catalog.AwsIamRoleRequest(role_arn=os.environ["TEST_METASTORE_DATA_ACCESS_ARN"]),
             )
             
-            by_name = w.storage_credentials.get(get=created.name)
+            by_name = w.storage_credentials.get(name=created.name)
             
             # cleanup
-            w.storage_credentials.delete(delete=created.name)
+            w.storage_credentials.delete(name=created.name)
 
         Gets a storage credential from the metastore. The caller must be a metastore admin, the owner of the
         storage credential, or have some permission on the storage credential.
@@ -123,10 +124,11 @@
         .. code-block::
 
             from databricks.sdk import WorkspaceClient
+            from databricks.sdk.service import catalog
             
             w = WorkspaceClient()
             
-            all = w.storage_credentials.list()
+            all = w.storage_credentials.list(catalog.ListStorageCredentialsRequest())
 
         Gets an array of storage credentials (as __StorageCredentialInfo__ objects). The array is limited to
         only those storage credentials the caller has permission to access. If the caller is a metastore

docs/workspace/ml/model_registry.rst

@@ -90,7 +90,7 @@
             
             w = WorkspaceClient()
             
-            model = w.model_registry.create_model(name=f"sdk-{time.time_ns()}")
+            created = w.model_registry.create_model(name=f"sdk-{time.time_ns()}")
 
         Creates a new registered model with the name specified in the request body. Throws
         `RESOURCE_ALREADY_EXISTS` if a registered model with the given name exists.
@@ -120,7 +120,7 @@
             
             model = w.model_registry.create_model(name=f"sdk-{time.time_ns()}")
             
-            created = w.model_registry.create_model_version(name=model.registered_model.name, source="dbfs:/tmp")
+            mv = w.model_registry.create_model_version(name=model.registered_model.name, source="dbfs:/tmp")
 
         Creates a model version.
 
@@ -734,14 +734,13 @@
             
             w = WorkspaceClient()
             
-            model = w.model_registry.create_model(name=f"sdk-{time.time_ns()}")
+            created = w.model_registry.create_model(name=f"sdk-{time.time_ns()}")
             
-            created = w.model_registry.create_model_version(name=model.registered_model.name, source="dbfs:/tmp")
+            model = w.model_registry.get_model(name=created.registered_model.name)
             
-            w.model_registry.update_model_version(
+            w.model_registry.update_model(
+                name=model.registered_model_databricks.name,
                 description=f"sdk-{time.time_ns()}",
-                name=created.model_version.name,
-                version=created.model_version.version,
             )
 
         Updates a registered model.

docs/workspace/workspace/workspace.rst

@@ -148,9 +148,9 @@
             
             w = WorkspaceClient()
             
-            notebook_path = f"/Users/{w.current_user.me().user_name}/sdk-{time.time_ns()}"
+            notebook = f"/Users/{w.current_user.me().user_name}/sdk-{time.time_ns()}"
             
-            obj = w.workspace.get_status(path=notebook_path)
+            get_status_response = w.workspace.get_status(path=notebook)
 
         Gets the status of an object or a directory. If `path` does not exist, this call returns an error
         `RESOURCE_DOES_NOT_EXIST`.
@@ -180,14 +180,12 @@
             
             w.workspace.import_(
                 path=notebook_path,
-                overwrite=true_,
+                overwrite=True,
                 format=workspace.ImportFormat.SOURCE,
                 language=workspace.Language.PYTHON,
                 content=base64.b64encode(
                     (
-                        """import time
-            time.sleep(10)
-            dbutils.notebook.exit('hello')
+                        """print(1)
             """
                     ).encode()
                 ).decode(),
@@ -234,14 +232,16 @@
 
         .. code-block::
 
+            import os
+            import time
+            
             from databricks.sdk import WorkspaceClient
             
             w = WorkspaceClient()
             
-            names = []
-            for i in w.workspace.list(f"/Users/{w.current_user.me().user_name}", recursive=True):
-                names.append(i.path)
-            assert len(names) > 0
+            notebook = f"/Users/{w.current_user.me().user_name}/sdk-{time.time_ns()}"
+            
+            objects = w.workspace.list(path=os.path.dirname(notebook))
 
         List workspace objects