databricks-sdk 0.64.0__py3-none-any.whl → 0.65.0__py3-none-any.whl

databricks/sdk/mixins/open_ai_client.py

@@ -31,7 +31,41 @@ class ServingEndpointsExt(ServingEndpointsAPI):
         http_client = httpx.Client(auth=databricks_token_auth)
         return http_client
 
-    def get_open_ai_client(self):
+    def get_open_ai_client(self, **kwargs):
+        """Create an OpenAI client configured for Databricks Model Serving.
+
+        Returns an OpenAI client instance that is pre-configured to send requests to
+        Databricks Model Serving endpoints. The client uses Databricks authentication
+        to query endpoints within the workspace associated with the current WorkspaceClient
+        instance.
+
+        Args:
+            **kwargs: Additional parameters to pass to the OpenAI client constructor.
+                Common parameters include:
+                - timeout (float): Request timeout in seconds (e.g., 30.0)
+                - max_retries (int): Maximum number of retries for failed requests (e.g., 3)
+                - default_headers (dict): Additional headers to include with requests
+                - default_query (dict): Additional query parameters to include with requests
+
+                Any parameter accepted by the OpenAI client constructor can be passed here,
+                except for the following parameters which are reserved for Databricks integration:
+                base_url, api_key, http_client
+
+        Returns:
+            OpenAI: An OpenAI client instance configured for Databricks Model Serving.
+
+        Raises:
+            ImportError: If the OpenAI library is not installed.
+            ValueError: If any reserved Databricks parameters are provided in kwargs.
+
+        Example:
+            >>> client = workspace_client.serving_endpoints.get_open_ai_client()
+            >>> # With custom timeout and retries
+            >>> client = workspace_client.serving_endpoints.get_open_ai_client(
+            ...     timeout=30.0,
+            ...     max_retries=5
+            ... )
+        """
         try:
             from openai import OpenAI
         except Exception:
@@ -39,11 +73,26 @@ class ServingEndpointsExt(ServingEndpointsAPI):
                 "Open AI is not installed. Please install the Databricks SDK with the following command `pip install databricks-sdk[openai]`"
             )
 
-        return OpenAI(
-            base_url=self._api._cfg.host + "/serving-endpoints",
-            api_key="no-token",  # Passing in a placeholder to pass validations, this will not be used
-            http_client=self._get_authorized_http_client(),
-        )
+        # Check for reserved parameters that should not be overridden
+        reserved_params = {"base_url", "api_key", "http_client"}
+        conflicting_params = reserved_params.intersection(kwargs.keys())
+        if conflicting_params:
+            raise ValueError(
+                f"Cannot override reserved Databricks parameters: {', '.join(sorted(conflicting_params))}. "
+                f"These parameters are automatically configured for Databricks Model Serving."
+            )
+
+        # Default parameters that are required for Databricks integration
+        client_params = {
+            "base_url": self._api._cfg.host + "/serving-endpoints",
+            "api_key": "no-token",  # Passing in a placeholder to pass validations, this will not be used
+            "http_client": self._get_authorized_http_client(),
+        }
+
+        # Update with any additional parameters passed by the user
+        client_params.update(kwargs)
+
+        return OpenAI(**client_params)
 
     def get_langchain_chat_open_ai_client(self, model):
         try:

databricks/sdk/service/cleanrooms.py

@@ -1509,17 +1509,18 @@ class CleanRoomAssetsAPI:
         clean_room_name: str,
         asset_type: CleanRoomAssetAssetType,
         name: str,
-        notebook_review: NotebookVersionReview,
+        *,
+        notebook_review: Optional[NotebookVersionReview] = None,
     ) -> CreateCleanRoomAssetReviewResponse:
         """Submit an asset review
 
         :param clean_room_name: str
           Name of the clean room
         :param asset_type: :class:`CleanRoomAssetAssetType`
-          Asset type. Can only be NOTEBOOK_FILE.
+          Asset type. Can either be NOTEBOOK_FILE or JAR_ANALYSIS.
         :param name: str
           Name of the asset
-        :param notebook_review: :class:`NotebookVersionReview`
+        :param notebook_review: :class:`NotebookVersionReview` (optional)
 
         :returns: :class:`CreateCleanRoomAssetReviewResponse`
         """

databricks/sdk/service/dashboards.py

@@ -1975,9 +1975,9 @@ class GenieAPI:
         space_id: str,
         conversation_id: str,
         message_id: str,
-        feedback_rating: GenieFeedbackRating,
+        rating: GenieFeedbackRating,
         *,
-        feedback_text: Optional[str] = None,
+        comment: Optional[str] = None,
     ):
         """Send feedback for a message.
 
@@ -1987,18 +1987,18 @@ class GenieAPI:
           The ID associated with the conversation.
         :param message_id: str
           The ID associated with the message to provide feedback for.
-        :param feedback_rating: :class:`GenieFeedbackRating`
+        :param rating: :class:`GenieFeedbackRating`
           The rating (POSITIVE, NEGATIVE, or NONE).
-        :param feedback_text: str (optional)
+        :param comment: str (optional)
           Optional text feedback that will be stored as a comment.
 
 
         """
         body = {}
-        if feedback_rating is not None:
-            body["feedback_rating"] = feedback_rating.value
-        if feedback_text is not None:
-            body["feedback_text"] = feedback_text
+        if comment is not None:
+            body["comment"] = comment
+        if rating is not None:
+            body["rating"] = rating.value
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/json",

databricks/sdk/service/database.py

@@ -125,6 +125,13 @@ class DatabaseInstance:
     creator: Optional[str] = None
     """The email of the creator of the instance."""
 
+    effective_enable_pg_native_login: Optional[bool] = None
+    """xref AIP-129. `enable_pg_native_login` is owned by the client, while
+    `effective_enable_pg_native_login` is owned by the server. `enable_pg_native_login` will only be
+    set in Create/Update response messages if and only if the user provides the field via the
+    request. `effective_enable_pg_native_login` on the other hand will always bet set in all
+    response messages (Create/Update/Get/List)."""
+
     effective_enable_readable_secondaries: Optional[bool] = None
     """xref AIP-129. `enable_readable_secondaries` is owned by the client, while
     `effective_enable_readable_secondaries` is owned by the server. `enable_readable_secondaries`
@@ -151,6 +158,9 @@ class DatabaseInstance:
     provides the field via the request. `effective_stopped` on the other hand will always bet set in
     all response messages (Create/Update/Get/List)."""
 
+    enable_pg_native_login: Optional[bool] = None
+    """Whether the instance has PG native password login enabled. Defaults to true."""
+
     enable_readable_secondaries: Optional[bool] = None
     """Whether to enable secondaries to serve read-only traffic. Defaults to false."""
 
@@ -197,6 +207,8 @@ class DatabaseInstance:
             body["creation_time"] = self.creation_time
         if self.creator is not None:
             body["creator"] = self.creator
+        if self.effective_enable_pg_native_login is not None:
+            body["effective_enable_pg_native_login"] = self.effective_enable_pg_native_login
         if self.effective_enable_readable_secondaries is not None:
             body["effective_enable_readable_secondaries"] = self.effective_enable_readable_secondaries
         if self.effective_node_count is not None:
@@ -205,6 +217,8 @@ class DatabaseInstance:
             body["effective_retention_window_in_days"] = self.effective_retention_window_in_days
         if self.effective_stopped is not None:
             body["effective_stopped"] = self.effective_stopped
+        if self.enable_pg_native_login is not None:
+            body["enable_pg_native_login"] = self.enable_pg_native_login
         if self.enable_readable_secondaries is not None:
             body["enable_readable_secondaries"] = self.enable_readable_secondaries
         if self.name is not None:
@@ -240,6 +254,8 @@ class DatabaseInstance:
             body["creation_time"] = self.creation_time
         if self.creator is not None:
             body["creator"] = self.creator
+        if self.effective_enable_pg_native_login is not None:
+            body["effective_enable_pg_native_login"] = self.effective_enable_pg_native_login
         if self.effective_enable_readable_secondaries is not None:
             body["effective_enable_readable_secondaries"] = self.effective_enable_readable_secondaries
         if self.effective_node_count is not None:
@@ -248,6 +264,8 @@ class DatabaseInstance:
             body["effective_retention_window_in_days"] = self.effective_retention_window_in_days
         if self.effective_stopped is not None:
             body["effective_stopped"] = self.effective_stopped
+        if self.enable_pg_native_login is not None:
+            body["enable_pg_native_login"] = self.enable_pg_native_login
         if self.enable_readable_secondaries is not None:
             body["enable_readable_secondaries"] = self.enable_readable_secondaries
         if self.name is not None:
@@ -280,10 +298,12 @@ class DatabaseInstance:
             child_instance_refs=_repeated_dict(d, "child_instance_refs", DatabaseInstanceRef),
             creation_time=d.get("creation_time", None),
             creator=d.get("creator", None),
+            effective_enable_pg_native_login=d.get("effective_enable_pg_native_login", None),
             effective_enable_readable_secondaries=d.get("effective_enable_readable_secondaries", None),
             effective_node_count=d.get("effective_node_count", None),
             effective_retention_window_in_days=d.get("effective_retention_window_in_days", None),
             effective_stopped=d.get("effective_stopped", None),
+            enable_pg_native_login=d.get("enable_pg_native_login", None),
             enable_readable_secondaries=d.get("enable_readable_secondaries", None),
             name=d.get("name", None),
             node_count=d.get("node_count", None),

databricks/sdk/service/iam.py

@@ -2193,7 +2193,9 @@ class AccountGroupsAPI:
         sort_order: Optional[ListSortOrder] = None,
         start_index: Optional[int] = None,
     ) -> Iterator[Group]:
-        """Gets all details of the groups associated with the Databricks account.
+        """Gets all details of the groups associated with the Databricks account. As of 08/22/2025, this endpoint
+        will not return members. Instead, members should be retrieved by iterating through `Get group
+        details`.
 
         :param attributes: str (optional)
           Comma-separated list of attributes to return in response.

databricks/sdk/service/jobs.py

@@ -873,11 +873,16 @@ class Continuous:
     pause_status: Optional[PauseStatus] = None
     """Indicate whether the continuous execution of the job is paused or not. Defaults to UNPAUSED."""
 
