databricks-sdk 0.27.1__py3-none-any.whl → 0.28.0__py3-none-any.whl

databricks/sdk/service/marketplace.py

@@ -1908,34 +1908,6 @@ class SortBy(Enum):
     SORT_BY_UNSPECIFIED = 'SORT_BY_UNSPECIFIED'
 
 
-@dataclass
-class SortBySpec:
-    sort_by: SortBy
-    """The field on which to sort the listing."""
-
-    sort_order: SortOrder
-    """The order in which to sort the listing."""
-
-    def as_dict(self) -> dict:
-        """Serializes the SortBySpec into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        if self.sort_by is not None: body['sort_by'] = self.sort_by.value
-        if self.sort_order is not None: body['sort_order'] = self.sort_order.value
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> SortBySpec:
-        """Deserializes the SortBySpec from a dictionary."""
-        return cls(sort_by=_enum(d, 'sort_by', SortBy), sort_order=_enum(d, 'sort_order', SortOrder))
-
-
-class SortOrder(Enum):
-
-    SORT_ORDER_ASCENDING = 'SORT_ORDER_ASCENDING'
-    SORT_ORDER_DESCENDING = 'SORT_ORDER_DESCENDING'
-    SORT_ORDER_UNSPECIFIED = 'SORT_ORDER_UNSPECIFIED'
-
-
 @dataclass
 class TokenDetail:
     bearer_token: Optional[str] = None
@@ -2579,13 +2551,14 @@ class ConsumerListingsAPI:
              *,
              assets: Optional[List[AssetType]] = None,
              categories: Optional[List[Category]] = None,
+             is_ascending: Optional[bool] = None,
              is_free: Optional[bool] = None,
              is_private_exchange: Optional[bool] = None,
              is_staff_pick: Optional[bool] = None,
              page_size: Optional[int] = None,
              page_token: Optional[str] = None,
              provider_ids: Optional[List[str]] = None,
-             sort_by_spec: Optional[SortBySpec] = None,
+             sort_by: Optional[SortBy] = None,
              tags: Optional[List[ListingTag]] = None) -> Iterator[Listing]:
         """List listings.
         
@@ -2595,6 +2568,7 @@ class ConsumerListingsAPI:
           Matches any of the following asset types
         :param categories: List[:class:`Category`] (optional)
           Matches any of the following categories
+        :param is_ascending: bool (optional)
         :param is_free: bool (optional)
           Filters each listing based on if it is free.
         :param is_private_exchange: bool (optional)
@@ -2605,7 +2579,7 @@ class ConsumerListingsAPI:
         :param page_token: str (optional)
         :param provider_ids: List[str] (optional)
           Matches any of the following provider ids
-        :param sort_by_spec: :class:`SortBySpec` (optional)
+        :param sort_by: :class:`SortBy` (optional)
           Criteria for sorting the resulting set of listings.
         :param tags: List[:class:`ListingTag`] (optional)
           Matches any of the following tags
@@ -2616,13 +2590,14 @@ class ConsumerListingsAPI:
         query = {}
         if assets is not None: query['assets'] = [v.value for v in assets]
         if categories is not None: query['categories'] = [v.value for v in categories]
+        if is_ascending is not None: query['is_ascending'] = is_ascending
         if is_free is not None: query['is_free'] = is_free
         if is_private_exchange is not None: query['is_private_exchange'] = is_private_exchange
         if is_staff_pick is not None: query['is_staff_pick'] = is_staff_pick
         if page_size is not None: query['page_size'] = page_size
         if page_token is not None: query['page_token'] = page_token
         if provider_ids is not None: query['provider_ids'] = [v for v in provider_ids]
-        if sort_by_spec is not None: query['sort_by_spec'] = sort_by_spec.as_dict()
+        if sort_by is not None: query['sort_by'] = sort_by.value
         if tags is not None: query['tags'] = [v.as_dict() for v in tags]
         headers = {'Accept': 'application/json', }
 
@@ -2640,6 +2615,7 @@ class ConsumerListingsAPI:
                *,
                assets: Optional[List[AssetType]] = None,
                categories: Optional[List[Category]] = None,
