databricks-sdk 0.69.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
@@ -863,17 +845,37 @@ class SecretInfo:
 
 @dataclass
 class TokenAccessPolicy:
+    absolute_session_lifetime_in_minutes: Optional[int] = None
+    """Absolute OAuth session TTL in minutes. Effective only when the single-use refresh token feature
+    is enabled. This is the absolute TTL of all refresh tokens issued in one OAuth session. When a
+    new refresh token is issued during refresh token rotation, it will inherit the same absolute TTL
+    as the old refresh token. In other words, this represents the maximum amount of time a user can
+    stay logged in without re-authenticating."""
+
     access_token_ttl_in_minutes: Optional[int] = None
     """access token time to live in minutes"""
 
+    enable_single_use_refresh_tokens: Optional[bool] = None
+    """Whether to enable single-use refresh tokens (refresh token rotation). If this feature is
+    enabled, upon successfully getting a new access token using a refresh token, Databricks will
+    issue a new refresh token along with the access token in the response and invalidate the old
+    refresh token. The client should use the new refresh token to get access tokens in future
+    requests."""
+
     refresh_token_ttl_in_minutes: Optional[int] = None
-    """refresh token time to live in minutes"""
+    """Refresh token time to live in minutes. When single-use refresh tokens are enabled, this
+    represents the TTL of an individual refresh token. If the refresh token is used before it
+    expires, a new one is issued with a renewed individual TTL."""
 
     def as_dict(self) -> dict:
         """Serializes the TokenAccessPolicy into a dictionary suitable for use as a JSON request body."""
         body = {}
+        if self.absolute_session_lifetime_in_minutes is not None:
+            body["absolute_session_lifetime_in_minutes"] = self.absolute_session_lifetime_in_minutes
         if self.access_token_ttl_in_minutes is not None:
             body["access_token_ttl_in_minutes"] = self.access_token_ttl_in_minutes
+        if self.enable_single_use_refresh_tokens is not None:
+            body["enable_single_use_refresh_tokens"] = self.enable_single_use_refresh_tokens
         if self.refresh_token_ttl_in_minutes is not None:
             body["refresh_token_ttl_in_minutes"] = self.refresh_token_ttl_in_minutes
         return body
@@ -881,8 +883,12 @@ class TokenAccessPolicy:
     def as_shallow_dict(self) -> dict:
         """Serializes the TokenAccessPolicy into a shallow dictionary of its immediate attributes."""
         body = {}
+        if self.absolute_session_lifetime_in_minutes is not None:
+            body["absolute_session_lifetime_in_minutes"] = self.absolute_session_lifetime_in_minutes
         if self.access_token_ttl_in_minutes is not None:
             body["access_token_ttl_in_minutes"] = self.access_token_ttl_in_minutes
+        if self.enable_single_use_refresh_tokens is not None:
+            body["enable_single_use_refresh_tokens"] = self.enable_single_use_refresh_tokens
         if self.refresh_token_ttl_in_minutes is not None:
             body["refresh_token_ttl_in_minutes"] = self.refresh_token_ttl_in_minutes
         return body
@@ -891,7 +897,9 @@ class TokenAccessPolicy:
     def from_dict(cls, d: Dict[str, Any]) -> TokenAccessPolicy:
         """Deserializes the TokenAccessPolicy from a dictionary."""
         return cls(
+            absolute_session_lifetime_in_minutes=d.get("absolute_session_lifetime_in_minutes", None),
             access_token_ttl_in_minutes=d.get("access_token_ttl_in_minutes", None),
+            enable_single_use_refresh_tokens=d.get("enable_single_use_refresh_tokens", None),
             refresh_token_ttl_in_minutes=d.get("refresh_token_ttl_in_minutes", None),
         )
 
@@ -987,6 +995,7 @@ class AccountFederationPolicyAPI:
 
         :returns: :class:`FederationPolicy`
         """
+
         body = policy.as_dict()
         query = {}
         if policy_id is not None:
@@ -1086,6 +1095,7 @@ class AccountFederationPolicyAPI:
 
         :returns: :class:`FederationPolicy`
         """
+
         body = policy.as_dict()
         query = {}
         if update_mask is not None:
@@ -1143,6 +1153,7 @@ class CustomAppIntegrationAPI:
 
         :returns: :class:`CreateCustomAppIntegrationOutput`
         """
+
         body = {}
         if confidential is not None:
             body["confidential"] = confidential
@@ -1275,6 +1286,7 @@ class CustomAppIntegrationAPI:
 
 
         """
+
         body = {}
         if redirect_urls is not None:
             body["redirect_urls"] = [v for v in redirect_urls]
