Merge branch 'main' into main

Author: jijosg

.codegen/_openapi_sha

@@ -1 +1 @@
-3ae6f76120079424c8654263eafbc30ec0551854
+6701524136501ef070774942ef5d6e01cfaafb88

.gitattributes

@@ -20,7 +20,9 @@ databricks/sdk/service/provisioning.py linguist-generated=true
 databricks/sdk/service/qualitymonitorv2.py linguist-generated=true
 databricks/sdk/service/serving.py linguist-generated=true
 databricks/sdk/service/settings.py linguist-generated=true
+databricks/sdk/service/settingsv2.py linguist-generated=true
 databricks/sdk/service/sharing.py linguist-generated=true
 databricks/sdk/service/sql.py linguist-generated=true
+databricks/sdk/service/tags.py linguist-generated=true
 databricks/sdk/service/vectorsearch.py linguist-generated=true
 databricks/sdk/service/workspace.py linguist-generated=true

.release_metadata.json

@@ -1,3 +1,3 @@
 {
-    "timestamp": "2025-08-13 08:59:01+0000"
+    "timestamp": "2025-08-20 11:46:15+0000"
 }

CHANGELOG.md

@@ -1,5 +1,30 @@
 # Version changelog
 
+## Release v0.64.0
+
+### API Changes
+* Added `databricks.sdk.service.settingsv2` and `databricks.sdk.service.tags` packages.
+* Added [w.apps_settings](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/apps/apps_settings.html) workspace-level service.
+* Added [w.entity_tag_assignments](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/catalog/entity_tag_assignments.html) workspace-level service and [w.rfa](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/catalog/rfa.html) workspace-level service.
+* Added [a.account_settings_v2](https://databricks-sdk-py.readthedocs.io/en/latest/account/settingsv2/account_settings_v2.html) account-level service and [w.workspace_settings_v2](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/settingsv2/workspace_settings_v2.html) workspace-level service.
+* Added [w.tag_policies](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/tags/tag_policies.html) workspace-level service.
+* Added `delete_conversation_message()`, `list_conversation_messages()` and `send_message_feedback()` methods for [w.genie](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/dashboards/genie.html) workspace-level service.
+* Added `include_all` field for `databricks.sdk.service.dashboards.GenieListConversationsRequest`.
+* Added `effective_usage_policy_id` field for `databricks.sdk.service.jobs.BaseJob`.
+* Added `effective_usage_policy_id` field for `databricks.sdk.service.jobs.BaseRun`.
+* Added `effective_usage_policy_id` field for `databricks.sdk.service.jobs.Job`.
+* Added `effective_usage_policy_id` field for `databricks.sdk.service.jobs.Run`.
+* Added `tokens` field for `databricks.sdk.service.serving.AiGatewayRateLimit`.
+* Added `usage_policy_id` field for `databricks.sdk.service.serving.ServingEndpoint`.
+* Added `effective_run_as` and `run_as` fields for `databricks.sdk.service.sql.AlertV2`.
+* Added `cache_query_id` field for `databricks.sdk.service.sql.QueryInfo`.
+* Added `model_endpoint_name_for_query` field for `databricks.sdk.service.vectorsearch.EmbeddingSourceColumn`.
+* [Breaking] Removed `environment_settings` field for `databricks.sdk.service.catalog.ConnectionInfo`.
+* [Breaking] Removed `environment_settings` field for `databricks.sdk.service.catalog.CreateConnection`.
+* [Breaking] Removed `environment_settings` field for `databricks.sdk.service.catalog.UpdateConnection`.
+* [Breaking] Removed `comment`, `display_name` and `tags` fields for `databricks.sdk.service.sharing.Share`.
+
+
 ## Release v0.63.0
 
 ### API Changes

NEXT_CHANGELOG.md

@@ -1,9 +1,11 @@
 # NEXT CHANGELOG
 
-## Release v0.64.0
+## Release v0.65.0
 
 ### New Features and Improvements
 
+* Added support for passing additional kwargs to `WorkspaceClient().serving_endpoints.get_open_ai_client()` ([#1025](https://github.com/databricks/databricks-sdk-py/pull/1025)). Users can now pass standard OpenAI client parameters like `timeout` and `max_retries` when creating an OpenAI client for Databricks Model Serving.
+
 ### Bug Fixes
 
 ### Documentation

databricks/sdk/__init__.py

@@ -31,19 +31,68 @@ def auth_flow(self, request: httpx.Request) -> httpx.Request:
         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:
             raise ImportError(
                 "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: