databricks-sdk 0.40.0__py3-none-any.whl → 0.42.0__py3-none-any.whl

databricks/sdk/service/billing.py

@@ -11,6 +11,8 @@ from ._internal import _enum, _from_dict, _repeated_dict
 
 _LOG = logging.getLogger('databricks.sdk')
 
+from databricks.sdk.service import compute
+
 # all definitions in this file are in alphabetical order
 
 
@@ -311,6 +313,44 @@ class BudgetConfigurationFilterWorkspaceIdClause:
                    values=d.get('values', None))
 
 
+@dataclass
+class BudgetPolicy:
+    """Contains the BudgetPolicy details."""
+
+    policy_id: str
+    """The Id of the policy. This field is generated by Databricks and globally unique."""
+
+    custom_tags: Optional[List[compute.CustomPolicyTag]] = None
+    """A list of tags defined by the customer. At most 20 entries are allowed per policy."""
+
+    policy_name: Optional[str] = None
+    """The name of the policy. - Must be unique among active policies. - Can contain only characters
+    from the ISO 8859-1 (latin1) set."""
+
+    def as_dict(self) -> dict:
+        """Serializes the BudgetPolicy into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.custom_tags: body['custom_tags'] = [v.as_dict() for v in self.custom_tags]
+        if self.policy_id is not None: body['policy_id'] = self.policy_id
+        if self.policy_name is not None: body['policy_name'] = self.policy_name
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the BudgetPolicy into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.custom_tags: body['custom_tags'] = self.custom_tags
+        if self.policy_id is not None: body['policy_id'] = self.policy_id
+        if self.policy_name is not None: body['policy_name'] = self.policy_name
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> BudgetPolicy:
+        """Deserializes the BudgetPolicy from a dictionary."""
+        return cls(custom_tags=_repeated_dict(d, 'custom_tags', compute.CustomPolicyTag),
+                   policy_id=d.get('policy_id', None),
+                   policy_name=d.get('policy_name', None))
+
+
 @dataclass
 class CreateBillingUsageDashboardRequest:
     dashboard_type: Optional[UsageDashboardType] = None
@@ -536,6 +576,45 @@ class CreateBudgetConfigurationResponse:
         return cls(budget=_from_dict(d, 'budget', BudgetConfiguration))
 
 
+@dataclass
+class CreateBudgetPolicyRequest:
+    """A request to create a BudgetPolicy."""
+
+    custom_tags: Optional[List[compute.CustomPolicyTag]] = None
+    """A list of tags defined by the customer. At most 40 entries are allowed per policy."""
+
+    policy_name: Optional[str] = None
+    """The name of the policy. - Must be unique among active policies. - Can contain only characters of
+    0-9, a-z, A-Z, -, =, ., :, /, @, _, +, whitespace."""
+
+    request_id: Optional[str] = None
+    """A unique identifier for this request. Restricted to 36 ASCII characters. A random UUID is
+    recommended. This request is only idempotent if a `request_id` is provided."""
+
+    def as_dict(self) -> dict:
+        """Serializes the CreateBudgetPolicyRequest into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.custom_tags: body['custom_tags'] = [v.as_dict() for v in self.custom_tags]
+        if self.policy_name is not None: body['policy_name'] = self.policy_name
+        if self.request_id is not None: body['request_id'] = self.request_id
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the CreateBudgetPolicyRequest into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.custom_tags: body['custom_tags'] = self.custom_tags
+        if self.policy_name is not None: body['policy_name'] = self.policy_name
+        if self.request_id is not None: body['request_id'] = self.request_id
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> CreateBudgetPolicyRequest:
+        """Deserializes the CreateBudgetPolicyRequest from a dictionary."""
+        return cls(custom_tags=_repeated_dict(d, 'custom_tags', compute.CustomPolicyTag),
+                   policy_name=d.get('policy_name', None),
+                   request_id=d.get('request_id', None))
+
+
 @dataclass
 class CreateLogDeliveryConfigurationParams:
     log_type: LogType