+    task_retry_mode: Optional[TaskRetryMode] = None
+    """Indicate whether the continuous job is applying task level retries or not. Defaults to NEVER."""
+
     def as_dict(self) -> dict:
         """Serializes the Continuous into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.pause_status is not None:
             body["pause_status"] = self.pause_status.value
+        if self.task_retry_mode is not None:
+            body["task_retry_mode"] = self.task_retry_mode.value
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -885,12 +890,17 @@ class Continuous:
         body = {}
         if self.pause_status is not None:
             body["pause_status"] = self.pause_status
+        if self.task_retry_mode is not None:
+            body["task_retry_mode"] = self.task_retry_mode
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> Continuous:
         """Deserializes the Continuous from a dictionary."""
-        return cls(pause_status=_enum(d, "pause_status", PauseStatus))
+        return cls(
+            pause_status=_enum(d, "pause_status", PauseStatus),
+            task_retry_mode=_enum(d, "task_retry_mode", TaskRetryMode),
+        )
 
 
 @dataclass
@@ -7891,6 +7901,16 @@ class TaskNotificationSettings:
         )
 
 
+class TaskRetryMode(Enum):
+    """task retry mode of the continuous job * NEVER: The failed task will not be retried. *
+    ON_FAILURE: Retry a failed task if at least one other task in the job is still running its first
+    attempt. When this condition is no longer met or the retry limit is reached, the job run is
+    cancelled and a new run is started."""
+
+    NEVER = "NEVER"
+    ON_FAILURE = "ON_FAILURE"
+
+
 class TerminationCodeCode(Enum):
     """The code indicates why the run was terminated. Additional codes might be introduced in future
     releases. * `SUCCESS`: The run was completed successfully. * `SUCCESS_WITH_FAILURES`: The run

databricks/sdk/service/ml.py

@@ -1210,51 +1210,6 @@ class ExperimentTag:
         return cls(key=d.get("key", None), value=d.get("value", None))
 
 
-@dataclass
-class Feature:
-    """Feature for model version."""
-
-    feature_name: Optional[str] = None
-    """Feature name"""
-
-    feature_table_id: Optional[str] = None
-    """Feature table id"""
-
-    feature_table_name: Optional[str] = None
-    """Feature table name"""
-
-    def as_dict(self) -> dict:
-        """Serializes the Feature into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        if self.feature_name is not None:
-            body["feature_name"] = self.feature_name
-        if self.feature_table_id is not None:
-            body["feature_table_id"] = self.feature_table_id
-        if self.feature_table_name is not None:
-            body["feature_table_name"] = self.feature_table_name
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the Feature into a shallow dictionary of its immediate attributes."""
-        body = {}
-        if self.feature_name is not None:
-            body["feature_name"] = self.feature_name
-        if self.feature_table_id is not None:
-            body["feature_table_id"] = self.feature_table_id
-        if self.feature_table_name is not None:
-            body["feature_table_name"] = self.feature_table_name
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> Feature:
-        """Deserializes the Feature from a dictionary."""
-        return cls(
-            feature_name=d.get("feature_name", None),
-            feature_table_id=d.get("feature_table_id", None),
-            feature_table_name=d.get("feature_table_name", None),
-        )
-
-
 @dataclass
 class FeatureLineage:
     feature_specs: Optional[List[FeatureLineageFeatureSpec]] = None
@@ -1391,7 +1346,7 @@ class FeatureLineageOnlineFeature:
 class FeatureList:
     """Feature list wrap all the features for a model version"""
 
-    features: Optional[List[Feature]] = None
+    features: Optional[List[LinkedFeature]] = None
 
     def as_dict(self) -> dict:
         """Serializes the FeatureList into a dictionary suitable for use as a JSON request body."""
@@ -1410,7 +1365,7 @@ class FeatureList:
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> FeatureList:
         """Deserializes the FeatureList from a dictionary."""
-        return cls(features=_repeated_dict(d, "features", Feature))
+        return cls(features=_repeated_dict(d, "features", LinkedFeature))
 
 
 @dataclass
@@ -2054,6 +2009,51 @@ class JobSpecWithoutSecret:
         return cls(job_id=d.get("job_id", None), workspace_url=d.get("workspace_url", None))
 
 
