databricks-sdk 0.44.0__py3-none-any.whl → 0.45.0__py3-none-any.whl

databricks/sdk/service/oauth2.py

@@ -4,11 +4,12 @@ from __future__ import annotations
 
 import logging
 from dataclasses import dataclass
-from typing import Dict, Iterator, List, Optional
+from typing import Any, Dict, Iterator, List, Optional
 
 from ._internal import _from_dict, _repeated_dict
 
-_LOG = logging.getLogger('databricks.sdk')
+_LOG = logging.getLogger("databricks.sdk")
+
 
 # all definitions in this file are in alphabetical order
 
@@ -38,35 +39,48 @@ class CreateCustomAppIntegration:
     def as_dict(self) -> dict:
         """Serializes the CreateCustomAppIntegration into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.confidential is not None: body['confidential'] = self.confidential
-        if self.name is not None: body['name'] = self.name
-        if self.redirect_urls: body['redirect_urls'] = [v for v in self.redirect_urls]
-        if self.scopes: body['scopes'] = [v for v in self.scopes]
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy.as_dict()
+        if self.confidential is not None:
+            body["confidential"] = self.confidential
+        if self.name is not None:
+            body["name"] = self.name
+        if self.redirect_urls:
+            body["redirect_urls"] = [v for v in self.redirect_urls]
+        if self.scopes:
+            body["scopes"] = [v for v in self.scopes]
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy.as_dict()
         if self.user_authorized_scopes:
-            body['user_authorized_scopes'] = [v for v in self.user_authorized_scopes]
+            body["user_authorized_scopes"] = [v for v in self.user_authorized_scopes]
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the CreateCustomAppIntegration into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.confidential is not None: body['confidential'] = self.confidential
-        if self.name is not None: body['name'] = self.name
-        if self.redirect_urls: body['redirect_urls'] = self.redirect_urls
-        if self.scopes: body['scopes'] = self.scopes
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy
-        if self.user_authorized_scopes: body['user_authorized_scopes'] = self.user_authorized_scopes
+        if self.confidential is not None:
+            body["confidential"] = self.confidential
+        if self.name is not None:
+            body["name"] = self.name
+        if self.redirect_urls:
+            body["redirect_urls"] = self.redirect_urls
+        if self.scopes:
+            body["scopes"] = self.scopes
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy
+        if self.user_authorized_scopes:
+            body["user_authorized_scopes"] = self.user_authorized_scopes
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> CreateCustomAppIntegration:
+    def from_dict(cls, d: Dict[str, Any]) -> CreateCustomAppIntegration:
         """Deserializes the CreateCustomAppIntegration from a dictionary."""
-        return cls(confidential=d.get('confidential', None),
-                   name=d.get('name', None),
-                   redirect_urls=d.get('redirect_urls', None),
-                   scopes=d.get('scopes', None),
-                   token_access_policy=_from_dict(d, 'token_access_policy', TokenAccessPolicy),
-                   user_authorized_scopes=d.get('user_authorized_scopes', None))
+        return cls(
+            confidential=d.get("confidential", None),
+            name=d.get("name", None),
+            redirect_urls=d.get("redirect_urls", None),
+            scopes=d.get("scopes", None),
+            token_access_policy=_from_dict(d, "token_access_policy", TokenAccessPolicy),
+            user_authorized_scopes=d.get("user_authorized_scopes", None),
+        )
 
 
 @dataclass
@@ -84,25 +98,33 @@ class CreateCustomAppIntegrationOutput:
     def as_dict(self) -> dict:
         """Serializes the CreateCustomAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.client_id is not None: body['client_id'] = self.client_id
-        if self.client_secret is not None: body['client_secret'] = self.client_secret
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
+        if self.client_id is not None:
+            body["client_id"] = self.client_id
+        if self.client_secret is not None:
+            body["client_secret"] = self.client_secret
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the CreateCustomAppIntegrationOutput into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.client_id is not None: body['client_id'] = self.client_id
-        if self.client_secret is not None: body['client_secret'] = self.client_secret
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
+        if self.client_id is not None:
+            body["client_id"] = self.client_id
+        if self.client_secret is not None:
+            body["client_secret"] = self.client_secret
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> CreateCustomAppIntegrationOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> CreateCustomAppIntegrationOutput:
         """Deserializes the CreateCustomAppIntegrationOutput from a dictionary."""
-        return cls(client_id=d.get('client_id', None),
-                   client_secret=d.get('client_secret', None),
-                   integration_id=d.get('integration_id', None))
+        return cls(
+            client_id=d.get("client_id", None),
+            client_secret=d.get("client_secret", None),
+            integration_id=d.get("integration_id", None),
+        )
 
 
 @dataclass
@@ -116,22 +138,27 @@ class CreatePublishedAppIntegration:
     def as_dict(self) -> dict:
         """Serializes the CreatePublishedAppIntegration into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.app_id is not None: body['app_id'] = self.app_id
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy.as_dict()
+        if self.app_id is not None:
+            body["app_id"] = self.app_id
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy.as_dict()
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the CreatePublishedAppIntegration into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.app_id is not None: body['app_id'] = self.app_id
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy
+        if self.app_id is not None:
+            body["app_id"] = self.app_id
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> CreatePublishedAppIntegration:
+    def from_dict(cls, d: Dict[str, Any]) -> CreatePublishedAppIntegration:
         """Deserializes the CreatePublishedAppIntegration from a dictionary."""
-        return cls(app_id=d.get('app_id', None),
-                   token_access_policy=_from_dict(d, 'token_access_policy', TokenAccessPolicy))
+        return cls(
+            app_id=d.get("app_id", None), token_access_policy=_from_dict(d, "token_access_policy", TokenAccessPolicy)
+        )
 
 
 @dataclass
@@ -142,19 +169,54 @@ class CreatePublishedAppIntegrationOutput:
     def as_dict(self) -> dict:
         """Serializes the CreatePublishedAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the CreatePublishedAppIntegrationOutput into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> CreatePublishedAppIntegrationOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> CreatePublishedAppIntegrationOutput:
         """Deserializes the CreatePublishedAppIntegrationOutput from a dictionary."""
-        return cls(integration_id=d.get('integration_id', None))
+        return cls(integration_id=d.get("integration_id", None))
+
+
+@dataclass
+class CreateServicePrincipalSecretRequest:
+    lifetime: Optional[str] = None
+    """The lifetime of the secret in seconds. If this parameter is not provided, the secret will have a
+    default lifetime of 730 days (63072000s)."""
+
+    service_principal_id: Optional[int] = None
+    """The service principal ID."""
+
+    def as_dict(self) -> dict:
+        """Serializes the CreateServicePrincipalSecretRequest into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.lifetime is not None:
+            body["lifetime"] = self.lifetime
+        if self.service_principal_id is not None:
+            body["service_principal_id"] = self.service_principal_id
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the CreateServicePrincipalSecretRequest into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.lifetime is not None:
+            body["lifetime"] = self.lifetime
+        if self.service_principal_id is not None:
+            body["service_principal_id"] = self.service_principal_id
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> CreateServicePrincipalSecretRequest:
+        """Deserializes the CreateServicePrincipalSecretRequest from a dictionary."""
+        return cls(lifetime=d.get("lifetime", None), service_principal_id=d.get("service_principal_id", None))
 
 
 @dataclass
@@ -162,6 +224,9 @@ class CreateServicePrincipalSecretResponse:
     create_time: Optional[str] = None
     """UTC time when the secret was created"""
 
+    expire_time: Optional[str] = None
+    """UTC time when the secret will expire. If the field is not present, the secret does not expire."""
+
     id: Optional[str] = None
     """ID of the secret"""
 
@@ -180,39 +245,57 @@ class CreateServicePrincipalSecretResponse:
     def as_dict(self) -> dict:
         """Serializes the CreateServicePrincipalSecretResponse into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.create_time is not None: body['create_time'] = self.create_time
-        if self.id is not None: body['id'] = self.id
-        if self.secret is not None: body['secret'] = self.secret
-        if self.secret_hash is not None: body['secret_hash'] = self.secret_hash
-        if self.status is not None: body['status'] = self.status
-        if self.update_time is not None: body['update_time'] = self.update_time
+        if self.create_time is not None:
+            body["create_time"] = self.create_time
+        if self.expire_time is not None:
+            body["expire_time"] = self.expire_time
+        if self.id is not None:
+            body["id"] = self.id
+        if self.secret is not None:
+            body["secret"] = self.secret
+        if self.secret_hash is not None:
+            body["secret_hash"] = self.secret_hash
+        if self.status is not None:
+            body["status"] = self.status
+        if self.update_time is not None:
+            body["update_time"] = self.update_time
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the CreateServicePrincipalSecretResponse into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.create_time is not None: body['create_time'] = self.create_time
-        if self.id is not None: body['id'] = self.id
-        if self.secret is not None: body['secret'] = self.secret
-        if self.secret_hash is not None: body['secret_hash'] = self.secret_hash
-        if self.status is not None: body['status'] = self.status
-        if self.update_time is not None: body['update_time'] = self.update_time
+        if self.create_time is not None:
+            body["create_time"] = self.create_time
+        if self.expire_time is not None:
+            body["expire_time"] = self.expire_time
+        if self.id is not None:
+            body["id"] = self.id
+        if self.secret is not None:
+            body["secret"] = self.secret
+        if self.secret_hash is not None:
+            body["secret_hash"] = self.secret_hash
+        if self.status is not None:
+            body["status"] = self.status
+        if self.update_time is not None:
+            body["update_time"] = self.update_time
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> CreateServicePrincipalSecretResponse:
+    def from_dict(cls, d: Dict[str, Any]) -> CreateServicePrincipalSecretResponse:
         """Deserializes the CreateServicePrincipalSecretResponse from a dictionary."""
