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

databricks/sdk/service/dashboards.py

@@ -594,12 +594,15 @@ class GenieMessage:
     `ASKING_AI`: Waiting for the LLM to respond to the user's question. * `PENDING_WAREHOUSE`:
     Waiting for warehouse before the SQL query can start executing. * `EXECUTING_QUERY`: Executing a
     generated SQL query. Get the SQL query result by calling
-    [getMessageQueryResult](:method:genie/getMessageQueryResult) API. * `FAILED`: The response
-    generation or query execution failed. See `error` field. * `COMPLETED`: Message processing is
-    completed. Results are in the `attachments` field. Get the SQL query result by calling
-    [getMessageQueryResult](:method:genie/getMessageQueryResult) API. * `SUBMITTED`: Message has
-    been submitted. * `QUERY_RESULT_EXPIRED`: SQL result is not available anymore. The user needs to
-    rerun the query. * `CANCELLED`: Message has been cancelled."""
+    [getMessageAttachmentQueryResult](:method:genie/getMessageAttachmentQueryResult) API. *
+    `FAILED`: The response generation or query execution failed. See `error` field. * `COMPLETED`:
+    Message processing is completed. Results are in the `attachments` field. Get the SQL query
+    result by calling
+    [getMessageAttachmentQueryResult](:method:genie/getMessageAttachmentQueryResult) API. *
+    `SUBMITTED`: Message has been submitted. * `QUERY_RESULT_EXPIRED`: SQL result is not available
+    anymore. The user needs to rerun the query. Rerun the SQL query result by calling
+    [executeMessageAttachmentQuery](:method:genie/executeMessageAttachmentQuery) API. * `CANCELLED`:
+    Message has been cancelled."""
 
     user_id: Optional[int] = None
     """ID of the user who created the message"""
@@ -697,6 +700,10 @@ class GenieQueryAttachment:
     query_result_metadata: Optional[GenieResultMetadata] = None
     """Metadata associated with the query result."""
 
+    statement_id: Optional[str] = None
+    """Statement Execution API statement id. Use [Get status, manifest, and result first
+    chunk](:method:statementexecution/getstatement) to get the full result data."""
+
     title: Optional[str] = None
     """Name of the query"""
 
@@ -713,6 +720,8 @@ class GenieQueryAttachment:
             body["query"] = self.query
         if self.query_result_metadata:
             body["query_result_metadata"] = self.query_result_metadata.as_dict()
+        if self.statement_id is not None:
+            body["statement_id"] = self.statement_id
         if self.title is not None:
             body["title"] = self.title
         return body
@@ -730,6 +739,8 @@ class GenieQueryAttachment:
             body["query"] = self.query
         if self.query_result_metadata:
             body["query_result_metadata"] = self.query_result_metadata
+        if self.statement_id is not None:
+            body["statement_id"] = self.statement_id
         if self.title is not None:
             body["title"] = self.title
         return body
@@ -743,6 +754,7 @@ class GenieQueryAttachment:
             last_updated_timestamp=d.get("last_updated_timestamp", None),
             query=d.get("query", None),
             query_result_metadata=_from_dict(d, "query_result_metadata", GenieResultMetadata),
+            statement_id=d.get("statement_id", None),
             title=d.get("title", None),
         )
 
@@ -1062,6 +1074,7 @@ class MessageErrorType(Enum):
     CHAT_COMPLETION_NETWORK_EXCEPTION = "CHAT_COMPLETION_NETWORK_EXCEPTION"
     CONTENT_FILTER_EXCEPTION = "CONTENT_FILTER_EXCEPTION"
     CONTEXT_EXCEEDED_EXCEPTION = "CONTEXT_EXCEEDED_EXCEPTION"
