fix auth profiles

Author: jh-db

cmd/auth/login.go

@@ -279,7 +279,7 @@ func setHostAndAccountId(ctx context.Context, cmd *cobra.Command, existingProfil
 
 	// If the account-id was not provided as a cmd line flag, try to read it from
 	// the specified profile.
-	isAccountHost := (&config.Config{Host: authArguments.Host, Experimental_IsUnifiedHost: authArguments.IsUnifiedHost}).GetHostType() != config.WorkspaceHost
+	isAccountHost := (&config.Config{Host: authArguments.Host, Experimental_IsUnifiedHost: authArguments.IsUnifiedHost}).HostType() != config.WorkspaceHost
 	accountID := authArguments.AccountID
 	if isAccountHost && accountID == "" {
 		if existingProfile != nil && existingProfile.AccountID != "" {

cmd/auth/profiles.go

@@ -51,7 +51,7 @@ func (c *profileMetadata) Load(ctx context.Context, configFilePath string, skipV
 		return
 	}
 
-	if cfg.GetHostType() != config.WorkspaceHost {
+	if cfg.ConfigType() == config.AccountConfig {
 		a, err := databricks.NewAccountClient((*databricks.Config)(cfg))
 		if err != nil {
 			return

go.mod

@@ -75,3 +75,5 @@ require (
 	google.golang.org/grpc v1.75.1 // indirect
 	google.golang.org/protobuf v1.36.9 // indirect
 )
+
+replace github.com/databricks/databricks-sdk-go => ../databricks-sdk-go

libs/auth/arguments.go

@@ -21,9 +21,9 @@ func (a AuthArguments) ToOAuthArgument() (u2m.OAuthArgument, error) {
 		Experimental_IsUnifiedHost: a.IsUnifiedHost,
 	}
 	host := cfg.CanonicalHostName()
-	if cfg.GetHostType() == config.AccountHost {
+	if cfg.HostType() == config.AccountHost {
 		return u2m.NewBasicAccountOAuthArgument(host, cfg.AccountID)
-	} else if cfg.GetHostType() == config.UnifiedHost {
+	} else if cfg.HostType() == config.UnifiedHost {
 		return u2m.NewBasicUnifiedOAuthArgument(host, cfg.AccountID)
 	}
 	return u2m.NewBasicWorkspaceOAuthArgument(host)