+               is_ascending: Optional[bool] = None,
                is_free: Optional[bool] = None,
                is_private_exchange: Optional[bool] = None,
                page_size: Optional[int] = None,
@@ -2657,6 +2633,7 @@ class ConsumerListingsAPI:
           Matches any of the following asset types
         :param categories: List[:class:`Category`] (optional)
           Matches any of the following categories
+        :param is_ascending: bool (optional)
         :param is_free: bool (optional)
         :param is_private_exchange: bool (optional)
         :param page_size: int (optional)
@@ -2671,6 +2648,7 @@ class ConsumerListingsAPI:
         query = {}
         if assets is not None: query['assets'] = [v.value for v in assets]
         if categories is not None: query['categories'] = [v.value for v in categories]
+        if is_ascending is not None: query['is_ascending'] = is_ascending
         if is_free is not None: query['is_free'] = is_free
         if is_private_exchange is not None: query['is_private_exchange'] = is_private_exchange
         if page_size is not None: query['page_size'] = page_size

databricks/sdk/service/pipelines.py

@@ -57,6 +57,9 @@ class CreatePipeline:
     filters: Optional[Filters] = None
     """Filters on which Pipeline packages to include in the deployed graph."""
 
+    gateway_definition: Optional[IngestionGatewayPipelineDefinition] = None
+    """The definition of a gateway pipeline to support CDC."""
+
     id: Optional[str] = None
     """Unique identifier for this pipeline."""
 
@@ -104,6 +107,7 @@ class CreatePipeline:
         if self.dry_run is not None: body['dry_run'] = self.dry_run
         if self.edition is not None: body['edition'] = self.edition
         if self.filters: body['filters'] = self.filters.as_dict()
+        if self.gateway_definition: body['gateway_definition'] = self.gateway_definition.as_dict()
         if self.id is not None: body['id'] = self.id
         if self.ingestion_definition: body['ingestion_definition'] = self.ingestion_definition.as_dict()
         if self.libraries: body['libraries'] = [v.as_dict() for v in self.libraries]
@@ -130,6 +134,7 @@ class CreatePipeline:
                    dry_run=d.get('dry_run', None),
                    edition=d.get('edition', None),
                    filters=_from_dict(d, 'filters', Filters),
+                   gateway_definition=_from_dict(d, 'gateway_definition', IngestionGatewayPipelineDefinition),
                    id=d.get('id', None),
                    ingestion_definition=_from_dict(d, 'ingestion_definition',
                                                    ManagedIngestionPipelineDefinition),
@@ -266,6 +271,9 @@ class EditPipeline:
     filters: Optional[Filters] = None
     """Filters on which Pipeline packages to include in the deployed graph."""
 
+    gateway_definition: Optional[IngestionGatewayPipelineDefinition] = None
+    """The definition of a gateway pipeline to support CDC."""
+
     id: Optional[str] = None
     """Unique identifier for this pipeline."""
 
@@ -317,6 +325,7 @@ class EditPipeline:
         if self.expected_last_modified is not None:
             body['expected_last_modified'] = self.expected_last_modified
         if self.filters: body['filters'] = self.filters.as_dict()
+        if self.gateway_definition: body['gateway_definition'] = self.gateway_definition.as_dict()
         if self.id is not None: body['id'] = self.id
         if self.ingestion_definition: body['ingestion_definition'] = self.ingestion_definition.as_dict()
         if self.libraries: body['libraries'] = [v.as_dict() for v in self.libraries]
@@ -344,6 +353,7 @@ class EditPipeline:
                    edition=d.get('edition', None),
                    expected_last_modified=d.get('expected_last_modified', None),
                    filters=_from_dict(d, 'filters', Filters),
+                   gateway_definition=_from_dict(d, 'gateway_definition', IngestionGatewayPipelineDefinition),
                    id=d.get('id', None),
                    ingestion_definition=_from_dict(d, 'ingestion_definition',
                                                    ManagedIngestionPipelineDefinition),
@@ -570,6 +580,43 @@ class IngestionConfig:
         return cls(schema=_from_dict(d, 'schema', SchemaSpec), table=_from_dict(d, 'table', TableSpec))
 
 