-        return cls(create_time=d.get('create_time', None),
-                   id=d.get('id', None),
-                   secret=d.get('secret', None),
-                   secret_hash=d.get('secret_hash', None),
-                   status=d.get('status', None),
-                   update_time=d.get('update_time', None))
+        return cls(
+            create_time=d.get("create_time", None),
+            expire_time=d.get("expire_time", None),
+            id=d.get("id", None),
+            secret=d.get("secret", None),
+            secret_hash=d.get("secret_hash", None),
+            status=d.get("status", None),
+            update_time=d.get("update_time", None),
+        )
 
 
 @dataclass
 class DeleteCustomAppIntegrationOutput:
-
     def as_dict(self) -> dict:
         """Serializes the DeleteCustomAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -224,14 +307,13 @@ class DeleteCustomAppIntegrationOutput:
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> DeleteCustomAppIntegrationOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> DeleteCustomAppIntegrationOutput:
         """Deserializes the DeleteCustomAppIntegrationOutput from a dictionary."""
         return cls()
 
 
 @dataclass
 class DeletePublishedAppIntegrationOutput:
-
     def as_dict(self) -> dict:
         """Serializes the DeletePublishedAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -243,14 +325,13 @@ class DeletePublishedAppIntegrationOutput:
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> DeletePublishedAppIntegrationOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> DeletePublishedAppIntegrationOutput:
         """Deserializes the DeletePublishedAppIntegrationOutput from a dictionary."""
         return cls()
 
 
 @dataclass
 class DeleteResponse:
-
     def as_dict(self) -> dict:
         """Serializes the DeleteResponse into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -262,7 +343,7 @@ class DeleteResponse:
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> DeleteResponse:
+    def from_dict(cls, d: Dict[str, Any]) -> DeleteResponse:
         """Deserializes the DeleteResponse from a dictionary."""
         return cls()
 
@@ -296,34 +377,48 @@ class FederationPolicy:
     def as_dict(self) -> dict:
         """Serializes the FederationPolicy into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.create_time is not None: body['create_time'] = self.create_time
-        if self.description is not None: body['description'] = self.description
-        if self.name is not None: body['name'] = self.name
-        if self.oidc_policy: body['oidc_policy'] = self.oidc_policy.as_dict()
-        if self.uid is not None: body['uid'] = self.uid
-        if self.update_time is not None: body['update_time'] = self.update_time
+        if self.create_time is not None:
+            body["create_time"] = self.create_time
+        if self.description is not None:
+            body["description"] = self.description
+        if self.name is not None:
+            body["name"] = self.name
+        if self.oidc_policy:
+            body["oidc_policy"] = self.oidc_policy.as_dict()
+        if self.uid is not None:
+            body["uid"] = self.uid
+        if self.update_time is not None:
+            body["update_time"] = self.update_time
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the FederationPolicy into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.create_time is not None: body['create_time'] = self.create_time
-        if self.description is not None: body['description'] = self.description
-        if self.name is not None: body['name'] = self.name
-        if self.oidc_policy: body['oidc_policy'] = self.oidc_policy
-        if self.uid is not None: body['uid'] = self.uid
-        if self.update_time is not None: body['update_time'] = self.update_time
+        if self.create_time is not None:
+            body["create_time"] = self.create_time
+        if self.description is not None:
+            body["description"] = self.description
+        if self.name is not None:
+            body["name"] = self.name
+        if self.oidc_policy:
+            body["oidc_policy"] = self.oidc_policy
+        if self.uid is not None:
+            body["uid"] = self.uid
+        if self.update_time is not None:
+            body["update_time"] = self.update_time
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> FederationPolicy:
+    def from_dict(cls, d: Dict[str, Any]) -> FederationPolicy:
         """Deserializes the FederationPolicy from a dictionary."""
-        return cls(create_time=d.get('create_time', None),
-                   description=d.get('description', None),
-                   name=d.get('name', None),
-                   oidc_policy=_from_dict(d, 'oidc_policy', OidcFederationPolicy),
-                   uid=d.get('uid', None),
-                   update_time=d.get('update_time', None))
+        return cls(
+            create_time=d.get("create_time", None),
+            description=d.get("description", None),
+            name=d.get("name", None),
+            oidc_policy=_from_dict(d, "oidc_policy", OidcFederationPolicy),
+            uid=d.get("uid", None),
+            update_time=d.get("update_time", None),
+        )
 
 
 @dataclass
@@ -361,50 +456,73 @@ class GetCustomAppIntegrationOutput:
     def as_dict(self) -> dict:
         """Serializes the GetCustomAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.client_id is not None: body['client_id'] = self.client_id
-        if self.confidential is not None: body['confidential'] = self.confidential
-        if self.create_time is not None: body['create_time'] = self.create_time
-        if self.created_by is not None: body['created_by'] = self.created_by
-        if self.creator_username is not None: body['creator_username'] = self.creator_username
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
-        if self.name is not None: body['name'] = self.name
-        if self.redirect_urls: body['redirect_urls'] = [v for v in self.redirect_urls]
-        if self.scopes: body['scopes'] = [v for v in self.scopes]
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy.as_dict()
+        if self.client_id is not None:
+            body["client_id"] = self.client_id
+        if self.confidential is not None:
+            body["confidential"] = self.confidential
+        if self.create_time is not None:
+            body["create_time"] = self.create_time
+        if self.created_by is not None:
+            body["created_by"] = self.created_by
+        if self.creator_username is not None:
+            body["creator_username"] = self.creator_username
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
+        if self.name is not None:
+            body["name"] = self.name
+        if self.redirect_urls:
+            body["redirect_urls"] = [v for v in self.redirect_urls]
+        if self.scopes:
+            body["scopes"] = [v for v in self.scopes]
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy.as_dict()
         if self.user_authorized_scopes:
-            body['user_authorized_scopes'] = [v for v in self.user_authorized_scopes]
+            body["user_authorized_scopes"] = [v for v in self.user_authorized_scopes]
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the GetCustomAppIntegrationOutput into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.client_id is not None: body['client_id'] = self.client_id
-        if self.confidential is not None: body['confidential'] = self.confidential
-        if self.create_time is not None: body['create_time'] = self.create_time
-        if self.created_by is not None: body['created_by'] = self.created_by
-        if self.creator_username is not None: body['creator_username'] = self.creator_username
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
-        if self.name is not None: body['name'] = self.name
-        if self.redirect_urls: body['redirect_urls'] = self.redirect_urls
-        if self.scopes: body['scopes'] = self.scopes
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy
-        if self.user_authorized_scopes: body['user_authorized_scopes'] = self.user_authorized_scopes
+        if self.client_id is not None:
+            body["client_id"] = self.client_id
+        if self.confidential is not None:
+            body["confidential"] = self.confidential
+        if self.create_time is not None:
+            body["create_time"] = self.create_time
+        if self.created_by is not None:
+            body["created_by"] = self.created_by
+        if self.creator_username is not None:
+            body["creator_username"] = self.creator_username
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
+        if self.name is not None:
+            body["name"] = self.name
+        if self.redirect_urls:
+            body["redirect_urls"] = self.redirect_urls
+        if self.scopes:
+            body["scopes"] = self.scopes
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy
+        if self.user_authorized_scopes:
+            body["user_authorized_scopes"] = self.user_authorized_scopes
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> GetCustomAppIntegrationOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> GetCustomAppIntegrationOutput:
         """Deserializes the GetCustomAppIntegrationOutput from a dictionary."""
-        return cls(client_id=d.get('client_id', None),
-                   confidential=d.get('confidential', None),
-                   create_time=d.get('create_time', None),
-                   created_by=d.get('created_by', None),
-                   creator_username=d.get('creator_username', None),
-                   integration_id=d.get('integration_id', None),
-                   name=d.get('name', None),
-                   redirect_urls=d.get('redirect_urls', None),
-                   scopes=d.get('scopes', None),
-                   token_access_policy=_from_dict(d, 'token_access_policy', TokenAccessPolicy),
-                   user_authorized_scopes=d.get('user_authorized_scopes', None))
+        return cls(
+            client_id=d.get("client_id", None),
+            confidential=d.get("confidential", None),
+            create_time=d.get("create_time", None),
+            created_by=d.get("created_by", None),
+            creator_username=d.get("creator_username", None),
+            integration_id=d.get("integration_id", None),
+            name=d.get("name", None),
+            redirect_urls=d.get("redirect_urls", None),
+            scopes=d.get("scopes", None),
+            token_access_policy=_from_dict(d, "token_access_policy", TokenAccessPolicy),
+            user_authorized_scopes=d.get("user_authorized_scopes", None),
+        )
 
 
 @dataclass
@@ -417,22 +535,28 @@ class GetCustomAppIntegrationsOutput:
     def as_dict(self) -> dict:
         """Serializes the GetCustomAppIntegrationsOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.apps: body['apps'] = [v.as_dict() for v in self.apps]
-        if self.next_page_token is not None: body['next_page_token'] = self.next_page_token
+        if self.apps:
+            body["apps"] = [v.as_dict() for v in self.apps]
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the GetCustomAppIntegrationsOutput into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.apps: body['apps'] = self.apps
-        if self.next_page_token is not None: body['next_page_token'] = self.next_page_token
+        if self.apps:
+            body["apps"] = self.apps
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> GetCustomAppIntegrationsOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> GetCustomAppIntegrationsOutput:
         """Deserializes the GetCustomAppIntegrationsOutput from a dictionary."""
-        return cls(apps=_repeated_dict(d, 'apps', GetCustomAppIntegrationOutput),
-                   next_page_token=d.get('next_page_token', None))
+        return cls(
+            apps=_repeated_dict(d, "apps", GetCustomAppIntegrationOutput),
+            next_page_token=d.get("next_page_token", None),
+        )
 
 
 @dataclass