+    COULD_NOT_GET_MODEL_DEPLOYMENTS_EXCEPTION = "COULD_NOT_GET_MODEL_DEPLOYMENTS_EXCEPTION"
     COULD_NOT_GET_UC_SCHEMA_EXCEPTION = "COULD_NOT_GET_UC_SCHEMA_EXCEPTION"
     DEPLOYMENT_NOT_FOUND_EXCEPTION = "DEPLOYMENT_NOT_FOUND_EXCEPTION"
     FUNCTIONS_NOT_AVAILABLE_EXCEPTION = "FUNCTIONS_NOT_AVAILABLE_EXCEPTION"
@@ -1069,6 +1082,7 @@ class MessageErrorType(Enum):
     FUNCTION_ARGUMENTS_INVALID_JSON_EXCEPTION = "FUNCTION_ARGUMENTS_INVALID_JSON_EXCEPTION"
     FUNCTION_ARGUMENTS_INVALID_TYPE_EXCEPTION = "FUNCTION_ARGUMENTS_INVALID_TYPE_EXCEPTION"
     FUNCTION_CALL_MISSING_PARAMETER_EXCEPTION = "FUNCTION_CALL_MISSING_PARAMETER_EXCEPTION"
+    GENERATED_SQL_QUERY_TOO_LONG_EXCEPTION = "GENERATED_SQL_QUERY_TOO_LONG_EXCEPTION"
     GENERIC_CHAT_COMPLETION_EXCEPTION = "GENERIC_CHAT_COMPLETION_EXCEPTION"
     GENERIC_CHAT_COMPLETION_SERVICE_EXCEPTION = "GENERIC_CHAT_COMPLETION_SERVICE_EXCEPTION"
     GENERIC_SQL_EXEC_API_CALL_EXCEPTION = "GENERIC_SQL_EXEC_API_CALL_EXCEPTION"
@@ -1083,6 +1097,7 @@ class MessageErrorType(Enum):
     MESSAGE_CANCELLED_WHILE_EXECUTING_EXCEPTION = "MESSAGE_CANCELLED_WHILE_EXECUTING_EXCEPTION"
     MESSAGE_DELETED_WHILE_EXECUTING_EXCEPTION = "MESSAGE_DELETED_WHILE_EXECUTING_EXCEPTION"
     MESSAGE_UPDATED_WHILE_EXECUTING_EXCEPTION = "MESSAGE_UPDATED_WHILE_EXECUTING_EXCEPTION"
+    MISSING_SQL_QUERY_EXCEPTION = "MISSING_SQL_QUERY_EXCEPTION"
     NO_DEPLOYMENTS_AVAILABLE_TO_WORKSPACE = "NO_DEPLOYMENTS_AVAILABLE_TO_WORKSPACE"
     NO_QUERY_TO_VISUALIZE_EXCEPTION = "NO_QUERY_TO_VISUALIZE_EXCEPTION"
     NO_TABLES_TO_QUERY_EXCEPTION = "NO_TABLES_TO_QUERY_EXCEPTION"
@@ -1107,12 +1122,15 @@ class MessageStatus(Enum):
     `ASKING_AI`: Waiting for the LLM to respond to the user's question. * `PENDING_WAREHOUSE`:
     Waiting for warehouse before the SQL query can start executing. * `EXECUTING_QUERY`: Executing a
     generated SQL query. Get the SQL query result by calling
-    [getMessageQueryResult](:method:genie/getMessageQueryResult) API. * `FAILED`: The response
-    generation or query execution failed. See `error` field. * `COMPLETED`: Message processing is
-    completed. Results are in the `attachments` field. Get the SQL query result by calling
-    [getMessageQueryResult](:method:genie/getMessageQueryResult) API. * `SUBMITTED`: Message has
-    been submitted. * `QUERY_RESULT_EXPIRED`: SQL result is not available anymore. The user needs to
-    rerun the query. * `CANCELLED`: Message has been cancelled."""
+    [getMessageAttachmentQueryResult](:method:genie/getMessageAttachmentQueryResult) API. *
+    `FAILED`: The response generation or query execution failed. See `error` field. * `COMPLETED`:
+    Message processing is completed. Results are in the `attachments` field. Get the SQL query
+    result by calling
+    [getMessageAttachmentQueryResult](:method:genie/getMessageAttachmentQueryResult) API. *
+    `SUBMITTED`: Message has been submitted. * `QUERY_RESULT_EXPIRED`: SQL result is not available
+    anymore. The user needs to rerun the query. Rerun the SQL query result by calling
+    [executeMessageAttachmentQuery](:method:genie/executeMessageAttachmentQuery) API. * `CANCELLED`:
+    Message has been cancelled."""
 
     ASKING_AI = "ASKING_AI"
     CANCELLED = "CANCELLED"
