From fc7cc7dc5ac9d53e307c493fe4dd0d25412f4a29 Mon Sep 17 00:00:00 2001
From: Hector Castejon Diaz <hector.castejon@databricks.com>
Date: Tue, 17 Feb 2026 10:53:02 +0000
Subject: [PATCH 1/2] sync logic with go

---
 .../com/databricks/sdk/core/DatabricksConfig.java   | 13 +++++--------
 .../databricks/sdk/DatabricksAuthManualTest.java    |  1 +
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java
index 84470959f..8388c218e 100644
--- a/databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java
+++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java
@@ -532,11 +532,7 @@ public DatabricksConfig setAzureTenantId(String azureTenantId) {
   }
 
   public AzureEnvironment getAzureEnvironment() {
-    String env = "PUBLIC";
-    if (azureEnvironment != null) {
-      env = azureEnvironment;
-    }
-    return AzureEnvironment.getEnvironment(env);
+    return getDatabricksEnvironment().getAzureEnvironment();
   }
 
   public DatabricksConfig setAzureEnvironment(String azureEnvironment) {
@@ -871,10 +867,11 @@ public DatabricksEnvironment getDatabricksEnvironment() {
       return this.databricksEnvironment;
     }
 
-    if (this.host == null && this.azureWorkspaceResourceId != null) {
+    if ((this.host == null || this.azureEnvironment != null)
+        && this.azureWorkspaceResourceId != null) {
       String azureEnv = "PUBLIC";
-      if (this.azureEnvironment != null) {
-        azureEnv = this.azureEnvironment;
+      if (this.azureEnvironment != null && !this.azureEnvironment.isEmpty()) {
+        azureEnv = this.azureEnvironment.toUpperCase();
       }
       for (DatabricksEnvironment env : DatabricksEnvironment.ALL_ENVIRONMENTS) {
         if (env.getCloud() != Cloud.AZURE) {
diff --git a/databricks-sdk-java/src/test/java/com/databricks/sdk/DatabricksAuthManualTest.java b/databricks-sdk-java/src/test/java/com/databricks/sdk/DatabricksAuthManualTest.java
index 288975445..0a66ac197 100644
--- a/databricks-sdk-java/src/test/java/com/databricks/sdk/DatabricksAuthManualTest.java
+++ b/databricks-sdk-java/src/test/java/com/databricks/sdk/DatabricksAuthManualTest.java
@@ -50,6 +50,7 @@ void azureCliUserWithManagementAccess() {
         createConfigWithMockClient()
             .setAuthType("azure-cli")
             .setHost("https://x")
+            .setAzureEnvironment("PUBLIC")
             .setAzureWorkspaceResourceId(azureWorkspaceResourceId);
     resolveConfig(config, env);
     Map<String, String> headers = config.authenticate();

From ccb19facbad292beeb2abff1243bb83ebe835955 Mon Sep 17 00:00:00 2001
From: Hector Castejon Diaz <hector.castejon@databricks.com>
Date: Tue, 17 Feb 2026 14:05:31 +0000
Subject: [PATCH 2/2] changelog

---
 NEXT_CHANGELOG.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/NEXT_CHANGELOG.md b/NEXT_CHANGELOG.md
index 2f99f0a27..ffeb58f9f 100644
--- a/NEXT_CHANGELOG.md
+++ b/NEXT_CHANGELOG.md
@@ -6,6 +6,8 @@
 
 ### Bug Fixes
 
+* Fixed inconsistent Azure environment resolution between `getAzureEnvironment()` and `getDatabricksEnvironment()` methods ([#XXX](https://github.com/databricks/databricks-sdk-java/pull/XXX))
+
 ### Security Vulnerabilities
 
 ### Documentation