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

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.1'

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: databricks-sdk
-Version: 0.43.0
+Version: 0.44.1
 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.1.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
@@ -7,7 +7,7 @@ databricks/sdk/casing.py,sha256=NKYPrfPbQjM7lU4hhNQK3z1jb_VEA29BfH4FEdby2tg,1137
 databricks/sdk/clock.py,sha256=Ivlow0r_TkXcTJ8UXkxSA0czKrY0GvwHAeOvjPkJnAQ,1360
 databricks/sdk/config.py,sha256=fUCRZX7tpY4LNvh3ilPtbaoC6wlFauWj9jyONsMDLWM,20314
 databricks/sdk/core.py,sha256=s8W2UlRg8y9vCencdiFocYs49hLS3bltswLi00_cn38,4086
-databricks/sdk/credentials_provider.py,sha256=IZyjoxdKGwDqhRJ6jTFT-mS1lf8l1LDpxNBbPSdDHlY,35311
+databricks/sdk/credentials_provider.py,sha256=ioH9ZYNc7e9L2KDCQRL2oOUiFifzHqqn11a2wG6kxj0,37817
 databricks/sdk/data_plane.py,sha256=Kr81rnBlgqnlwNlDbV4JWzzTA5NSVlf0J-DQOylp0w4,2582
 databricks/sdk/dbutils.py,sha256=HFCuB-el6SFKhF8qRfJxYANtyLTm-VG9GtQuQgZXFkM,15741
 databricks/sdk/environments.py,sha256=5KoVuVfF-ZX17rua1sH3EJCCtniVrREXBXsMNDEV-UU,4293
@@ -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=-1tkgtTZjJHtveKwFUf8Bh2OZhXwVDjVJDxyd7uV6Ic,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
@@ -34,19 +34,19 @@ databricks/sdk/logger/round_trip_logger.py,sha256=SMtHDfdqy5Noge2iZO-LpuEm92rz3A
 databricks/sdk/mixins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databricks/sdk/mixins/compute.py,sha256=khb00BzBckc4RLUF4-GnNMCSO5lXKt_XYMM3IhiUxlA,11237
 databricks/sdk/mixins/files.py,sha256=G5Tv0lbQGgOECwvFxmACX5X24ZMP6ZQD7ZudtkORbDw,28283
-databricks/sdk/mixins/jobs.py,sha256=PqiYWi0O-drOjAD6fsptHuAj6IofCdR5e_F3I2BzHqQ,2622
+databricks/sdk/mixins/jobs.py,sha256=xDXHKG5dSTF-K9tlyfwSu47get_7PF8koBnhHXnROLw,2966
 databricks/sdk/mixins/open_ai_client.py,sha256=gvh8xKm1O8B23gNmIv6vxSxk8tNmphUqvS-XkkS260U,4776
 databricks/sdk/mixins/workspace.py,sha256=dWMNvuEi8jJ5wMhrDt1LiqxNdWSsmEuDTzrcZR-eJzY,4896
 databricks/sdk/runtime/__init__.py,sha256=9NnZkBzeZXZRQxcE1qKzAszQEzcpIgpL7lQzW3_kxEU,7266
 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.1.dist-info/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
+databricks_sdk-0.44.1.dist-info/METADATA,sha256=iCnThrfuodH8SezEZUmnw5lHzZcHnUjK-nanS8oNJh0,38301
+databricks_sdk-0.44.1.dist-info/NOTICE,sha256=tkRcQYA1k68wDLcnOWbg2xJDsUOJw8G8DGBhb8dnI3w,1588
+databricks_sdk-0.44.1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+databricks_sdk-0.44.1.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
+databricks_sdk-0.44.1.dist-info/RECORD,,