+@dataclass
+class IngestionGatewayPipelineDefinition:
+    connection_id: Optional[str] = None
+    """Immutable. The Unity Catalog connection this gateway pipeline uses to communicate with the
+    source."""
+
+    gateway_storage_catalog: Optional[str] = None
+    """Required, Immutable. The name of the catalog for the gateway pipeline's storage location."""
+
+    gateway_storage_name: Optional[str] = None
+    """Required. The Unity Catalog-compatible naming for the gateway storage location. This is the
+    destination to use for the data that is extracted by the gateway. Delta Live Tables system will
+    automatically create the storage location under the catalog and schema."""
+
+    gateway_storage_schema: Optional[str] = None
+    """Required, Immutable. The name of the schema for the gateway pipelines's storage location."""
+
+    def as_dict(self) -> dict:
+        """Serializes the IngestionGatewayPipelineDefinition into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.connection_id is not None: body['connection_id'] = self.connection_id
+        if self.gateway_storage_catalog is not None:
+            body['gateway_storage_catalog'] = self.gateway_storage_catalog
+        if self.gateway_storage_name is not None: body['gateway_storage_name'] = self.gateway_storage_name
+        if self.gateway_storage_schema is not None:
+            body['gateway_storage_schema'] = self.gateway_storage_schema
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> IngestionGatewayPipelineDefinition:
+        """Deserializes the IngestionGatewayPipelineDefinition from a dictionary."""
+        return cls(connection_id=d.get('connection_id', None),
+                   gateway_storage_catalog=d.get('gateway_storage_catalog', None),
+                   gateway_storage_name=d.get('gateway_storage_name', None),
+                   gateway_storage_schema=d.get('gateway_storage_schema', None))
+
+
 @dataclass
 class ListPipelineEventsResponse:
     events: Optional[List[PipelineEvent]] = None
@@ -659,12 +706,17 @@ class ManagedIngestionPipelineDefinition:
     objects: Optional[List[IngestionConfig]] = None
     """Required. Settings specifying tables to replicate and the destination for the replicated tables."""
 
+    table_configuration: Optional[TableSpecificConfig] = None
+    """Configuration settings to control the ingestion of tables. These settings are applied to all
+    tables in the pipeline."""
+
     def as_dict(self) -> dict:
         """Serializes the ManagedIngestionPipelineDefinition into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.connection_name is not None: body['connection_name'] = self.connection_name
         if self.ingestion_gateway_id is not None: body['ingestion_gateway_id'] = self.ingestion_gateway_id
         if self.objects: body['objects'] = [v.as_dict() for v in self.objects]
+        if self.table_configuration: body['table_configuration'] = self.table_configuration.as_dict()
         return body
 
     @classmethod
@@ -672,7 +724,8 @@ class ManagedIngestionPipelineDefinition:
         """Deserializes the ManagedIngestionPipelineDefinition from a dictionary."""
         return cls(connection_name=d.get('connection_name', None),
                    ingestion_gateway_id=d.get('ingestion_gateway_id', None),
-                   objects=_repeated_dict(d, 'objects', IngestionConfig))
+                   objects=_repeated_dict(d, 'objects', IngestionConfig),
+                   table_configuration=_from_dict(d, 'table_configuration', TableSpecificConfig))
 
 
 @dataclass
@@ -1344,6 +1397,9 @@ class PipelineSpec:
     filters: Optional[Filters] = None
     """Filters on which Pipeline packages to include in the deployed graph."""
 
+    gateway_definition: Optional[IngestionGatewayPipelineDefinition] = None
+    """The definition of a gateway pipeline to support CDC."""
+
     id: Optional[str] = None
     """Unique identifier for this pipeline."""
 
@@ -1389,6 +1445,7 @@ class PipelineSpec:
         if self.development is not None: body['development'] = self.development
         if self.edition is not None: body['edition'] = self.edition
         if self.filters: body['filters'] = self.filters.as_dict()
+        if self.gateway_definition: body['gateway_definition'] = self.gateway_definition.as_dict()
         if self.id is not None: body['id'] = self.id
         if self.ingestion_definition: body['ingestion_definition'] = self.ingestion_definition.as_dict()
         if self.libraries: body['libraries'] = [v.as_dict() for v in self.libraries]
