databricks-sdk 0.61.0__py3-none-any.whl → 0.63.0__py3-none-any.whl

databricks/sdk/service/jobs.py

@@ -3037,6 +3037,11 @@ class JobSettings:
     the job runs only when triggered by clicking “Run Now” in the Jobs UI or sending an API
     request to `runNow`."""
 
+    usage_policy_id: Optional[str] = None
+    """The id of the user specified usage policy to use for this job. If not specified, a default usage
+    policy may be applied when creating or modifying the job. See `effective_budget_policy_id` for
+    the budget policy used by this workload."""
+
     webhook_notifications: Optional[WebhookNotifications] = None
     """A collection of system notification IDs to notify when runs of this job begin or complete."""
 
@@ -3089,6 +3094,8 @@ class JobSettings:
             body["timeout_seconds"] = self.timeout_seconds
         if self.trigger:
             body["trigger"] = self.trigger.as_dict()
+        if self.usage_policy_id is not None:
+            body["usage_policy_id"] = self.usage_policy_id
         if self.webhook_notifications:
             body["webhook_notifications"] = self.webhook_notifications.as_dict()
         return body
@@ -3142,6 +3149,8 @@ class JobSettings:
             body["timeout_seconds"] = self.timeout_seconds
         if self.trigger:
             body["trigger"] = self.trigger
+        if self.usage_policy_id is not None:
+            body["usage_policy_id"] = self.usage_policy_id
         if self.webhook_notifications:
             body["webhook_notifications"] = self.webhook_notifications
         return body
@@ -3173,6 +3182,7 @@ class JobSettings:
             tasks=_repeated_dict(d, "tasks", Task),
             timeout_seconds=d.get("timeout_seconds", None),
             trigger=_from_dict(d, "trigger", TriggerSettings),
+            usage_policy_id=d.get("usage_policy_id", None),
             webhook_notifications=_from_dict(d, "webhook_notifications", WebhookNotifications),
         )
 
@@ -7223,6 +7233,73 @@ class SubscriptionSubscriber:
         return cls(destination_id=d.get("destination_id", None), user_name=d.get("user_name", None))
 
 
+@dataclass
+class TableState:
+    has_seen_updates: Optional[bool] = None
+    """Whether or not the table has seen updates since either the creation of the trigger or the last
+    successful evaluation of the trigger"""
+
+    table_name: Optional[str] = None
+    """Full table name of the table to monitor, e.g. `mycatalog.myschema.mytable`"""
+
+    def as_dict(self) -> dict:
+        """Serializes the TableState into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.has_seen_updates is not None:
+            body["has_seen_updates"] = self.has_seen_updates
+        if self.table_name is not None:
+            body["table_name"] = self.table_name
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the TableState into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.has_seen_updates is not None:
+            body["has_seen_updates"] = self.has_seen_updates
+        if self.table_name is not None:
+            body["table_name"] = self.table_name
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> TableState:
+        """Deserializes the TableState from a dictionary."""
+        return cls(has_seen_updates=d.get("has_seen_updates", None), table_name=d.get("table_name", None))
+
+
+@dataclass
+class TableTriggerState:
+    last_seen_table_states: Optional[List[TableState]] = None
+
+    using_scalable_monitoring: Optional[bool] = None
+    """Indicates whether the trigger is using scalable monitoring."""
+
+    def as_dict(self) -> dict:
+        """Serializes the TableTriggerState into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.last_seen_table_states:
+            body["last_seen_table_states"] = [v.as_dict() for v in self.last_seen_table_states]
+        if self.using_scalable_monitoring is not None:
+            body["using_scalable_monitoring"] = self.using_scalable_monitoring
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the TableTriggerState into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.last_seen_table_states:
+            body["last_seen_table_states"] = self.last_seen_table_states
+        if self.using_scalable_monitoring is not None:
+            body["using_scalable_monitoring"] = self.using_scalable_monitoring
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> TableTriggerState:
+        """Deserializes the TableTriggerState from a dictionary."""
+        return cls(
+            last_seen_table_states=_repeated_dict(d, "last_seen_table_states", TableState),
+            using_scalable_monitoring=d.get("using_scalable_monitoring", None),
+        )
+
+
 @dataclass
 class TableUpdateTriggerConfiguration:
     condition: Optional[Condition] = None
@@ -7817,6 +7894,8 @@ class TerminationCodeCode(Enum):
     run failed due to a cloud provider issue. Refer to the state message for further details. *
     `MAX_JOB_QUEUE_SIZE_EXCEEDED`: The run was skipped due to reaching the job level queue size
     limit. * `DISABLED`: The run was never executed because it was disabled explicitly by the user.
+    * `BREAKING_CHANGE`: Run failed because of an intentional breaking change in Spark, but it will
+    be retried with a mitigation config.
 
     [Link]: https://kb.databricks.com/en_US/notebooks/too-many-execution-contexts-are-open-right-now"""
 
