Merge pull request #274 from matthisholleville/fix/server-authentication

fix: use the aiProvider object when launching the server

cmd/serve/serve.go

@@ -1,12 +1,13 @@
 package serve
 
 import (
-	"fmt"
+	"os"
+
 	"github.com/fatih/color"
+	"github.com/k8sgpt-ai/k8sgpt/pkg/ai"
 	k8sgptserver "github.com/k8sgpt-ai/k8sgpt/pkg/server"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
-	"os"
 )
 
 var (
@@ -21,30 +22,38 @@ var ServeCmd = &cobra.Command{
 	Long:  `Runs k8sgpt as a server to allow for easy integration with other applications.`,
 	Run: func(cmd *cobra.Command, args []string) {
 
-		backendType := viper.GetString("backend_type")
-		if backendType == "" {
-			color.Red("No backend set. Please run k8sgpt auth")
+		var configAI ai.AIConfiguration
+		err := viper.UnmarshalKey("ai", &configAI)
+		if err != nil {
+			color.Red("Error: %v", err)
 			os.Exit(1)
 		}
 
-		if backend != "" {
-			backendType = backend
+		if len(configAI.Providers) == 0 {
+			color.Red("Error: AI provider not specified in configuration. Please run k8sgpt auth")
+			os.Exit(1)
 		}
 
-		token := viper.GetString(fmt.Sprintf("%s_key", backendType))
-		// check if nil
-		if token == "" {
-			color.Red("No %s key set. Please run k8sgpt auth", backendType)
+		var aiProvider ai.AIProvider
+		for _, provider := range configAI.Providers {
+			if backend == provider.Name {
+				aiProvider = provider
+				break
+			}
+		}
+
+		if aiProvider.Name == "" {
+			color.Red("Error: AI provider %s not specified in configuration. Please run k8sgpt auth", backend)
 			os.Exit(1)
 		}
 
 		server := k8sgptserver.Config{
-			Backend: backend,
+			Backend: aiProvider.Name,
 			Port:    port,
-			Token:   token,
+			Token:   aiProvider.Password,
 		}
 
-		err := server.Serve()
+		err = server.Serve()
 		if err != nil {
 			color.Red("Error: %v", err)
 			os.Exit(1)