[Internal] Decouple oauth2 and serving (#855)

## What changes are proposed in this pull request? This PR removes serving.py's indirect dependency on oauth2.py by moving DataPlaneInfo to serving.py. This change is also reflected in the OpenAPI specification. It narrows down import scopes to prevent circular dependencies. ## How is this tested? Existing Unit tests.
databricks · Jan 10, 2025 · cbae014 · cbae014
1 parent d907c0c
commit cbae014
Show file tree

Hide file tree

Showing 10 changed files with 254 additions and 84 deletions.
diff --git a/.codegen/_openapi_sha b/.codegen/_openapi_sha
@@ -1 +1 @@
-a6a317df8327c9b1e5cb59a03a42ffa2aabeef6d
+779817ed8d63031f5ea761fbd25ee84f38feec0d
diff --git a/databricks/sdk/__init__.py b/databricks/sdk/__init__.py
diff --git a/databricks/sdk/data_plane.py b/databricks/sdk/data_plane.py
@@ -3,7 +3,6 @@
 from typing import Callable, List
 
 from databricks.sdk.oauth import Token
-from databricks.sdk.service.oauth2 import DataPlaneInfo
 
 
 @dataclass
@@ -19,6 +18,7 @@ class DataPlaneDetails:
 
 class DataPlaneService:
     """Helper class to fetch and manage DataPlane details."""
+    from .service.serving import DataPlaneInfo
 
     def __init__(self):
         self._data_plane_info = {}

diff --git a/databricks/sdk/service/apps.py b/databricks/sdk/service/apps.py
diff --git a/databricks/sdk/service/catalog.py b/databricks/sdk/service/catalog.py
diff --git a/databricks/sdk/service/jobs.py b/databricks/sdk/service/jobs.py
diff --git a/databricks/sdk/service/oauth2.py b/databricks/sdk/service/oauth2.py
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		a6a317df8327c9b1e5cb59a03a42ffa2aabeef6d
		779817ed8d63031f5ea761fbd25ee84f38feec0d