databricks-sdk 0.53.0__py3-none-any.whl → 0.54.0__py3-none-any.whl

databricks/sdk/service/cleanrooms.py

@@ -338,6 +338,15 @@ 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_collaborator_aliases: Optional[List[str]] = 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 = {}
@@ -345,6 +354,12 @@ class CleanRoomAssetNotebook:
             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_collaborator_aliases:
+            body["runner_collaborator_aliases"] = [v for v in self.runner_collaborator_aliases]
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -354,12 +369,24 @@ class CleanRoomAssetNotebook:
             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_collaborator_aliases:
+            body["runner_collaborator_aliases"] = self.runner_collaborator_aliases
         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_collaborator_aliases=d.get("runner_collaborator_aliases", None),
+        )
 
 
 class CleanRoomAssetStatusEnum(Enum):
@@ -585,6 +612,78 @@ class CleanRoomCollaborator:
         )
 
 
+@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"""
+
+    review_sub_reason: Optional[CleanRoomNotebookReviewNotebookReviewSubReason] = None
+    """specified when the review was not explicitly made by a user"""
+
+    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.review_sub_reason is not None:
+            body["review_sub_reason"] = self.review_sub_reason.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.review_sub_reason is not None:
+            body["review_sub_reason"] = self.review_sub_reason
+        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),
+            review_sub_reason=_enum(d, "review_sub_reason", CleanRoomNotebookReviewNotebookReviewSubReason),
+            reviewer_collaborator_alias=d.get("reviewer_collaborator_alias", None),
+        )
+
+
+class CleanRoomNotebookReviewNotebookReviewState(Enum):
+
+    APPROVED = "APPROVED"
+    PENDING = "PENDING"
+    REJECTED = "REJECTED"
+
+
+class CleanRoomNotebookReviewNotebookReviewSubReason(Enum):
+
+    AUTO_APPROVED = "AUTO_APPROVED"
+    BACKFILLED = "BACKFILLED"
+
+
 @dataclass
 class CleanRoomNotebookTaskRun:
     """Stores information about a single task run."""
@@ -594,12 +693,18 @@ class CleanRoomNotebookTaskRun:
     LIST API. if the task was run within the same workspace the API is being called. If the task run
     was in a different workspace under the same metastore, only the workspace_id is included."""
 
+    notebook_etag: Optional[str] = None
+    """Etag of the notebook executed in this task run, used to identify the notebook version."""
+
     notebook_job_run_state: Optional[jobs.CleanRoomTaskRunState] = None
     """State of the task run."""
 
     notebook_name: Optional[str] = None
     """Asset name of the notebook executed in this task run."""
 
+    notebook_updated_at: Optional[int] = None
+    """The timestamp of when the notebook was last updated."""
+
     output_schema_expiration_time: Optional[int] = None
     """Expiration time of the output schema of the task run (if any), in epoch milliseconds."""
 
@@ -617,10 +722,14 @@ class CleanRoomNotebookTaskRun:
         body = {}
         if self.collaborator_job_run_info:
             body["collaborator_job_run_info"] = self.collaborator_job_run_info.as_dict()
+        if self.notebook_etag is not None:
+            body["notebook_etag"] = self.notebook_etag
         if self.notebook_job_run_state:
             body["notebook_job_run_state"] = self.notebook_job_run_state.as_dict()
         if self.notebook_name is not None:
             body["notebook_name"] = self.notebook_name
+        if self.notebook_updated_at is not None:
+            body["notebook_updated_at"] = self.notebook_updated_at
         if self.output_schema_expiration_time is not None:
             body["output_schema_expiration_time"] = self.output_schema_expiration_time
         if self.output_schema_name is not None:
@@ -636,10 +745,14 @@ class CleanRoomNotebookTaskRun:
         body = {}
         if self.collaborator_job_run_info:
             body["collaborator_job_run_info"] = self.collaborator_job_run_info
+        if self.notebook_etag is not None:
+            body["notebook_etag"] = self.notebook_etag
         if self.notebook_job_run_state:
             body["notebook_job_run_state"] = self.notebook_job_run_state
         if self.notebook_name is not None:
             body["notebook_name"] = self.notebook_name
+        if self.notebook_updated_at is not None:
+            body["notebook_updated_at"] = self.notebook_updated_at
         if self.output_schema_expiration_time is not None:
             body["output_schema_expiration_time"] = self.output_schema_expiration_time
         if self.output_schema_name is not None:
@@ -655,8 +768,10 @@ class CleanRoomNotebookTaskRun:
         """Deserializes the CleanRoomNotebookTaskRun from a dictionary."""
         return cls(
             collaborator_job_run_info=_from_dict(d, "collaborator_job_run_info", CollaboratorJobRunInfo),
+            notebook_etag=d.get("notebook_etag", None),
             notebook_job_run_state=_from_dict(d, "notebook_job_run_state", jobs.CleanRoomTaskRunState),
             notebook_name=d.get("notebook_name", None),
+            notebook_updated_at=d.get("notebook_updated_at", None),
             output_schema_expiration_time=d.get("output_schema_expiration_time", None),
             output_schema_name=d.get("output_schema_name", None),
             run_duration=d.get("run_duration", None),

databricks/sdk/service/compute.py

@@ -729,7 +729,8 @@ class ClusterAttributes:
 
     cluster_name: Optional[str] = None
     """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."""
+    creation, the cluster name will be an empty string. For job clusters, the cluster name is
+    automatically set based on the job and job run IDs."""
 
     custom_tags: Optional[Dict[str, str]] = None
     """Additional tags for cluster resources. Databricks will tag all cluster resources (e.g., AWS
@@ -1118,7 +1119,8 @@ class ClusterDetails:
 
     cluster_name: Optional[str] = None
     """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."""
+    creation, the cluster name will be an empty string. For job clusters, the cluster name is
+    automatically set based on the job and job run IDs."""
 
     cluster_source: Optional[ClusterSource] = None
     """Determines whether the cluster was created by a user through the UI, created by the Databricks
@@ -2300,7 +2302,8 @@ class ClusterSpec:
 
     cluster_name: Optional[str] = None
     """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."""
+    creation, the cluster name will be an empty string. For job clusters, the cluster name is
+    automatically set based on the job and job run IDs."""
 
     custom_tags: Optional[Dict[str, str]] = None
     """Additional tags for cluster resources. Databricks will tag all cluster resources (e.g., AWS
@@ -2803,7 +2806,8 @@ class CreateCluster:
 
     cluster_name: Optional[str] = None
     """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."""
+    creation, the cluster name will be an empty string. For job clusters, the cluster name is
+    automatically set based on the job and job run IDs."""
 
     custom_tags: Optional[Dict[str, str]] = None
     """Additional tags for cluster resources. Databricks will tag all cluster resources (e.g., AWS
@@ -4117,7 +4121,8 @@ class EditCluster:
 
     cluster_name: Optional[str] = None
     """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."""
+    creation, the cluster name will be an empty string. For job clusters, the cluster name is
+    automatically set based on the job and job run IDs."""
 
     custom_tags: Optional[Dict[str, str]] = None
     """Additional tags for cluster resources. Databricks will tag all cluster resources (e.g., AWS
@@ -4499,10 +4504,6 @@ class EditInstancePool:
     min_idle_instances: Optional[int] = None
     """Minimum number of idle instances to keep in the instance pool"""
 
-    node_type_flexibility: Optional[NodeTypeFlexibility] = None
-    """For Fleet-pool V2, this object contains the information about the alternate node type ids to use
-    when attempting to launch a cluster if the node type id is not available."""
-
     def as_dict(self) -> dict:
         """Serializes the EditInstancePool into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -4518,8 +4519,6 @@ class EditInstancePool:
             body["max_capacity"] = self.max_capacity
         if self.min_idle_instances is not None:
             body["min_idle_instances"] = self.min_idle_instances
-        if self.node_type_flexibility:
-            body["node_type_flexibility"] = self.node_type_flexibility.as_dict()
         if self.node_type_id is not None:
             body["node_type_id"] = self.node_type_id
         return body
@@ -4539,8 +4538,6 @@ class EditInstancePool:
             body["max_capacity"] = self.max_capacity
         if self.min_idle_instances is not None:
             body["min_idle_instances"] = self.min_idle_instances
-        if self.node_type_flexibility:
-            body["node_type_flexibility"] = self.node_type_flexibility
         if self.node_type_id is not None:
             body["node_type_id"] = self.node_type_id
         return body
@@ -4555,7 +4552,6 @@ class EditInstancePool:
             instance_pool_name=d.get("instance_pool_name", None),
             max_capacity=d.get("max_capacity", None),
             min_idle_instances=d.get("min_idle_instances", None),
-            node_type_flexibility=_from_dict(d, "node_type_flexibility", NodeTypeFlexibility),
             node_type_id=d.get("node_type_id", None),
         )
 
@@ -4782,9 +4778,7 @@ class EnforceClusterComplianceResponse:
 @dataclass
 class Environment:
     """The environment entity used to preserve serverless environment side panel, jobs' environment for
-    non-notebook task, and DLT's environment for classic and serverless pipelines. (Note: DLT uses a
-    copied version of the Environment proto below, at
-    //spark/pipelines/api/protos/copied/libraries-environments-copy.proto) In this minimal
+    non-notebook task, and DLT's environment for classic and serverless pipelines. In this minimal
     environment spec, only pip dependencies are supported."""
 
     client: str
@@ -4800,6 +4794,13 @@ class Environment:
     Databricks), <vcs project url> E.g. dependencies: ["foo==0.0.1", "-r
     /Workspace/test/requirements.txt"]"""
 
+    environment_version: Optional[str] = None
+    """We renamed `client` to `environment_version` in notebook exports. This field is meant solely so
+    that imported notebooks with `environment_version` can be deserialized correctly, in a
+    backwards-compatible way (i.e. if `client` is specified instead of `environment_version`, it
+    will be deserialized correctly). Do NOT use this field for any other purpose, e.g. notebook
+    storage. This field is not yet exposed to customers (e.g. in the jobs API)."""
+
     jar_dependencies: Optional[List[str]] = None
     """List of jar dependencies, should be string representing volume paths. For example:
     `/Volumes/path/to/test.jar`."""
@@ -4811,6 +4812,8 @@ class Environment:
             body["client"] = self.client
         if self.dependencies:
             body["dependencies"] = [v for v in self.dependencies]
+        if self.environment_version is not None:
+            body["environment_version"] = self.environment_version
         if self.jar_dependencies:
             body["jar_dependencies"] = [v for v in self.jar_dependencies]
         return body
@@ -4822,6 +4825,8 @@ class Environment:
             body["client"] = self.client
         if self.dependencies:
             body["dependencies"] = self.dependencies
+        if self.environment_version is not None:
+            body["environment_version"] = self.environment_version
         if self.jar_dependencies:
             body["jar_dependencies"] = self.jar_dependencies
         return body
@@ -4832,6 +4837,7 @@ class Environment:
         return cls(
             client=d.get("client", None),
             dependencies=d.get("dependencies", None),
+            environment_version=d.get("environment_version", None),
             jar_dependencies=d.get("jar_dependencies", None),
         )
 
@@ -5497,10 +5503,6 @@ class GetInstancePool:
     min_idle_instances: Optional[int] = None
     """Minimum number of idle instances to keep in the instance pool"""
 
-    node_type_flexibility: Optional[NodeTypeFlexibility] = None
-    """For Fleet-pool V2, this object contains the information about the alternate node type ids to use
-    when attempting to launch a cluster if the node type id is not available."""
-
     node_type_id: Optional[str] = None
     """This field encodes, through a single value, the resources available to each of the Spark nodes
     in this cluster. For example, the Spark nodes can be provisioned and optimized for memory or
@@ -5551,8 +5553,6 @@ class GetInstancePool:
             body["max_capacity"] = self.max_capacity
         if self.min_idle_instances is not None:
             body["min_idle_instances"] = self.min_idle_instances
-        if self.node_type_flexibility:
-            body["node_type_flexibility"] = self.node_type_flexibility.as_dict()
         if self.node_type_id is not None:
             body["node_type_id"] = self.node_type_id
         if self.preloaded_docker_images:
@@ -5594,8 +5594,6 @@ class GetInstancePool:
             body["max_capacity"] = self.max_capacity
         if self.min_idle_instances is not None:
             body["min_idle_instances"] = self.min_idle_instances
-        if self.node_type_flexibility:
-            body["node_type_flexibility"] = self.node_type_flexibility
         if self.node_type_id is not None:
             body["node_type_id"] = self.node_type_id
         if self.preloaded_docker_images:
@@ -5626,7 +5624,6 @@ class GetInstancePool:
             instance_pool_name=d.get("instance_pool_name", None),
             max_capacity=d.get("max_capacity", None),
             min_idle_instances=d.get("min_idle_instances", None),
-            node_type_flexibility=_from_dict(d, "node_type_flexibility", NodeTypeFlexibility),
             node_type_id=d.get("node_type_id", None),
             preloaded_docker_images=_repeated_dict(d, "preloaded_docker_images", DockerImage),
             preloaded_spark_versions=d.get("preloaded_spark_versions", None),
@@ -6461,10 +6458,6 @@ class InstancePoolAndStats:
     min_idle_instances: Optional[int] = None
     """Minimum number of idle instances to keep in the instance pool"""
 
-    node_type_flexibility: Optional[NodeTypeFlexibility] = None
-    """For Fleet-pool V2, this object contains the information about the alternate node type ids to use
-    when attempting to launch a cluster if the node type id is not available."""
-
     node_type_id: Optional[str] = None
     """This field encodes, through a single value, the resources available to each of the Spark nodes
     in this cluster. For example, the Spark nodes can be provisioned and optimized for memory or
@@ -6515,8 +6508,6 @@ class InstancePoolAndStats:
             body["max_capacity"] = self.max_capacity
         if self.min_idle_instances is not None:
             body["min_idle_instances"] = self.min_idle_instances
-        if self.node_type_flexibility:
-            body["node_type_flexibility"] = self.node_type_flexibility.as_dict()
         if self.node_type_id is not None:
             body["node_type_id"] = self.node_type_id
         if self.preloaded_docker_images:
@@ -6558,8 +6549,6 @@ class InstancePoolAndStats:
             body["max_capacity"] = self.max_capacity
         if self.min_idle_instances is not None:
             body["min_idle_instances"] = self.min_idle_instances
-        if self.node_type_flexibility:
-            body["node_type_flexibility"] = self.node_type_flexibility
         if self.node_type_id is not None:
             body["node_type_id"] = self.node_type_id
         if self.preloaded_docker_images:
@@ -6590,7 +6579,6 @@ class InstancePoolAndStats:
             instance_pool_name=d.get("instance_pool_name", None),
             max_capacity=d.get("max_capacity", None),
             min_idle_instances=d.get("min_idle_instances", None),
-            node_type_flexibility=_from_dict(d, "node_type_flexibility", NodeTypeFlexibility),
             node_type_id=d.get("node_type_id", None),
             preloaded_docker_images=_repeated_dict(d, "preloaded_docker_images", DockerImage),
             preloaded_spark_versions=d.get("preloaded_spark_versions", None),
@@ -8053,28 +8041,6 @@ class NodeType:
         )
 
 
-@dataclass
-class NodeTypeFlexibility:
-    """For Fleet-V2 using classic clusters, this object contains the information about the alternate
-    node type ids to use when attempting to launch a cluster. It can be used with both the driver
-    and worker node types."""
-
-    def as_dict(self) -> dict:
-        """Serializes the NodeTypeFlexibility into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the NodeTypeFlexibility into a shallow dictionary of its immediate attributes."""
-        body = {}
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> NodeTypeFlexibility:
-        """Deserializes the NodeTypeFlexibility from a dictionary."""
-        return cls()
-
-
 @dataclass
 class PendingInstanceError:
     """Error message of a failed pending instances"""
@@ -9404,7 +9370,8 @@ class UpdateClusterResource:
 
     cluster_name: Optional[str] = None
     """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."""
+    creation, the cluster name will be an empty string. For job clusters, the cluster name is
+    automatically set based on the job and job run IDs."""
 
     custom_tags: Optional[Dict[str, str]] = None
     """Additional tags for cluster resources. Databricks will tag all cluster resources (e.g., AWS
@@ -10374,7 +10341,8 @@ class ClustersAPI:
           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.
+          the cluster name will be an empty string. For job clusters, the cluster name is automatically set
+          based on the job and job run IDs.
         :param custom_tags: Dict[str,str] (optional)
           Additional tags for cluster resources. Databricks will tag all cluster resources (e.g., AWS
           instances and EBS volumes) with these tags in addition to `default_tags`. Notes:
@@ -10766,7 +10734,8 @@ class ClustersAPI:
           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.
+          the cluster name will be an empty string. For job clusters, the cluster name is automatically set
+          based on the job and job run IDs.
         :param custom_tags: Dict[str,str] (optional)
           Additional tags for cluster resources. Databricks will tag all cluster resources (e.g., AWS
           instances and EBS volumes) with these tags in addition to `default_tags`. Notes:
@@ -12227,7 +12196,6 @@ class InstancePoolsAPI:
         idle_instance_autotermination_minutes: Optional[int] = None,
         max_capacity: Optional[int] = None,
         min_idle_instances: Optional[int] = None,
-        node_type_flexibility: Optional[NodeTypeFlexibility] = None,
     ):
         """Edit an existing instance pool.
 
@@ -12260,9 +12228,6 @@ class InstancePoolsAPI:
           upsize requests.
         :param min_idle_instances: int (optional)
           Minimum number of idle instances to keep in the instance pool
-        :param node_type_flexibility: :class:`NodeTypeFlexibility` (optional)
-          For Fleet-pool V2, this object contains the information about the alternate node type ids to use
-          when attempting to launch a cluster if the node type id is not available.
 
 
         """
@@ -12279,8 +12244,6 @@ class InstancePoolsAPI:
             body["max_capacity"] = max_capacity
         if min_idle_instances is not None:
             body["min_idle_instances"] = min_idle_instances
-        if node_type_flexibility is not None:
-            body["node_type_flexibility"] = node_type_flexibility.as_dict()
         if node_type_id is not None:
             body["node_type_id"] = node_type_id
         headers = {
@@ -12440,8 +12403,10 @@ class InstanceProfilesAPI:
     ):
         """Register an instance profile.
 
-        In the UI, you can select the instance profile when launching clusters. This API is only available to
-        admin users.
+        Registers an instance profile in Databricks. In the UI, you can then give users the permission to use
+        this instance profile when launching clusters.
+
+        This API is only available to admin users.
 
         :param instance_profile_arn: str
           The AWS ARN of the instance profile to register with Databricks. This field is required.

databricks/sdk/service/dashboards.py

@@ -1255,6 +1255,9 @@ class MessageErrorType(Enum):
     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"
+    DESCRIBE_QUERY_INVALID_SQL_ERROR = "DESCRIBE_QUERY_INVALID_SQL_ERROR"
+    DESCRIBE_QUERY_TIMEOUT = "DESCRIBE_QUERY_TIMEOUT"
+    DESCRIBE_QUERY_UNEXPECTED_FAILURE = "DESCRIBE_QUERY_UNEXPECTED_FAILURE"
     FUNCTIONS_NOT_AVAILABLE_EXCEPTION = "FUNCTIONS_NOT_AVAILABLE_EXCEPTION"
     FUNCTION_ARGUMENTS_INVALID_EXCEPTION = "FUNCTION_ARGUMENTS_INVALID_EXCEPTION"
     FUNCTION_ARGUMENTS_INVALID_JSON_EXCEPTION = "FUNCTION_ARGUMENTS_INVALID_JSON_EXCEPTION"
@@ -1267,9 +1270,13 @@ class MessageErrorType(Enum):
     ILLEGAL_PARAMETER_DEFINITION_EXCEPTION = "ILLEGAL_PARAMETER_DEFINITION_EXCEPTION"
     INVALID_CERTIFIED_ANSWER_FUNCTION_EXCEPTION = "INVALID_CERTIFIED_ANSWER_FUNCTION_EXCEPTION"
     INVALID_CERTIFIED_ANSWER_IDENTIFIER_EXCEPTION = "INVALID_CERTIFIED_ANSWER_IDENTIFIER_EXCEPTION"
+    INVALID_CHAT_COMPLETION_ARGUMENTS_JSON_EXCEPTION = "INVALID_CHAT_COMPLETION_ARGUMENTS_JSON_EXCEPTION"
     INVALID_CHAT_COMPLETION_JSON_EXCEPTION = "INVALID_CHAT_COMPLETION_JSON_EXCEPTION"
     INVALID_COMPLETION_REQUEST_EXCEPTION = "INVALID_COMPLETION_REQUEST_EXCEPTION"
     INVALID_FUNCTION_CALL_EXCEPTION = "INVALID_FUNCTION_CALL_EXCEPTION"
+    INVALID_SQL_MULTIPLE_DATASET_REFERENCES_EXCEPTION = "INVALID_SQL_MULTIPLE_DATASET_REFERENCES_EXCEPTION"
+    INVALID_SQL_MULTIPLE_STATEMENTS_EXCEPTION = "INVALID_SQL_MULTIPLE_STATEMENTS_EXCEPTION"
+    INVALID_SQL_UNKNOWN_TABLE_EXCEPTION = "INVALID_SQL_UNKNOWN_TABLE_EXCEPTION"
     INVALID_TABLE_IDENTIFIER_EXCEPTION = "INVALID_TABLE_IDENTIFIER_EXCEPTION"
     LOCAL_CONTEXT_EXCEEDED_EXCEPTION = "LOCAL_CONTEXT_EXCEEDED_EXCEPTION"
     MESSAGE_CANCELLED_WHILE_EXECUTING_EXCEPTION = "MESSAGE_CANCELLED_WHILE_EXECUTING_EXCEPTION"