unified prof

Author: hectorcast-db

databricks/sdk/core.py

@@ -23,19 +23,11 @@ class ApiClient:
     def __init__(self, cfg: Config):
         self._cfg = cfg
 
-        # Create header factory that includes both auth and org ID headers
-        def combined_header_factory():
-            headers = cfg.authenticate()
-            # Add X-Databricks-Org-Id header for workspace clients on unified hosts
-            if cfg.workspace_id and cfg.host_type == HostType.UNIFIED:
-                headers["X-Databricks-Org-Id"] = cfg.workspace_id
-            return headers
-
         self._api_client = _BaseClient(
             debug_truncate_bytes=cfg.debug_truncate_bytes,
             retry_timeout_seconds=cfg.retry_timeout_seconds,
             user_agent_base=cfg.user_agent,
-            header_factory=combined_header_factory,
+            header_factory=cfg.authenticate,
             max_connection_pools=cfg.max_connection_pools,
             max_connections_per_pool=cfg.max_connections_per_pool,
             pool_block=True,