@@ -7993,11 +8072,15 @@ class TriggerSettings:
 class TriggerStateProto:
     file_arrival: Optional[FileArrivalTriggerState] = None
 
+    table: Optional[TableTriggerState] = None
+
     def as_dict(self) -> dict:
         """Serializes the TriggerStateProto into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.file_arrival:
             body["file_arrival"] = self.file_arrival.as_dict()
+        if self.table:
+            body["table"] = self.table.as_dict()
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -8005,12 +8088,17 @@ class TriggerStateProto:
         body = {}
         if self.file_arrival:
             body["file_arrival"] = self.file_arrival
+        if self.table:
+            body["table"] = self.table
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> TriggerStateProto:
         """Deserializes the TriggerStateProto from a dictionary."""
-        return cls(file_arrival=_from_dict(d, "file_arrival", FileArrivalTriggerState))
+        return cls(
+            file_arrival=_from_dict(d, "file_arrival", FileArrivalTriggerState),
+            table=_from_dict(d, "table", TableTriggerState),
+        )
 
 
 class TriggerType(Enum):
@@ -8361,6 +8449,7 @@ class JobsAPI:
         tasks: Optional[List[Task]] = None,
         timeout_seconds: Optional[int] = None,
         trigger: Optional[TriggerSettings] = None,
+        usage_policy_id: Optional[str] = None,
         webhook_notifications: Optional[WebhookNotifications] = None,
     ) -> CreateResponse:
         """Create a new job.
@@ -8455,6 +8544,10 @@ class JobsAPI:
           A configuration to trigger a run when certain conditions are met. The default behavior is that the
           job runs only when triggered by clicking “Run Now” in the Jobs UI or sending an API request to
           `runNow`.
+        :param usage_policy_id: str (optional)
+          The id of the user specified usage policy to use for this job. If not specified, a default usage
+          policy may be applied when creating or modifying the job. See `effective_budget_policy_id` for the
+          budget policy used by this workload.
         :param webhook_notifications: :class:`WebhookNotifications` (optional)
           A collection of system notification IDs to notify when runs of this job begin or complete.
 
@@ -8509,6 +8602,8 @@ class JobsAPI:
             body["timeout_seconds"] = timeout_seconds
         if trigger is not None:
             body["trigger"] = trigger.as_dict()
