Fix highlighting issue due to multi-segment imports in WorkspaceClient/AccountClient (#979)

## What changes are proposed in this pull request?

This PR fixes a reported highlighting problem with the way API clients
are imported in `WorkspaceClient/AccountClient`. Specifically, it now
imports the API clients using their module as reference instead of a
multi-segment reference.

See highlighting issue:


![image](https://github.com/user-attachments/assets/98829599-8ec4-4e4f-8a36-3f526bf0f493)

This PR regenerates the SDK documentation that wasn't generated in the
previous release.

## How is this tested?

Unit and integration tests + verified locally that the objects are now
properly identified.

Author: renaudhartert-db

.codegen.json

@@ -5,9 +5,9 @@
     "databricks/sdk/version.py": "__version__ = \"$VERSION\""
   },
   "toolchain": {
-    "required": ["python3"],
+    "required": ["python3.12"],
     "pre_setup": [
-      "python3 -m venv .databricks"
+      "python3.12 -m venv .databricks"
     ],
     "prepend_path": ".databricks/bin",
     "setup": [
@@ -17,7 +17,7 @@
       "make fmt",
       "pytest -m 'not integration' --cov=databricks --cov-report html tests",
       "pip install .",
-      "python docs/gen-client-docs.py"
+      "python3.12 docs/gen-client-docs.py"
     ]
   }
 }

NEXT_CHANGELOG.md

@@ -11,6 +11,9 @@
 
 ### Bug Fixes
 
