mirror of
https://github.com/k8sgpt-ai/k8sgpt.git
synced 2025-07-07 20:39:30 +00:00
9b243cdcaa
This commit adds support for a new configuration format that is not backwards compatible with the previous format. This is a breaking change and requires users to update their configuration files to use the new format. BREAKING CHANGE: The format of the configuration file has changed. Users must update their configuration files to use the new format. Signed-off-by: Matthis Holleville <matthish29@gmail.com>
104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
package auth
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/k8sgpt-ai/k8sgpt/pkg/ai"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
var (
|
|
backend string
|
|
password string
|
|
model string
|
|
)
|
|
|
|
// authCmd represents the auth command
|
|
var AuthCmd = &cobra.Command{
|
|
Use: "auth",
|
|
Short: "Authenticate with your chosen backend",
|
|
Long: `Provide the necessary credentials to authenticate with your chosen backend.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
|
|
// get ai configuration
|
|
var configAI ai.AIConfiguration
|
|
err := viper.UnmarshalKey("ai", &configAI)
|
|
if err != nil {
|
|
color.Red("Error: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// search for provider with same name
|
|
providerIndex := -1
|
|
for i, provider := range configAI.Providers {
|
|
if backend == provider.Name {
|
|
providerIndex = i
|
|
break
|
|
}
|
|
}
|
|
|
|
// check if backend is not empty
|
|
if backend == "" {
|
|
color.Red("Error: Backend AI cannot be empty.")
|
|
os.Exit(1)
|
|
}
|
|
|
|
color.Green("Using %s as backend AI provider", backend)
|
|
|
|
// check if model is not empty
|
|
if model == "" {
|
|
color.Red("Error: Model cannot be empty.")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if password == "" {
|
|
fmt.Printf("Enter %s Key: ", backend)
|
|
bytePassword, err := term.ReadPassword(int(syscall.Stdin))
|
|
if err != nil {
|
|
color.Red("Error reading %s Key from stdin: %s", backend,
|
|
err.Error())
|
|
os.Exit(1)
|
|
}
|
|
password = strings.TrimSpace(string(bytePassword))
|
|
}
|
|
|
|
// create new provider object
|
|
newProvider := ai.AIProvider{
|
|
Name: backend,
|
|
Model: model,
|
|
Password: password,
|
|
}
|
|
|
|
if providerIndex == -1 {
|
|
// provider with same name does not exist, add new provider to list
|
|
configAI.Providers = append(configAI.Providers, newProvider)
|
|
color.Green("New provider added")
|
|
} else {
|
|
// provider with same name exists, update provider info
|
|
configAI.Providers[providerIndex] = newProvider
|
|
color.Green("Provider updated")
|
|
}
|
|
viper.Set("ai", configAI)
|
|
if err := viper.WriteConfig(); err != nil {
|
|
color.Red("Error writing config file: %s", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
color.Green("key added")
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
// add flag for backend
|
|
AuthCmd.Flags().StringVarP(&backend, "backend", "b", "openai", "Backend AI provider")
|
|
// add flag for model
|
|
AuthCmd.Flags().StringVarP(&model, "model", "m", "gpt-3.5-turbo", "Backend AI model")
|
|
// add flag for password
|
|
AuthCmd.Flags().StringVarP(&password, "password", "p", "", "Backend AI password")
|
|
}
|