+@dataclass
+class LinkedFeature:
+    """Feature for model version. ([ML-57150] Renamed from Feature to LinkedFeature)"""
+
+    feature_name: Optional[str] = None
+    """Feature name"""
+
+    feature_table_id: Optional[str] = None
+    """Feature table id"""
+
+    feature_table_name: Optional[str] = None
+    """Feature table name"""
+
+    def as_dict(self) -> dict:
+        """Serializes the LinkedFeature into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.feature_name is not None:
+            body["feature_name"] = self.feature_name
+        if self.feature_table_id is not None:
+            body["feature_table_id"] = self.feature_table_id
+        if self.feature_table_name is not None:
+            body["feature_table_name"] = self.feature_table_name
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the LinkedFeature into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.feature_name is not None:
+            body["feature_name"] = self.feature_name
+        if self.feature_table_id is not None:
+            body["feature_table_id"] = self.feature_table_id
+        if self.feature_table_name is not None:
+            body["feature_table_name"] = self.feature_table_name
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> LinkedFeature:
+        """Deserializes the LinkedFeature from a dictionary."""
+        return cls(
+            feature_name=d.get("feature_name", None),
+            feature_table_id=d.get("feature_table_id", None),
+            feature_table_name=d.get("feature_table_name", None),
+        )
+
+
 @dataclass
 class ListArtifactsResponse:
     files: Optional[List[FileInfo]] = None

databricks/sdk/service/pipelines.py

@@ -613,6 +613,9 @@ class IngestionPipelineDefinition:
     objects: Optional[List[IngestionConfig]] = None
     """Required. Settings specifying tables to replicate and the destination for the replicated tables."""
 
+    source_configurations: Optional[List[SourceConfig]] = None
+    """Top-level source configurations"""
+
     source_type: Optional[IngestionSourceType] = None
     """The type of the foreign source. The source type will be inferred from the source connection or
     ingestion gateway. This field is output only and will be ignored if provided."""
@@ -630,6 +633,8 @@ class IngestionPipelineDefinition:
             body["ingestion_gateway_id"] = self.ingestion_gateway_id
         if self.objects:
             body["objects"] = [v.as_dict() for v in self.objects]
+        if self.source_configurations:
+            body["source_configurations"] = [v.as_dict() for v in self.source_configurations]
         if self.source_type is not None:
             body["source_type"] = self.source_type.value
         if self.table_configuration:
@@ -645,6 +650,8 @@ class IngestionPipelineDefinition:
             body["ingestion_gateway_id"] = self.ingestion_gateway_id
         if self.objects:
             body["objects"] = self.objects
+        if self.source_configurations:
+            body["source_configurations"] = self.source_configurations
         if self.source_type is not None:
             body["source_type"] = self.source_type
         if self.table_configuration:
@@ -658,6 +665,7 @@ class IngestionPipelineDefinition:
             connection_name=d.get("connection_name", None),
             ingestion_gateway_id=d.get("ingestion_gateway_id", None),
             objects=_repeated_dict(d, "objects", IngestionConfig),
+            source_configurations=_repeated_dict(d, "source_configurations", SourceConfig),
             source_type=_enum(d, "source_type", IngestionSourceType),
             table_configuration=_from_dict(d, "table_configuration", TableSpecificConfig),
         )
@@ -2239,6 +2247,67 @@ class PipelinesEnvironment:
         return cls(dependencies=d.get("dependencies", None))
 
 
+@dataclass
+class PostgresCatalogConfig:
+    """PG-specific catalog-level configuration parameters"""
+
+    slot_config: Optional[PostgresSlotConfig] = None
+    """Optional. The Postgres slot configuration to use for logical replication"""
+
+    def as_dict(self) -> dict:
+        """Serializes the PostgresCatalogConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.slot_config:
+            body["slot_config"] = self.slot_config.as_dict()
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the PostgresCatalogConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.slot_config:
+            body["slot_config"] = self.slot_config
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> PostgresCatalogConfig:
+        """Deserializes the PostgresCatalogConfig from a dictionary."""
+        return cls(slot_config=_from_dict(d, "slot_config", PostgresSlotConfig))
+
+
+@dataclass
+class PostgresSlotConfig:
+    """PostgresSlotConfig contains the configuration for a Postgres logical replication slot"""
+
+    publication_name: Optional[str] = None
+    """The name of the publication to use for the Postgres source"""
+
+    slot_name: Optional[str] = None
+    """The name of the logical replication slot to use for the Postgres source"""
+
+    def as_dict(self) -> dict:
+        """Serializes the PostgresSlotConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.publication_name is not None:
+            body["publication_name"] = self.publication_name
+        if self.slot_name is not None:
+            body["slot_name"] = self.slot_name
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the PostgresSlotConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.publication_name is not None:
+            body["publication_name"] = self.publication_name
+        if self.slot_name is not None:
+            body["slot_name"] = self.slot_name
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> PostgresSlotConfig:
+        """Deserializes the PostgresSlotConfig from a dictionary."""
+        return cls(publication_name=d.get("publication_name", None), slot_name=d.get("slot_name", None))
+
+
 @dataclass
 class ReportSpec:
     source_url: str
@@ -2527,6 +2596,67 @@ class SerializedException:
         )
 
 
+@dataclass
+class SourceCatalogConfig:
+    """SourceCatalogConfig contains catalog-level custom configuration parameters for each source"""
+
+    postgres: Optional[PostgresCatalogConfig] = None
+    """Postgres-specific catalog-level configuration parameters"""
+
+    source_catalog: Optional[str] = None
+    """Source catalog name"""
+
+    def as_dict(self) -> dict:
+        """Serializes the SourceCatalogConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.postgres:
+            body["postgres"] = self.postgres.as_dict()
+        if self.source_catalog is not None:
+            body["source_catalog"] = self.source_catalog
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the SourceCatalogConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.postgres:
+            body["postgres"] = self.postgres
+        if self.source_catalog is not None:
+            body["source_catalog"] = self.source_catalog
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> SourceCatalogConfig:
+        """Deserializes the SourceCatalogConfig from a dictionary."""
+        return cls(
+            postgres=_from_dict(d, "postgres", PostgresCatalogConfig), source_catalog=d.get("source_catalog", None)
+        )
+
+
+@dataclass
+class SourceConfig:
+    catalog: Optional[SourceCatalogConfig] = None
+    """Catalog-level source configuration parameters"""
+
+    def as_dict(self) -> dict:
+        """Serializes the SourceConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.catalog:
+            body["catalog"] = self.catalog.as_dict()
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the SourceConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.catalog:
+            body["catalog"] = self.catalog
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> SourceConfig:
+        """Deserializes the SourceConfig from a dictionary."""
+        return cls(catalog=_from_dict(d, "catalog", SourceCatalogConfig))
+
+
 @dataclass
 class StackFrame:
     declaring_class: Optional[str] = None

databricks/sdk/service/settings.py

@@ -3587,8 +3587,32 @@ class LlmProxyPartnerPoweredWorkspace:
 
 @dataclass
 class MicrosoftTeamsConfig:
+    app_id: Optional[str] = None
+    """[Input-Only] App ID for Microsoft Teams App."""
+
+    app_id_set: Optional[bool] = None
+    """[Output-Only] Whether App ID is set."""
+
+    auth_secret: Optional[str] = None
+    """[Input-Only] Secret for Microsoft Teams App authentication."""
+
+    auth_secret_set: Optional[bool] = None
+    """[Output-Only] Whether secret is set."""
+
+    channel_url: Optional[str] = None
+    """[Input-Only] Channel URL for Microsoft Teams App."""
+
+    channel_url_set: Optional[bool] = None
+    """[Output-Only] Whether Channel URL is set."""
+
+    tenant_id: Optional[str] = None
+    """[Input-Only] Tenant ID for Microsoft Teams App."""
+
+    tenant_id_set: Optional[bool] = None
+    """[Output-Only] Whether Tenant ID is set."""
+
     url: Optional[str] = None
-    """[Input-Only] URL for Microsoft Teams."""
+    """[Input-Only] URL for Microsoft Teams webhook."""
 
     url_set: Optional[bool] = None
     """[Output-Only] Whether URL is set."""
@@ -3596,6 +3620,22 @@ class MicrosoftTeamsConfig:
     def as_dict(self) -> dict:
         """Serializes the MicrosoftTeamsConfig into a dictionary suitable for use as a JSON request body."""
         body = {}
+        if self.app_id is not None:
+            body["app_id"] = self.app_id
+        if self.app_id_set is not None:
+            body["app_id_set"] = self.app_id_set
+        if self.auth_secret is not None:
+            body["auth_secret"] = self.auth_secret
+        if self.auth_secret_set is not None:
+            body["auth_secret_set"] = self.auth_secret_set
+        if self.channel_url is not None:
+            body["channel_url"] = self.channel_url
+        if self.channel_url_set is not None:
+            body["channel_url_set"] = self.channel_url_set
+        if self.tenant_id is not None:
+            body["tenant_id"] = self.tenant_id
+        if self.tenant_id_set is not None:
+            body["tenant_id_set"] = self.tenant_id_set
         if self.url is not None:
             body["url"] = self.url
         if self.url_set is not None:
@@ -3605,6 +3645,22 @@ class MicrosoftTeamsConfig:
     def as_shallow_dict(self) -> dict:
         """Serializes the MicrosoftTeamsConfig into a shallow dictionary of its immediate attributes."""
         body = {}
+        if self.app_id is not None:
+            body["app_id"] = self.app_id
+        if self.app_id_set is not None:
+            body["app_id_set"] = self.app_id_set
+        if self.auth_secret is not None:
+            body["auth_secret"] = self.auth_secret
+        if self.auth_secret_set is not None:
+            body["auth_secret_set"] = self.auth_secret_set
+        if self.channel_url is not None:
+            body["channel_url"] = self.channel_url
+        if self.channel_url_set is not None:
+            body["channel_url_set"] = self.channel_url_set
+        if self.tenant_id is not None:
+            body["tenant_id"] = self.tenant_id
+        if self.tenant_id_set is not None:
+            body["tenant_id_set"] = self.tenant_id_set
         if self.url is not None:
             body["url"] = self.url
         if self.url_set is not None:
@@ -3614,7 +3670,18 @@ class MicrosoftTeamsConfig:
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> MicrosoftTeamsConfig:
         """Deserializes the MicrosoftTeamsConfig from a dictionary."""
-        return cls(url=d.get("url", None), url_set=d.get("url_set", None))
+        return cls(
+            app_id=d.get("app_id", None),
+            app_id_set=d.get("app_id_set", None),
+            auth_secret=d.get("auth_secret", None),
+            auth_secret_set=d.get("auth_secret_set", None),
+            channel_url=d.get("channel_url", None),
+            channel_url_set=d.get("channel_url_set", None),
+            tenant_id=d.get("tenant_id", None),
+            tenant_id_set=d.get("tenant_id_set", None),
+            url=d.get("url", None),
+            url_set=d.get("url_set", None),
+        )
 
 
 @dataclass

databricks/sdk/service/vectorsearch.py

@@ -872,6 +872,60 @@ class QueryVectorIndexResponse:
         )
 
 
+@dataclass
+class RerankerConfig:
+    model: Optional[str] = None
+
+    parameters: Optional[RerankerConfigRerankerParameters] = None
+
+    def as_dict(self) -> dict:
+        """Serializes the RerankerConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.model is not None:
+            body["model"] = self.model
+        if self.parameters:
+            body["parameters"] = self.parameters.as_dict()
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the RerankerConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.model is not None:
+            body["model"] = self.model
+        if self.parameters:
+            body["parameters"] = self.parameters
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> RerankerConfig:
+        """Deserializes the RerankerConfig from a dictionary."""
+        return cls(model=d.get("model", None), parameters=_from_dict(d, "parameters", RerankerConfigRerankerParameters))
+
+
+@dataclass
+class RerankerConfigRerankerParameters:
+    columns_to_rerank: Optional[List[str]] = None
+
+    def as_dict(self) -> dict:
+        """Serializes the RerankerConfigRerankerParameters into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.columns_to_rerank:
+            body["columns_to_rerank"] = [v for v in self.columns_to_rerank]
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the RerankerConfigRerankerParameters into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.columns_to_rerank:
+            body["columns_to_rerank"] = self.columns_to_rerank
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> RerankerConfigRerankerParameters:
+        """Deserializes the RerankerConfigRerankerParameters from a dictionary."""
+        return cls(columns_to_rerank=d.get("columns_to_rerank", None))
+
+
 @dataclass
 class ResultData:
     """Data returned in the query result."""
@@ -1468,7 +1522,8 @@ class VectorSearchEndpointsAPI:
         :param endpoint_name: str
           Name of the vector search endpoint
         :param budget_policy_id: str
-          The budget policy id to be applied
+          The budget policy id to be applied (hima-sheth) TODO: remove this once we've migrated to usage
+          policies
 
         :returns: :class:`PatchEndpointBudgetPolicyResponse`
         """