@@ -456,34 +580,48 @@ class GetPublishedAppIntegrationOutput:
     def as_dict(self) -> dict:
         """Serializes the GetPublishedAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.app_id is not None: body['app_id'] = self.app_id
-        if self.create_time is not None: body['create_time'] = self.create_time
-        if self.created_by is not None: body['created_by'] = self.created_by
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
-        if self.name is not None: body['name'] = self.name
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy.as_dict()
+        if self.app_id is not None:
+            body["app_id"] = self.app_id
+        if self.create_time is not None:
+            body["create_time"] = self.create_time
+        if self.created_by is not None:
+            body["created_by"] = self.created_by
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
+        if self.name is not None:
+            body["name"] = self.name
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy.as_dict()
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the GetPublishedAppIntegrationOutput into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.app_id is not None: body['app_id'] = self.app_id
-        if self.create_time is not None: body['create_time'] = self.create_time
-        if self.created_by is not None: body['created_by'] = self.created_by
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
-        if self.name is not None: body['name'] = self.name
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy
+        if self.app_id is not None:
+            body["app_id"] = self.app_id
+        if self.create_time is not None:
+            body["create_time"] = self.create_time
+        if self.created_by is not None:
+            body["created_by"] = self.created_by
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
+        if self.name is not None:
+            body["name"] = self.name
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> GetPublishedAppIntegrationOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> GetPublishedAppIntegrationOutput:
         """Deserializes the GetPublishedAppIntegrationOutput from a dictionary."""
-        return cls(app_id=d.get('app_id', None),
-                   create_time=d.get('create_time', None),
-                   created_by=d.get('created_by', None),
-                   integration_id=d.get('integration_id', None),
-                   name=d.get('name', None),
-                   token_access_policy=_from_dict(d, 'token_access_policy', TokenAccessPolicy))
+        return cls(
+            app_id=d.get("app_id", None),
+            create_time=d.get("create_time", None),
+            created_by=d.get("created_by", None),
+            integration_id=d.get("integration_id", None),
+            name=d.get("name", None),
+            token_access_policy=_from_dict(d, "token_access_policy", TokenAccessPolicy),
+        )
 
 
 @dataclass
@@ -496,22 +634,28 @@ class GetPublishedAppIntegrationsOutput:
     def as_dict(self) -> dict:
         """Serializes the GetPublishedAppIntegrationsOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.apps: body['apps'] = [v.as_dict() for v in self.apps]
-        if self.next_page_token is not None: body['next_page_token'] = self.next_page_token
+        if self.apps:
+            body["apps"] = [v.as_dict() for v in self.apps]
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the GetPublishedAppIntegrationsOutput into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.apps: body['apps'] = self.apps
-        if self.next_page_token is not None: body['next_page_token'] = self.next_page_token
+        if self.apps:
+            body["apps"] = self.apps
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> GetPublishedAppIntegrationsOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> GetPublishedAppIntegrationsOutput:
         """Deserializes the GetPublishedAppIntegrationsOutput from a dictionary."""
-        return cls(apps=_repeated_dict(d, 'apps', GetPublishedAppIntegrationOutput),
-                   next_page_token=d.get('next_page_token', None))
+        return cls(
+            apps=_repeated_dict(d, "apps", GetPublishedAppIntegrationOutput),
+            next_page_token=d.get("next_page_token", None),
+        )
 
 
 @dataclass
@@ -526,22 +670,25 @@ class GetPublishedAppsOutput:
     def as_dict(self) -> dict:
         """Serializes the GetPublishedAppsOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.apps: body['apps'] = [v.as_dict() for v in self.apps]
-        if self.next_page_token is not None: body['next_page_token'] = self.next_page_token
+        if self.apps:
+            body["apps"] = [v.as_dict() for v in self.apps]
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the GetPublishedAppsOutput into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.apps: body['apps'] = self.apps
-        if self.next_page_token is not None: body['next_page_token'] = self.next_page_token
+        if self.apps:
+            body["apps"] = self.apps
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> GetPublishedAppsOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> GetPublishedAppsOutput:
         """Deserializes the GetPublishedAppsOutput from a dictionary."""
-        return cls(apps=_repeated_dict(d, 'apps', PublishedAppOutput),
-                   next_page_token=d.get('next_page_token', None))
+        return cls(apps=_repeated_dict(d, "apps", PublishedAppOutput), next_page_token=d.get("next_page_token", None))
 
 
 @dataclass
@@ -553,22 +700,27 @@ class ListFederationPoliciesResponse:
     def as_dict(self) -> dict:
         """Serializes the ListFederationPoliciesResponse into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.next_page_token is not None: body['next_page_token'] = self.next_page_token
-        if self.policies: body['policies'] = [v.as_dict() for v in self.policies]
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
+        if self.policies:
+            body["policies"] = [v.as_dict() for v in self.policies]
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the ListFederationPoliciesResponse into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.next_page_token is not None: body['next_page_token'] = self.next_page_token
-        if self.policies: body['policies'] = self.policies
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
+        if self.policies:
+            body["policies"] = self.policies
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> ListFederationPoliciesResponse:
+    def from_dict(cls, d: Dict[str, Any]) -> ListFederationPoliciesResponse:
         """Deserializes the ListFederationPoliciesResponse from a dictionary."""
-        return cls(next_page_token=d.get('next_page_token', None),
-                   policies=_repeated_dict(d, 'policies', FederationPolicy))
+        return cls(
+            next_page_token=d.get("next_page_token", None), policies=_repeated_dict(d, "policies", FederationPolicy)
+        )
 
 
 @dataclass
@@ -582,22 +734,25 @@ class ListServicePrincipalSecretsResponse:
     def as_dict(self) -> dict:
         """Serializes the ListServicePrincipalSecretsResponse into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.next_page_token is not None: body['next_page_token'] = self.next_page_token
-        if self.secrets: body['secrets'] = [v.as_dict() for v in self.secrets]
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
+        if self.secrets:
+            body["secrets"] = [v.as_dict() for v in self.secrets]
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the ListServicePrincipalSecretsResponse into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.next_page_token is not None: body['next_page_token'] = self.next_page_token
-        if self.secrets: body['secrets'] = self.secrets
+        if self.next_page_token is not None:
+            body["next_page_token"] = self.next_page_token
+        if self.secrets:
+            body["secrets"] = self.secrets
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> ListServicePrincipalSecretsResponse:
+    def from_dict(cls, d: Dict[str, Any]) -> ListServicePrincipalSecretsResponse:
         """Deserializes the ListServicePrincipalSecretsResponse from a dictionary."""
-        return cls(next_page_token=d.get('next_page_token', None),
-                   secrets=_repeated_dict(d, 'secrets', SecretInfo))
+        return cls(next_page_token=d.get("next_page_token", None), secrets=_repeated_dict(d, "secrets", SecretInfo))
 
 
 @dataclass
@@ -615,10 +770,18 @@ class OidcFederationPolicy:
     """The required token issuer, as specified in the 'iss' claim of federated tokens."""
 
     jwks_json: Optional[str] = None
-    """The public keys used to validate the signature of federated tokens, in JWKS format. If
-    unspecified (recommended), Databricks automatically fetches the public keys from your issuer’s
-    well known endpoint. Databricks strongly recommends relying on your issuer’s well known
-    endpoint for discovering public keys."""
+    """The public keys used to validate the signature of federated tokens, in JWKS format. Most use
+    cases should not need to specify this field. If jwks_uri and jwks_json are both unspecified
+    (recommended), Databricks automatically fetches the public keys from your issuer’s well known
+    endpoint. Databricks strongly recommends relying on your issuer’s well known endpoint for
+    discovering public keys."""
+
+    jwks_uri: Optional[str] = None
+    """URL of the public keys used to validate the signature of federated tokens, in JWKS format. Most
+    use cases should not need to specify this field. If jwks_uri and jwks_json are both unspecified
+    (recommended), Databricks automatically fetches the public keys from your issuer’s well known
+    endpoint. Databricks strongly recommends relying on your issuer’s well known endpoint for
+    discovering public keys."""
 
     subject: Optional[str] = None
     """The required token subject, as specified in the subject claim of federated tokens. Must be
@@ -631,31 +794,48 @@ class OidcFederationPolicy:
     def as_dict(self) -> dict:
         """Serializes the OidcFederationPolicy into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.audiences: body['audiences'] = [v for v in self.audiences]
-        if self.issuer is not None: body['issuer'] = self.issuer
-        if self.jwks_json is not None: body['jwks_json'] = self.jwks_json
-        if self.subject is not None: body['subject'] = self.subject
-        if self.subject_claim is not None: body['subject_claim'] = self.subject_claim
+        if self.audiences:
+            body["audiences"] = [v for v in self.audiences]
+        if self.issuer is not None:
+            body["issuer"] = self.issuer
+        if self.jwks_json is not None:
+            body["jwks_json"] = self.jwks_json
+        if self.jwks_uri is not None:
+            body["jwks_uri"] = self.jwks_uri
+        if self.subject is not None:
+            body["subject"] = self.subject
+        if self.subject_claim is not None:
+            body["subject_claim"] = self.subject_claim
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the OidcFederationPolicy into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.audiences: body['audiences'] = self.audiences
-        if self.issuer is not None: body['issuer'] = self.issuer
-        if self.jwks_json is not None: body['jwks_json'] = self.jwks_json
-        if self.subject is not None: body['subject'] = self.subject
-        if self.subject_claim is not None: body['subject_claim'] = self.subject_claim
+        if self.audiences:
+            body["audiences"] = self.audiences
+        if self.issuer is not None:
+            body["issuer"] = self.issuer
+        if self.jwks_json is not None:
+            body["jwks_json"] = self.jwks_json
+        if self.jwks_uri is not None:
+            body["jwks_uri"] = self.jwks_uri
+        if self.subject is not None:
+            body["subject"] = self.subject
+        if self.subject_claim is not None:
+            body["subject_claim"] = self.subject_claim
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> OidcFederationPolicy:
+    def from_dict(cls, d: Dict[str, Any]) -> OidcFederationPolicy:
         """Deserializes the OidcFederationPolicy from a dictionary."""
-        return cls(audiences=d.get('audiences', None),
-                   issuer=d.get('issuer', None),
-                   jwks_json=d.get('jwks_json', None),
-                   subject=d.get('subject', None),
-                   subject_claim=d.get('subject_claim', None))
+        return cls(
+            audiences=d.get("audiences", None),
+            issuer=d.get("issuer", None),
+            jwks_json=d.get("jwks_json", None),
+            jwks_uri=d.get("jwks_uri", None),
+            subject=d.get("subject", None),
+            subject_claim=d.get("subject_claim", None),
+        )
 
 
 @dataclass
@@ -685,39 +865,53 @@ class PublishedAppOutput:
     def as_dict(self) -> dict:
         """Serializes the PublishedAppOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.app_id is not None: body['app_id'] = self.app_id