+- Fix a reported highlighting problem with the way API clients are imported in WorkspaceClient/AccountClient 
+  ([#979](https://github.com/databricks/databricks-sdk-py/pull/979)).
+
 ### Documentation
 
 ### Internal Changes

databricks/sdk/__init__.py

@@ -18,6 +18,11 @@
         :param resource: str
           The resource name for which assignable roles will be listed.
 
+          Examples | Summary :--- | :--- `resource=accounts/<ACCOUNT_ID>` | A resource name for the account.
+          `resource=accounts/<ACCOUNT_ID>/groups/<GROUP_ID>` | A resource name for the group.
+          `resource=accounts/<ACCOUNT_ID>/servicePrincipals/<SP_ID>` | A resource name for the service
+          principal.
+
         :returns: :class:`GetAssignableRolesForResourceResponse`
 
 
@@ -30,6 +35,12 @@
 
         :param name: str
           The ruleset name associated with the request.
+
+          Examples | Summary :--- | :--- `name=accounts/<ACCOUNT_ID>/ruleSets/default` | A name for a rule set
+          on the account. `name=accounts/<ACCOUNT_ID>/groups/<GROUP_ID>/ruleSets/default` | A name for a rule
+          set on the group.
+          `name=accounts/<ACCOUNT_ID>/servicePrincipals/<SERVICE_PRINCIPAL_APPLICATION_ID>/ruleSets/default` |
+          A name for a rule set on the service principal.
         :param etag: str
           Etag used for versioning. The response is at least as fresh as the eTag provided. Etag is used for
           optimistic concurrency control as a way to help prevent simultaneous updates of a rule set from
@@ -38,6 +49,10 @@
           etag from a GET rule set request, and pass it with the PUT update request to identify the rule set
           version you are updating.
 
+          Examples | Summary :--- | :--- `etag=` | An empty etag can only be used in GET to indicate no
+          freshness requirements. `etag=RENUAAABhSweA4NvVmmUYdiU717H3Tgy0UJdor3gE4a+mq/oj9NjAf8ZsQ==` | An
+          etag encoded a specific version of the rule set to get or to be updated.
+
         :returns: :class:`RuleSetResponse`
 
 

docs/account/iam/access_control.rst

@@ -47,9 +47,9 @@
             
             a = AccountClient()
             
-            workspace_id = os.environ["TEST_WORKSPACE_ID"]
+            workspace_id = os.environ["DUMMY_WORKSPACE_ID"]
             
-            all = a.workspace_assignment.list(list=workspace_id)
+            all = a.workspace_assignment.list(workspace_id=workspace_id)
 
         Get permission assignments.
 
@@ -80,9 +80,9 @@
             
             spn_id = spn.id
             
-            workspace_id = os.environ["TEST_WORKSPACE_ID"]
+            workspace_id = os.environ["DUMMY_WORKSPACE_ID"]
             
-            a.workspace_assignment.update(
+            _ = a.workspace_assignment.update(
                 workspace_id=workspace_id,
                 principal_id=spn_id,
                 permissions=[iam.WorkspacePermission.USER],

docs/account/iam/workspace_assignment.rst

@@ -16,6 +16,7 @@
 
         .. code-block::
 
+            import os
             import time
             
             from databricks.sdk import AccountClient
@@ -25,8 +26,11 @@
             
             storage = a.storage.create(
                 storage_configuration_name=f"sdk-{time.time_ns()}",
-                root_bucket_info=provisioning.RootBucketInfo(bucket_name=f"sdk-{time.time_ns()}"),
+                root_bucket_info=provisioning.RootBucketInfo(bucket_name=os.environ["TEST_ROOT_BUCKET"]),
             )
+            
+            # cleanup
+            a.storage.delete(storage_configuration_id=storage.storage_configuration_id)
 
         Create new storage configuration.
 

docs/account/provisioning/storage.rst

@@ -9,9 +9,13 @@ Manage security settings for Accounts and Workspaces
 
    ip_access_lists
    network_connectivity
+   network_policies
    settings
    csp_enablement_account
    disable_legacy_features
    enable_ip_access_lists
    esm_enablement_account
-   personal_compute
+   llm_proxy_partner_powered_account
+   llm_proxy_partner_powered_enforce
+   personal_compute
+   workspace_network_configuration

docs/account/settings/index.rst

@@ -0,0 +1,46 @@
+``a.settings.llm_proxy_partner_powered_account``: Enable Partner Powered AI Features for Account
+================================================================================================
+.. currentmodule:: databricks.sdk.service.settings
+
+.. py:class:: LlmProxyPartnerPoweredAccountAPI
+
+    Determines if partner powered models are enabled or not for a specific account
+
+    .. py:method:: get( [, etag: Optional[str]]) -> LlmProxyPartnerPoweredAccount
+
+        Get the enable partner powered AI features account setting.
+
+        Gets the enable partner powered AI features account setting.
+
+        :param etag: str (optional)
+          etag used for versioning. The response is at least as fresh as the eTag provided. This is used for
+          optimistic concurrency control as a way to help prevent simultaneous writes of a setting overwriting
+          each other. It is strongly suggested that systems make use of the etag in the read -> delete pattern
+          to perform setting deletions in order to avoid race conditions. That is, get an etag from a GET
+          request, and pass it with the DELETE request to identify the rule set version you are deleting.
+
+        :returns: :class:`LlmProxyPartnerPoweredAccount`
+        
+
+    .. py:method:: update(allow_missing: bool, setting: LlmProxyPartnerPoweredAccount, field_mask: str) -> LlmProxyPartnerPoweredAccount
+
+        Update the enable partner powered AI features account setting.
+
+        Updates the enable partner powered AI features account setting.
+
+        :param allow_missing: bool
+          This should always be set to true for Settings API. Added for AIP compliance.
+        :param setting: :class:`LlmProxyPartnerPoweredAccount`
+        :param field_mask: str
+          The field mask must be a single string, with multiple fields separated by commas (no spaces). The
+          field path is relative to the resource object, using a dot (`.`) to navigate sub-fields (e.g.,
+          `author.given_name`). Specification of elements in sequence or map fields is not allowed, as only
+          the entire collection field can be specified. Field names must exactly match the resource field
+          names.
+
+          A field mask of `*` indicates full replacement. It’s recommended to always explicitly list the
+          fields being updated and avoid using `*` wildcards, as it can lead to unintended results if the API
+          changes in the future.
+
+        :returns: :class:`LlmProxyPartnerPoweredAccount`
+        

docs/account/settings/llm_proxy_partner_powered_account.rst

@@ -0,0 +1,47 @@
+``a.settings.llm_proxy_partner_powered_enforce``: Enable Enforcement of Partner Powered AI Features
+===================================================================================================
+.. currentmodule:: databricks.sdk.service.settings
+
+.. py:class:: LlmProxyPartnerPoweredEnforceAPI
+
+    Determines if the account-level partner-powered setting value is enforced upon the workspace-level
+    partner-powered setting
+
+    .. py:method:: get( [, etag: Optional[str]]) -> LlmProxyPartnerPoweredEnforce
+
+        Get the enforcement status of partner powered AI features account setting.
+
+        Gets the enforcement status of partner powered AI features account setting.
+
+        :param etag: str (optional)
+          etag used for versioning. The response is at least as fresh as the eTag provided. This is used for
+          optimistic concurrency control as a way to help prevent simultaneous writes of a setting overwriting
+          each other. It is strongly suggested that systems make use of the etag in the read -> delete pattern
+          to perform setting deletions in order to avoid race conditions. That is, get an etag from a GET
+          request, and pass it with the DELETE request to identify the rule set version you are deleting.
+
+        :returns: :class:`LlmProxyPartnerPoweredEnforce`
+        
+
+    .. py:method:: update(allow_missing: bool, setting: LlmProxyPartnerPoweredEnforce, field_mask: str) -> LlmProxyPartnerPoweredEnforce
+
+        Update the enforcement status of partner powered AI features account setting.
+
+        Updates the enable enforcement status of partner powered AI features account setting.
+
+        :param allow_missing: bool
+          This should always be set to true for Settings API. Added for AIP compliance.
+        :param setting: :class:`LlmProxyPartnerPoweredEnforce`
+        :param field_mask: str
+          The field mask must be a single string, with multiple fields separated by commas (no spaces). The
+          field path is relative to the resource object, using a dot (`.`) to navigate sub-fields (e.g.,
+          `author.given_name`). Specification of elements in sequence or map fields is not allowed, as only
+          the entire collection field can be specified. Field names must exactly match the resource field
+          names.
+
+          A field mask of `*` indicates full replacement. It’s recommended to always explicitly list the
+          fields being updated and avoid using `*` wildcards, as it can lead to unintended results if the API
+          changes in the future.
+
+        :returns: :class:`LlmProxyPartnerPoweredEnforce`
+        

docs/account/settings/llm_proxy_partner_powered_enforce.rst

@@ -0,0 +1,73 @@
+``a.network_policies``: Network Policies
+========================================
+.. currentmodule:: databricks.sdk.service.settings
+
+.. py:class:: NetworkPoliciesAPI
+
+    These APIs manage network policies for this account. Network policies control which network destinations
+    can be accessed from the Databricks environment. Each Databricks account includes a default policy named
+    'default-policy'. 'default-policy' is associated with any workspace lacking an explicit network policy
+    assignment, and is automatically associated with each newly created workspace. 'default-policy' is
+    reserved and cannot be deleted, but it can be updated to customize the default network access rules for
+    your account.
+
+    .. py:method:: create_network_policy_rpc(network_policy: AccountNetworkPolicy) -> AccountNetworkPolicy
+
+        Create a network policy.
+
+        Creates a new network policy to manage which network destinations can be accessed from the Databricks
+        environment.
+
+        :param network_policy: :class:`AccountNetworkPolicy`
+
+        :returns: :class:`AccountNetworkPolicy`
+        
+
+    .. py:method:: delete_network_policy_rpc(network_policy_id: str)
+
+        Delete a network policy.
+
+        Deletes a network policy. Cannot be called on 'default-policy'.
+
+        :param network_policy_id: str
+          The unique identifier of the network policy to delete.
+
+
+        
+
+    .. py:method:: get_network_policy_rpc(network_policy_id: str) -> AccountNetworkPolicy
+
+        Get a network policy.
+
+        Gets a network policy.
+
+        :param network_policy_id: str
+          The unique identifier of the network policy to retrieve.
+
+        :returns: :class:`AccountNetworkPolicy`
+        
+
+    .. py:method:: list_network_policies_rpc( [, page_token: Optional[str]]) -> Iterator[AccountNetworkPolicy]
+
+        List network policies.
+
+        Gets an array of network policies.
+
+        :param page_token: str (optional)
+          Pagination token to go to next page based on previous query.
+
+        :returns: Iterator over :class:`AccountNetworkPolicy`
+        
+
+    .. py:method:: update_network_policy_rpc(network_policy_id: str, network_policy: AccountNetworkPolicy) -> AccountNetworkPolicy
+
+        Update a network policy.
+
+        Updates a network policy. This allows you to modify the configuration of a network policy.
+
+        :param network_policy_id: str
+          The unique identifier for the network policy.
+        :param network_policy: :class:`AccountNetworkPolicy`
+
+        :returns: :class:`AccountNetworkPolicy`
+