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

databricks/sdk/service/apps.py

@@ -45,6 +45,9 @@ class App:
     description: Optional[str] = None
     """The description of the app."""
 
+    id: Optional[str] = None
+    """The unique identifier of the app."""
+
     pending_deployment: Optional[AppDeployment] = None
     """The pending deployment of the app. A deployment is considered pending when it is being prepared
     for deployment to the app compute."""
@@ -78,6 +81,7 @@ class App:
         if self.default_source_code_path is not None:
             body['default_source_code_path'] = self.default_source_code_path
         if self.description is not None: body['description'] = self.description
+        if self.id is not None: body['id'] = self.id
         if self.name is not None: body['name'] = self.name
         if self.pending_deployment: body['pending_deployment'] = self.pending_deployment.as_dict()
         if self.resources: body['resources'] = [v.as_dict() for v in self.resources]
@@ -102,6 +106,7 @@ class App:
         if self.default_source_code_path is not None:
             body['default_source_code_path'] = self.default_source_code_path
         if self.description is not None: body['description'] = self.description
+        if self.id is not None: body['id'] = self.id
         if self.name is not None: body['name'] = self.name
         if self.pending_deployment: body['pending_deployment'] = self.pending_deployment
         if self.resources: body['resources'] = self.resources
@@ -125,6 +130,7 @@ class App:
                    creator=d.get('creator', None),
                    default_source_code_path=d.get('default_source_code_path', None),
                    description=d.get('description', None),
+                   id=d.get('id', None),
                    name=d.get('name', None),
                    pending_deployment=_from_dict(d, 'pending_deployment', AppDeployment),
                    resources=_repeated_dict(d, 'resources', AppResource),

databricks/sdk/service/billing.py

@@ -894,6 +894,27 @@ class GetBudgetConfigurationResponse:
         return cls(budget=_from_dict(d, 'budget', BudgetConfiguration))
 
 
+@dataclass
+class LimitConfig:
+    """The limit configuration of the policy. Limit configuration provide a budget policy level cost
+    control by enforcing the limit."""
+
+    def as_dict(self) -> dict:
+        """Serializes the LimitConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the LimitConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> LimitConfig:
+        """Deserializes the LimitConfig from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ListBudgetConfigurationsResponse:
     budgets: Optional[List[BudgetConfiguration]] = None
@@ -1641,23 +1662,32 @@ class BudgetPolicyAPI:
                 return
             query['page_token'] = json['next_page_token']
 