-        if self.client_id is not None: body['client_id'] = self.client_id
-        if self.description is not None: body['description'] = self.description
+        if self.app_id is not None:
+            body["app_id"] = self.app_id
+        if self.client_id is not None:
+            body["client_id"] = self.client_id
+        if self.description is not None:
+            body["description"] = self.description
         if self.is_confidential_client is not None:
-            body['is_confidential_client'] = self.is_confidential_client
-        if self.name is not None: body['name'] = self.name
-        if self.redirect_urls: body['redirect_urls'] = [v for v in self.redirect_urls]
-        if self.scopes: body['scopes'] = [v for v in self.scopes]
+            body["is_confidential_client"] = self.is_confidential_client
+        if self.name is not None:
+            body["name"] = self.name
+        if self.redirect_urls:
+            body["redirect_urls"] = [v for v in self.redirect_urls]
+        if self.scopes:
+            body["scopes"] = [v for v in self.scopes]
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the PublishedAppOutput into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.app_id is not None: body['app_id'] = self.app_id
-        if self.client_id is not None: body['client_id'] = self.client_id
-        if self.description is not None: body['description'] = self.description
+        if self.app_id is not None:
+            body["app_id"] = self.app_id
+        if self.client_id is not None:
+            body["client_id"] = self.client_id
+        if self.description is not None:
+            body["description"] = self.description
         if self.is_confidential_client is not None:
-            body['is_confidential_client'] = self.is_confidential_client
-        if self.name is not None: body['name'] = self.name
-        if self.redirect_urls: body['redirect_urls'] = self.redirect_urls
-        if self.scopes: body['scopes'] = self.scopes
+            body["is_confidential_client"] = self.is_confidential_client
+        if self.name is not None:
+            body["name"] = self.name
+        if self.redirect_urls:
+            body["redirect_urls"] = self.redirect_urls
+        if self.scopes:
+            body["scopes"] = self.scopes
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> PublishedAppOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> PublishedAppOutput:
         """Deserializes the PublishedAppOutput from a dictionary."""
-        return cls(app_id=d.get('app_id', None),
-                   client_id=d.get('client_id', None),
-                   description=d.get('description', None),
-                   is_confidential_client=d.get('is_confidential_client', None),
-                   name=d.get('name', None),
-                   redirect_urls=d.get('redirect_urls', None),
-                   scopes=d.get('scopes', None))
+        return cls(
+            app_id=d.get("app_id", None),
+            client_id=d.get("client_id", None),
+            description=d.get("description", None),
+            is_confidential_client=d.get("is_confidential_client", None),
+            name=d.get("name", None),
+            redirect_urls=d.get("redirect_urls", None),
+            scopes=d.get("scopes", None),
+        )
 
 
 @dataclass
@@ -725,6 +919,9 @@ class SecretInfo:
     create_time: Optional[str] = None
     """UTC time when the secret was created"""
 
+    expire_time: Optional[str] = None
+    """UTC time when the secret will expire. If the field is not present, the secret does not expire."""
+
     id: Optional[str] = None
     """ID of the secret"""
 
@@ -740,31 +937,48 @@ class SecretInfo:
     def as_dict(self) -> dict:
         """Serializes the SecretInfo into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.create_time is not None: body['create_time'] = self.create_time
-        if self.id is not None: body['id'] = self.id
-        if self.secret_hash is not None: body['secret_hash'] = self.secret_hash
-        if self.status is not None: body['status'] = self.status
-        if self.update_time is not None: body['update_time'] = self.update_time
+        if self.create_time is not None:
+            body["create_time"] = self.create_time
+        if self.expire_time is not None:
+            body["expire_time"] = self.expire_time
+        if self.id is not None:
+            body["id"] = self.id
+        if self.secret_hash is not None:
+            body["secret_hash"] = self.secret_hash
+        if self.status is not None:
+            body["status"] = self.status
+        if self.update_time is not None:
+            body["update_time"] = self.update_time
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the SecretInfo into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.create_time is not None: body['create_time'] = self.create_time
-        if self.id is not None: body['id'] = self.id
-        if self.secret_hash is not None: body['secret_hash'] = self.secret_hash
-        if self.status is not None: body['status'] = self.status
-        if self.update_time is not None: body['update_time'] = self.update_time
+        if self.create_time is not None:
+            body["create_time"] = self.create_time
+        if self.expire_time is not None:
+            body["expire_time"] = self.expire_time
+        if self.id is not None:
+            body["id"] = self.id
+        if self.secret_hash is not None:
+            body["secret_hash"] = self.secret_hash
+        if self.status is not None:
+            body["status"] = self.status
+        if self.update_time is not None:
+            body["update_time"] = self.update_time
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> SecretInfo:
+    def from_dict(cls, d: Dict[str, Any]) -> SecretInfo:
         """Deserializes the SecretInfo from a dictionary."""
-        return cls(create_time=d.get('create_time', None),
-                   id=d.get('id', None),
-                   secret_hash=d.get('secret_hash', None),
-                   status=d.get('status', None),
-                   update_time=d.get('update_time', None))
+        return cls(
+            create_time=d.get("create_time", None),
+            expire_time=d.get("expire_time", None),
+            id=d.get("id", None),
+            secret_hash=d.get("secret_hash", None),
+            status=d.get("status", None),
+            update_time=d.get("update_time", None),
+        )
 
 
 @dataclass
@@ -779,25 +993,27 @@ class TokenAccessPolicy:
         """Serializes the TokenAccessPolicy into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.access_token_ttl_in_minutes is not None:
-            body['access_token_ttl_in_minutes'] = self.access_token_ttl_in_minutes
+            body["access_token_ttl_in_minutes"] = self.access_token_ttl_in_minutes
         if self.refresh_token_ttl_in_minutes is not None:
-            body['refresh_token_ttl_in_minutes'] = self.refresh_token_ttl_in_minutes
+            body["refresh_token_ttl_in_minutes"] = self.refresh_token_ttl_in_minutes
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the TokenAccessPolicy into a shallow dictionary of its immediate attributes."""
         body = {}
         if self.access_token_ttl_in_minutes is not None:
-            body['access_token_ttl_in_minutes'] = self.access_token_ttl_in_minutes
+            body["access_token_ttl_in_minutes"] = self.access_token_ttl_in_minutes
         if self.refresh_token_ttl_in_minutes is not None:
-            body['refresh_token_ttl_in_minutes'] = self.refresh_token_ttl_in_minutes
+            body["refresh_token_ttl_in_minutes"] = self.refresh_token_ttl_in_minutes
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> TokenAccessPolicy:
+    def from_dict(cls, d: Dict[str, Any]) -> TokenAccessPolicy:
         """Deserializes the TokenAccessPolicy from a dictionary."""
-        return cls(access_token_ttl_in_minutes=d.get('access_token_ttl_in_minutes', None),
-                   refresh_token_ttl_in_minutes=d.get('refresh_token_ttl_in_minutes', None))
+        return cls(
+            access_token_ttl_in_minutes=d.get("access_token_ttl_in_minutes", None),
+            refresh_token_ttl_in_minutes=d.get("refresh_token_ttl_in_minutes", None),
+        )
 
 
 @dataclass
@@ -821,37 +1037,47 @@ class UpdateCustomAppIntegration:
     def as_dict(self) -> dict:
         """Serializes the UpdateCustomAppIntegration into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
-        if self.redirect_urls: body['redirect_urls'] = [v for v in self.redirect_urls]
-        if self.scopes: body['scopes'] = [v for v in self.scopes]
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy.as_dict()
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
+        if self.redirect_urls:
+            body["redirect_urls"] = [v for v in self.redirect_urls]
+        if self.scopes:
+            body["scopes"] = [v for v in self.scopes]
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy.as_dict()
         if self.user_authorized_scopes:
-            body['user_authorized_scopes'] = [v for v in self.user_authorized_scopes]
+            body["user_authorized_scopes"] = [v for v in self.user_authorized_scopes]
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the UpdateCustomAppIntegration into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
-        if self.redirect_urls: body['redirect_urls'] = self.redirect_urls
-        if self.scopes: body['scopes'] = self.scopes
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy
-        if self.user_authorized_scopes: body['user_authorized_scopes'] = self.user_authorized_scopes
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
+        if self.redirect_urls:
+            body["redirect_urls"] = self.redirect_urls
+        if self.scopes:
+            body["scopes"] = self.scopes
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy
+        if self.user_authorized_scopes:
+            body["user_authorized_scopes"] = self.user_authorized_scopes
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> UpdateCustomAppIntegration:
+    def from_dict(cls, d: Dict[str, Any]) -> UpdateCustomAppIntegration:
         """Deserializes the UpdateCustomAppIntegration from a dictionary."""
