Skip to content

Commit ce06c21

Browse files
committed
[Feature] CLI Profile support for unified host
1 parent bc06fb5 commit ce06c21

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksCliCredentialsProvider.java

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,18 @@ private CliTokenSource getDatabricksCliTokenSource(DatabricksConfig config) {
3131
}
3232
List<String> cmd =
3333
new ArrayList<>(Arrays.asList(cliPath, "auth", "token", "--host", config.getHost()));
34-
if (config.getClientType() == ClientType.ACCOUNT) {
34+
if (config.getExperimentalIsUnifiedHost() != null && config.getExperimentalIsUnifiedHost()) {
35+
// For unified hosts, pass account_id, workspace_id, and experimental flag
36+
cmd.add("--experimental-is-unified-host");
37+
if (config.getAccountId() != null) {
38+
cmd.add("--account-id");
39+
cmd.add(config.getAccountId());
40+
}
41+
if (config.getWorkspaceId() != null) {
42+
cmd.add("--workspace-id");
43+
cmd.add(config.getWorkspaceId());
44+
}
45+
} else if (config.getClientType() == ClientType.ACCOUNT) {
3546
cmd.add("--account-id");
3647
cmd.add(config.getAccountId());
3748
}

0 commit comments

Comments
 (0)