+        if usage_policy_id is not None:
+            body["usage_policy_id"] = usage_policy_id
         if webhook_notifications is not None:
             body["webhook_notifications"] = webhook_notifications.as_dict()
         headers = {
@@ -9305,6 +9400,7 @@ class JobsAPI:
         run_name: Optional[str] = None,
         tasks: Optional[List[SubmitTask]] = None,
         timeout_seconds: Optional[int] = None,
+        usage_policy_id: Optional[str] = None,
         webhook_notifications: Optional[WebhookNotifications] = None,
     ) -> Wait[Run]:
         """Submit a one-time run. This endpoint allows you to submit a workload directly without creating a job.
@@ -9356,6 +9452,9 @@ class JobsAPI:
         :param tasks: List[:class:`SubmitTask`] (optional)
         :param timeout_seconds: int (optional)
           An optional timeout applied to each run of this job. A value of `0` means no timeout.
+        :param usage_policy_id: str (optional)
+          The user specified id of the usage policy to use for this one-time run. If not specified, a default
+          usage policy may be applied when creating or modifying the job.
         :param webhook_notifications: :class:`WebhookNotifications` (optional)
           A collection of system notification IDs to notify when the run begins or completes.
 
@@ -9390,6 +9489,8 @@ class JobsAPI:
             body["tasks"] = [v.as_dict() for v in tasks]
         if timeout_seconds is not None:
             body["timeout_seconds"] = timeout_seconds
+        if usage_policy_id is not None:
+            body["usage_policy_id"] = usage_policy_id
         if webhook_notifications is not None:
             body["webhook_notifications"] = webhook_notifications.as_dict()
         headers = {
@@ -9420,6 +9521,7 @@ class JobsAPI:
         run_name: Optional[str] = None,
         tasks: Optional[List[SubmitTask]] = None,
         timeout_seconds: Optional[int] = None,
+        usage_policy_id: Optional[str] = None,
         webhook_notifications: Optional[WebhookNotifications] = None,
         timeout=timedelta(minutes=20),
     ) -> Run:
@@ -9437,6 +9539,7 @@ class JobsAPI:
             run_name=run_name,
             tasks=tasks,
             timeout_seconds=timeout_seconds,
+            usage_policy_id=usage_policy_id,
             webhook_notifications=webhook_notifications,
         ).result(timeout=timeout)
 

databricks/sdk/service/ml.py

@@ -3494,10 +3494,8 @@ class PublishSpec:
     online_table_name: str
     """The full three-part (catalog, schema, table) name of the online table."""
 
-    publish_mode: Optional[PublishSpecPublishMode] = None
-    """The publish mode of the pipeline that syncs the online table with the source table. Defaults to
-    TRIGGERED if not specified. All publish modes require the source table to have Change Data Feed
-    (CDF) enabled."""
+    publish_mode: PublishSpecPublishMode
+    """The publish mode of the pipeline that syncs the online table with the source table."""
 
     def as_dict(self) -> dict:
         """Serializes the PublishSpec into a dictionary suitable for use as a JSON request body."""
@@ -3534,6 +3532,7 @@ class PublishSpec:
 class PublishSpecPublishMode(Enum):
 
     CONTINUOUS = "CONTINUOUS"
+    SNAPSHOT = "SNAPSHOT"
     TRIGGERED = "TRIGGERED"
 
 

databricks/sdk/service/serving.py

@@ -990,10 +990,13 @@ class DatabricksModelServingConfig:
 @dataclass
 class DataframeSplitInput:
     columns: Optional[List[Any]] = None
+    """Columns array for the dataframe"""
 
     data: Optional[List[Any]] = None
+    """Data array for the dataframe"""
 
     index: Optional[List[int]] = None
+    """Index array for the dataframe"""
 
     def as_dict(self) -> dict:
         """Serializes the DataframeSplitInput into a dictionary suitable for use as a JSON request body."""
@@ -1041,9 +1044,46 @@ class DeleteResponse:
         return cls()
 
 
+@dataclass
+class EmailNotifications:
+    on_update_failure: Optional[List[str]] = None
+    """A list of email addresses to be notified when an endpoint fails to update its configuration or
+    state."""
+
+    on_update_success: Optional[List[str]] = None
+    """A list of email addresses to be notified when an endpoint successfully updates its configuration
+    or state."""
+
+    def as_dict(self) -> dict:
+        """Serializes the EmailNotifications into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.on_update_failure:
+            body["on_update_failure"] = [v for v in self.on_update_failure]
+        if self.on_update_success:
+            body["on_update_success"] = [v for v in self.on_update_success]
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the EmailNotifications into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.on_update_failure:
+            body["on_update_failure"] = self.on_update_failure
+        if self.on_update_success:
+            body["on_update_success"] = self.on_update_success
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> EmailNotifications:
+        """Deserializes the EmailNotifications from a dictionary."""
+        return cls(
+            on_update_failure=d.get("on_update_failure", None), on_update_success=d.get("on_update_success", None)
+        )
+
+
 @dataclass
 class EmbeddingsV1ResponseEmbeddingElement:
     embedding: Optional[List[float]] = None