@@ -1360,6 +1372,7 @@ class PublishedAppIntegrationAPI:
 
         :returns: :class:`CreatePublishedAppIntegrationOutput`
         """
+
         body = {}
         if app_id is not None:
             body["app_id"] = app_id
@@ -1460,6 +1473,7 @@ class PublishedAppIntegrationAPI:
 
 
         """
+
         body = {}
         if token_access_policy is not None:
             body["token_access_policy"] = token_access_policy.as_dict()
@@ -1535,6 +1549,7 @@ class ServicePrincipalFederationPolicyAPI:
 
         :returns: :class:`FederationPolicy`
         """
+
         body = policy.as_dict()
         query = {}
         if policy_id is not None:
@@ -1651,6 +1666,7 @@ class ServicePrincipalFederationPolicyAPI:
 
         :returns: :class:`FederationPolicy`
         """
+
         body = policy.as_dict()
         query = {}
         if update_mask is not None:
@@ -1700,6 +1716,7 @@ class ServicePrincipalSecretsAPI:
 
         :returns: :class:`CreateServicePrincipalSecretResponse`
         """
+
         body = {}
         if lifetime is not None:
             body["lifetime"] = lifetime
@@ -1810,6 +1827,7 @@ class ServicePrincipalSecretsProxyAPI:
 
         :returns: :class:`CreateServicePrincipalSecretResponse`
         """
+
         body = {}
         if lifetime is not None:
             body["lifetime"] = lifetime

databricks/sdk/service/pipelines.py

@@ -119,8 +119,8 @@ class DataPlaneId:
 
 
 class DayOfWeek(Enum):
-    """Days of week in which the restart is allowed to happen (within a five-hour window starting at
-    start_hour). If not specified all days of the week will be used."""
+    """Days of week in which the window is allowed to happen. If not specified all days of the week
+    will be used."""
 
     FRIDAY = "FRIDAY"
     MONDAY = "MONDAY"
@@ -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

@@ -1436,7 +1436,7 @@ class Workspace:
     azure_workspace_info: Optional[AzureWorkspaceInfo] = None
 
     cloud: Optional[str] = None
-    """The cloud name. This field always has the value `gcp`."""
+    """The cloud name. This field can have values like `azure`, `gcp`."""
 
     cloud_resource_container: Optional[CloudResourceContainer] = None
 
@@ -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
@@ -2487,6 +2494,7 @@ class WorkspacesAPI:
         gke_config: Optional[GkeConfig] = None,
         location: Optional[str] = None,
         managed_services_customer_managed_key_id: Optional[str] = None,
+        network_connectivity_config_id: Optional[str] = None,
         network_id: Optional[str] = None,
         pricing_tier: Optional[PricingTier] = None,
         private_access_settings_id: Optional[str] = None,
@@ -2565,6 +2573,10 @@ class WorkspacesAPI:
           The ID of the workspace's managed services encryption key configuration object. This is used to help
           protect and control access to the workspace's notebooks, secrets, Databricks SQL queries, and query
           history. The provided key configuration object property use_cases must contain MANAGED_SERVICES.
+        :param network_connectivity_config_id: str (optional)
+          The object ID of network connectivity config. Once assigned, the workspace serverless compute
+          resources use the same set of stable IP CIDR blocks and optional private link to access your
+          resources.
         :param network_id: str (optional)
           The ID of the workspace's network configuration object. To use AWS PrivateLink, this field is
           required.
@@ -2590,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
@@ -2613,6 +2626,8 @@ class WorkspacesAPI:
             body["location"] = location
         if managed_services_customer_managed_key_id is not None:
             body["managed_services_customer_managed_key_id"] = managed_services_customer_managed_key_id
+        if network_connectivity_config_id is not None:
+            body["network_connectivity_config_id"] = network_connectivity_config_id
         if network_id is not None:
             body["network_id"] = network_id
         if pricing_tier is not None:
@@ -2653,6 +2668,7 @@ class WorkspacesAPI:
         gke_config: Optional[GkeConfig] = None,
         location: Optional[str] = None,
         managed_services_customer_managed_key_id: Optional[str] = None,
+        network_connectivity_config_id: Optional[str] = None,
         network_id: Optional[str] = None,
         pricing_tier: Optional[PricingTier] = None,
         private_access_settings_id: Optional[str] = None,
@@ -2673,6 +2689,7 @@ class WorkspacesAPI:
             gke_config=gke_config,
             location=location,
             managed_services_customer_managed_key_id=managed_services_customer_managed_key_id,
+            network_connectivity_config_id=network_connectivity_config_id,
             network_id=network_id,
             pricing_tier=pricing_tier,
             private_access_settings_id=private_access_settings_id,
@@ -2759,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