@@ -670,6 +749,25 @@ class DeleteBudgetConfigurationResponse:
         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 = {}
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the DeleteResponse into a shallow dictionary of its immediate attributes."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteResponse:
+        """Deserializes the DeleteResponse from a dictionary."""
+        return cls()
+
+
 class DeliveryStatus(Enum):
     """The status string for log delivery. Possible values are: * `CREATED`: There were no log delivery
     attempts since the config was created. * `SUCCEEDED`: The latest attempt of log delivery has
@@ -708,6 +806,44 @@ class DownloadResponse:
         return cls(contents=d.get('contents', None))
 
 
+@dataclass
+class Filter:
+    """Structured representation of a filter to be applied to a list of policies. All specified filters
+    will be applied in conjunction."""
+
+    creator_user_id: Optional[int] = None
+    """The policy creator user id to be filtered on. If unspecified, all policies will be returned."""
+
+    creator_user_name: Optional[str] = None
+    """The policy creator user name to be filtered on. If unspecified, all policies will be returned."""
+
+    policy_name: Optional[str] = None
+    """The partial name of policies to be filtered on. If unspecified, all policies will be returned."""
+
+    def as_dict(self) -> dict:
+        """Serializes the Filter into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.creator_user_id is not None: body['creator_user_id'] = self.creator_user_id
+        if self.creator_user_name is not None: body['creator_user_name'] = self.creator_user_name
+        if self.policy_name is not None: body['policy_name'] = self.policy_name
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the Filter into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.creator_user_id is not None: body['creator_user_id'] = self.creator_user_id
+        if self.creator_user_name is not None: body['creator_user_name'] = self.creator_user_name
+        if self.policy_name is not None: body['policy_name'] = self.policy_name
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> Filter:
+        """Deserializes the Filter from a dictionary."""
+        return cls(creator_user_id=d.get('creator_user_id', None),
+                   creator_user_name=d.get('creator_user_name', None),
+                   policy_name=d.get('policy_name', None))
+
+
 @dataclass
 class GetBillingUsageDashboardResponse:
     dashboard_id: Optional[str] = None
@@ -787,6 +923,44 @@ class ListBudgetConfigurationsResponse:
                    next_page_token=d.get('next_page_token', None))
 
 
+@dataclass
+class ListBudgetPoliciesResponse:
+    """A list of policies."""
+
+    next_page_token: Optional[str] = None
+    """A token that can be sent as `page_token` to retrieve the next page. If this field is omitted,
+    there are no subsequent pages."""
+
+    policies: Optional[List[BudgetPolicy]] = None
+
+    previous_page_token: Optional[str] = None
+    """A token that can be sent as `page_token` to retrieve the previous page. In this field is
+    omitted, there are no previous pages."""
+
+    def as_dict(self) -> dict:
+        """Serializes the ListBudgetPoliciesResponse 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.previous_page_token is not None: body['previous_page_token'] = self.previous_page_token
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the ListBudgetPoliciesResponse 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.previous_page_token is not None: body['previous_page_token'] = self.previous_page_token
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> ListBudgetPoliciesResponse:
+        """Deserializes the ListBudgetPoliciesResponse from a dictionary."""
+        return cls(next_page_token=d.get('next_page_token', None),
+                   policies=_repeated_dict(d, 'policies', BudgetPolicy),
+                   previous_page_token=d.get('previous_page_token', None))
+
+
 class LogDeliveryConfigStatus(Enum):
     """Status of log delivery configuration. Set to `ENABLED` (enabled) or `DISABLED` (disabled).
     Defaults to `ENABLED`. You can [enable or disable the
@@ -1046,6 +1220,39 @@ class PatchStatusResponse:
         return cls()
 
 
+@dataclass
+class SortSpec:
+    descending: Optional[bool] = None
+    """Whether to sort in descending order."""
+
+    field: Optional[SortSpecField] = None
+    """The filed to sort by"""
+
+    def as_dict(self) -> dict:
+        """Serializes the SortSpec into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.descending is not None: body['descending'] = self.descending
+        if self.field is not None: body['field'] = self.field.value
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the SortSpec into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.descending is not None: body['descending'] = self.descending
+        if self.field is not None: body['field'] = self.field
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> SortSpec:
+        """Deserializes the SortSpec from a dictionary."""
+        return cls(descending=d.get('descending', None), field=_enum(d, 'field', SortSpecField))
+
+
+class SortSpecField(Enum):
+
+    POLICY_NAME = 'POLICY_NAME'
+
+
 @dataclass
 class UpdateBudgetConfigurationBudget:
     account_id: Optional[str] = None
@@ -1315,6 +1522,147 @@ class BillableUsageAPI:
         return DownloadResponse.from_dict(res)
 
 
+class BudgetPolicyAPI:
+    """A service serves REST API about Budget policies"""
+
+    def __init__(self, api_client):
+        self._api = api_client
+
+    def create(self,
+               *,
+               custom_tags: Optional[List[compute.CustomPolicyTag]] = None,
+               policy_name: Optional[str] = None,
+               request_id: Optional[str] = None) -> BudgetPolicy:
+        """Create a budget policy.
+        
+        Creates a new policy.
+        
+        :param custom_tags: List[:class:`CustomPolicyTag`] (optional)
+          A list of tags defined by the customer. At most 40 entries are allowed per policy.
+        :param policy_name: str (optional)
+          The name of the policy. - Must be unique among active policies. - Can contain only characters of
+          0-9, a-z, A-Z, -, =, ., :, /, @, _, +, whitespace.
+        :param request_id: str (optional)
+          A unique identifier for this request. Restricted to 36 ASCII characters. A random UUID is
+          recommended. This request is only idempotent if a `request_id` is provided.
+        
+        :returns: :class:`BudgetPolicy`
+        """
+        body = {}
+        if custom_tags is not None: body['custom_tags'] = [v.as_dict() for v in custom_tags]
+        if policy_name is not None: body['policy_name'] = policy_name
+        if request_id is not None: body['request_id'] = request_id
+        headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
+        res = self._api.do('POST',
+                           f'/api/2.1/accounts/{self._api.account_id}/budget-policies',
+                           body=body,
+                           headers=headers)
+        return BudgetPolicy.from_dict(res)
+
+    def delete(self, policy_id: str):
+        """Delete a budget policy.
+        
+        Deletes a policy
+        
+        :param policy_id: str
+          The Id of the policy.
+        
+        
+        """
+
+        headers = {'Accept': 'application/json', }
+
+        self._api.do('DELETE',
+                     f'/api/2.1/accounts/{self._api.account_id}/budget-policies/{policy_id}',
+                     headers=headers)
+
+    def get(self, policy_id: str) -> BudgetPolicy:
+        """Get a budget policy.
+        
+        Retrieves a policy by it's ID.
+        
+        :param policy_id: str
+          The Id of the policy.
+        
+        :returns: :class:`BudgetPolicy`
+        """
+
+        headers = {'Accept': 'application/json', }
+
+        res = self._api.do('GET',
+                           f'/api/2.1/accounts/{self._api.account_id}/budget-policies/{policy_id}',
+                           headers=headers)
+        return BudgetPolicy.from_dict(res)
+
+    def list(self,
+             *,
+             filter_by: Optional[Filter] = None,
+             page_size: Optional[int] = None,
+             page_token: Optional[str] = None,
+             sort_spec: Optional[SortSpec] = None) -> Iterator[BudgetPolicy]:
+        """List policies.
+        
+        Lists all policies. Policies are returned in the alphabetically ascending order of their names.
+        
+        :param filter_by: :class:`Filter` (optional)
+          A filter to apply to the list of policies.
+        :param page_size: int (optional)
+          The maximum number of budget policies to return. If unspecified, at most 100 budget policies will be
+          returned. The maximum value is 1000; values above 1000 will be coerced to 1000.
+        :param page_token: str (optional)
+          A page token, received from a previous `ListServerlessPolicies` call. Provide this to retrieve the
+          subsequent page. If unspecified, the first page will be returned.
+          
+          When paginating, all other parameters provided to `ListServerlessPoliciesRequest` must match the
+          call that provided the page token.
+        :param sort_spec: :class:`SortSpec` (optional)
+          The sort specification.
+        
+        :returns: Iterator over :class:`BudgetPolicy`
+        """
+
+        query = {}
+        if filter_by is not None: query['filter_by'] = filter_by.as_dict()
+        if page_size is not None: query['page_size'] = page_size
+        if page_token is not None: query['page_token'] = page_token
+        if sort_spec is not None: query['sort_spec'] = sort_spec.as_dict()
+        headers = {'Accept': 'application/json', }
+
+        while True:
+            json = self._api.do('GET',
+                                f'/api/2.1/accounts/{self._api.account_id}/budget-policies',
+                                query=query,
+                                headers=headers)
+            if 'policies' in json:
+                for v in json['policies']:
+                    yield BudgetPolicy.from_dict(v)
+            if 'next_page_token' not in json or not json['next_page_token']:
+                return
+            query['page_token'] = json['next_page_token']
+
+    def update(self, policy_id: str, *, policy: Optional[BudgetPolicy] = None) -> BudgetPolicy:
+        """Update a budget policy.
+        
+        Updates a policy
+        
+        :param policy_id: str
+          The Id of the policy. This field is generated by Databricks and globally unique.
+        :param policy: :class:`BudgetPolicy` (optional)
+          Contains the BudgetPolicy details.
+        
+        :returns: :class:`BudgetPolicy`
+        """
+        body = policy.as_dict()
+        headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
+        res = self._api.do('PATCH',
+                           f'/api/2.1/accounts/{self._api.account_id}/budget-policies/{policy_id}',
+                           body=body,
+                           headers=headers)
+        return BudgetPolicy.from_dict(res)
+
+
 class BudgetsAPI:
     """These APIs manage budget configurations for this account. Budgets enable you to monitor usage across your
     account. You can set up budgets to either track account-wide spending, or apply filters to track the

databricks/sdk/service/catalog.py

@@ -787,9 +787,6 @@ class CatalogInfo:
     provisioning_info: Optional[ProvisioningInfo] = None
     """Status of an asynchronously provisioned resource."""
 
-    securable_kind: Optional[CatalogInfoSecurableKind] = None
-    """Kind of catalog securable."""
-
     securable_type: Optional[str] = None
 
     share_name: Optional[str] = None
@@ -831,7 +828,6 @@ class CatalogInfo:
         if self.properties: body['properties'] = self.properties
         if self.provider_name is not None: body['provider_name'] = self.provider_name
         if self.provisioning_info: body['provisioning_info'] = self.provisioning_info.as_dict()
-        if self.securable_kind is not None: body['securable_kind'] = self.securable_kind.value
         if self.securable_type is not None: body['securable_type'] = self.securable_type
         if self.share_name is not None: body['share_name'] = self.share_name
         if self.storage_location is not None: body['storage_location'] = self.storage_location
@@ -862,7 +858,6 @@ class CatalogInfo:
         if self.properties: body['properties'] = self.properties
         if self.provider_name is not None: body['provider_name'] = self.provider_name
         if self.provisioning_info: body['provisioning_info'] = self.provisioning_info
-        if self.securable_kind is not None: body['securable_kind'] = self.securable_kind
         if self.securable_type is not None: body['securable_type'] = self.securable_type
         if self.share_name is not None: body['share_name'] = self.share_name
         if self.storage_location is not None: body['storage_location'] = self.storage_location
@@ -893,7 +888,6 @@ class CatalogInfo:
                    properties=d.get('properties', None),
                    provider_name=d.get('provider_name', None),
                    provisioning_info=_from_dict(d, 'provisioning_info', ProvisioningInfo),
-                   securable_kind=_enum(d, 'securable_kind', CatalogInfoSecurableKind),
                    securable_type=d.get('securable_type', None),
                    share_name=d.get('share_name', None),
                    storage_location=d.get('storage_location', None),
@@ -902,24 +896,6 @@ class CatalogInfo:
                    updated_by=d.get('updated_by', None))
 
 