+    """The embedding vector"""
 
     index: Optional[int] = None
     """The index of the embedding in the response."""
@@ -3262,11 +3302,11 @@ class ServedModelState:
 
 class ServedModelStateDeployment(Enum):
 
-    ABORTED = "DEPLOYMENT_ABORTED"
-    CREATING = "DEPLOYMENT_CREATING"
-    FAILED = "DEPLOYMENT_FAILED"
-    READY = "DEPLOYMENT_READY"
-    RECOVERING = "DEPLOYMENT_RECOVERING"
+    DEPLOYMENT_ABORTED = "DEPLOYMENT_ABORTED"
+    DEPLOYMENT_CREATING = "DEPLOYMENT_CREATING"
+    DEPLOYMENT_FAILED = "DEPLOYMENT_FAILED"
+    DEPLOYMENT_READY = "DEPLOYMENT_READY"
+    DEPLOYMENT_RECOVERING = "DEPLOYMENT_RECOVERING"
 
 
 @dataclass
@@ -3545,6 +3585,9 @@ class ServingEndpointDetailed:
     description: Optional[str] = None
     """Description of the serving model"""
 
+    email_notifications: Optional[EmailNotifications] = None
+    """Email notification settings."""
+
     endpoint_url: Optional[str] = None
     """Endpoint invocation url if route optimization is enabled for endpoint"""
 
@@ -3593,6 +3636,8 @@ class ServingEndpointDetailed:
             body["data_plane_info"] = self.data_plane_info.as_dict()
         if self.description is not None:
             body["description"] = self.description
+        if self.email_notifications:
+            body["email_notifications"] = self.email_notifications.as_dict()
         if self.endpoint_url is not None:
             body["endpoint_url"] = self.endpoint_url
         if self.id is not None:
@@ -3632,6 +3677,8 @@ class ServingEndpointDetailed:
             body["data_plane_info"] = self.data_plane_info
         if self.description is not None:
             body["description"] = self.description
+        if self.email_notifications:
+            body["email_notifications"] = self.email_notifications
         if self.endpoint_url is not None:
             body["endpoint_url"] = self.endpoint_url
         if self.id is not None:
@@ -3665,6 +3712,7 @@ class ServingEndpointDetailed:
             creator=d.get("creator", None),
             data_plane_info=_from_dict(d, "data_plane_info", ModelDataPlaneInfo),
             description=d.get("description", None),
+            email_notifications=_from_dict(d, "email_notifications", EmailNotifications),
             endpoint_url=d.get("endpoint_url", None),
             id=d.get("id", None),
             last_updated_timestamp=d.get("last_updated_timestamp", None),
@@ -3978,6 +4026,7 @@ class ServingEndpointsAPI:
         budget_policy_id: Optional[str] = None,
         config: Optional[EndpointCoreConfigInput] = None,
         description: Optional[str] = None,
+        email_notifications: Optional[EmailNotifications] = None,
         rate_limits: Optional[List[RateLimit]] = None,
         route_optimized: Optional[bool] = None,
         tags: Optional[List[EndpointTag]] = None,
@@ -3996,6 +4045,8 @@ class ServingEndpointsAPI:
         :param config: :class:`EndpointCoreConfigInput` (optional)
           The core config of the serving endpoint.
         :param description: str (optional)
+        :param email_notifications: :class:`EmailNotifications` (optional)
+          Email notification settings.
         :param rate_limits: List[:class:`RateLimit`] (optional)
           Rate limits to be applied to the serving endpoint. NOTE: this field is deprecated, please use AI
           Gateway to manage rate limits.
