update sdk to latest spec

Author: Divyansh-db

.gitattributes

@@ -34,4 +34,3 @@ databricks/sdk/service/vectorsearch.py linguist-generated=true
 databricks/sdk/service/workspace.py linguist-generated=true
 test_http_call.py linguist-generated=true
 test_json_marshall.py linguist-generated=true
-test_lro_call.py linguist-generated=true

databricks/sdk/service/billing.py

@@ -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.
@@ -156,12 +156,13 @@
             import time
             
             from databricks.sdk import WorkspaceClient
+            from databricks.sdk.service import catalog
             
             w = WorkspaceClient()
             
             created = w.catalogs.create(name=f"sdk-{time.time_ns()}")
             
-            _ = w.catalogs.update(name=created.name, comment="updated")
+            _ = w.catalogs.update(name=created.name, isolation_mode=catalog.CatalogIsolationMode.ISOLATED)
             
             # cleanup
             w.catalogs.delete(name=created.name, force=True)

databricks/sdk/service/cleanrooms.py

@@ -30,22 +30,20 @@
             
             w = WorkspaceClient()
             
-            storage_credential = w.storage_credentials.create(
+            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",
             )
             
-            external_location = w.external_locations.create(
+            created = w.external_locations.create(
                 name=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()}",
+                credential_name=credential.name,
+                url="s3://%s/%s" % (os.environ["TEST_BUCKET"], f"sdk-{time.time_ns()}"),
             )
             
             # cleanup
-            w.storage_credentials.delete(name=storage_credential.name)
-            w.external_locations.delete(name=external_location.name)
+            w.storage_credentials.delete(name=credential.name)
+            w.external_locations.delete(name=created.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
@@ -107,20 +105,20 @@
             
             credential = 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"]),
             )
             
             created = w.external_locations.create(
                 name=f"sdk-{time.time_ns()}",
                 credential_name=credential.name,
-                url=f's3://{os.environ["TEST_BUCKET"]}/sdk-{time.time_ns()}',
+                url="s3://%s/%s" % (os.environ["TEST_BUCKET"], f"sdk-{time.time_ns()}"),
             )
             
-            _ = w.external_locations.get(get=created.name)
+            _ = w.external_locations.get(name=created.name)
             
             # cleanup
-            w.storage_credentials.delete(delete=credential.name)
-            w.external_locations.delete(delete=created.name)
+            w.storage_credentials.delete(name=credential.name)
+            w.external_locations.delete(name=created.name)
 
         Gets an external location from the metastore. The caller must be either a metastore admin, the owner
         of the external location, or a user that has some privilege on the external location.
@@ -194,24 +192,24 @@
             
             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"]),
+                aws_iam_role=catalog.AwsIamRole(role_arn=os.environ["TEST_METASTORE_DATA_ACCESS_ARN"]),
             )
             
             created = 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()}"),
+                url=f's3://{os.environ["TEST_BUCKET"]}/sdk-{time.time_ns()}',
             )
             
             _ = w.external_locations.update(
                 name=created.name,
                 credential_name=credential.name,
-                url="s3://%s/%s" % (os.environ["TEST_BUCKET"], f"sdk-{time.time_ns()}"),
+                url=f's3://{os.environ["TEST_BUCKET"]}/sdk-{time.time_ns()}',
             )
             
             # cleanup
-            w.storage_credentials.delete(name=credential.name)
-            w.external_locations.delete(name=created.name)
+            w.storage_credentials.delete(delete=credential.name)
+            w.external_locations.delete(delete=created.name)
 
         Updates an external location in the metastore. The caller must be the owner of the external location,
         or be a metastore admin. In the second case, the admin can only update the name of the external

databricks/sdk/service/dashboards.py

@@ -32,11 +32,11 @@
             
             created = w.storage_credentials.create(
                 name=f"sdk-{time.time_ns()}",
-                aws_iam_role=catalog.AwsIamRoleRequest(role_arn=os.environ["TEST_METASTORE_DATA_ACCESS_ARN"]),
+                aws_iam_role=catalog.AwsIamRole(role_arn=os.environ["TEST_METASTORE_DATA_ACCESS_ARN"]),
             )
             
             # cleanup
-            w.storage_credentials.delete(name=created.name)
+            w.storage_credentials.delete(delete=created.name)
 
         Creates a new storage credential.
 
@@ -98,13 +98,13 @@
             
             created = w.storage_credentials.create(
                 name=f"sdk-{time.time_ns()}",
-                aws_iam_role=catalog.AwsIamRoleRequest(role_arn=os.environ["TEST_METASTORE_DATA_ACCESS_ARN"]),
+                aws_iam_role=catalog.AwsIamRole(role_arn=os.environ["TEST_METASTORE_DATA_ACCESS_ARN"]),
             )
             
-            by_name = w.storage_credentials.get(name=created.name)
+            by_name = w.storage_credentials.get(get=created.name)
             
             # cleanup
-            w.storage_credentials.delete(name=created.name)
+            w.storage_credentials.delete(delete=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.