-class CatalogInfoSecurableKind(Enum):
-    """Kind of catalog securable."""
-
-    CATALOG_DELTASHARING = 'CATALOG_DELTASHARING'
-    CATALOG_FOREIGN_BIGQUERY = 'CATALOG_FOREIGN_BIGQUERY'
-    CATALOG_FOREIGN_DATABRICKS = 'CATALOG_FOREIGN_DATABRICKS'
-    CATALOG_FOREIGN_MYSQL = 'CATALOG_FOREIGN_MYSQL'
-    CATALOG_FOREIGN_POSTGRESQL = 'CATALOG_FOREIGN_POSTGRESQL'
-    CATALOG_FOREIGN_REDSHIFT = 'CATALOG_FOREIGN_REDSHIFT'
-    CATALOG_FOREIGN_SNOWFLAKE = 'CATALOG_FOREIGN_SNOWFLAKE'
-    CATALOG_FOREIGN_SQLDW = 'CATALOG_FOREIGN_SQLDW'
-    CATALOG_FOREIGN_SQLSERVER = 'CATALOG_FOREIGN_SQLSERVER'
-    CATALOG_INTERNAL = 'CATALOG_INTERNAL'
-    CATALOG_STANDARD = 'CATALOG_STANDARD'
-    CATALOG_SYSTEM = 'CATALOG_SYSTEM'
-    CATALOG_SYSTEM_DELTASHARING = 'CATALOG_SYSTEM_DELTASHARING'
-
-
 class CatalogIsolationMode(Enum):
     """Whether the current securable is accessible from all workspaces or a specific set of workspaces."""
 