@@ -4017,6 +4068,8 @@ class ServingEndpointsAPI:
             body["config"] = config.as_dict()
         if description is not None:
             body["description"] = description
+        if email_notifications is not None:
+            body["email_notifications"] = email_notifications.as_dict()
         if name is not None:
             body["name"] = name
         if rate_limits is not None:
@@ -4045,6 +4098,7 @@ class ServingEndpointsAPI:
         budget_policy_id: Optional[str] = None,
         config: Optional[EndpointCoreConfigInput] = None,
         description: Optional[str] = None,
+        email_notifications: Optional[EmailNotifications] = None,
         rate_limits: Optional[List[RateLimit]] = None,
         route_optimized: Optional[bool] = None,
         tags: Optional[List[EndpointTag]] = None,
@@ -4055,6 +4109,7 @@ class ServingEndpointsAPI:
             budget_policy_id=budget_policy_id,
             config=config,
             description=description,
+            email_notifications=email_notifications,
             name=name,
             rate_limits=rate_limits,
             route_optimized=route_optimized,
@@ -4068,6 +4123,7 @@ class ServingEndpointsAPI:
         *,
         ai_gateway: Optional[AiGatewayConfig] = None,
         budget_policy_id: Optional[str] = None,
+        email_notifications: Optional[EmailNotifications] = None,
         tags: Optional[List[EndpointTag]] = None,
     ) -> Wait[ServingEndpointDetailed]:
         """Create a new PT serving endpoint.
@@ -4081,6 +4137,8 @@ class ServingEndpointsAPI:
           The AI Gateway configuration for the serving endpoint.
         :param budget_policy_id: str (optional)
           The budget policy associated with the endpoint.
+        :param email_notifications: :class:`EmailNotifications` (optional)
+          Email notification settings.
         :param tags: List[:class:`EndpointTag`] (optional)
           Tags to be attached to the serving endpoint and automatically propagated to billing logs.
 
@@ -4095,6 +4153,8 @@ class ServingEndpointsAPI:
             body["budget_policy_id"] = budget_policy_id
         if config is not None:
             body["config"] = config.as_dict()
+        if email_notifications is not None:
+            body["email_notifications"] = email_notifications.as_dict()
         if name is not None:
             body["name"] = name
         if tags is not None:
@@ -4118,11 +4178,17 @@ class ServingEndpointsAPI:
         *,
         ai_gateway: Optional[AiGatewayConfig] = None,
         budget_policy_id: Optional[str] = None,
+        email_notifications: Optional[EmailNotifications] = None,
         tags: Optional[List[EndpointTag]] = None,
         timeout=timedelta(minutes=20),
     ) -> ServingEndpointDetailed:
         return self.create_provisioned_throughput_endpoint(
-            ai_gateway=ai_gateway, budget_policy_id=budget_policy_id, config=config, name=name, tags=tags
+            ai_gateway=ai_gateway,
+            budget_policy_id=budget_policy_id,
+            config=config,
+            email_notifications=email_notifications,
+            name=name,
+            tags=tags,
         ).result(timeout=timeout)
 
     def delete(self, name: str):
@@ -4408,6 +4474,7 @@ class ServingEndpointsAPI:
         self,
         name: str,
         *,
+        client_request_id: Optional[str] = None,
         dataframe_records: Optional[List[Any]] = None,
         dataframe_split: Optional[DataframeSplitInput] = None,
         extra_params: Optional[Dict[str, str]] = None,
@@ -4421,11 +4488,15 @@ class ServingEndpointsAPI:
         stop: Optional[List[str]] = None,
         stream: Optional[bool] = None,
         temperature: Optional[float] = None,
+        usage_context: Optional[Dict[str, str]] = None,
     ) -> QueryEndpointResponse:
-        """Query a serving endpoint.
+        """Query a serving endpoint
 
         :param name: str
-          The name of the serving endpoint. This field is required.
+          The name of the serving endpoint. This field is required and is provided via the path parameter.
+        :param client_request_id: str (optional)
+          Optional user-provided request identifier that will be recorded in the inference table and the usage
+          tracking table.
         :param dataframe_records: List[Any] (optional)
           Pandas Dataframe input in the records orientation.
         :param dataframe_split: :class:`DataframeSplitInput` (optional)
@@ -4446,8 +4517,8 @@ class ServingEndpointsAPI:
           The max tokens field used ONLY for __completions__ and __chat external & foundation model__ serving
           endpoints. This is an integer and should only be used with other chat/completions query fields.
         :param messages: List[:class:`ChatMessage`] (optional)
-          The messages field used ONLY for __chat external & foundation model__ serving endpoints. This is a
-          map of strings and should only be used with other chat query fields.
+          The messages field used ONLY for __chat external & foundation model__ serving endpoints. This is an
+          array of ChatMessage objects and should only be used with other chat query fields.
         :param n: int (optional)
           The n (number of candidates) field used ONLY for __completions__ and __chat external & foundation
           model__ serving endpoints. This is an integer between 1 and 5 with a default of 1 and should only be
@@ -4467,10 +4538,14 @@ class ServingEndpointsAPI:
           The temperature field used ONLY for __completions__ and __chat external & foundation model__ serving
           endpoints. This is a float between 0.0 and 2.0 with a default of 1.0 and should only be used with
           other chat/completions query fields.
+        :param usage_context: Dict[str,str] (optional)
+          Optional user-provided context that will be recorded in the usage tracking table.
 
         :returns: :class:`QueryEndpointResponse`
         """
         body = {}
