Skip to content

Commit 67f040f

Browse files
2003LKLauren Kam
andauthored
2026-01-01 API Upgrade (#9584)
* Editing tests so they run in personal subscription * Updating API version * Updating HISTORY.rst * Updating release version * Addressing comments * Setting default SKU value to be premium * Default sku value only set for serverless * test results * Passing all tests with aaz * Setting version --------- Co-authored-by: Lauren Kam <laurenkam@microsoft.com>
1 parent 83ea7bf commit 67f040f

44 files changed

Lines changed: 7569 additions & 6226 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

src/databricks/HISTORY.rst

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,19 @@
22
33
Release History
44
===============
5+
1.3.1
6+
+++++
7+
* Upgrading to 2026-01-01 API version
8+
* Set default SKU to Premium for serverless workspaces
9+
510
1.3.0
611
+++++
712
* Revert the change of upgrading to 2026-01-01 API version
813

14+
1.2.1
15+
+++++
16+
* Upgrade API version from 2025-10-01-preview to 2026-01-01
17+
918
1.2.0
1019
+++++
1120
* az databricks workspace create: Add --compute-mode to set the compute mode for the workspace. Allowed values: 'Hybrid', 'Serverless'

src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_create.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,9 @@ class Create(AAZCommand):
2222
"""
2323

2424
_aaz_info = {
25-
"version": "2025-10-01-preview",
25+
"version": "2026-01-01",
2626
"resources": [
27-
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2025-10-01-preview"],
27+
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2026-01-01"],
2828
]
2929
}
3030

@@ -173,7 +173,7 @@ def url_parameters(self):
173173
def query_parameters(self):
174174
parameters = {
175175
**self.serialize_query_param(
176-
"api-version", "2025-10-01-preview",
176+
"api-version", "2026-01-01",
177177
required=True,
178178
),
179179
}

src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_delete.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,9 @@ class Delete(AAZCommand):
2222
"""
2323

2424
_aaz_info = {
25-
"version": "2025-10-01-preview",
25+
"version": "2026-01-01",
2626
"resources": [
27-
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2025-10-01-preview"],
27+
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2026-01-01"],
2828
]
2929
}
3030

@@ -146,7 +146,7 @@ def url_parameters(self):
146146
def query_parameters(self):
147147
parameters = {
148148
**self.serialize_query_param(
149-
"api-version", "2025-10-01-preview",
149+
"api-version", "2026-01-01",
150150
required=True,
151151
),
152152
}

src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_list.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,10 @@ class List(AAZCommand):
2525
"""
2626

2727
_aaz_info = {
28-
"version": "2025-10-01-preview",
28+
"version": "2026-01-01",
2929
"resources": [
30-
["mgmt-plane", "/subscriptions/{}/providers/microsoft.databricks/accessconnectors", "2025-10-01-preview"],
31-
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors", "2025-10-01-preview"],
30+
["mgmt-plane", "/subscriptions/{}/providers/microsoft.databricks/accessconnectors", "2026-01-01"],
31+
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors", "2026-01-01"],
3232
]
3333
}
3434

@@ -119,7 +119,7 @@ def url_parameters(self):
119119
def query_parameters(self):
120120
parameters = {
121121
**self.serialize_query_param(
122-
"api-version", "2025-10-01-preview",
122+
"api-version", "2026-01-01",
123123
required=True,
124124
),
125125
}
@@ -288,7 +288,7 @@ def url_parameters(self):
288288
def query_parameters(self):
289289
parameters = {
290290
**self.serialize_query_param(
291-
"api-version", "2025-10-01-preview",
291+
"api-version", "2026-01-01",
292292
required=True,
293293
),
294294
}

src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_show.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,9 @@ class Show(AAZCommand):
2222
"""
2323

2424
_aaz_info = {
25-
"version": "2025-10-01-preview",
25+
"version": "2026-01-01",
2626
"resources": [
27-
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2025-10-01-preview"],
27+
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2026-01-01"],
2828
]
2929
}
3030

@@ -124,7 +124,7 @@ def url_parameters(self):
124124
def query_parameters(self):
125125
parameters = {
126126
**self.serialize_query_param(
127-
"api-version", "2025-10-01-preview",
127+
"api-version", "2026-01-01",
128128
required=True,
129129
),
130130
}

src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_update.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,9 @@ class Update(AAZCommand):
2222
"""
2323

2424
_aaz_info = {
25-
"version": "2025-10-01-preview",
25+
"version": "2026-01-01",
2626
"resources": [
27-
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2025-10-01-preview"],
27+
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2026-01-01"],
2828
]
2929
}
3030

@@ -171,7 +171,7 @@ def url_parameters(self):
171171
def query_parameters(self):
172172
parameters = {
173173
**self.serialize_query_param(
174-
"api-version", "2025-10-01-preview",
174+
"api-version", "2026-01-01",
175175
required=True,
176176
),
177177
}
@@ -270,7 +270,7 @@ def url_parameters(self):
270270
def query_parameters(self):
271271
parameters = {
272272
**self.serialize_query_param(
273-
"api-version", "2025-10-01-preview",
273+
"api-version", "2026-01-01",
274274
required=True,
275275
),
276276
}

src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_wait.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ class Wait(AAZWaitCommand):
2020

2121
_aaz_info = {
2222
"resources": [
23-
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2025-10-01-preview"],
23+
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2026-01-01"],
2424
]
2525
}
2626

@@ -120,7 +120,7 @@ def url_parameters(self):
120120
def query_parameters(self):
121121
parameters = {
122122
**self.serialize_query_param(
123-
"api-version", "2025-10-01-preview",
123+
"api-version", "2026-01-01",
124124
required=True,
125125
),
126126
}

src/databricks/azext_databricks/aaz/latest/databricks/workspace/_create.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,9 @@ class Create(AAZCommand):
3636
"""
3737

3838
_aaz_info = {
39-
"version": "2025-10-01-preview",
39+
"version": "2026-01-01",
4040
"resources": [
41-
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/workspaces/{}", "2025-10-01-preview"],
41+
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/workspaces/{}", "2026-01-01"],
4242
]
4343
}
4444

@@ -440,7 +440,7 @@ def url_parameters(self):
440440
def query_parameters(self):
441441
parameters = {
442442
**self.serialize_query_param(
443-
"api-version", "2025-10-01-preview",
443+
"api-version", "2026-01-01",
444444
required=True,
445445
),
446446
}
@@ -628,8 +628,14 @@ def content(self):
628628
if require_infrastructure_encryption is not None:
629629
require_infrastructure_encryption.set_prop("value", AAZBoolType, ".require_infrastructure_encryption", typ_kwargs={"flags": {"required": True}})
630630

631+
# Set sku.name to 'premium' if sku is not set and compute_mode is 'Serverless'
631632
sku = _builder.get(".sku")
632-
if sku is not None:
633+
compute_mode = args.get("compute_mode", None)
634+
sku_value = args.get("sku", None)
635+
if (not sku_value or sku_value == "") and compute_mode and compute_mode.lower() == "serverless":
636+
if sku is not None:
637+
sku.set_const("name", "premium", AAZStrType)
638+
elif sku is not None:
633639
sku.set_prop("name", AAZStrType, ".sku", typ_kwargs={"flags": {"required": True}})
634640

635641
tags = _builder.get(".tags")

src/databricks/azext_databricks/aaz/latest/databricks/workspace/_delete.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,9 @@ class Delete(AAZCommand):
2323
"""
2424

2525
_aaz_info = {
26-
"version": "2025-10-01-preview",
26+
"version": "2026-01-01",
2727
"resources": [
28-
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/workspaces/{}", "2025-10-01-preview"],
28+
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/workspaces/{}", "2026-01-01"],
2929
]
3030
}
3131

@@ -155,7 +155,7 @@ def query_parameters(self):
155155
"forceDeletion", self.ctx.args.force_deletion,
156156
),
157157
**self.serialize_query_param(
158-
"api-version", "2025-10-01-preview",
158+
"api-version", "2026-01-01",
159159
required=True,
160160
),
161161
}

src/databricks/azext_databricks/aaz/latest/databricks/workspace/_list.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,10 @@ class List(AAZCommand):
2525
"""
2626

2727
_aaz_info = {
28-
"version": "2025-10-01-preview",
28+
"version": "2026-01-01",
2929
"resources": [
30-
["mgmt-plane", "/subscriptions/{}/providers/microsoft.databricks/workspaces", "2025-10-01-preview"],
31-
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/workspaces", "2025-10-01-preview"],
30+
["mgmt-plane", "/subscriptions/{}/providers/microsoft.databricks/workspaces", "2026-01-01"],
31+
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/workspaces", "2026-01-01"],
3232
]
3333
}
3434

@@ -119,7 +119,7 @@ def url_parameters(self):
119119
def query_parameters(self):
120120
parameters = {
121121
**self.serialize_query_param(
122-
"api-version", "2025-10-01-preview",
122+
"api-version", "2026-01-01",
123123
required=True,
124124
),
125125
}
@@ -596,7 +596,7 @@ def url_parameters(self):
596596
def query_parameters(self):
597597
parameters = {
598598
**self.serialize_query_param(
599-
"api-version", "2025-10-01-preview",
599+
"api-version", "2026-01-01",
600600
required=True,
601601
),
602602
}

0 commit comments

Comments
 (0)