@@ -1608,20 +1663,27 @@ class VectorSearchIndexesAPI:
 
         self._api.do("DELETE", f"/api/2.0/vector-search/indexes/{index_name}", headers=headers)
 
-    def get_index(self, index_name: str) -> VectorIndex:
+    def get_index(self, index_name: str, *, ensure_reranker_compatible: Optional[bool] = None) -> VectorIndex:
         """Get an index.
 
         :param index_name: str
           Name of the index
+        :param ensure_reranker_compatible: bool (optional)
+          If true, the URL returned for the index is guaranteed to be compatible with the reranker. Currently
+          this means we return the CP URL regardless of how the index is being accessed. If not set or set to
+          false, the URL may still be compatible with the reranker depending on what URL we return.
 
         :returns: :class:`VectorIndex`
         """
 
+        query = {}
+        if ensure_reranker_compatible is not None:
+            query["ensure_reranker_compatible"] = ensure_reranker_compatible
         headers = {
             "Accept": "application/json",
         }
 
-        res = self._api.do("GET", f"/api/2.0/vector-search/indexes/{index_name}", headers=headers)
+        res = self._api.do("GET", f"/api/2.0/vector-search/indexes/{index_name}", query=query, headers=headers)
         return VectorIndex.from_dict(res)
 
     def list_indexes(self, endpoint_name: str, *, page_token: Optional[str] = None) -> Iterator[MiniVectorIndex]:
@@ -1664,6 +1726,7 @@ class VectorSearchIndexesAPI:
         query_text: Optional[str] = None,
         query_type: Optional[str] = None,
         query_vector: Optional[List[float]] = None,
+        reranker: Optional[RerankerConfig] = None,
         score_threshold: Optional[float] = None,
     ) -> QueryVectorIndexResponse:
         """Query the specified vector index.
@@ -1691,6 +1754,7 @@ class VectorSearchIndexesAPI:
         :param query_vector: List[float] (optional)
           Query vector. Required for Direct Vector Access Index and Delta Sync Index using self-managed
           vectors.
+        :param reranker: :class:`RerankerConfig` (optional)
         :param score_threshold: float (optional)
           Threshold for the approximate nearest neighbor search. Defaults to 0.0.
 
@@ -1711,6 +1775,8 @@ class VectorSearchIndexesAPI:
             body["query_type"] = query_type
         if query_vector is not None:
             body["query_vector"] = [v for v in query_vector]
+        if reranker is not None:
+            body["reranker"] = reranker.as_dict()
         if score_threshold is not None:
             body["score_threshold"] = score_threshold
         headers = {

databricks/sdk/version.py

@@ -1 +1 @@
-__version__ = "0.64.0"
+__version__ = "0.65.0"

{databricks_sdk-0.64.0.dist-info → databricks_sdk-0.65.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-sdk
-Version: 0.64.0
+Version: 0.65.0
 Summary: Databricks SDK for Python (Beta)
 Project-URL: Documentation, https://databricks-sdk-py.readthedocs.io
 Keywords: databricks,sdk
@@ -26,7 +26,7 @@ Requires-Dist: google-auth~=2.0
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
-Requires-Dist: pytest-xdist; extra == "dev"
+Requires-Dist: pytest-xdist<4.0,>=3.6.1; extra == "dev"
 Requires-Dist: pytest-mock; extra == "dev"
 Requires-Dist: black; extra == "dev"
 Requires-Dist: pycodestyle; extra == "dev"

{databricks_sdk-0.64.0.dist-info → databricks_sdk-0.65.0.dist-info}/RECORD

@@ -17,7 +17,7 @@ databricks/sdk/oidc_token_supplier.py,sha256=QrO6J0QY4yFfcdQDL5h2OfxMxvBZJPtPmPe
 databricks/sdk/py.typed,sha256=pSvaHpbY1UPNEXyVFUjlgBhjPFZMmVC_UNrPC7eMOHI,74
 databricks/sdk/retries.py,sha256=7k2kEexGqGKXHNAWHbPFSZSugU8UIU0qtyly_hix22Q,2581
 databricks/sdk/useragent.py,sha256=boEgzTv-Zmo6boipZKjSopNy0CXg4GShC1_lTKpJgqs,7361
-databricks/sdk/version.py,sha256=4aEOIqCiMfp1lfhnZ0isYinRITBjjNflprA4nXib3RQ,23
+databricks/sdk/version.py,sha256=eauVKKmGRrPjcW6OLz78zXL8jRoYaQPOL0tbet7E8Vo,23
 databricks/sdk/_widgets/__init__.py,sha256=VhI-VvLlr3rKUT1nbROslHJIbmZX_tPJ9rRhrdFsYUA,2811
 databricks/sdk/_widgets/default_widgets_utils.py,sha256=_hwCbptLbRzWEmknco0H1wQNAYcuy2pjFO9NiRbvFeo,1127
 databricks/sdk/_widgets/ipywidgets_utils.py,sha256=mg3rEPG9z76e0yVjGgcLybUvd_zSuN5ziGeKiZ-c8Ew,2927
@@ -38,7 +38,7 @@ databricks/sdk/mixins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 databricks/sdk/mixins/compute.py,sha256=76Fhc7cDQfOf2IHkPtHZpAnxNfrSLMKl9dbQ6KswXaM,11066
 databricks/sdk/mixins/files.py,sha256=Y3IvOnB9Yogf6Ndr2uJ-HLGm57sHACoic4N3MXXmx38,57605
 databricks/sdk/mixins/jobs.py,sha256=4ywi0dZ8mEN8KZWLmZBFfdbejTP6JATvf9wCCRkdJBw,11558
-databricks/sdk/mixins/open_ai_client.py,sha256=FxMatC648dY6_PU-ClO172Iowivt4LzLgem00tNPPvQ,5552
+databricks/sdk/mixins/open_ai_client.py,sha256=Tur77AvlUJd-gDSfLb1mtMJhWuTwp1ufon9-7HGGOnQ,7969
 databricks/sdk/mixins/workspace.py,sha256=sgahprJIPLAxTvikHd9Wq2ifBW1Mcc5qz9u6EB-qm7w,4958
 databricks/sdk/runtime/__init__.py,sha256=6nthZxeYY1HjHieQcP7kXVLIId7w2yfHpZRXXtDLDAc,7333
 databricks/sdk/runtime/dbutils_stub.py,sha256=S_pgWyGmwp3Ay-pMDEXccYsPwNVqCtz7MpD3fZVlHUA,11408
@@ -48,30 +48,30 @@ databricks/sdk/service/agentbricks.py,sha256=cGl_d1MEFG2XY48WN9veCJ4UgVMadC9KOsD
 databricks/sdk/service/apps.py,sha256=RXuwG12MU62W9B855dP9N99R47hAZkNJlXcdmCNe5Jo,77562
 databricks/sdk/service/billing.py,sha256=Y1tuA7x-wl720TCA98merqUqwrhA4KGd92oWCv679ps,90880
 databricks/sdk/service/catalog.py,sha256=JjvVZdQ8my2JnYMrTgK55ixb8Bs6I2C0pALB0Kx_S0Y,657446
-databricks/sdk/service/cleanrooms.py,sha256=QpmlhVjy-DMWa7ggAK8n0Zspz317i2N1sPXb5RLprpA,81091
+databricks/sdk/service/cleanrooms.py,sha256=EOsSMl103re7FJPGGvyrnAo-HLS1RUwL09yM7PyQaso,81148
 databricks/sdk/service/compute.py,sha256=ZRCPSF2CkJaT5RNWb1YWSYVj8_EjQft13xPj0vDFQpc,468481
-databricks/sdk/service/dashboards.py,sha256=uz604lHCvx-YMPPhrUFAqiBD2nycAriruCjofZJnYb0,98027
-databricks/sdk/service/database.py,sha256=Xcfuc7sdgoZRA0PwfF3aS73_t_FuxLMkTjGdYaH808c,85977
+databricks/sdk/service/dashboards.py,sha256=_sgmqKwSVShGJiKzD5eBrU6GkMDvF0OlrwIFTFCLs2I,97952
+databricks/sdk/service/database.py,sha256=UAdYY0n0LADLRZW7bzB4EcAp8SgbOOgv8_8CGbEif8U,87319
 databricks/sdk/service/files.py,sha256=k28gM20L9bw_RmKcIm8IcNFfHVtENCbakptM3p8bgbw,38403
-databricks/sdk/service/iam.py,sha256=cSKocvBtKqC9IARCBpYpCfUZAnmGdRis2vVdzx6X_WQ,165768
-databricks/sdk/service/jobs.py,sha256=3S5dZAX2bmw1BCNY1Ks-RTAHSWLc7Lafan0b8IKnI0g,431714
+databricks/sdk/service/iam.py,sha256=seqnHS8P840m6bnx4Bp9ftArBQHb_e2IlI7dWE_FW7Y,165920
+databricks/sdk/service/jobs.py,sha256=SkChfnKaGsV5kg1S7ZM8U313bApFqr0w0RscX92uSrw,432593
 databricks/sdk/service/marketplace.py,sha256=8MpP8Y65R6bGyvuWW4ZU6b-6__a4WLZVcDScLh0to4g,153028
-databricks/sdk/service/ml.py,sha256=y-I4JLzCGNZu3nW_PHlTDBbcTKCFIcnwCTHaRZr12fQ,303773
+databricks/sdk/service/ml.py,sha256=7QyoeyT_FRNhwA3zD8faUlnOL6aKgkTo4tLg3oGw2EA,303866
 databricks/sdk/service/oauth2.py,sha256=6yoa5zmpJ68lCIIFyqcrM3fiSaWvPtf7Pl1dNhXL9pU,75330
-databricks/sdk/service/pipelines.py,sha256=CblPuyVF6z5cLEbxuU0JpYLPDtGavvUN8vyiaAYGOPA,154614
+databricks/sdk/service/pipelines.py,sha256=4eMigB0xIVKX7nPZCHUbp8ymq3sOV3qqmVqJwoJHP84,159629
 databricks/sdk/service/provisioning.py,sha256=zZm_4lDO_mhDAOcaNoMDDz4Die_MXoapNa4NuwLJYww,129980
 databricks/sdk/service/qualitymonitorv2.py,sha256=82IUD7oTDNPwMcIDE_v59-nr2I3gpL65Ih7UfB5eInY,9202
 databricks/sdk/service/serving.py,sha256=xDddrGyr7836whZCf2Cfr7dmWHye1-r7EZ3YorEgWlw,214037
-databricks/sdk/service/settings.py,sha256=Dqv0yPC6kcMrxvjx6RzaRonpkhkdUgGsJHOGLCqhrts,372788
+databricks/sdk/service/settings.py,sha256=032ZhPMCCEA6fMnEdWWIg6qcsdC5Oc7pFBjvhvzT0ns,375492
 databricks/sdk/service/settingsv2.py,sha256=l6uKjlMwXb8z4GX37cQceDvgug6hx1sjZTQVr3TTAzg,39673
 databricks/sdk/service/sharing.py,sha256=7B4wnHx4bu6PqOCST2Q8GUcZH7yVInj0L645Z5RfGE8,142394
 databricks/sdk/service/sql.py,sha256=C2OhRU9_z8K7Jr4po86rxGYAgMVEbyHp4IhlC4HNbQg,387360
 databricks/sdk/service/tags.py,sha256=sx1sMAfBE9Y5yu2mBSWXZKW3l_NTA1ujPQ3oHEljc_E,7817
-databricks/sdk/service/vectorsearch.py,sha256=PWbZ9XIC5VocdpVpVFJP8TAtSJkYhYQOAYs-JE0STBo,69716
+databricks/sdk/service/vectorsearch.py,sha256=y2GK_kgArrgPwf1gXtHfuIt3nn5GD9GapSmQTemdhYE,72592
 databricks/sdk/service/workspace.py,sha256=iss6wuYvMDSMrgwks0FuRRBeJSZFmWNOCkPIMJAzMgY,111868
-databricks_sdk-0.64.0.dist-info/licenses/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
-databricks_sdk-0.64.0.dist-info/licenses/NOTICE,sha256=tkRcQYA1k68wDLcnOWbg2xJDsUOJw8G8DGBhb8dnI3w,1588
-databricks_sdk-0.64.0.dist-info/METADATA,sha256=ZcqjNGQKK9zVQnftUJU7axAeDYPVgmfKTd1AkHR72KQ,39397
-databricks_sdk-0.64.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-databricks_sdk-0.64.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
-databricks_sdk-0.64.0.dist-info/RECORD,,
+databricks_sdk-0.65.0.dist-info/licenses/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
+databricks_sdk-0.65.0.dist-info/licenses/NOTICE,sha256=tkRcQYA1k68wDLcnOWbg2xJDsUOJw8G8DGBhb8dnI3w,1588
+databricks_sdk-0.65.0.dist-info/METADATA,sha256=MMtxLGRv6dxA70T1shr1a-5CYOvuOlt8EQFaWUcketM,39409
+databricks_sdk-0.65.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+databricks_sdk-0.65.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
+databricks_sdk-0.65.0.dist-info/RECORD,,