databricks-sdk 0.70.0__py3-none-any.whl → 0.71.0__py3-none-any.whl

databricks/sdk/service/oauth2.py

@@ -194,24 +194,6 @@ class DeletePublishedAppIntegrationOutput:
         return cls()
 
 
-@dataclass
-class DeleteResponse:
-    def as_dict(self) -> dict:
-        """Serializes the DeleteResponse into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the DeleteResponse into a shallow dictionary of its immediate attributes."""
-        body = {}
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> DeleteResponse:
-        """Deserializes the DeleteResponse from a dictionary."""
-        return cls()
-
-
 @dataclass
 class FederationPolicy:
     create_time: Optional[str] = None
@@ -1013,6 +995,7 @@ class AccountFederationPolicyAPI:
 
         :returns: :class:`FederationPolicy`
         """
+
         body = policy.as_dict()
         query = {}
         if policy_id is not None:
@@ -1112,6 +1095,7 @@ class AccountFederationPolicyAPI:
 
         :returns: :class:`FederationPolicy`
         """
+
         body = policy.as_dict()
         query = {}
         if update_mask is not None:
@@ -1169,6 +1153,7 @@ class CustomAppIntegrationAPI:
 
         :returns: :class:`CreateCustomAppIntegrationOutput`
         """
+
         body = {}
         if confidential is not None:
             body["confidential"] = confidential
@@ -1301,6 +1286,7 @@ class CustomAppIntegrationAPI:
 
 
         """
+
         body = {}
         if redirect_urls is not None:
             body["redirect_urls"] = [v for v in redirect_urls]
@@ -1386,6 +1372,7 @@ class PublishedAppIntegrationAPI:
 
         :returns: :class:`CreatePublishedAppIntegrationOutput`
         """
+
         body = {}
         if app_id is not None:
             body["app_id"] = app_id
@@ -1486,6 +1473,7 @@ class PublishedAppIntegrationAPI:
 
 
         """
+
         body = {}
         if token_access_policy is not None:
             body["token_access_policy"] = token_access_policy.as_dict()
@@ -1561,6 +1549,7 @@ class ServicePrincipalFederationPolicyAPI:
 
         :returns: :class:`FederationPolicy`
         """
+
         body = policy.as_dict()
         query = {}
         if policy_id is not None:
@@ -1677,6 +1666,7 @@ class ServicePrincipalFederationPolicyAPI:
 
         :returns: :class:`FederationPolicy`
         """
+
         body = policy.as_dict()
         query = {}
         if update_mask is not None:
@@ -1726,6 +1716,7 @@ class ServicePrincipalSecretsAPI:
 
         :returns: :class:`CreateServicePrincipalSecretResponse`
         """
+
         body = {}
         if lifetime is not None:
             body["lifetime"] = lifetime
@@ -1836,6 +1827,7 @@ class ServicePrincipalSecretsProxyAPI:
 
         :returns: :class:`CreateServicePrincipalSecretResponse`
         """
+
         body = {}
         if lifetime is not None:
             body["lifetime"] = lifetime

databricks/sdk/service/pipelines.py

@@ -2024,6 +2024,9 @@ class PipelineSpec:
     trigger: Optional[PipelineTrigger] = None
     """Which pipeline trigger to use. Deprecated: Use `continuous` instead."""
 
+    usage_policy_id: Optional[str] = None
+    """Usage policy of this pipeline."""
+
     def as_dict(self) -> dict:
         """Serializes the PipelineSpec into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -2081,6 +2084,8 @@ class PipelineSpec:
             body["target"] = self.target
         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
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -2140,6 +2145,8 @@ class PipelineSpec:
             body["target"] = self.target
         if self.trigger:
             body["trigger"] = self.trigger
+        if self.usage_policy_id is not None:
+            body["usage_policy_id"] = self.usage_policy_id
         return body
 
     @classmethod
@@ -2173,6 +2180,7 @@ class PipelineSpec:
             tags=d.get("tags", None),
             target=d.get("target", None),
             trigger=_from_dict(d, "trigger", PipelineTrigger),
+            usage_policy_id=d.get("usage_policy_id", None),
         )
 
 
@@ -3318,6 +3326,7 @@ class PipelinesAPI:
         tags: Optional[Dict[str, str]] = None,
         target: Optional[str] = None,
         trigger: Optional[PipelineTrigger] = None,
+        usage_policy_id: Optional[str] = None,
     ) -> CreatePipelineResponse:
         """Creates a new data processing pipeline based on the requested configuration. If successful, this
         method returns the ID of the new pipeline.
@@ -3388,9 +3397,12 @@ class PipelinesAPI:
           for pipeline creation in favor of the `schema` field.
         :param trigger: :class:`PipelineTrigger` (optional)
           Which pipeline trigger to use. Deprecated: Use `continuous` instead.