-    def update(self, policy_id: str, *, policy: Optional[BudgetPolicy] = None) -> BudgetPolicy:
+    def update(self,
+               policy_id: str,
+               *,
+               limit_config: Optional[LimitConfig] = None,
+               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 limit_config: :class:`LimitConfig` (optional)
+          DEPRECATED. This is redundant field as LimitConfig is part of the BudgetPolicy
         :param policy: :class:`BudgetPolicy` (optional)
           Contains the BudgetPolicy details.
         
         :returns: :class:`BudgetPolicy`
         """
         body = policy.as_dict()
+        query = {}
+        if limit_config is not None: query['limit_config'] = limit_config.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}',
+                           query=query,
                            body=body,
                            headers=headers)
         return BudgetPolicy.from_dict(res)

databricks/sdk/service/catalog.py

@@ -8983,6 +8983,7 @@ class CatalogsAPI:
         if page_token is not None: query['page_token'] = page_token
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET', '/api/2.1/unity-catalog/catalogs', query=query, headers=headers)
             if 'catalogs' in json:
@@ -9151,6 +9152,7 @@ class ConnectionsAPI:
         if page_token is not None: query['page_token'] = page_token
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET', '/api/2.1/unity-catalog/connections', query=query, headers=headers)
             if 'connections' in json:
@@ -9656,6 +9658,7 @@ class ExternalLocationsAPI:
         if page_token is not None: query['page_token'] = page_token
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET',
                                 '/api/2.1/unity-catalog/external-locations',
@@ -11389,6 +11392,7 @@ class SchemasAPI:
         if page_token is not None: query['page_token'] = page_token
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET', '/api/2.1/unity-catalog/schemas', query=query, headers=headers)
             if 'schemas' in json:
@@ -11578,6 +11582,7 @@ class StorageCredentialsAPI:
         if page_token is not None: query['page_token'] = page_token
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET',
                                 '/api/2.1/unity-catalog/storage-credentials',
@@ -11802,6 +11807,7 @@ class SystemSchemasAPI:
         if page_token is not None: query['page_token'] = page_token
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET',
                                 f'/api/2.1/unity-catalog/metastores/{metastore_id}/systemschemas',
@@ -12044,6 +12050,7 @@ class TablesAPI:
         if schema_name is not None: query['schema_name'] = schema_name
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET', '/api/2.1/unity-catalog/tables', query=query, headers=headers)
             if 'tables' in json:
@@ -12104,6 +12111,7 @@ class TablesAPI:
         if table_name_pattern is not None: query['table_name_pattern'] = table_name_pattern
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET', '/api/2.1/unity-catalog/table-summaries', query=query, headers=headers)
             if 'tables' in json:

databricks/sdk/service/cleanrooms.py

@@ -289,24 +289,11 @@ 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:
@@ -314,19 +301,12 @@ 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),
-                   review_state=_enum(d, 'review_state', CleanRoomNotebookReviewNotebookReviewState),
-                   reviews=_repeated_dict(d, 'reviews', CleanRoomNotebookReview),
-                   runner_collaborators=_repeated_dict(d, 'runner_collaborators', CleanRoomCollaborator))
+        return cls(etag=d.get('etag', None), notebook_content=d.get('notebook_content', None))
 
 
 class CleanRoomAssetStatusEnum(Enum):
@@ -531,56 +511,6 @@ 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."""

databricks/sdk/service/compute.py

@@ -637,11 +637,11 @@ class ClusterAttributes:
     a set of default values will be used."""
 
     cluster_log_conf: Optional[ClusterLogConf] = None
-    """The configuration for delivering spark logs to a long-term storage destination. Two kinds of
-    destinations (dbfs and s3) are supported. Only one destination can be specified for one cluster.
-    If the conf is given, the logs will be delivered to the destination every `5 mins`. The
-    destination of driver logs is `$destination/$clusterId/driver`, while the destination of
-    executor logs is `$destination/$clusterId/executor`."""
+    """The configuration for delivering spark logs to a long-term storage destination. Three kinds of
+    destinations (DBFS, S3 and Unity Catalog volumes) are supported. Only one destination can be
+    specified for one cluster. If the conf is given, the logs will be delivered to the destination
+    every `5 mins`. The destination of driver logs is `$destination/$clusterId/driver`, while the
+    destination of executor logs is `$destination/$clusterId/executor`."""
 
     cluster_name: Optional[str] = None
     """Cluster name requested by the user. This doesn't have to be unique. If not specified at
@@ -947,11 +947,11 @@ class ClusterDetails:
     while each new cluster has a globally unique id."""
 
     cluster_log_conf: Optional[ClusterLogConf] = None
-    """The configuration for delivering spark logs to a long-term storage destination. Two kinds of
-    destinations (dbfs and s3) are supported. Only one destination can be specified for one cluster.
-    If the conf is given, the logs will be delivered to the destination every `5 mins`. The
-    destination of driver logs is `$destination/$clusterId/driver`, while the destination of
-    executor logs is `$destination/$clusterId/executor`."""
+    """The configuration for delivering spark logs to a long-term storage destination. Three kinds of
+    destinations (DBFS, S3 and Unity Catalog volumes) are supported. Only one destination can be
+    specified for one cluster. If the conf is given, the logs will be delivered to the destination
+    every `5 mins`. The destination of driver logs is `$destination/$clusterId/driver`, while the
+    destination of executor logs is `$destination/$clusterId/executor`."""
 
     cluster_log_status: Optional[LogSyncStatus] = None
     """Cluster log delivery status."""
@@ -1428,11 +1428,16 @@ class ClusterLogConf:
     access s3, please make sure the cluster iam role in `instance_profile_arn` has permission to
     write data to the s3 destination."""
 
+    volumes: Optional[VolumesStorageInfo] = None
+    """destination needs to be provided. e.g. `{ "volumes" : { "destination" :
+    "/Volumes/catalog/schema/volume/cluster_log" } }`"""
+
     def as_dict(self) -> dict:
         """Serializes the ClusterLogConf into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.dbfs: body['dbfs'] = self.dbfs.as_dict()
         if self.s3: body['s3'] = self.s3.as_dict()
+        if self.volumes: body['volumes'] = self.volumes.as_dict()
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -1440,12 +1445,15 @@ class ClusterLogConf:
         body = {}
         if self.dbfs: body['dbfs'] = self.dbfs
         if self.s3: body['s3'] = self.s3
+        if self.volumes: body['volumes'] = self.volumes
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, any]) -> ClusterLogConf:
         """Deserializes the ClusterLogConf from a dictionary."""
-        return cls(dbfs=_from_dict(d, 'dbfs', DbfsStorageInfo), s3=_from_dict(d, 's3', S3StorageInfo))
+        return cls(dbfs=_from_dict(d, 'dbfs', DbfsStorageInfo),
+                   s3=_from_dict(d, 's3', S3StorageInfo),
+                   volumes=_from_dict(d, 'volumes', VolumesStorageInfo))
 
 
 @dataclass
@@ -1918,11 +1926,11 @@ class ClusterSpec:
     a set of default values will be used."""
 
     cluster_log_conf: Optional[ClusterLogConf] = None
-    """The configuration for delivering spark logs to a long-term storage destination. Two kinds of
-    destinations (dbfs and s3) are supported. Only one destination can be specified for one cluster.
-    If the conf is given, the logs will be delivered to the destination every `5 mins`. The
-    destination of driver logs is `$destination/$clusterId/driver`, while the destination of
-    executor logs is `$destination/$clusterId/executor`."""
+    """The configuration for delivering spark logs to a long-term storage destination. Three kinds of
+    destinations (DBFS, S3 and Unity Catalog volumes) are supported. Only one destination can be
+    specified for one cluster. If the conf is given, the logs will be delivered to the destination
+    every `5 mins`. The destination of driver logs is `$destination/$clusterId/driver`, while the
+    destination of executor logs is `$destination/$clusterId/executor`."""
 
     cluster_name: Optional[str] = None
     """Cluster name requested by the user. This doesn't have to be unique. If not specified at
@@ -2334,11 +2342,11 @@ class CreateCluster:
     cluster."""
 
     cluster_log_conf: Optional[ClusterLogConf] = None
-    """The configuration for delivering spark logs to a long-term storage destination. Two kinds of
-    destinations (dbfs and s3) are supported. Only one destination can be specified for one cluster.
-    If the conf is given, the logs will be delivered to the destination every `5 mins`. The
-    destination of driver logs is `$destination/$clusterId/driver`, while the destination of
-    executor logs is `$destination/$clusterId/executor`."""
+    """The configuration for delivering spark logs to a long-term storage destination. Three kinds of
+    destinations (DBFS, S3 and Unity Catalog volumes) are supported. Only one destination can be
+    specified for one cluster. If the conf is given, the logs will be delivered to the destination
+    every `5 mins`. The destination of driver logs is `$destination/$clusterId/driver`, while the
+    destination of executor logs is `$destination/$clusterId/executor`."""
 
     cluster_name: Optional[str] = None
     """Cluster name requested by the user. This doesn't have to be unique. If not specified at
@@ -3469,11 +3477,11 @@ class EditCluster:
     a set of default values will be used."""
 
     cluster_log_conf: Optional[ClusterLogConf] = None
-    """The configuration for delivering spark logs to a long-term storage destination. Two kinds of
-    destinations (dbfs and s3) are supported. Only one destination can be specified for one cluster.
-    If the conf is given, the logs will be delivered to the destination every `5 mins`. The
-    destination of driver logs is `$destination/$clusterId/driver`, while the destination of
-    executor logs is `$destination/$clusterId/executor`."""
+    """The configuration for delivering spark logs to a long-term storage destination. Three kinds of
+    destinations (DBFS, S3 and Unity Catalog volumes) are supported. Only one destination can be
+    specified for one cluster. If the conf is given, the logs will be delivered to the destination
+    every `5 mins`. The destination of driver logs is `$destination/$clusterId/driver`, while the
+    destination of executor logs is `$destination/$clusterId/executor`."""
 
     cluster_name: Optional[str] = None
     """Cluster name requested by the user. This doesn't have to be unique. If not specified at
@@ -7773,11 +7781,11 @@ class UpdateClusterResource:
     a set of default values will be used."""
 
     cluster_log_conf: Optional[ClusterLogConf] = None
-    """The configuration for delivering spark logs to a long-term storage destination. Two kinds of
-    destinations (dbfs and s3) are supported. Only one destination can be specified for one cluster.
-    If the conf is given, the logs will be delivered to the destination every `5 mins`. The
-    destination of driver logs is `$destination/$clusterId/driver`, while the destination of
-    executor logs is `$destination/$clusterId/executor`."""
+    """The configuration for delivering spark logs to a long-term storage destination. Three kinds of
+    destinations (DBFS, S3 and Unity Catalog volumes) are supported. Only one destination can be
+    specified for one cluster. If the conf is given, the logs will be delivered to the destination
+    every `5 mins`. The destination of driver logs is `$destination/$clusterId/driver`, while the
+    destination of executor logs is `$destination/$clusterId/executor`."""
 
     cluster_name: Optional[str] = None
     """Cluster name requested by the user. This doesn't have to be unique. If not specified at
@@ -8077,7 +8085,7 @@ class UpdateResponse:
 @dataclass
 class VolumesStorageInfo:
     destination: str
-    """Unity Catalog Volumes file destination, e.g. `/Volumes/my-init.sh`"""
+    """Unity Catalog volumes file destination, e.g. `/Volumes/catalog/schema/volume/dir/file`"""
 
     def as_dict(self) -> dict:
         """Serializes the VolumesStorageInfo into a dictionary suitable for use as a JSON request body."""
@@ -8619,11 +8627,11 @@ class ClustersAPI:
         :param clone_from: :class:`CloneCluster` (optional)
           When specified, this clones libraries from a source cluster during the creation of a new cluster.
         :param cluster_log_conf: :class:`ClusterLogConf` (optional)
-          The configuration for delivering spark logs to a long-term storage destination. Two kinds of
-          destinations (dbfs and s3) are supported. Only one destination can be specified for one cluster. If
-          the conf is given, the logs will be delivered to the destination every `5 mins`. The destination of
-          driver logs is `$destination/$clusterId/driver`, while the destination of executor logs is
-          `$destination/$clusterId/executor`.
+          The configuration for delivering spark logs to a long-term storage destination. Three kinds of
+          destinations (DBFS, S3 and Unity Catalog volumes) are supported. Only one destination can be
+          specified for one cluster. If the conf is given, the logs will be delivered to the destination every
+          `5 mins`. The destination of driver logs is `$destination/$clusterId/driver`, while the destination
+          of executor logs is `$destination/$clusterId/executor`.
         :param cluster_name: str (optional)
           Cluster name requested by the user. This doesn't have to be unique. If not specified at creation,
           the cluster name will be an empty string.
@@ -8952,11 +8960,11 @@ class ClustersAPI:
           Attributes related to clusters running on Microsoft Azure. If not specified at cluster creation, a
           set of default values will be used.
         :param cluster_log_conf: :class:`ClusterLogConf` (optional)
-          The configuration for delivering spark logs to a long-term storage destination. Two kinds of
-          destinations (dbfs and s3) are supported. Only one destination can be specified for one cluster. If
-          the conf is given, the logs will be delivered to the destination every `5 mins`. The destination of
-          driver logs is `$destination/$clusterId/driver`, while the destination of executor logs is
-          `$destination/$clusterId/executor`.
+          The configuration for delivering spark logs to a long-term storage destination. Three kinds of
+          destinations (DBFS, S3 and Unity Catalog volumes) are supported. Only one destination can be
+          specified for one cluster. If the conf is given, the logs will be delivered to the destination every
+          `5 mins`. The destination of driver logs is `$destination/$clusterId/driver`, while the destination
+          of executor logs is `$destination/$clusterId/executor`.
         :param cluster_name: str (optional)
           Cluster name requested by the user. This doesn't have to be unique. If not specified at creation,
           the cluster name will be an empty string.

databricks/sdk/service/dashboards.py

@@ -827,6 +827,7 @@ class MessageErrorType(Enum):
     REPLY_PROCESS_TIMEOUT_EXCEPTION = 'REPLY_PROCESS_TIMEOUT_EXCEPTION'
     RETRYABLE_PROCESSING_EXCEPTION = 'RETRYABLE_PROCESSING_EXCEPTION'
     SQL_EXECUTION_EXCEPTION = 'SQL_EXECUTION_EXCEPTION'
+    STOP_PROCESS_DUE_TO_AUTO_REGENERATE = 'STOP_PROCESS_DUE_TO_AUTO_REGENERATE'
     TABLES_MISSING_EXCEPTION = 'TABLES_MISSING_EXCEPTION'
     TOO_MANY_CERTIFIED_ANSWERS_EXCEPTION = 'TOO_MANY_CERTIFIED_ANSWERS_EXCEPTION'
     TOO_MANY_TABLES_EXCEPTION = 'TOO_MANY_TABLES_EXCEPTION'
@@ -1743,6 +1744,33 @@ class GenieAPI:
             headers=headers)
         return GenieGetMessageQueryResultResponse.from_dict(res)
 
+    def get_message_query_result_by_attachment(self, space_id: str, conversation_id: str, message_id: str,
+                                               attachment_id: str) -> GenieGetMessageQueryResultResponse:
+        """Get conversation message SQL query result by attachment id.
+        
+        Get the result of SQL query by attachment id This is only available if a message has a query
+        attachment and the message status is `EXECUTING_QUERY`.
+        
+        :param space_id: str
+          Genie space ID
+        :param conversation_id: str
+          Conversation ID
+        :param message_id: str
+          Message ID
+        :param attachment_id: str
+          Attachment ID
+        
+        :returns: :class:`GenieGetMessageQueryResultResponse`
+        """
+
+        headers = {'Accept': 'application/json', }
+
+        res = self._api.do(
+            'GET',
+            f'/api/2.0/genie/spaces/{space_id}/conversations/{conversation_id}/messages/{message_id}/query-result/{attachment_id}',
+            headers=headers)
+        return GenieGetMessageQueryResultResponse.from_dict(res)
+
     def start_conversation(self, space_id: str, content: str) -> Wait[GenieMessage]:
         """Start conversation.
         

databricks/sdk/service/serving.py

@@ -1247,7 +1247,7 @@ class ExternalModel:
     provider: ExternalModelProvider
     """The name of the provider for the external model. Currently, the supported providers are
     'ai21labs', 'anthropic', 'amazon-bedrock', 'cohere', 'databricks-model-serving',
-    'google-cloud-vertex-ai', 'openai', and 'palm'."""
+    'google-cloud-vertex-ai', 'openai', 'palm', and 'custom'."""
 
     name: str
     """The name of the external model."""

databricks/sdk/service/sharing.py

@@ -1658,6 +1658,7 @@ class ProvidersAPI:
         if page_token is not None: query['page_token'] = page_token
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET', '/api/2.1/unity-catalog/providers', query=query, headers=headers)
             if 'providers' in json:
@@ -1699,12 +1700,18 @@ class ProvidersAPI:
         if page_token is not None: query['page_token'] = page_token
         headers = {'Accept': 'application/json', }
 
-        json = self._api.do('GET',
-                            f'/api/2.1/unity-catalog/providers/{name}/shares',
-                            query=query,
-                            headers=headers)
-        parsed = ListProviderSharesResponse.from_dict(json).shares
-        return parsed if parsed is not None else []
+        if "max_results" not in query: query['max_results'] = 0
+        while True:
+            json = self._api.do('GET',
+                                f'/api/2.1/unity-catalog/providers/{name}/shares',
+                                query=query,
+                                headers=headers)
+            if 'shares' in json:
+                for v in json['shares']:
+                    yield ProviderShare.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,
                name: str,
@@ -1937,6 +1944,7 @@ class RecipientsAPI:
         if page_token is not None: query['page_token'] = page_token
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET', '/api/2.1/unity-catalog/recipients', query=query, headers=headers)
             if 'recipients' in json:
@@ -2157,6 +2165,7 @@ class SharesAPI:
         if page_token is not None: query['page_token'] = page_token
         headers = {'Accept': 'application/json', }
 
+        if "max_results" not in query: query['max_results'] = 0
         while True:
             json = self._api.do('GET', '/api/2.1/unity-catalog/shares', query=query, headers=headers)
             if 'shares' in json:

databricks/sdk/version.py

@@ -1 +1 @@
-__version__ = '0.43.0'
+__version__ = '0.44.0'

{databricks_sdk-0.43.0.dist-info → databricks_sdk-0.44.0.dist-info}/METADATA

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

{databricks_sdk-0.43.0.dist-info → databricks_sdk-0.44.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 databricks/__init__.py,sha256=CF2MJcZFwbpn9TwQER8qnCDhkPooBGQNVkX4v7g6p3g,537
-databricks/sdk/__init__.py,sha256=5AKsc4AyROmjCeNDrVmUCDV8z6c0M67u4bke_g4VcL4,52230
+databricks/sdk/__init__.py,sha256=PEAdNX4VkzmzgAqhh89lIy42WmW-JM2oAnj69XThCUM,56239
 databricks/sdk/_base_client.py,sha256=FwKMk4pN0AXt8S8RML2OHFYV4yxyhgd9NMjxJKiSIUs,16071
 databricks/sdk/_property.py,sha256=sGjsipeFrjMBSVPjtIb0HNCRcMIhFpVx6wq4BkC3LWs,1636
 databricks/sdk/azure.py,sha256=8P7nEdun0hbQCap9Ojo7yZse_JHxnhYsE6ApojnPz7Q,1009
@@ -15,7 +15,7 @@ databricks/sdk/oauth.py,sha256=ZlIzEGlKTUgGGgLfv5NQJr3Y_mWpKgTr8-hUEwwqfEE,23861
 databricks/sdk/py.typed,sha256=pSvaHpbY1UPNEXyVFUjlgBhjPFZMmVC_UNrPC7eMOHI,74
 databricks/sdk/retries.py,sha256=kdCKGIJjSkGLZYmQ0oI_hiGj7FP7MIqK9-nIr7WbykU,2574
 databricks/sdk/useragent.py,sha256=o9cojoaVwI7C6tbIZy6jcQ8QiYuUmdL5_zATu6IZSaw,7373
-databricks/sdk/version.py,sha256=uLdCe5IU74Eq2OM-KK7s23arKNEHXWY8B1o0eK5WeNc,23
+databricks/sdk/version.py,sha256=MMPeQ2-kYJhgLEqI-j8q_6GpHFGW12eIl3b2Nt7TutU,23
 databricks/sdk/_widgets/__init__.py,sha256=Qm3JB8LmdPgEn_-VgxKkodTO4gn6OdaDPwsYcDmeIRI,2667
 databricks/sdk/_widgets/default_widgets_utils.py,sha256=Rk59AFzVYVpOektB_yC_7j-vSt5OdtZA85IlG0kw0xA,1202
 databricks/sdk/_widgets/ipywidgets_utils.py,sha256=P-AyGeahPiX3S59mxpAMgffi4gyJ0irEOY7Ekkn9nQ0,2850
@@ -41,12 +41,12 @@ databricks/sdk/runtime/__init__.py,sha256=9NnZkBzeZXZRQxcE1qKzAszQEzcpIgpL7lQzW3
 databricks/sdk/runtime/dbutils_stub.py,sha256=UFbRZF-bBcwxjbv_pxma00bjNtktLLaYpo8oHRc4-9g,11421
 databricks/sdk/service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databricks/sdk/service/_internal.py,sha256=nWbJfW5eJCQgAZ3TmA26xoWb6SNZ5N76ZA8bO1N4AsU,1961
-databricks/sdk/service/apps.py,sha256=xxBPanIqAvN0iPZmnEyxY98n4RQr8pQWt2IXcnhk8L8,52535
-databricks/sdk/service/billing.py,sha256=8Zt0mD-StF-V5iICMtcRwxu-cKtpKdyY2g9CgUoT11E,97365
-databricks/sdk/service/catalog.py,sha256=btcUKO_7Ks7OCeyNmqQ74xX8Avf3vUKcXbigkWG33Rc,589904
-databricks/sdk/service/cleanrooms.py,sha256=UnVuLAjbVJ-rvxyGY4ArHhJ4zrXnk0z3XhZ3bDd8xXY,61330
-databricks/sdk/service/compute.py,sha256=rlzr-_sy6gzMqggd3p_wAuBdEYd0Cf6YpY3zpAMrDq8,535852
-databricks/sdk/service/dashboards.py,sha256=y7OBnRYLe3sLMda3Tsb8lTXTnRMlvtde6TaAg8xbT9c,100210
+databricks/sdk/service/apps.py,sha256=TVEqapzmon-QbWwgxWtC50qYt6BLDa1hKNYz4v2BeXU,52756
+databricks/sdk/service/billing.py,sha256=I7c6xdysk8HU9PD7JCO-foJrGr8P208wXv2A5qq8XJw,98437
+databricks/sdk/service/catalog.py,sha256=jB3WyXFw6ox7_C6jiXhl1tWWsO-dGmdRSovR1DWpe2U,590416
+databricks/sdk/service/cleanrooms.py,sha256=pfCxkGNI6XHgcHPNab7EqkcRHzemK8CKShCi1bl2r9M,57959
+databricks/sdk/service/compute.py,sha256=Q5hmR9vVlbJtd-CwL_IEHj4cw6TtmNEuwxKI1TOarWY,536472
+databricks/sdk/service/dashboards.py,sha256=KrVP9mo3cubJNom2mKdemhq3_MPU-TeEGP5cKgh7QKI,101393
 databricks/sdk/service/files.py,sha256=KewI3yw9HsqHKTlJlAkeO0CBszvaMrdBeyxTKORK9rk,45392
 databricks/sdk/service/iam.py,sha256=ez1G4m8AihZtip2On0o5aADTba25SWBcpYGf1SssJu0,172829
 databricks/sdk/service/jobs.py,sha256=q0NY-SBp1Amo4M8CmxUQ-wwNIOfi9TJrqYDWyupgFZI,426794
@@ -55,15 +55,15 @@ databricks/sdk/service/ml.py,sha256=wvheyoVzDUczufsWOjrUvBkK3KKwV1ZSJ6kXWQN4y_M,
 databricks/sdk/service/oauth2.py,sha256=pXU8MnsZFxmLCr3lLoLO1bCCJld7925jSjT6L4xxKKw,75768
 databricks/sdk/service/pipelines.py,sha256=ZAtYNEaVqkMpa4uWAH1pbXH0Rx2DDxNVdKqsADcQeS8,161720
 databricks/sdk/service/provisioning.py,sha256=QAFKTjRP6rh9gPIP17ownqhAFY2XE0HvqNfTsf3D27w,168727
-databricks/sdk/service/serving.py,sha256=bk2ZnhzcOmnYdSMgXoXTBILRrlPAi0U-PDokYmCSrwA,196924
+databricks/sdk/service/serving.py,sha256=kH9Y8fRcwkKg1wx5m3Gih2CcOsdbAnDXgkCeSOB4QY8,196934
 databricks/sdk/service/settings.py,sha256=Y21HbrWLwaJw60tS2oghbd28Z2W77zfCzFZ0vs4uhVs,319026
-databricks/sdk/service/sharing.py,sha256=ymeJWblpB00vZrvflEVHa0joxWM9tafu6y09rR0I57k,104961
+databricks/sdk/service/sharing.py,sha256=vs69Y2FE28A4XnLxDGKL2HvV-LKgB-DV7nSnbYc_y60,105419
 databricks/sdk/service/sql.py,sha256=GU_8ALx1r3lgu_FCmEJs8zTcWFKtJhKQ2CJcIcJgxzo,399590
 databricks/sdk/service/vectorsearch.py,sha256=5p5pW94Bv_Q2tw4j8kFb35nAoFa9GUG5FIHTdfAHWps,77997
 databricks/sdk/service/workspace.py,sha256=BCoi43R1L2eJI9DYq9vwCVdjbMsdLuzDebN6AZvT4kg,128751
-databricks_sdk-0.43.0.dist-info/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
-databricks_sdk-0.43.0.dist-info/METADATA,sha256=0xu8M1qFM0O86f49-ZQhhNVOaUfdMpIAleEXm9NdShU,38301
-databricks_sdk-0.43.0.dist-info/NOTICE,sha256=tkRcQYA1k68wDLcnOWbg2xJDsUOJw8G8DGBhb8dnI3w,1588
-databricks_sdk-0.43.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-databricks_sdk-0.43.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
-databricks_sdk-0.43.0.dist-info/RECORD,,
+databricks_sdk-0.44.0.dist-info/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
+databricks_sdk-0.44.0.dist-info/METADATA,sha256=2Y3c4mHOJ-s54GaO-Up6IP8WToQdcTnC59B2v3F6Ah4,38301
+databricks_sdk-0.44.0.dist-info/NOTICE,sha256=tkRcQYA1k68wDLcnOWbg2xJDsUOJw8G8DGBhb8dnI3w,1588
+databricks_sdk-0.44.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+databricks_sdk-0.44.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
+databricks_sdk-0.44.0.dist-info/RECORD,,