@@ -1158,9 +1134,6 @@ class ConnectionInfo:
     read_only: Optional[bool] = None
     """If the connection is read only."""
 
-    securable_kind: Optional[ConnectionInfoSecurableKind] = None
-    """Kind of connection securable."""
-
     securable_type: Optional[str] = None
 
     updated_at: Optional[int] = None
@@ -1189,7 +1162,6 @@ class ConnectionInfo:
         if self.properties: body['properties'] = self.properties
         if self.provisioning_info: body['provisioning_info'] = self.provisioning_info.as_dict()
         if self.read_only is not None: body['read_only'] = self.read_only
-        if self.securable_kind is not None: body['securable_kind'] = self.securable_kind.value
         if self.securable_type is not None: body['securable_type'] = self.securable_type
         if self.updated_at is not None: body['updated_at'] = self.updated_at
         if self.updated_by is not None: body['updated_by'] = self.updated_by
@@ -1213,7 +1185,6 @@ class ConnectionInfo:
         if self.properties: body['properties'] = self.properties
         if self.provisioning_info: body['provisioning_info'] = self.provisioning_info
         if self.read_only is not None: body['read_only'] = self.read_only
-        if self.securable_kind is not None: body['securable_kind'] = self.securable_kind
         if self.securable_type is not None: body['securable_type'] = self.securable_type
         if self.updated_at is not None: body['updated_at'] = self.updated_at
         if self.updated_by is not None: body['updated_by'] = self.updated_by
@@ -1237,31 +1208,12 @@ class ConnectionInfo:
                    properties=d.get('properties', None),
                    provisioning_info=_from_dict(d, 'provisioning_info', ProvisioningInfo),
                    read_only=d.get('read_only', None),
-                   securable_kind=_enum(d, 'securable_kind', ConnectionInfoSecurableKind),
                    securable_type=d.get('securable_type', None),
                    updated_at=d.get('updated_at', None),
                    updated_by=d.get('updated_by', None),
                    url=d.get('url', None))
 
 
