diff --git a/lib/connection/auth/DatabricksOAuth/OAuthManager.ts b/lib/connection/auth/DatabricksOAuth/OAuthManager.ts index 02bbde1..db7e7c6 100644 --- a/lib/connection/auth/DatabricksOAuth/OAuthManager.ts +++ b/lib/connection/auth/DatabricksOAuth/OAuthManager.ts @@ -219,7 +219,7 @@ export default abstract class OAuthManager { } if (options.useDatabricksOAuthInAzure) { - const azureDomains = ['.azuredatabricks.net']; + const azureDomains = ['.azuredatabricks.net', '.databricks.azure.cn']; const isAzureDomain = azureDomains.some((domain) => host.endsWith(domain)); if (isAzureDomain) { // eslint-disable-next-line @typescript-eslint/no-use-before-define