+        :param usage_policy_id: str (optional)
+          Usage policy of this pipeline.
 
         :returns: :class:`CreatePipelineResponse`
         """
+
         body = {}
         if allow_duplicate_names is not None:
             body["allow_duplicate_names"] = allow_duplicate_names
@@ -3452,6 +3464,8 @@ class PipelinesAPI:
             body["target"] = target
         if trigger is not None:
             body["trigger"] = trigger.as_dict()
+        if usage_policy_id is not None:
+            body["usage_policy_id"] = usage_policy_id
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/json",
@@ -3699,6 +3713,7 @@ class PipelinesAPI:
 
         :returns: :class:`PipelinePermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -3741,6 +3756,7 @@ class PipelinesAPI:
 
         :returns: :class:`StartUpdateResponse`
         """
+
         body = {}
         if cause is not None:
             body["cause"] = cause.value
@@ -3815,6 +3831,7 @@ class PipelinesAPI:
         tags: Optional[Dict[str, str]] = None,
         target: Optional[str] = None,
         trigger: Optional[PipelineTrigger] = None,
+        usage_policy_id: Optional[str] = None,
     ):
         """Updates a pipeline with the supplied configuration.
 
@@ -3888,9 +3905,12 @@ class PipelinesAPI:
           for pipeline creation in favor of the `schema` field.
         :param trigger: :class:`PipelineTrigger` (optional)
           Which pipeline trigger to use. Deprecated: Use `continuous` instead.
+        :param usage_policy_id: str (optional)
+          Usage policy of this pipeline.
 
 
         """
+
         body = {}
         if allow_duplicate_names is not None:
             body["allow_duplicate_names"] = allow_duplicate_names
@@ -3952,6 +3972,8 @@ class PipelinesAPI:
             body["target"] = target
         if trigger is not None:
             body["trigger"] = trigger.as_dict()