-class ConnectionInfoSecurableKind(Enum):
-    """Kind of connection securable."""
-
-    CONNECTION_BIGQUERY = 'CONNECTION_BIGQUERY'
-    CONNECTION_BUILTIN_HIVE_METASTORE = 'CONNECTION_BUILTIN_HIVE_METASTORE'
-    CONNECTION_DATABRICKS = 'CONNECTION_DATABRICKS'
-    CONNECTION_EXTERNAL_HIVE_METASTORE = 'CONNECTION_EXTERNAL_HIVE_METASTORE'
-    CONNECTION_GLUE = 'CONNECTION_GLUE'
-    CONNECTION_HTTP_BEARER = 'CONNECTION_HTTP_BEARER'
-    CONNECTION_MYSQL = 'CONNECTION_MYSQL'
-    CONNECTION_ONLINE_CATALOG = 'CONNECTION_ONLINE_CATALOG'
-    CONNECTION_POSTGRESQL = 'CONNECTION_POSTGRESQL'
-    CONNECTION_REDSHIFT = 'CONNECTION_REDSHIFT'
-    CONNECTION_SNOWFLAKE = 'CONNECTION_SNOWFLAKE'
-    CONNECTION_SQLDW = 'CONNECTION_SQLDW'
-    CONNECTION_SQLSERVER = 'CONNECTION_SQLSERVER'
-
-
 class ConnectionType(Enum):
     """The type of connection."""
 
@@ -5810,6 +5762,7 @@ class ProvisioningInfo:
 class ProvisioningInfoState(Enum):
 
     ACTIVE = 'ACTIVE'
+    DEGRADED = 'DEGRADED'
     DELETING = 'DELETING'
     FAILED = 'FAILED'
     PROVISIONING = 'PROVISIONING'
@@ -6268,20 +6221,21 @@ SecurablePropertiesMap = Dict[str, str]
 class SecurableType(Enum):
     """The type of Unity Catalog securable"""
 
-    CATALOG = 'catalog'
-    CONNECTION = 'connection'
-    CREDENTIAL = 'credential'
-    EXTERNAL_LOCATION = 'external_location'
-    FUNCTION = 'function'
-    METASTORE = 'metastore'
-    PIPELINE = 'pipeline'
-    PROVIDER = 'provider'
-    RECIPIENT = 'recipient'
-    SCHEMA = 'schema'
-    SHARE = 'share'
-    STORAGE_CREDENTIAL = 'storage_credential'
-    TABLE = 'table'
-    VOLUME = 'volume'
+    CATALOG = 'CATALOG'
+    CLEAN_ROOM = 'CLEAN_ROOM'
+    CONNECTION = 'CONNECTION'
+    CREDENTIAL = 'CREDENTIAL'
+    EXTERNAL_LOCATION = 'EXTERNAL_LOCATION'
+    FUNCTION = 'FUNCTION'
+    METASTORE = 'METASTORE'
+    PIPELINE = 'PIPELINE'
+    PROVIDER = 'PROVIDER'
+    RECIPIENT = 'RECIPIENT'
+    SCHEMA = 'SCHEMA'
+    SHARE = 'SHARE'
+    STORAGE_CREDENTIAL = 'STORAGE_CREDENTIAL'
+    TABLE = 'TABLE'
+    VOLUME = 'VOLUME'
 
 
 @dataclass

databricks/sdk/service/cleanrooms.py

@@ -289,11 +289,24 @@ class CleanRoomAssetNotebook:
     """Base 64 representation of the notebook contents. This is the same format as returned by
     :method:workspace/export with the format of **HTML**."""
 
