fix: Resolve comments.

Signed-off-by: Aris Boutselis <aris.boutselis@senseon.io>

cmd/auth/auth.go

@@ -60,9 +60,8 @@ var AuthCmd = &cobra.Command{
 		}
 
 		// check if backend is not empty and a valid value
-		backends := util.Backends()
-		if backend == "" || !util.ValidBackend(backends, backend) {
-			color.Red("Error: Backend AI cannot be empty and accepted values are '%v'", strings.Join(backends, ", "))
+		if backend == "" || !util.ValidBackend(ai.Backends, backend) {
+			color.Red("Error: Backend AI cannot be empty and accepted values are '%v'", strings.Join(ai.Backends, ", "))
 			os.Exit(1)
 		}
 

pkg/ai/iai.go

@@ -19,6 +19,21 @@ import (
 	"github.com/k8sgpt-ai/k8sgpt/pkg/cache"
 )
 
+var (
+	clients = []IAI{
+		&OpenAIClient{},
+		&AzureAIClient{},
+		&LocalAIClient{},
+		&NoOpAIClient{},
+	}
+	Backends = []string{
+		"openai",
+		"localai",
+		"azureopenai",
+		"noopai",
+	}
+)
+
 type IAI interface {
 	Configure(config IAIConfig, language string) error
 	GetCompletion(ctx context.Context, prompt string) (string, error)
@@ -34,18 +49,13 @@ type IAIConfig interface {
 }
 
 func NewClient(provider string) IAI {
-	switch provider {
-	case "openai":
-		return &OpenAIClient{}
-	case "azureopenai":
-		return &AzureAIClient{}
-	case "localai":
-		return &LocalAIClient{}
-	case "noopai":
-		return &NoOpAIClient{}
-	default:
-		return &OpenAIClient{}
+	for _, c := range clients {
+		if provider == c.GetName() {
+			return c
+		}
 	}
+	// default client
+	return &OpenAIClient{}
 }
 
 type AIConfiguration struct {

pkg/util/util.go

@@ -186,10 +186,6 @@ func EnsureDirExists(dir string) error {
 	return err
 }
 
-func Backends() []string {
-	return []string{"openai", "localai", "azureopenai", "noopai"}
-}
-
 func ValidBackend(validBackends []string, backend string) bool {
 	for _, b := range validBackends {
 		if b == backend {