fix: missing validation for backend option in remove command (#429)

Signed-off-by: Matthis Holleville <matthish29@gmail.com>
k8sgpt-ai · May 16, 2023 · af826d5 · af826d5
1 parent cbe2fb4
commit af826d5
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cmd/auth/remove.go b/cmd/auth/remove.go
@@ -32,6 +32,11 @@ var removeCmd = &cobra.Command{
 			os.Exit(1)
 		}
 
+		if backend == "" {
+			color.Red("Error: --backend option must be set.")
+			os.Exit(1)
+		}
+
 		foundBackend := false
 		for i, provider := range configAI.Providers {
 			if backend == provider.Name {
@@ -55,5 +60,5 @@ var removeCmd = &cobra.Command{
 
 func init() {
 	// add flag for backend
-	removeCmd.Flags().StringVarP(&backend, "backend", "b", "openai", "Backend AI provider")
+	removeCmd.Flags().StringVarP(&backend, "backend", "b", "", "Backend AI provider")
 }