+        if client_request_id is not None:
+            body["client_request_id"] = client_request_id
         if dataframe_records is not None:
             body["dataframe_records"] = [v for v in dataframe_records]
         if dataframe_split is not None:
@@ -4497,6 +4572,8 @@ class ServingEndpointsAPI:
             body["stream"] = stream
         if temperature is not None:
             body["temperature"] = temperature
+        if usage_context is not None:
+            body["usage_context"] = usage_context
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/json",
@@ -4711,6 +4788,7 @@ class ServingEndpointsDataPlaneAPI:
         self,
         name: str,
         *,
+        client_request_id: Optional[str] = None,
         dataframe_records: Optional[List[Any]] = None,
         dataframe_split: Optional[DataframeSplitInput] = None,
         extra_params: Optional[Dict[str, str]] = None,
@@ -4724,11 +4802,15 @@ class ServingEndpointsDataPlaneAPI:
         stop: Optional[List[str]] = None,
         stream: Optional[bool] = None,
         temperature: Optional[float] = None,
+        usage_context: Optional[Dict[str, str]] = None,
     ) -> QueryEndpointResponse:
-        """Query a serving endpoint.
+        """Query a serving endpoint
 
         :param name: str
-          The name of the serving endpoint. This field is required.
+          The name of the serving endpoint. This field is required and is provided via the path parameter.
+        :param client_request_id: str (optional)
+          Optional user-provided request identifier that will be recorded in the inference table and the usage
+          tracking table.
         :param dataframe_records: List[Any] (optional)
           Pandas Dataframe input in the records orientation.
         :param dataframe_split: :class:`DataframeSplitInput` (optional)
@@ -4749,8 +4831,8 @@ class ServingEndpointsDataPlaneAPI:
           The max tokens field used ONLY for __completions__ and __chat external & foundation model__ serving
           endpoints. This is an integer and should only be used with other chat/completions query fields.
         :param messages: List[:class:`ChatMessage`] (optional)
-          The messages field used ONLY for __chat external & foundation model__ serving endpoints. This is a
-          map of strings and should only be used with other chat query fields.
+          The messages field used ONLY for __chat external & foundation model__ serving endpoints. This is an
+          array of ChatMessage objects and should only be used with other chat query fields.
         :param n: int (optional)
           The n (number of candidates) field used ONLY for __completions__ and __chat external & foundation
           model__ serving endpoints. This is an integer between 1 and 5 with a default of 1 and should only be