+    review_state: Optional[CleanRoomNotebookReviewNotebookReviewState] = None
+    """top-level status derived from all reviews"""
+
+    reviews: Optional[List[CleanRoomNotebookReview]] = None
+    """All existing approvals or rejections"""
+
+    runner_collaborators: Optional[List[CleanRoomCollaborator]] = None
+    """collaborators that can run the notebook"""
+
     def as_dict(self) -> dict:
         """Serializes the CleanRoomAssetNotebook into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.etag is not None: body['etag'] = self.etag
         if self.notebook_content is not None: body['notebook_content'] = self.notebook_content
+        if self.review_state is not None: body['review_state'] = self.review_state.value
+        if self.reviews: body['reviews'] = [v.as_dict() for v in self.reviews]
+        if self.runner_collaborators:
+            body['runner_collaborators'] = [v.as_dict() for v in self.runner_collaborators]
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -301,17 +314,25 @@ class CleanRoomAssetNotebook:
         body = {}
         if self.etag is not None: body['etag'] = self.etag
         if self.notebook_content is not None: body['notebook_content'] = self.notebook_content
+        if self.review_state is not None: body['review_state'] = self.review_state
+        if self.reviews: body['reviews'] = self.reviews
+        if self.runner_collaborators: body['runner_collaborators'] = self.runner_collaborators
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, any]) -> CleanRoomAssetNotebook:
         """Deserializes the CleanRoomAssetNotebook from a dictionary."""
-        return cls(etag=d.get('etag', None), notebook_content=d.get('notebook_content', None))
+        return cls(etag=d.get('etag', None),
+                   notebook_content=d.get('notebook_content', None),
+                   review_state=_enum(d, 'review_state', CleanRoomNotebookReviewNotebookReviewState),
+                   reviews=_repeated_dict(d, 'reviews', CleanRoomNotebookReview),
+                   runner_collaborators=_repeated_dict(d, 'runner_collaborators', CleanRoomCollaborator))
 
 
 class CleanRoomAssetStatusEnum(Enum):
 
     ACTIVE = 'ACTIVE'
+    PENDING = 'PENDING'
     PERMISSION_DENIED = 'PERMISSION_DENIED'
 
 
@@ -443,7 +464,7 @@ class CleanRoomAssetVolumeLocalDetails:
 class CleanRoomCollaborator:
     """Publicly visible clean room collaborator."""
 
-    collaborator_alias: Optional[str] = None
+    collaborator_alias: str
     """Collaborator alias specified by the clean room creator. It is unique across all collaborators of
     this clean room, and used to derive multiple values internally such as catalog alias and clean
     room name for single metastore clean rooms. It should follow [UC securable naming requirements].
@@ -510,6 +531,56 @@ class CleanRoomCollaborator:
                    organization_name=d.get('organization_name', None))
 
 
+@dataclass
+class CleanRoomNotebookReview:
+    comment: Optional[str] = None
+    """review comment"""
+
+    created_at_millis: Optional[int] = None
+    """timestamp of when the review was submitted"""
+
+    review_state: Optional[CleanRoomNotebookReviewNotebookReviewState] = None
+    """review outcome"""
+
+    reviewer_collaborator_alias: Optional[str] = None
+    """collaborator alias of the reviewer"""
+
+    def as_dict(self) -> dict:
+        """Serializes the CleanRoomNotebookReview into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.comment is not None: body['comment'] = self.comment
+        if self.created_at_millis is not None: body['created_at_millis'] = self.created_at_millis
+        if self.review_state is not None: body['review_state'] = self.review_state.value
+        if self.reviewer_collaborator_alias is not None:
+            body['reviewer_collaborator_alias'] = self.reviewer_collaborator_alias
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the CleanRoomNotebookReview into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.comment is not None: body['comment'] = self.comment
+        if self.created_at_millis is not None: body['created_at_millis'] = self.created_at_millis
+        if self.review_state is not None: body['review_state'] = self.review_state
+        if self.reviewer_collaborator_alias is not None:
+            body['reviewer_collaborator_alias'] = self.reviewer_collaborator_alias
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> CleanRoomNotebookReview:
+        """Deserializes the CleanRoomNotebookReview from a dictionary."""
+        return cls(comment=d.get('comment', None),
+                   created_at_millis=d.get('created_at_millis', None),
+                   review_state=_enum(d, 'review_state', CleanRoomNotebookReviewNotebookReviewState),
+                   reviewer_collaborator_alias=d.get('reviewer_collaborator_alias', None))
+
+
+class CleanRoomNotebookReviewNotebookReviewState(Enum):
+
+    APPROVED = 'APPROVED'
+    PENDING = 'PENDING'
+    REJECTED = 'REJECTED'
+
+
 @dataclass
 class CleanRoomNotebookTaskRun:
     """Stores information about a single task run."""