+        if usage_policy_id is not None:
+            body["usage_policy_id"] = usage_policy_id
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/json",
@@ -3970,6 +3992,7 @@ class PipelinesAPI:
 
         :returns: :class:`PipelinePermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]

databricks/sdk/service/provisioning.py

@@ -1765,6 +1765,7 @@ class CredentialsAPI:
 
         :returns: :class:`Credential`
         """
+
         body = {}
         if aws_credentials is not None:
             body["aws_credentials"] = aws_credentials.as_dict()
@@ -1876,6 +1877,7 @@ class EncryptionKeysAPI:
 
         :returns: :class:`CustomerManagedKey`
         """
+
         body = {}
         if aws_key_info is not None:
             body["aws_key_info"] = aws_key_info.as_dict()
@@ -1996,6 +1998,7 @@ class NetworksAPI:
 
         :returns: :class:`Network`
         """
+
         body = {}
         if gcp_network_info is not None:
             body["gcp_network_info"] = gcp_network_info.as_dict()
@@ -2111,6 +2114,7 @@ class PrivateAccessAPI:
 
         :returns: :class:`PrivateAccessSettings`
         """
+
         body = {}
         if allowed_vpc_endpoint_ids is not None:
             body["allowed_vpc_endpoint_ids"] = [v for v in allowed_vpc_endpoint_ids]
@@ -2205,6 +2209,7 @@ class PrivateAccessAPI:
 
         :returns: :class:`PrivateAccessSettings`
         """
+
         body = customer_facing_private_access_settings.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2247,6 +2252,7 @@ class StorageAPI:
 
         :returns: :class:`StorageConfiguration`
         """
+
         body = {}
         if role_arn is not None:
             body["role_arn"] = role_arn
@@ -2356,6 +2362,7 @@ class VpcEndpointsAPI:
 
         :returns: :class:`VpcEndpoint`
         """
+
         body = {}
         if aws_vpc_endpoint_id is not None:
             body["aws_vpc_endpoint_id"] = aws_vpc_endpoint_id
@@ -2595,6 +2602,7 @@ class WorkspacesAPI:
           Long-running operation waiter for :class:`Workspace`.
           See :method:wait_get_workspace_running for more details.
         """
+
         body = {}
         if aws_region is not None:
             body["aws_region"] = aws_region
@@ -2768,6 +2776,7 @@ class WorkspacesAPI:
           Long-running operation waiter for :class:`Workspace`.
           See :method:wait_get_workspace_running for more details.
         """
+
         body = customer_facing_workspace.as_dict()
         query = {}
         if update_mask is not None:

databricks/sdk/service/qualitymonitorv2.py

@@ -151,6 +151,7 @@ class QualityMonitorV2API:
 
         :returns: :class:`QualityMonitor`
         """
+
         body = quality_monitor.as_dict()
         headers = {
             "Accept": "application/json",
@@ -237,6 +238,7 @@ class QualityMonitorV2API:
 
         :returns: :class:`QualityMonitor`
         """
+
         body = quality_monitor.as_dict()
         headers = {
             "Accept": "application/json",

databricks/sdk/service/serving.py

@@ -1036,24 +1036,6 @@ class DataframeSplitInput:
         return cls(columns=d.get("columns", None), data=d.get("data", None), index=d.get("index", None))
 
 
-@dataclass
-class DeleteResponse:
-    def as_dict(self) -> dict:
-        """Serializes the DeleteResponse into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the DeleteResponse into a shallow dictionary of its immediate attributes."""
-        body = {}
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> DeleteResponse:
-        """Deserializes the DeleteResponse from a dictionary."""
-        return cls()
-
-
 @dataclass
 class EmailNotifications:
     on_update_failure: Optional[List[str]] = None
@@ -1141,15 +1123,15 @@ class EmbeddingsV1ResponseEmbeddingElementObject(Enum):
 
 @dataclass
 class EndpointCoreConfigInput:
+    name: str
+    """The name of the serving endpoint to update. This field is required."""
+
     auto_capture_config: Optional[AutoCaptureConfigInput] = None
     """Configuration for Inference Tables which automatically logs requests and responses to Unity
     Catalog. Note: this field is deprecated for creating new provisioned throughput endpoints, or
     updating existing provisioned throughput endpoints that never have inference table configured;
     in these cases please use AI Gateway to manage inference tables."""
 
-    name: Optional[str] = None
-    """The name of the serving endpoint to update. This field is required."""
-
     served_entities: Optional[List[ServedEntityInput]] = None
     """The list of served entities under the serving endpoint config."""
 
@@ -4109,6 +4091,7 @@ class ServingEndpointsAPI:
           Long-running operation waiter for :class:`ServingEndpointDetailed`.
           See :method:wait_get_serving_endpoint_not_updating for more details.
         """
+
         body = {}
         if ai_gateway is not None:
             body["ai_gateway"] = ai_gateway.as_dict()
@@ -4196,6 +4179,7 @@ class ServingEndpointsAPI:
           Long-running operation waiter for :class:`ServingEndpointDetailed`.
           See :method:wait_get_serving_endpoint_not_updating for more details.
         """
+
         body = {}
         if ai_gateway is not None:
             body["ai_gateway"] = ai_gateway.as_dict()
@@ -4366,6 +4350,7 @@ class ServingEndpointsAPI:
 
         :returns: :class:`HttpRequestResponse`
         """
+
         body = {}
         if connection_name is not None:
             body["connection_name"] = connection_name
@@ -4436,6 +4421,7 @@ class ServingEndpointsAPI:
 
         :returns: :class:`EndpointTags`
         """
+
         body = {}
         if add_tags is not None:
             body["add_tags"] = [v.as_dict() for v in add_tags]
@@ -4459,6 +4445,7 @@ class ServingEndpointsAPI:
 
         :returns: :class:`PutResponse`
         """
+
         body = {}
         if rate_limits is not None:
             body["rate_limits"] = [v.as_dict() for v in rate_limits]
@@ -4501,6 +4488,7 @@ class ServingEndpointsAPI:
 
         :returns: :class:`PutAiGatewayResponse`
         """
+
         body = {}
         if fallback_config is not None:
             body["fallback_config"] = fallback_config.as_dict()
@@ -4593,6 +4581,7 @@ class ServingEndpointsAPI:
 
         :returns: :class:`QueryEndpointResponse`
         """
+
         body = {}
         if client_request_id is not None:
             body["client_request_id"] = client_request_id
@@ -4656,6 +4645,7 @@ class ServingEndpointsAPI:
 
         :returns: :class:`ServingEndpointPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -4701,6 +4691,7 @@ class ServingEndpointsAPI:
           Long-running operation waiter for :class:`ServingEndpointDetailed`.
           See :method:wait_get_serving_endpoint_not_updating for more details.
         """
+
         body = {}
         if auto_capture_config is not None:
             body["auto_capture_config"] = auto_capture_config.as_dict()
@@ -4753,6 +4744,7 @@ class ServingEndpointsAPI:
 
         :returns: :class:`UpdateInferenceEndpointNotificationsResponse`
         """
+
         body = {}
         if email_notifications is not None:
             body["email_notifications"] = email_notifications.as_dict()
@@ -4779,6 +4771,7 @@ class ServingEndpointsAPI:
 
         :returns: :class:`ServingEndpointPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -4807,6 +4800,7 @@ class ServingEndpointsAPI:
           Long-running operation waiter for :class:`ServingEndpointDetailed`.
           See :method:wait_get_serving_endpoint_not_updating for more details.
         """
+
         body = {}
         if config is not None:
             body["config"] = config.as_dict()
@@ -4931,6 +4925,7 @@ class ServingEndpointsDataPlaneAPI:
 
         :returns: :class:`QueryEndpointResponse`
         """
+
         body = {}
         if client_request_id is not None:
             body["client_request_id"] = client_request_id