@@ -1917,7 +1935,8 @@ class GenieAPI:
     ) -> GenieGetMessageQueryResultResponse:
         """Execute message attachment SQL query.
 
-        Execute the SQL for a message query attachment.
+        Execute the SQL for a message query attachment. Use this API when the query attachment has expired and
+        needs to be re-executed.
 
         :param space_id: str
           Genie space ID
@@ -1945,7 +1964,7 @@ class GenieAPI:
     def execute_message_query(
         self, space_id: str, conversation_id: str, message_id: str
     ) -> GenieGetMessageQueryResultResponse:
-        """Execute SQL query in a conversation message.
+        """[Deprecated] Execute SQL query in a conversation message.
 
         Execute the SQL query in the message.
 
@@ -2059,7 +2078,7 @@ class GenieAPI:
     def get_message_query_result_by_attachment(
         self, space_id: str, conversation_id: str, message_id: str, attachment_id: str
     ) -> GenieGetMessageQueryResultResponse:
-        """[deprecated] Get conversation message SQL query result.
+        """[Deprecated] Get conversation message SQL query result.
 
         Get the result of SQL query if the message has a query attachment. This is only available if a message
         has a query attachment and the message status is `EXECUTING_QUERY` OR `COMPLETED`.
@@ -2088,9 +2107,9 @@ class GenieAPI:
         return GenieGetMessageQueryResultResponse.from_dict(res)
 
     def get_space(self, space_id: str) -> GenieSpace:
-        """Get details of a Genie Space.
+        """Get Genie Space.
 
-        Get a Genie Space.
+        Get details of a Genie Space.
 
         :param space_id: str
           The ID associated with the Genie space

databricks/sdk/service/files.py

@@ -314,12 +314,14 @@ class DirectoryEntry:
 @dataclass
 class DownloadResponse:
     content_length: Optional[int] = None
+    """The length of the HTTP response body in bytes."""
 
     content_type: Optional[str] = None
 
     contents: Optional[BinaryIO] = None
 
     last_modified: Optional[str] = None
+    """The last modified time of the file in HTTP-date (RFC 7231) format."""
 
     def as_dict(self) -> dict:
         """Serializes the DownloadResponse into a dictionary suitable for use as a JSON request body."""
@@ -430,10 +432,12 @@ class GetDirectoryMetadataResponse:
 @dataclass
 class GetMetadataResponse:
     content_length: Optional[int] = None
+    """The length of the HTTP response body in bytes."""
 
     content_type: Optional[str] = None
 
     last_modified: Optional[str] = None
+    """The last modified time of the file in HTTP-date (RFC 7231) format."""
 
     def as_dict(self) -> dict:
         """Serializes the GetMetadataResponse into a dictionary suitable for use as a JSON request body."""

databricks/sdk/service/iam.py

@@ -846,7 +846,7 @@ class ObjectPermissions:
 @dataclass
 class PartialUpdate:
     id: Optional[str] = None
-    """Unique ID for a user in the Databricks workspace."""
+    """Unique ID in the Databricks workspace."""
 
     operations: Optional[List[Patch]] = None
 
@@ -1918,8 +1918,7 @@ class User:
     groups: Optional[List[ComplexValue]] = None
 
     id: Optional[str] = None
-    """Databricks user ID. This is automatically set by Databricks. Any value provided by the client
-    will be ignored."""
+    """Databricks user ID."""
 
     name: Optional[Name] = None
 
@@ -2480,7 +2479,7 @@ class AccountGroupsAPI:
         Partially updates the details of a group.
 
         :param id: str
-          Unique ID for a group in the Databricks account.
+          Unique ID in the Databricks workspace.
         :param operations: List[:class:`Patch`] (optional)
         :param schemas: List[:class:`PatchSchema`] (optional)
           The schema of the patch request. Must be ["urn:ietf:params:scim:api:messages:2.0:PatchOp"].
@@ -2493,7 +2492,6 @@ class AccountGroupsAPI:
         if schemas is not None:
             body["schemas"] = [v.value for v in schemas]
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -2557,7 +2555,6 @@ class AccountGroupsAPI:
         if schemas is not None:
             body["schemas"] = [v.value for v in schemas]
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -2765,7 +2762,7 @@ class AccountServicePrincipalsAPI:
         Partially updates the details of a single service principal in the Databricks account.
 
         :param id: str
-          Unique ID for a service principal in the Databricks account.
+          Unique ID in the Databricks workspace.
         :param operations: List[:class:`Patch`] (optional)
         :param schemas: List[:class:`PatchSchema`] (optional)
           The schema of the patch request. Must be ["urn:ietf:params:scim:api:messages:2.0:PatchOp"].
@@ -2778,7 +2775,6 @@ class AccountServicePrincipalsAPI:
         if schemas is not None:
             body["schemas"] = [v.value for v in schemas]
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -2848,7 +2844,6 @@ class AccountServicePrincipalsAPI:
         if schemas is not None:
             body["schemas"] = [v.value for v in schemas]
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -2912,8 +2907,7 @@ class AccountUsersAPI:
           External ID is not currently supported. It is reserved for future use.
         :param groups: List[:class:`ComplexValue`] (optional)
         :param id: str (optional)
-          Databricks user ID. This is automatically set by Databricks. Any value provided by the client will
-          be ignored.
+          Databricks user ID.
         :param name: :class:`Name` (optional)
         :param roles: List[:class:`ComplexValue`] (optional)
           Corresponds to AWS instance profile/arn role.
@@ -3123,7 +3117,7 @@ class AccountUsersAPI:
         Partially updates a user resource by applying the supplied operations on specific user attributes.
 
         :param id: str
-          Unique ID for a user in the Databricks account.
+          Unique ID in the Databricks workspace.
         :param operations: List[:class:`Patch`] (optional)
         :param schemas: List[:class:`PatchSchema`] (optional)
           The schema of the patch request. Must be ["urn:ietf:params:scim:api:messages:2.0:PatchOp"].
@@ -3136,7 +3130,6 @@ class AccountUsersAPI:
         if schemas is not None:
             body["schemas"] = [v.value for v in schemas]
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -3164,8 +3157,7 @@ class AccountUsersAPI:
         Replaces a user's information with the data supplied in request.
 
         :param id: str
-          Databricks user ID. This is automatically set by Databricks. Any value provided by the client will
-          be ignored.
+          Databricks user ID.
         :param active: bool (optional)
           If this user is active
         :param display_name: str (optional)
@@ -3215,7 +3207,6 @@ class AccountUsersAPI:
         if user_name is not None:
             body["userName"] = user_name
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -3434,7 +3425,7 @@ class GroupsAPI:
         Partially updates the details of a group.
 
         :param id: str
-          Unique ID for a group in the Databricks workspace.
+          Unique ID in the Databricks workspace.
         :param operations: List[:class:`Patch`] (optional)
         :param schemas: List[:class:`PatchSchema`] (optional)
           The schema of the patch request. Must be ["urn:ietf:params:scim:api:messages:2.0:PatchOp"].
@@ -3447,7 +3438,6 @@ class GroupsAPI:
         if schemas is not None:
             body["schemas"] = [v.value for v in schemas]
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -3509,7 +3499,6 @@ class GroupsAPI:
         if schemas is not None:
             body["schemas"] = [v.value for v in schemas]
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -3922,7 +3911,7 @@ class ServicePrincipalsAPI:
         Partially updates the details of a single service principal in the Databricks workspace.
 
         :param id: str
-          Unique ID for a service principal in the Databricks workspace.
+          Unique ID in the Databricks workspace.
         :param operations: List[:class:`Patch`] (optional)
         :param schemas: List[:class:`PatchSchema`] (optional)
           The schema of the patch request. Must be ["urn:ietf:params:scim:api:messages:2.0:PatchOp"].
@@ -3935,7 +3924,6 @@ class ServicePrincipalsAPI:
         if schemas is not None:
             body["schemas"] = [v.value for v in schemas]
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -4000,7 +3988,6 @@ class ServicePrincipalsAPI:
         if schemas is not None:
             body["schemas"] = [v.value for v in schemas]
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -4059,8 +4046,7 @@ class UsersAPI:
           External ID is not currently supported. It is reserved for future use.
         :param groups: List[:class:`ComplexValue`] (optional)
         :param id: str (optional)
-          Databricks user ID. This is automatically set by Databricks. Any value provided by the client will
-          be ignored.
+          Databricks user ID.
         :param name: :class:`Name` (optional)
         :param roles: List[:class:`ComplexValue`] (optional)
           Corresponds to AWS instance profile/arn role.
@@ -4294,7 +4280,7 @@ class UsersAPI:
         Partially updates a user resource by applying the supplied operations on specific user attributes.
 
         :param id: str
-          Unique ID for a user in the Databricks workspace.
+          Unique ID in the Databricks workspace.
         :param operations: List[:class:`Patch`] (optional)
         :param schemas: List[:class:`PatchSchema`] (optional)
           The schema of the patch request. Must be ["urn:ietf:params:scim:api:messages:2.0:PatchOp"].
@@ -4307,7 +4293,6 @@ class UsersAPI:
         if schemas is not None:
             body["schemas"] = [v.value for v in schemas]
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
@@ -4356,8 +4341,7 @@ class UsersAPI:
         Replaces a user's information with the data supplied in request.
 
         :param id: str
-          Databricks user ID. This is automatically set by Databricks. Any value provided by the client will
-          be ignored.
+          Databricks user ID.
         :param active: bool (optional)
           If this user is active
         :param display_name: str (optional)
@@ -4407,7 +4391,6 @@ class UsersAPI:
         if user_name is not None:
             body["userName"] = user_name
         headers = {
-            "Accept": "application/json",
             "Content-Type": "application/json",
         }
 

databricks/sdk/service/jobs.py

@@ -3659,6 +3659,7 @@ class PerformanceTarget(Enum):
     on serverless compute should be. The performance mode on the job or pipeline should map to a
     performance setting that is passed to Cluster Manager (see cluster-common PerformanceTarget)."""
 
+    BALANCED = "BALANCED"
     COST_OPTIMIZED = "COST_OPTIMIZED"
     PERFORMANCE_OPTIMIZED = "PERFORMANCE_OPTIMIZED"
 

databricks/sdk/service/marketplace.py

@@ -1192,6 +1192,7 @@ class FileParent:
 class FileParentType(Enum):
 
     LISTING = "LISTING"
+    LISTING_RESOURCE = "LISTING_RESOURCE"
     PROVIDER = "PROVIDER"
 
 
@@ -2452,6 +2453,7 @@ class ListingType(Enum):
 
 class MarketplaceFileType(Enum):
 
+    APP = "APP"
     EMBEDDED_NOTEBOOK = "EMBEDDED_NOTEBOOK"
     PROVIDER_ICON = "PROVIDER_ICON"