@@ -1413,6 +1470,7 @@ class PipelineSpec:
                    development=d.get('development', None),
                    edition=d.get('edition', None),
                    filters=_from_dict(d, 'filters', Filters),
+                   gateway_definition=_from_dict(d, 'gateway_definition', IngestionGatewayPipelineDefinition),
                    id=d.get('id', None),
                    ingestion_definition=_from_dict(d, 'ingestion_definition',
                                                    ManagedIngestionPipelineDefinition),
@@ -1523,6 +1581,11 @@ class SchemaSpec:
     source_schema: Optional[str] = None
     """Required. Schema name in the source database."""
 
+    table_configuration: Optional[TableSpecificConfig] = None
+    """Configuration settings to control the ingestion of tables. These settings are applied to all
+    tables in this schema and override the table_configuration defined in the
+    ManagedIngestionPipelineDefinition object."""
+
     def as_dict(self) -> dict:
         """Serializes the SchemaSpec into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -1530,6 +1593,7 @@ class SchemaSpec:
         if self.destination_schema is not None: body['destination_schema'] = self.destination_schema
         if self.source_catalog is not None: body['source_catalog'] = self.source_catalog
         if self.source_schema is not None: body['source_schema'] = self.source_schema
+        if self.table_configuration: body['table_configuration'] = self.table_configuration.as_dict()
         return body
 
     @classmethod
@@ -1538,7 +1602,8 @@ class SchemaSpec:
         return cls(destination_catalog=d.get('destination_catalog', None),
                    destination_schema=d.get('destination_schema', None),
                    source_catalog=d.get('source_catalog', None),
-                   source_schema=d.get('source_schema', None))
+                   source_schema=d.get('source_schema', None),
+                   table_configuration=_from_dict(d, 'table_configuration', TableSpecificConfig))
 
 
 @dataclass
@@ -1729,6 +1794,10 @@ class TableSpec:
     source_table: Optional[str] = None
     """Required. Table name in the source database."""
 
+    table_configuration: Optional[TableSpecificConfig] = None
+    """Configuration settings to control the ingestion of tables. These settings override the
+    table_configuration defined in the ManagedIngestionPipelineDefinition object and the SchemaSpec."""
+
     def as_dict(self) -> dict:
         """Serializes the TableSpec into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -1738,6 +1807,7 @@ class TableSpec:
         if self.source_catalog is not None: body['source_catalog'] = self.source_catalog
         if self.source_schema is not None: body['source_schema'] = self.source_schema
         if self.source_table is not None: body['source_table'] = self.source_table
+        if self.table_configuration: body['table_configuration'] = self.table_configuration.as_dict()
         return body
 
     @classmethod
@@ -1748,7 +1818,44 @@ class TableSpec:
                    destination_table=d.get('destination_table', None),
                    source_catalog=d.get('source_catalog', None),
                    source_schema=d.get('source_schema', None),