@@ -4770,10 +4852,14 @@ class ServingEndpointsDataPlaneAPI:
           The temperature field used ONLY for __completions__ and __chat external & foundation model__ serving
           endpoints. This is a float between 0.0 and 2.0 with a default of 1.0 and should only be used with
           other chat/completions query fields.
+        :param usage_context: Dict[str,str] (optional)
+          Optional user-provided context that will be recorded in the usage tracking table.
 
         :returns: :class:`QueryEndpointResponse`
         """
         body = {}
+        if client_request_id is not None:
+            body["client_request_id"] = client_request_id
         if dataframe_records is not None:
             body["dataframe_records"] = [v for v in dataframe_records]
         if dataframe_split is not None:
@@ -4800,6 +4886,8 @@ class ServingEndpointsDataPlaneAPI:
             body["stream"] = stream
         if temperature is not None:
             body["temperature"] = temperature
+        if usage_context is not None:
+            body["usage_context"] = usage_context
         data_plane_info = self._data_plane_info_query(
             name=name,
         )

databricks/sdk/service/settings.py

@@ -4617,6 +4617,18 @@ class SetStatusResponse:
 
 @dataclass
 class SlackConfig:
+    channel_id: Optional[str] = None
+    """[Input-Only] Slack channel ID for notifications."""
+
+    channel_id_set: Optional[bool] = None
+    """[Output-Only] Whether channel ID is set."""
+
+    oauth_token: Optional[str] = None
+    """[Input-Only] OAuth token for Slack authentication."""
+
+    oauth_token_set: Optional[bool] = None
+    """[Output-Only] Whether OAuth token is set."""
+
     url: Optional[str] = None
     """[Input-Only] URL for Slack destination."""
 
@@ -4626,6 +4638,14 @@ class SlackConfig:
     def as_dict(self) -> dict:
         """Serializes the SlackConfig into a dictionary suitable for use as a JSON request body."""
         body = {}
+        if self.channel_id is not None:
+            body["channel_id"] = self.channel_id
+        if self.channel_id_set is not None:
+            body["channel_id_set"] = self.channel_id_set
+        if self.oauth_token is not None:
+            body["oauth_token"] = self.oauth_token
+        if self.oauth_token_set is not None:
+            body["oauth_token_set"] = self.oauth_token_set
         if self.url is not None:
             body["url"] = self.url
         if self.url_set is not None:
@@ -4635,6 +4655,14 @@ class SlackConfig:
     def as_shallow_dict(self) -> dict:
         """Serializes the SlackConfig into a shallow dictionary of its immediate attributes."""
         body = {}
+        if self.channel_id is not None:
+            body["channel_id"] = self.channel_id
+        if self.channel_id_set is not None:
+            body["channel_id_set"] = self.channel_id_set
+        if self.oauth_token is not None:
+            body["oauth_token"] = self.oauth_token
+        if self.oauth_token_set is not None:
+            body["oauth_token_set"] = self.oauth_token_set
         if self.url is not None:
             body["url"] = self.url
         if self.url_set is not None:
@@ -4644,7 +4672,14 @@ class SlackConfig:
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> SlackConfig:
         """Deserializes the SlackConfig from a dictionary."""
-        return cls(url=d.get("url", None), url_set=d.get("url_set", None))
+        return cls(
+            channel_id=d.get("channel_id", None),
+            channel_id_set=d.get("channel_id_set", None),
+            oauth_token=d.get("oauth_token", None),
+            oauth_token_set=d.get("oauth_token_set", None),
+            url=d.get("url", None),
+            url_set=d.get("url_set", None),
+        )
 
 
 @dataclass

databricks/sdk/version.py

@@ -1 +1 @@
-__version__ = "0.61.0"
+__version__ = "0.63.0"

{databricks_sdk-0.61.0.dist-info → databricks_sdk-0.63.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-sdk
-Version: 0.61.0
+Version: 0.63.0
 Summary: Databricks SDK for Python (Beta)
 Project-URL: Documentation, https://databricks-sdk-py.readthedocs.io
 Keywords: databricks,sdk