-        return cls(integration_id=d.get('integration_id', None),
-                   redirect_urls=d.get('redirect_urls', None),
-                   scopes=d.get('scopes', None),
-                   token_access_policy=_from_dict(d, 'token_access_policy', TokenAccessPolicy),
-                   user_authorized_scopes=d.get('user_authorized_scopes', None))
+        return cls(
+            integration_id=d.get("integration_id", None),
+            redirect_urls=d.get("redirect_urls", None),
+            scopes=d.get("scopes", None),
+            token_access_policy=_from_dict(d, "token_access_policy", TokenAccessPolicy),
+            user_authorized_scopes=d.get("user_authorized_scopes", None),
+        )
 
 
 @dataclass
 class UpdateCustomAppIntegrationOutput:
-
     def as_dict(self) -> dict:
         """Serializes the UpdateCustomAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -863,7 +1089,7 @@ class UpdateCustomAppIntegrationOutput:
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> UpdateCustomAppIntegrationOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> UpdateCustomAppIntegrationOutput:
         """Deserializes the UpdateCustomAppIntegrationOutput from a dictionary."""
         return cls()
 
@@ -878,27 +1104,32 @@ class UpdatePublishedAppIntegration:
     def as_dict(self) -> dict:
         """Serializes the UpdatePublishedAppIntegration into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy.as_dict()
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy.as_dict()
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the UpdatePublishedAppIntegration into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.integration_id is not None: body['integration_id'] = self.integration_id
-        if self.token_access_policy: body['token_access_policy'] = self.token_access_policy
+        if self.integration_id is not None:
+            body["integration_id"] = self.integration_id
+        if self.token_access_policy:
+            body["token_access_policy"] = self.token_access_policy
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> UpdatePublishedAppIntegration:
+    def from_dict(cls, d: Dict[str, Any]) -> UpdatePublishedAppIntegration:
         """Deserializes the UpdatePublishedAppIntegration from a dictionary."""
-        return cls(integration_id=d.get('integration_id', None),
-                   token_access_policy=_from_dict(d, 'token_access_policy', TokenAccessPolicy))
+        return cls(
+            integration_id=d.get("integration_id", None),
+            token_access_policy=_from_dict(d, "token_access_policy", TokenAccessPolicy),
+        )
 
 
 @dataclass
 class UpdatePublishedAppIntegrationOutput:
-
     def as_dict(self) -> dict:
         """Serializes the UpdatePublishedAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -910,27 +1141,27 @@ class UpdatePublishedAppIntegrationOutput:
         return body
 
     @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> UpdatePublishedAppIntegrationOutput:
+    def from_dict(cls, d: Dict[str, Any]) -> UpdatePublishedAppIntegrationOutput:
         """Deserializes the UpdatePublishedAppIntegrationOutput from a dictionary."""
         return cls()
 
 
 class AccountFederationPolicyAPI:
     """These APIs manage account federation policies.
-    
+
     Account federation policies allow users and service principals in your Databricks account to securely
     access Databricks APIs using tokens from your trusted identity providers (IdPs).
-    
+
     With token federation, your users and service principals can exchange tokens from your IdP for Databricks
     OAuth tokens, which can be used to access Databricks APIs. Token federation eliminates the need to manage
     Databricks secrets, and allows you to centralize management of token issuance policies in your IdP.
     Databricks token federation is typically used in combination with [SCIM], so users in your IdP are
     synchronized into your Databricks account.
-    
+
     Token federation is configured in your Databricks account using an account federation policy. An account
     federation policy specifies: * which IdP, or issuer, your Databricks account should accept tokens from *
     how to determine which Databricks user, or subject, a token is issued for
-    
+
     To configure a federation policy, you provide the following: * The required token __issuer__, as specified
     in the “iss” claim of your tokens. The issuer is an https URL that identifies your IdP. * The allowed
     token __audiences__, as specified in the “aud” claim of your tokens. This identifier is intended to
@@ -941,117 +1172,122 @@ class AccountFederationPolicyAPI:
     public keys used to validate the signature of your tokens, in JWKS format. If unspecified (recommended),
     Databricks automatically fetches the public keys from your issuer’s well known endpoint. Databricks
     strongly recommends relying on your issuer’s well known endpoint for discovering public keys.
-    
+
     An example federation policy is: ``` issuer: "https://idp.mycompany.com/oidc" audiences: ["databricks"]
     subject_claim: "sub" ```
-    
+
     An example JWT token body that matches this policy and could be used to authenticate to Databricks as user
     `username@mycompany.com` is: ``` { "iss": "https://idp.mycompany.com/oidc", "aud": "databricks", "sub":
     "username@mycompany.com" } ```
-    
+
     You may also need to configure your IdP to generate tokens for your users to exchange with Databricks, if
     your users do not already have the ability to generate tokens that are compatible with your federation
     policy.
-    
+
     You do not need to configure an OAuth application in Databricks to use token federation.
-    
+
     [SCIM]: https://docs.databricks.com/admin/users-groups/scim/index.html"""
 
     def __init__(self, api_client):
         self._api = api_client
 
-    def create(self,
-               *,
-               policy: Optional[FederationPolicy] = None,
-               policy_id: Optional[str] = None) -> FederationPolicy:
+    def create(self, *, policy: Optional[FederationPolicy] = None, policy_id: Optional[str] = None) -> FederationPolicy:
         """Create account federation policy.
-        
+
         :param policy: :class:`FederationPolicy` (optional)
         :param policy_id: str (optional)
           The identifier for the federation policy. The identifier must contain only lowercase alphanumeric
           characters, numbers, hyphens, and slashes. If unspecified, the id will be assigned by Databricks.
-        
+
         :returns: :class:`FederationPolicy`
         """
         body = policy.as_dict()
         query = {}
-        if policy_id is not None: query['policy_id'] = policy_id
-        headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
-
-        res = self._api.do('POST',
-                           f'/api/2.0/accounts/{self._api.account_id}/federationPolicies',
-                           query=query,
-                           body=body,
-                           headers=headers)
+        if policy_id is not None:
+            query["policy_id"] = policy_id
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
+
+        res = self._api.do(
+            "POST",
+            f"/api/2.0/accounts/{self._api.account_id}/federationPolicies",
+            query=query,
+            body=body,
+            headers=headers,
+        )
         return FederationPolicy.from_dict(res)
 
     def delete(self, policy_id: str):
         """Delete account federation policy.
-        
+
         :param policy_id: str
           The identifier for the federation policy.
-        
-        
+
+
         """
 
-        headers = {'Accept': 'application/json', }
+        headers = {
+            "Accept": "application/json",
+        }
 
-        self._api.do('DELETE',
-                     f'/api/2.0/accounts/{self._api.account_id}/federationPolicies/{policy_id}',
-                     headers=headers)
+        self._api.do(
+            "DELETE", f"/api/2.0/accounts/{self._api.account_id}/federationPolicies/{policy_id}", headers=headers
+        )
 
     def get(self, policy_id: str) -> FederationPolicy:
         """Get account federation policy.
-        
+
         :param policy_id: str
           The identifier for the federation policy.
-        
+
         :returns: :class:`FederationPolicy`
         """
 
-        headers = {'Accept': 'application/json', }
+        headers = {
+            "Accept": "application/json",
+        }
 
-        res = self._api.do('GET',
-                           f'/api/2.0/accounts/{self._api.account_id}/federationPolicies/{policy_id}',
-                           headers=headers)
+        res = self._api.do(
+            "GET", f"/api/2.0/accounts/{self._api.account_id}/federationPolicies/{policy_id}", headers=headers
+        )
         return FederationPolicy.from_dict(res)
 
-    def list(self,
-             *,
-             page_size: Optional[int] = None,
-             page_token: Optional[str] = None) -> Iterator[FederationPolicy]:
+    def list(self, *, page_size: Optional[int] = None, page_token: Optional[str] = None) -> Iterator[FederationPolicy]:
         """List account federation policies.
-        
+
         :param page_size: int (optional)
         :param page_token: str (optional)
-        
+
         :returns: Iterator over :class:`FederationPolicy`
         """
 
         query = {}
-        if page_size is not None: query['page_size'] = page_size
-        if page_token is not None: query['page_token'] = page_token
-        headers = {'Accept': 'application/json', }
+        if page_size is not None:
+            query["page_size"] = page_size
+        if page_token is not None:
+            query["page_token"] = page_token
+        headers = {
+            "Accept": "application/json",
+        }
 
         while True:
-            json = self._api.do('GET',
-                                f'/api/2.0/accounts/{self._api.account_id}/federationPolicies',
-                                query=query,
-                                headers=headers)
-            if 'policies' in json:
-                for v in json['policies']:
+            json = self._api.do(
+                "GET", f"/api/2.0/accounts/{self._api.account_id}/federationPolicies", query=query, headers=headers
+            )
+            if "policies" in json:
+                for v in json["policies"]:
                     yield FederationPolicy.from_dict(v)
-            if 'next_page_token' not in json or not json['next_page_token']:
+            if "next_page_token" not in json or not json["next_page_token"]:
                 return
-            query['page_token'] = json['next_page_token']
+            query["page_token"] = json["next_page_token"]
 