-                   source_table=d.get('source_table', None))
+                   source_table=d.get('source_table', None),
+                   table_configuration=_from_dict(d, 'table_configuration', TableSpecificConfig))
+
+
+@dataclass
+class TableSpecificConfig:
+    primary_keys: Optional[List[str]] = None
+    """The primary key of the table used to apply changes."""
+
+    salesforce_include_formula_fields: Optional[bool] = None
+    """If true, formula fields defined in the table are included in the ingestion. This setting is only
+    valid for the Salesforce connector"""
+
+    scd_type: Optional[TableSpecificConfigScdType] = None
+    """The SCD type to use to ingest the table."""
+
+    def as_dict(self) -> dict:
+        """Serializes the TableSpecificConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.primary_keys: body['primary_keys'] = [v for v in self.primary_keys]
+        if self.salesforce_include_formula_fields is not None:
+            body['salesforce_include_formula_fields'] = self.salesforce_include_formula_fields
+        if self.scd_type is not None: body['scd_type'] = self.scd_type.value
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> TableSpecificConfig:
+        """Deserializes the TableSpecificConfig from a dictionary."""
+        return cls(primary_keys=d.get('primary_keys', None),
+                   salesforce_include_formula_fields=d.get('salesforce_include_formula_fields', None),
+                   scd_type=_enum(d, 'scd_type', TableSpecificConfigScdType))
+
+
+class TableSpecificConfigScdType(Enum):
+    """The SCD type to use to ingest the table."""
+
+    SCD_TYPE_1 = 'SCD_TYPE_1'
+    SCD_TYPE_2 = 'SCD_TYPE_2'
 
 
 @dataclass
@@ -1981,6 +2088,7 @@ class PipelinesAPI:
                dry_run: Optional[bool] = None,
                edition: Optional[str] = None,
                filters: Optional[Filters] = None,
+               gateway_definition: Optional[IngestionGatewayPipelineDefinition] = None,
                id: Optional[str] = None,
                ingestion_definition: Optional[ManagedIngestionPipelineDefinition] = None,
                libraries: Optional[List[PipelineLibrary]] = None,
@@ -2019,6 +2127,8 @@ class PipelinesAPI:
           Pipeline product edition.
         :param filters: :class:`Filters` (optional)
           Filters on which Pipeline packages to include in the deployed graph.
+        :param gateway_definition: :class:`IngestionGatewayPipelineDefinition` (optional)
+          The definition of a gateway pipeline to support CDC.
         :param id: str (optional)
           Unique identifier for this pipeline.
         :param ingestion_definition: :class:`ManagedIngestionPipelineDefinition` (optional)
@@ -2056,6 +2166,7 @@ class PipelinesAPI:
         if dry_run is not None: body['dry_run'] = dry_run
         if edition is not None: body['edition'] = edition
         if filters is not None: body['filters'] = filters.as_dict()
+        if gateway_definition is not None: body['gateway_definition'] = gateway_definition.as_dict()
         if id is not None: body['id'] = id
         if ingestion_definition is not None: body['ingestion_definition'] = ingestion_definition.as_dict()
         if libraries is not None: body['libraries'] = [v.as_dict() for v in libraries]
@@ -2385,6 +2496,7 @@ class PipelinesAPI:
                edition: Optional[str] = None,
                expected_last_modified: Optional[int] = None,
                filters: Optional[Filters] = None,
+               gateway_definition: Optional[IngestionGatewayPipelineDefinition] = None,
                id: Optional[str] = None,
                ingestion_definition: Optional[ManagedIngestionPipelineDefinition] = None,
                libraries: Optional[List[PipelineLibrary]] = None,
@@ -2426,6 +2538,8 @@ class PipelinesAPI:
           modified after that time, then the request will fail with a conflict.
         :param filters: :class:`Filters` (optional)
           Filters on which Pipeline packages to include in the deployed graph.
+        :param gateway_definition: :class:`IngestionGatewayPipelineDefinition` (optional)
+          The definition of a gateway pipeline to support CDC.
         :param id: str (optional)
           Unique identifier for this pipeline.
         :param ingestion_definition: :class:`ManagedIngestionPipelineDefinition` (optional)
@@ -2463,6 +2577,7 @@ class PipelinesAPI:
         if edition is not None: body['edition'] = edition
         if expected_last_modified is not None: body['expected_last_modified'] = expected_last_modified
         if filters is not None: body['filters'] = filters.as_dict()
+        if gateway_definition is not None: body['gateway_definition'] = gateway_definition.as_dict()
         if id is not None: body['id'] = id
         if ingestion_definition is not None: body['ingestion_definition'] = ingestion_definition.as_dict()
         if libraries is not None: body['libraries'] = [v.as_dict() for v in libraries]

databricks/sdk/service/serving.py

@@ -8,7 +8,7 @@ import time
 from dataclasses import dataclass
 from datetime import timedelta
 from enum import Enum
-from typing import Any, Callable, Dict, Iterator, List, Optional
+from typing import Any, BinaryIO, Callable, Dict, Iterator, List, Optional
 
 from ..errors import OperationFailed
 from ._internal import Wait, _enum, _from_dict, _repeated_dict
@@ -170,6 +170,9 @@ class AppDeployment:
     creator: Optional[str] = None
     """The email of the user creates the deployment."""
 
+    deployment_artifacts: Optional[AppDeploymentArtifacts] = None
+    """The deployment artifacts for an app."""
+
     deployment_id: Optional[str] = None
     """The unique id of the deployment."""
 
@@ -184,6 +187,7 @@ class AppDeployment:
         body = {}
         if self.create_time is not None: body['create_time'] = self.create_time
         if self.creator is not None: body['creator'] = self.creator
+        if self.deployment_artifacts: body['deployment_artifacts'] = self.deployment_artifacts.as_dict()
         if self.deployment_id is not None: body['deployment_id'] = self.deployment_id
         if self.source_code_path is not None: body['source_code_path'] = self.source_code_path
         if self.status: body['status'] = self.status.as_dict()
@@ -195,18 +199,37 @@ class AppDeployment:
         """Deserializes the AppDeployment from a dictionary."""
         return cls(create_time=d.get('create_time', None),
                    creator=d.get('creator', None),
+                   deployment_artifacts=_from_dict(d, 'deployment_artifacts', AppDeploymentArtifacts),
                    deployment_id=d.get('deployment_id', None),
                    source_code_path=d.get('source_code_path', None),
                    status=_from_dict(d, 'status', AppDeploymentStatus),
                    update_time=d.get('update_time', None))
 
 
+@dataclass
+class AppDeploymentArtifacts:
+    source_code_path: Optional[str] = None
+    """The source code of the deployment."""
+
+    def as_dict(self) -> dict:
+        """Serializes the AppDeploymentArtifacts into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.source_code_path is not None: body['source_code_path'] = self.source_code_path
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> AppDeploymentArtifacts:
+        """Deserializes the AppDeploymentArtifacts from a dictionary."""
+        return cls(source_code_path=d.get('source_code_path', None))
+
+
 class AppDeploymentState(Enum):
 
     CANCELLED = 'CANCELLED'
     FAILED = 'FAILED'
     IN_PROGRESS = 'IN_PROGRESS'
     STATE_UNSPECIFIED = 'STATE_UNSPECIFIED'
