fix validation

brevdev · Nov 15, 2024 · 80ce94f · 80ce94f
1 parent d08c877
commit 80ce94f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/cmd/login/login.go b/pkg/cmd/login/login.go
@@ -157,12 +157,12 @@ func (o LoginOptions) RunLogin(t *terminal.Terminal, loginToken string, skipBrow
 		return breverrors.NewValidationError("auth provider must be nvidia or legacy")
 	}
 
-	if emailFlag != "" && authProviderFlag != "nvidia" {
+	authenticator := auth.StandardLogin(authProviderFlag, emailFlag, tokens)
+
+	if emailFlag != "" && authenticator.GetCredentialProvider() != auth.CredentialProviderKAS {
 		return breverrors.NewValidationError("email flag can only be used with nvidia auth provider")
 	}
 
-	authenticator := auth.StandardLogin(authProviderFlag, emailFlag, tokens)
-
 	o.Auth = auth.NewAuth(o.LoginStore, authenticator)
 
 	caretType := color.New(color.FgGreen, color.Bold).SprintFunc()