-    def update(self,
-               policy_id: str,
-               *,
-               policy: Optional[FederationPolicy] = None,
-               update_mask: Optional[str] = None) -> FederationPolicy:
+    def update(
+        self, policy_id: str, *, policy: Optional[FederationPolicy] = None, update_mask: Optional[str] = None
+    ) -> FederationPolicy:
         """Update account federation policy.
-        
+
         :param policy_id: str
           The identifier for the federation policy.
         :param policy: :class:`FederationPolicy` (optional)
@@ -1061,19 +1297,25 @@ class AccountFederationPolicyAPI:
           should be updated (full replacement). If unspecified, all fields that are set in the policy provided
           in the update request will overwrite the corresponding fields in the existing policy. Example value:
           'description,oidc_policy.audiences'.
-        
+
         :returns: :class:`FederationPolicy`
         """
         body = policy.as_dict()
         query = {}
-        if update_mask is not None: query['update_mask'] = update_mask
-        headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
-
-        res = self._api.do('PATCH',
-                           f'/api/2.0/accounts/{self._api.account_id}/federationPolicies/{policy_id}',
-                           query=query,
-                           body=body,
-                           headers=headers)
+        if update_mask is not None:
+            query["update_mask"] = update_mask
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
+
+        res = self._api.do(
+            "PATCH",
+            f"/api/2.0/accounts/{self._api.account_id}/federationPolicies/{policy_id}",
+            query=query,
+            body=body,
+            headers=headers,
+        )
         return FederationPolicy.from_dict(res)
 
 
@@ -1084,20 +1326,22 @@ class CustomAppIntegrationAPI:
     def __init__(self, api_client):
         self._api = api_client
 
-    def create(self,
-               *,
-               confidential: Optional[bool] = None,
-               name: Optional[str] = None,
-               redirect_urls: Optional[List[str]] = None,
-               scopes: Optional[List[str]] = None,
-               token_access_policy: Optional[TokenAccessPolicy] = None,
-               user_authorized_scopes: Optional[List[str]] = None) -> CreateCustomAppIntegrationOutput:
+    def create(
+        self,
+        *,
+        confidential: Optional[bool] = None,
+        name: Optional[str] = None,
+        redirect_urls: Optional[List[str]] = None,
+        scopes: Optional[List[str]] = None,
+        token_access_policy: Optional[TokenAccessPolicy] = None,
+        user_authorized_scopes: Optional[List[str]] = None,
+    ) -> CreateCustomAppIntegrationOutput:
         """Create Custom OAuth App Integration.
-        
+
         Create Custom OAuth App Integration.
-        
+
         You can retrieve the custom OAuth app integration via :method:CustomAppIntegration/get.
-        
+
         :param confidential: bool (optional)
           This field indicates whether an OAuth client secret is required to authenticate this client.
         :param name: str (optional)
@@ -1112,108 +1356,135 @@ class CustomAppIntegrationAPI:
         :param user_authorized_scopes: List[str] (optional)
           Scopes that will need to be consented by end user to mint the access token. If the user does not
           authorize the access token will not be minted. Must be a subset of scopes.
-        
+
         :returns: :class:`CreateCustomAppIntegrationOutput`
         """
         body = {}
-        if confidential is not None: body['confidential'] = confidential
-        if name is not None: body['name'] = name
-        if redirect_urls is not None: body['redirect_urls'] = [v for v in redirect_urls]
-        if scopes is not None: body['scopes'] = [v for v in scopes]
-        if token_access_policy is not None: body['token_access_policy'] = token_access_policy.as_dict()
+        if confidential is not None:
+            body["confidential"] = confidential
+        if name is not None:
+            body["name"] = name
+        if redirect_urls is not None:
+            body["redirect_urls"] = [v for v in redirect_urls]
+        if scopes is not None:
+            body["scopes"] = [v for v in scopes]
+        if token_access_policy is not None:
+            body["token_access_policy"] = token_access_policy.as_dict()
         if user_authorized_scopes is not None:
-            body['user_authorized_scopes'] = [v for v in user_authorized_scopes]
-        headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+            body["user_authorized_scopes"] = [v for v in user_authorized_scopes]
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
 
-        res = self._api.do('POST',
-                           f'/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations',
-                           body=body,
-                           headers=headers)
+        res = self._api.do(
+            "POST",
+            f"/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations",
+            body=body,
+            headers=headers,
+        )
         return CreateCustomAppIntegrationOutput.from_dict(res)
 
     def delete(self, integration_id: str):
         """Delete Custom OAuth App Integration.
-        
+
         Delete an existing Custom OAuth App Integration. You can retrieve the custom OAuth app integration via
         :method:CustomAppIntegration/get.
-        
+
         :param integration_id: str
-        
-        
+
+
         """
 
-        headers = {'Accept': 'application/json', }
+        headers = {
+            "Accept": "application/json",
+        }
 
         self._api.do(
-            'DELETE',
-            f'/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations/{integration_id}',
-            headers=headers)
+            "DELETE",
+            f"/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations/{integration_id}",
+            headers=headers,
+        )
 
     def get(self, integration_id: str) -> GetCustomAppIntegrationOutput:
         """Get OAuth Custom App Integration.
-        
+
         Gets the Custom OAuth App Integration for the given integration id.
-        
+
         :param integration_id: str
           The OAuth app integration ID.
-        
+
         :returns: :class:`GetCustomAppIntegrationOutput`
         """
 
-        headers = {'Accept': 'application/json', }
+        headers = {
+            "Accept": "application/json",
+        }
 
         res = self._api.do(
-            'GET',
-            f'/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations/{integration_id}',
-            headers=headers)
+            "GET",
+            f"/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations/{integration_id}",
+            headers=headers,
+        )
         return GetCustomAppIntegrationOutput.from_dict(res)
 
-    def list(self,
-             *,
-             include_creator_username: Optional[bool] = None,
-             page_size: Optional[int] = None,
-             page_token: Optional[str] = None) -> Iterator[GetCustomAppIntegrationOutput]:
+    def list(
+        self,
+        *,
+        include_creator_username: Optional[bool] = None,
+        page_size: Optional[int] = None,
+        page_token: Optional[str] = None,
+    ) -> Iterator[GetCustomAppIntegrationOutput]:
         """Get custom oauth app integrations.
-        
+
         Get the list of custom OAuth app integrations for the specified Databricks account
-        
+
         :param include_creator_username: bool (optional)
         :param page_size: int (optional)
         :param page_token: str (optional)
-        
+
         :returns: Iterator over :class:`GetCustomAppIntegrationOutput`
         """
 
         query = {}
-        if include_creator_username is not None: query['include_creator_username'] = include_creator_username
-        if page_size is not None: query['page_size'] = page_size
-        if page_token is not None: query['page_token'] = page_token
-        headers = {'Accept': 'application/json', }
+        if include_creator_username is not None:
+            query["include_creator_username"] = include_creator_username
+        if page_size is not None:
+            query["page_size"] = page_size
+        if page_token is not None:
+            query["page_token"] = page_token
+        headers = {
+            "Accept": "application/json",
+        }
 
         while True:
-            json = self._api.do('GET',
-                                f'/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations',
-                                query=query,
-                                headers=headers)
-            if 'apps' in json:
-                for v in json['apps']:
+            json = self._api.do(
+                "GET",
+                f"/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations",
+                query=query,
+                headers=headers,
+            )
+            if "apps" in json:
+                for v in json["apps"]:
                     yield GetCustomAppIntegrationOutput.from_dict(v)
-            if 'next_page_token' not in json or not json['next_page_token']:
+            if "next_page_token" not in json or not json["next_page_token"]:
                 return
-            query['page_token'] = json['next_page_token']
-
-    def update(self,
-               integration_id: str,
-               *,
-               redirect_urls: Optional[List[str]] = None,
-               scopes: Optional[List[str]] = None,
-               token_access_policy: Optional[TokenAccessPolicy] = None,
-               user_authorized_scopes: Optional[List[str]] = None):
+            query["page_token"] = json["next_page_token"]
+
+    def update(
+        self,
+        integration_id: str,
+        *,
+        redirect_urls: Optional[List[str]] = None,
+        scopes: Optional[List[str]] = None,
+        token_access_policy: Optional[TokenAccessPolicy] = None,
+        user_authorized_scopes: Optional[List[str]] = None,
+    ):
         """Updates Custom OAuth App Integration.
-        
+
         Updates an existing custom OAuth App Integration. You can retrieve the custom OAuth app integration
         via :method:CustomAppIntegration/get.
-        
+
         :param integration_id: str
         :param redirect_urls: List[str] (optional)
           List of OAuth redirect urls to be updated in the custom OAuth app integration
@@ -1225,22 +1496,29 @@ class CustomAppIntegrationAPI:
         :param user_authorized_scopes: List[str] (optional)
           Scopes that will need to be consented by end user to mint the access token. If the user does not
           authorize the access token will not be minted. Must be a subset of scopes.
-        
-        
+
+
         """
         body = {}
-        if redirect_urls is not None: body['redirect_urls'] = [v for v in redirect_urls]
-        if scopes is not None: body['scopes'] = [v for v in scopes]
-        if token_access_policy is not None: body['token_access_policy'] = token_access_policy.as_dict()
+        if redirect_urls is not None:
+            body["redirect_urls"] = [v for v in redirect_urls]
+        if scopes is not None:
+            body["scopes"] = [v for v in scopes]
+        if token_access_policy is not None:
+            body["token_access_policy"] = token_access_policy.as_dict()
         if user_authorized_scopes is not None:
-            body['user_authorized_scopes'] = [v for v in user_authorized_scopes]
-        headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+            body["user_authorized_scopes"] = [v for v in user_authorized_scopes]
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
 
         self._api.do(
-            'PATCH',
-            f'/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations/{integration_id}',
+            "PATCH",
+            f"/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations/{integration_id}",
             body=body,
-            headers=headers)
+            headers=headers,
+        )
 
 
 class OAuthPublishedAppsAPI:
@@ -1251,38 +1529,40 @@ class OAuthPublishedAppsAPI:
     def __init__(self, api_client):
         self._api = api_client
 
-    def list(self,
-             *,
-             page_size: Optional[int] = None,
-             page_token: Optional[str] = None) -> Iterator[PublishedAppOutput]:
+    def list(
+        self, *, page_size: Optional[int] = None, page_token: Optional[str] = None
+    ) -> Iterator[PublishedAppOutput]:
         """Get all the published OAuth apps.
-        
+
         Get all the available published OAuth apps in Databricks.
-        
+
         :param page_size: int (optional)
           The max number of OAuth published apps to return in one page.
         :param page_token: str (optional)
           A token that can be used to get the next page of results.
-        
+
         :returns: Iterator over :class:`PublishedAppOutput`
         """
 
         query = {}
-        if page_size is not None: query['page_size'] = page_size
-        if page_token is not None: query['page_token'] = page_token
-        headers = {'Accept': 'application/json', }
+        if page_size is not None:
+            query["page_size"] = page_size
+        if page_token is not None:
+            query["page_token"] = page_token
+        headers = {
+            "Accept": "application/json",
+        }
 
         while True:
-            json = self._api.do('GET',
-                                f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-apps',
-                                query=query,
-                                headers=headers)
-            if 'apps' in json:
-                for v in json['apps']:
+            json = self._api.do(
+                "GET", f"/api/2.0/accounts/{self._api.account_id}/oauth2/published-apps", query=query, headers=headers
+            )
+            if "apps" in json:
+                for v in json["apps"]:
                     yield PublishedAppOutput.from_dict(v)
-            if 'next_page_token' not in json or not json['next_page_token']:
+            if "next_page_token" not in json or not json["next_page_token"]:
                 return
-            query['page_token'] = json['next_page_token']
+            query["page_token"] = json["next_page_token"]
 
 
 class PublishedAppIntegrationAPI:
@@ -1293,142 +1573,163 @@ class PublishedAppIntegrationAPI:
         self._api = api_client
 
     def create(
-            self,
-            *,
-            app_id: Optional[str] = None,
-            token_access_policy: Optional[TokenAccessPolicy] = None) -> CreatePublishedAppIntegrationOutput:
+        self, *, app_id: Optional[str] = None, token_access_policy: Optional[TokenAccessPolicy] = None
+    ) -> CreatePublishedAppIntegrationOutput:
         """Create Published OAuth App Integration.
-        
+
         Create Published OAuth App Integration.
-        
+
         You can retrieve the published OAuth app integration via :method:PublishedAppIntegration/get.
-        
+
         :param app_id: str (optional)
           App id of the OAuth published app integration. For example power-bi, tableau-deskop
         :param token_access_policy: :class:`TokenAccessPolicy` (optional)
           Token access policy
-        
+
         :returns: :class:`CreatePublishedAppIntegrationOutput`
         """
         body = {}
-        if app_id is not None: body['app_id'] = app_id
-        if token_access_policy is not None: body['token_access_policy'] = token_access_policy.as_dict()
-        headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
-
-        res = self._api.do('POST',
-                           f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations',
-                           body=body,
-                           headers=headers)
+        if app_id is not None:
+            body["app_id"] = app_id
+        if token_access_policy is not None:
+            body["token_access_policy"] = token_access_policy.as_dict()
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
+
+        res = self._api.do(
+            "POST",
+            f"/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations",
+            body=body,
+            headers=headers,
+        )
         return CreatePublishedAppIntegrationOutput.from_dict(res)
 
     def delete(self, integration_id: str):
         """Delete Published OAuth App Integration.
-        
+
         Delete an existing Published OAuth App Integration. You can retrieve the published OAuth app
         integration via :method:PublishedAppIntegration/get.
-        
+
         :param integration_id: str
-        
-        
+
+
         """
 
-        headers = {'Accept': 'application/json', }
+        headers = {
+            "Accept": "application/json",
+        }
 
         self._api.do(
-            'DELETE',
-            f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations/{integration_id}',
-            headers=headers)
+            "DELETE",
+            f"/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations/{integration_id}",
+            headers=headers,
+        )
 
     def get(self, integration_id: str) -> GetPublishedAppIntegrationOutput:
         """Get OAuth Published App Integration.
-        
+
         Gets the Published OAuth App Integration for the given integration id.
-        
+
         :param integration_id: str
-        
+
         :returns: :class:`GetPublishedAppIntegrationOutput`
         """
 
-        headers = {'Accept': 'application/json', }
+        headers = {
+            "Accept": "application/json",
+        }
 
         res = self._api.do(
-            'GET',
-            f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations/{integration_id}',
-            headers=headers)
+            "GET",
+            f"/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations/{integration_id}",
+            headers=headers,
+        )
         return GetPublishedAppIntegrationOutput.from_dict(res)
 
-    def list(self,
-             *,
-             page_size: Optional[int] = None,
-             page_token: Optional[str] = None) -> Iterator[GetPublishedAppIntegrationOutput]:
+    def list(
+        self, *, page_size: Optional[int] = None, page_token: Optional[str] = None
+    ) -> Iterator[GetPublishedAppIntegrationOutput]:
         """Get published oauth app integrations.
-        
+
         Get the list of published OAuth app integrations for the specified Databricks account
-        
+
         :param page_size: int (optional)
         :param page_token: str (optional)
-        
+
         :returns: Iterator over :class:`GetPublishedAppIntegrationOutput`
         """
 
         query = {}
-        if page_size is not None: query['page_size'] = page_size
-        if page_token is not None: query['page_token'] = page_token
-        headers = {'Accept': 'application/json', }
+        if page_size is not None:
+            query["page_size"] = page_size
+        if page_token is not None:
+            query["page_token"] = page_token
+        headers = {
+            "Accept": "application/json",
+        }
 
         while True:
-            json = self._api.do('GET',
-                                f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations',
-                                query=query,
-                                headers=headers)
-            if 'apps' in json:
-                for v in json['apps']:
+            json = self._api.do(
+                "GET",
+                f"/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations",
+                query=query,
+                headers=headers,
+            )
+            if "apps" in json:
+                for v in json["apps"]:
                     yield GetPublishedAppIntegrationOutput.from_dict(v)
-            if 'next_page_token' not in json or not json['next_page_token']:
+            if "next_page_token" not in json or not json["next_page_token"]:
                 return
-            query['page_token'] = json['next_page_token']
+            query["page_token"] = json["next_page_token"]
 
     def update(self, integration_id: str, *, token_access_policy: Optional[TokenAccessPolicy] = None):
         """Updates Published OAuth App Integration.
-        
+
         Updates an existing published OAuth App Integration. You can retrieve the published OAuth app
         integration via :method:PublishedAppIntegration/get.
-        
+
         :param integration_id: str
         :param token_access_policy: :class:`TokenAccessPolicy` (optional)
           Token access policy to be updated in the published OAuth app integration
-        
-        
+
+
         """
         body = {}
-        if token_access_policy is not None: body['token_access_policy'] = token_access_policy.as_dict()
-        headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+        if token_access_policy is not None:
+            body["token_access_policy"] = token_access_policy.as_dict()
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
 
         self._api.do(
-            'PATCH',
-            f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations/{integration_id}',
+            "PATCH",
+            f"/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations/{integration_id}",
             body=body,
-            headers=headers)
+            headers=headers,
+        )
 
 
 class ServicePrincipalFederationPolicyAPI:
     """These APIs manage service principal federation policies.
-    
+
     Service principal federation, also known as Workload Identity Federation, allows your automated workloads
     running outside of Databricks to securely access Databricks APIs without the need for Databricks secrets.
     With Workload Identity Federation, your application (or workload) authenticates to Databricks as a
     Databricks service principal, using tokens provided by the workload runtime.
-    
+
     Databricks strongly recommends using Workload Identity Federation to authenticate to Databricks from
     automated workloads, over alternatives such as OAuth client secrets or Personal Access Tokens, whenever
     possible. Workload Identity Federation is supported by many popular services, including Github Actions,
     Azure DevOps, GitLab, Terraform Cloud, and Kubernetes clusters, among others.
-    
+
     Workload identity federation is configured in your Databricks account using a service principal federation
     policy. A service principal federation policy specifies: * which IdP, or issuer, the service principal is
     allowed to authenticate from * which workload identity, or subject, is allowed to authenticate as the
     Databricks service principal
-    
+
     To configure a federation policy, you provide the following: * The required token __issuer__, as specified
     in the “iss” claim of workload identity tokens. The issuer is an https URL that identifies the
     workload identity provider. * The required token __subject__, as specified in the “sub” claim of
@@ -1440,129 +1741,143 @@ class ServicePrincipalFederationPolicyAPI:
     of the workload identity tokens, in JWKS format. If unspecified (recommended), Databricks automatically
     fetches the public keys from the issuer’s well known endpoint. Databricks strongly recommends relying on
     the issuer’s well known endpoint for discovering public keys.
-    
+
     An example service principal federation policy, for a Github Actions workload, is: ``` issuer:
     "https://token.actions.githubusercontent.com" audiences: ["https://github.com/my-github-org"] subject:
     "repo:my-github-org/my-repo:environment:prod" ```
-    
+
     An example JWT token body that matches this policy and could be used to authenticate to Databricks is: ```
     { "iss": "https://token.actions.githubusercontent.com", "aud": "https://github.com/my-github-org", "sub":
     "repo:my-github-org/my-repo:environment:prod" } ```
-    
+
     You may also need to configure the workload runtime to generate tokens for your workloads.
-    
+
     You do not need to configure an OAuth application in Databricks to use token federation."""
 
     def __init__(self, api_client):
         self._api = api_client
 