+    STOPPED = 'STOPPED'
     SUCCEEDED = 'SUCCEEDED'
 
 
@@ -491,6 +514,9 @@ class CreateServingEndpoint:
     """Rate limits to be applied to the serving endpoint. NOTE: only external and foundation model
     endpoints are supported as of now."""
 
+    route_optimized: Optional[bool] = None
+    """Enable route optimization for the serving endpoint."""
+
     tags: Optional[List[EndpointTag]] = None
     """Tags to be attached to the serving endpoint and automatically propagated to billing logs."""
 
@@ -500,6 +526,7 @@ class CreateServingEndpoint:
         if self.config: body['config'] = self.config.as_dict()
         if self.name is not None: body['name'] = self.name
         if self.rate_limits: body['rate_limits'] = [v.as_dict() for v in self.rate_limits]
+        if self.route_optimized is not None: body['route_optimized'] = self.route_optimized
         if self.tags: body['tags'] = [v.as_dict() for v in self.tags]
         return body
 
@@ -509,6 +536,7 @@ class CreateServingEndpoint:
         return cls(config=_from_dict(d, 'config', EndpointCoreConfigInput),
                    name=d.get('name', None),
                    rate_limits=_repeated_dict(d, 'rate_limits', RateLimit),
+                   route_optimized=d.get('route_optimized', None),
                    tags=_repeated_dict(d, 'tags', EndpointTag))
 
 
@@ -844,16 +872,18 @@ class EnvVariable:
 
 @dataclass
 class ExportMetricsResponse:
+    contents: Optional[BinaryIO] = None
 
     def as_dict(self) -> dict:
         """Serializes the ExportMetricsResponse into a dictionary suitable for use as a JSON request body."""
         body = {}
+        if self.contents: body['contents'] = self.contents
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, any]) -> ExportMetricsResponse:
         """Deserializes the ExportMetricsResponse from a dictionary."""
-        return cls()
+        return cls(contents=d.get('contents', None))
 
 
 @dataclass
@@ -1090,14 +1120,24 @@ class ListEndpointsResponse:
 
 @dataclass
 class OpenAiConfig:
-    openai_api_key: str
-    """The Databricks secret key reference for an OpenAI or Azure OpenAI API key."""
+    microsoft_entra_client_id: Optional[str] = None
+    """This field is only required for Azure AD OpenAI and is the Microsoft Entra Client ID."""
+
+    microsoft_entra_client_secret: Optional[str] = None
+    """The Databricks secret key reference for the Microsoft Entra Client Secret that is only required
+    for Azure AD OpenAI."""
+
+    microsoft_entra_tenant_id: Optional[str] = None
+    """This field is only required for Azure AD OpenAI and is the Microsoft Entra Tenant ID."""
 
     openai_api_base: Optional[str] = None
     """This is the base URL for the OpenAI API (default: "https://api.openai.com/v1"). For Azure
     OpenAI, this field is required, and is the base URL for the Azure OpenAI API service provided by
     Azure."""
 
+    openai_api_key: Optional[str] = None
+    """The Databricks secret key reference for an OpenAI or Azure OpenAI API key."""
+
     openai_api_type: Optional[str] = None
     """This is an optional field to specify the type of OpenAI API to use. For Azure OpenAI, this field
     is required, and adjust this parameter to represent the preferred security access validation
@@ -1118,6 +1158,12 @@ class OpenAiConfig:
     def as_dict(self) -> dict:
         """Serializes the OpenAiConfig into a dictionary suitable for use as a JSON request body."""
         body = {}
+        if self.microsoft_entra_client_id is not None:
+            body['microsoft_entra_client_id'] = self.microsoft_entra_client_id
+        if self.microsoft_entra_client_secret is not None:
+            body['microsoft_entra_client_secret'] = self.microsoft_entra_client_secret
+        if self.microsoft_entra_tenant_id is not None:
+            body['microsoft_entra_tenant_id'] = self.microsoft_entra_tenant_id
         if self.openai_api_base is not None: body['openai_api_base'] = self.openai_api_base
         if self.openai_api_key is not None: body['openai_api_key'] = self.openai_api_key
         if self.openai_api_type is not None: body['openai_api_type'] = self.openai_api_type
@@ -1130,7 +1176,10 @@ class OpenAiConfig:
     @classmethod
     def from_dict(cls, d: Dict[str, any]) -> OpenAiConfig:
         """Deserializes the OpenAiConfig from a dictionary."""
-        return cls(openai_api_base=d.get('openai_api_base', None),
+        return cls(microsoft_entra_client_id=d.get('microsoft_entra_client_id', None),
+                   microsoft_entra_client_secret=d.get('microsoft_entra_client_secret', None),
+                   microsoft_entra_tenant_id=d.get('microsoft_entra_tenant_id', None),
+                   openai_api_base=d.get('openai_api_base', None),
                    openai_api_key=d.get('openai_api_key', None),
                    openai_api_type=d.get('openai_api_type', None),
                    openai_api_version=d.get('openai_api_version', None),
@@ -2123,6 +2172,9 @@ class ServingEndpointDetailed:
     creator: Optional[str] = None
     """The email of the user who created the serving endpoint."""
 
+    endpoint_url: Optional[str] = None
+    """Endpoint invocation url if route optimization is enabled for endpoint"""
+
     id: Optional[str] = None
     """System-generated ID of the endpoint. This is used to refer to the endpoint in the Permissions
     API"""
@@ -2139,6 +2191,9 @@ class ServingEndpointDetailed:
     permission_level: Optional[ServingEndpointDetailedPermissionLevel] = None
     """The permission level of the principal making the request."""
 
+    route_optimized: Optional[bool] = None
+    """Boolean representing if route optimization has been enabled for the endpoint"""
+
     state: Optional[EndpointState] = None
     """Information corresponding to the state of the serving endpoint."""
 
@@ -2154,12 +2209,14 @@ class ServingEndpointDetailed:
         if self.config: body['config'] = self.config.as_dict()
         if self.creation_timestamp is not None: body['creation_timestamp'] = self.creation_timestamp
         if self.creator is not None: body['creator'] = self.creator
+        if self.endpoint_url is not None: body['endpoint_url'] = self.endpoint_url
         if self.id is not None: body['id'] = self.id
         if self.last_updated_timestamp is not None:
             body['last_updated_timestamp'] = self.last_updated_timestamp
         if self.name is not None: body['name'] = self.name
         if self.pending_config: body['pending_config'] = self.pending_config.as_dict()
         if self.permission_level is not None: body['permission_level'] = self.permission_level.value
+        if self.route_optimized is not None: body['route_optimized'] = self.route_optimized
         if self.state: body['state'] = self.state.as_dict()
         if self.tags: body['tags'] = [v.as_dict() for v in self.tags]
         if self.task is not None: body['task'] = self.task
@@ -2171,11 +2228,13 @@ class ServingEndpointDetailed:
         return cls(config=_from_dict(d, 'config', EndpointCoreConfigOutput),
                    creation_timestamp=d.get('creation_timestamp', None),
                    creator=d.get('creator', None),
+                   endpoint_url=d.get('endpoint_url', None),
                    id=d.get('id', None),
                    last_updated_timestamp=d.get('last_updated_timestamp', None),
                    name=d.get('name', None),
                    pending_config=_from_dict(d, 'pending_config', EndpointPendingConfig),
                    permission_level=_enum(d, 'permission_level', ServingEndpointDetailedPermissionLevel),
+                   route_optimized=d.get('route_optimized', None),
                    state=_from_dict(d, 'state', EndpointState),
                    tags=_repeated_dict(d, 'tags', EndpointTag),
                    task=d.get('task', None))
@@ -2760,6 +2819,7 @@ class ServingEndpointsAPI:
                config: EndpointCoreConfigInput,
                *,
                rate_limits: Optional[List[RateLimit]] = None,
+               route_optimized: Optional[bool] = None,
                tags: Optional[List[EndpointTag]] = None) -> Wait[ServingEndpointDetailed]:
         """Create a new serving endpoint.
         
@@ -2771,6 +2831,8 @@ class ServingEndpointsAPI:
         :param rate_limits: List[:class:`RateLimit`] (optional)
           Rate limits to be applied to the serving endpoint. NOTE: only external and foundation model
           endpoints are supported as of now.
+        :param route_optimized: bool (optional)
+          Enable route optimization for the serving endpoint.
         :param tags: List[:class:`EndpointTag`] (optional)
           Tags to be attached to the serving endpoint and automatically propagated to billing logs.
         
@@ -2782,6 +2844,7 @@ class ServingEndpointsAPI:
         if config is not None: body['config'] = config.as_dict()
         if name is not None: body['name'] = name
         if rate_limits is not None: body['rate_limits'] = [v.as_dict() for v in rate_limits]
+        if route_optimized is not None: body['route_optimized'] = route_optimized
         if tags is not None: body['tags'] = [v.as_dict() for v in tags]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
 
@@ -2796,9 +2859,13 @@ class ServingEndpointsAPI:
         config: EndpointCoreConfigInput,
         *,
         rate_limits: Optional[List[RateLimit]] = None,
+        route_optimized: Optional[bool] = None,
         tags: Optional[List[EndpointTag]] = None,
         timeout=timedelta(minutes=20)) -> ServingEndpointDetailed:
-        return self.create(config=config, name=name, rate_limits=rate_limits,
+        return self.create(config=config,
+                           name=name,
+                           rate_limits=rate_limits,
+                           route_optimized=route_optimized,
                            tags=tags).result(timeout=timeout)
 
     def delete(self, name: str):
@@ -2814,7 +2881,7 @@ class ServingEndpointsAPI:
 
         self._api.do('DELETE', f'/api/2.0/serving-endpoints/{name}', headers=headers)
 
-    def export_metrics(self, name: str):
+    def export_metrics(self, name: str) -> ExportMetricsResponse:
         """Get metrics of a serving endpoint.
         
         Retrieves the metrics associated with the provided serving endpoint in either Prometheus or
@@ -2823,12 +2890,13 @@ class ServingEndpointsAPI:
         :param name: str
           The name of the serving endpoint to retrieve metrics for. This field is required.
         
-        
+        :returns: :class:`ExportMetricsResponse`
         """
 
-        headers = {}
+        headers = {'Accept': 'text/plain', }
 
-        self._api.do('GET', f'/api/2.0/serving-endpoints/{name}/metrics', headers=headers)
+        res = self._api.do('GET', f'/api/2.0/serving-endpoints/{name}/metrics', headers=headers, raw=True)
+        return ExportMetricsResponse.from_dict(res)
 
     def get(self, name: str) -> ServingEndpointDetailed:
         """Get a single serving endpoint.