-    def create(self,
-               service_principal_id: int,
-               *,
-               policy: Optional[FederationPolicy] = None,
-               policy_id: Optional[str] = None) -> FederationPolicy:
+    def create(
+        self, service_principal_id: int, *, policy: Optional[FederationPolicy] = None, policy_id: Optional[str] = None
+    ) -> FederationPolicy:
         """Create service principal federation policy.
-        
+
         :param service_principal_id: int
           The service principal id for the federation policy.
         :param policy: :class:`FederationPolicy` (optional)
         :param policy_id: str (optional)
           The identifier for the federation policy. The identifier must contain only lowercase alphanumeric
           characters, numbers, hyphens, and slashes. If unspecified, the id will be assigned by Databricks.
-        
+
         :returns: :class:`FederationPolicy`
         """
         body = policy.as_dict()
         query = {}
-        if policy_id is not None: query['policy_id'] = policy_id
-        headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+        if policy_id is not None:
+            query["policy_id"] = policy_id
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
 
         res = self._api.do(
-            'POST',
-            f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/federationPolicies',
+            "POST",
+            f"/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/federationPolicies",
             query=query,
             body=body,
-            headers=headers)
+            headers=headers,
+        )
         return FederationPolicy.from_dict(res)
 
     def delete(self, service_principal_id: int, policy_id: str):
         """Delete service principal federation policy.
-        
+
         :param service_principal_id: int
           The service principal id for the federation policy.
         :param policy_id: str
           The identifier for the federation policy.
-        
-        
+
+
         """
 
-        headers = {'Accept': 'application/json', }
+        headers = {
+            "Accept": "application/json",
+        }
 
         self._api.do(
-            'DELETE',
-            f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/federationPolicies/{policy_id}',
-            headers=headers)
+            "DELETE",
+            f"/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/federationPolicies/{policy_id}",
+            headers=headers,
+        )
 
     def get(self, service_principal_id: int, policy_id: str) -> FederationPolicy:
         """Get service principal federation policy.
-        
+
         :param service_principal_id: int
           The service principal id for the federation policy.
         :param policy_id: str
           The identifier for the federation policy.
-        
+
         :returns: :class:`FederationPolicy`
         """
 
-        headers = {'Accept': 'application/json', }
+        headers = {
+            "Accept": "application/json",
+        }
 
         res = self._api.do(
-            'GET',
-            f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/federationPolicies/{policy_id}',
-            headers=headers)
+            "GET",
+            f"/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/federationPolicies/{policy_id}",
+            headers=headers,
+        )
         return FederationPolicy.from_dict(res)
 
-    def list(self,
-             service_principal_id: int,
-             *,
-             page_size: Optional[int] = None,
-             page_token: Optional[str] = None) -> Iterator[FederationPolicy]:
+    def list(
+        self, service_principal_id: int, *, page_size: Optional[int] = None, page_token: Optional[str] = None
+    ) -> Iterator[FederationPolicy]:
         """List service principal federation policies.
-        
+
         :param service_principal_id: int
           The service principal id for the federation policy.
         :param page_size: int (optional)
         :param page_token: str (optional)
-        
+
         :returns: Iterator over :class:`FederationPolicy`
         """
 
         query = {}
-        if page_size is not None: query['page_size'] = page_size
-        if page_token is not None: query['page_token'] = page_token
-        headers = {'Accept': 'application/json', }
+        if page_size is not None:
+            query["page_size"] = page_size
+        if page_token is not None:
+            query["page_token"] = page_token
+        headers = {
+            "Accept": "application/json",
+        }
 
         while True:
             json = self._api.do(
-                'GET',
-                f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/federationPolicies',
+                "GET",
+                f"/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/federationPolicies",
                 query=query,
-                headers=headers)
-            if 'policies' in json:
-                for v in json['policies']:
+                headers=headers,
+            )
+            if "policies" in json:
+                for v in json["policies"]:
                     yield FederationPolicy.from_dict(v)
-            if 'next_page_token' not in json or not json['next_page_token']:
+            if "next_page_token" not in json or not json["next_page_token"]:
                 return
-            query['page_token'] = json['next_page_token']
-
-    def update(self,
-               service_principal_id: int,
-               policy_id: str,
-               *,
-               policy: Optional[FederationPolicy] = None,
-               update_mask: Optional[str] = None) -> FederationPolicy:
+            query["page_token"] = json["next_page_token"]
+
+    def update(
+        self,
+        service_principal_id: int,
+        policy_id: str,
+        *,
+        policy: Optional[FederationPolicy] = None,
+        update_mask: Optional[str] = None,
+    ) -> FederationPolicy:
         """Update service principal federation policy.
-        
+
         :param service_principal_id: int
           The service principal id for the federation policy.
         :param policy_id: str
@@ -1574,84 +1889,103 @@ class ServicePrincipalFederationPolicyAPI:
           should be updated (full replacement). If unspecified, all fields that are set in the policy provided
           in the update request will overwrite the corresponding fields in the existing policy. Example value:
           'description,oidc_policy.audiences'.
-        
+
         :returns: :class:`FederationPolicy`
         """
         body = policy.as_dict()
         query = {}
-        if update_mask is not None: query['update_mask'] = update_mask
-        headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+        if update_mask is not None:
+            query["update_mask"] = update_mask
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
 
         res = self._api.do(
-            'PATCH',
-            f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/federationPolicies/{policy_id}',
+            "PATCH",
+            f"/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/federationPolicies/{policy_id}",
             query=query,
             body=body,
-            headers=headers)
+            headers=headers,
+        )
         return FederationPolicy.from_dict(res)
 
 
 class ServicePrincipalSecretsAPI:
     """These APIs enable administrators to manage service principal secrets.
-    
+
     You can use the generated secrets to obtain OAuth access tokens for a service principal, which can then be
     used to access Databricks Accounts and Workspace APIs. For more information, see [Authentication using
     OAuth tokens for service principals],
-    
+
     In addition, the generated secrets can be used to configure the Databricks Terraform Provider to
     authenticate with the service principal. For more information, see [Databricks Terraform Provider].
-    
+
     [Authentication using OAuth tokens for service principals]: https://docs.databricks.com/dev-tools/authentication-oauth.html
-    [Databricks Terraform Provider]: https://github.com/databricks/terraform-provider-databricks/blob/master/docs/index.md#authenticating-with-service-principal"""
+    [Databricks Terraform Provider]: https://github.com/databricks/terraform-provider-databricks/blob/master/docs/index.md#authenticating-with-service-principal
+    """
 
     def __init__(self, api_client):
         self._api = api_client
 
-    def create(self, service_principal_id: int) -> CreateServicePrincipalSecretResponse:
+    def create(
+        self, service_principal_id: int, *, lifetime: Optional[str] = None
+    ) -> CreateServicePrincipalSecretResponse:
         """Create service principal secret.
-        
+
         Create a secret for the given service principal.
-        
+
         :param service_principal_id: int
           The service principal ID.
-        
+        :param lifetime: str (optional)
+          The lifetime of the secret in seconds. If this parameter is not provided, the secret will have a
+          default lifetime of 730 days (63072000s).
+
         :returns: :class:`CreateServicePrincipalSecretResponse`
         """
-
-        headers = {'Accept': 'application/json', }
+        body = {}
+        if lifetime is not None:
+            body["lifetime"] = lifetime
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
 
         res = self._api.do(
-            'POST',
-            f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/credentials/secrets',
-            headers=headers)
+            "POST",
+            f"/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/credentials/secrets",
+            body=body,
+            headers=headers,
+        )
         return CreateServicePrincipalSecretResponse.from_dict(res)
 
     def delete(self, service_principal_id: int, secret_id: str):
         """Delete service principal secret.
-        
+
         Delete a secret from the given service principal.
-        
+
         :param service_principal_id: int
           The service principal ID.
         :param secret_id: str
           The secret ID.
-        
-        
+
+
         """
 
         headers = {}
 
         self._api.do(
-            'DELETE',
-            f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/credentials/secrets/{secret_id}',
-            headers=headers)
+            "DELETE",
+            f"/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/credentials/secrets/{secret_id}",
+            headers=headers,
+        )
 
     def list(self, service_principal_id: int, *, page_token: Optional[str] = None) -> Iterator[SecretInfo]:
         """List service principal secrets.
-        
+
         List all secrets associated with the given service principal. This operation only returns information
         about the secrets themselves and does not include the secret values.
-        
+
         :param service_principal_id: int
           The service principal ID.
         :param page_token: str (optional)
@@ -1661,23 +1995,27 @@ class ServicePrincipalSecretsAPI:
           previous request. To list all of the secrets for a service principal, it is necessary to continue
           requesting pages of entries until the response contains no `next_page_token`. Note that the number
           of entries returned must not be used to determine when the listing is complete.
-        
+
         :returns: Iterator over :class:`SecretInfo`
         """
 
         query = {}
-        if page_token is not None: query['page_token'] = page_token
-        headers = {'Accept': 'application/json', }
+        if page_token is not None:
+            query["page_token"] = page_token
+        headers = {
+            "Accept": "application/json",
+        }
 
         while True:
             json = self._api.do(
-                'GET',
-                f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/credentials/secrets',
+                "GET",
+                f"/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/credentials/secrets",
                 query=query,
-                headers=headers)
-            if 'secrets' in json:
-                for v in json['secrets']:
+                headers=headers,
+            )
+            if "secrets" in json:
+                for v in json["secrets"]:
                     yield SecretInfo.from_dict(v)
-            if 'next_page_token' not in json or not json['next_page_token']:
+            if "next_page_token" not in json or not json["next_page_token"]:
                 return
-            query['page_token'] = json['next_page_token']
+            query["page_token"] = json["next_page_token"]