github/k8sgpt
1
0
Fork 0
mirror of https://github.com/k8sgpt-ai/k8sgpt.git synced 2025-07-13 07:05:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: envs to initialise server

Browse Source 
Signed-off-by: Alex Jones <alexsimonjones@gmail.com>
This commit is contained in:
Alex Jones 2023-04-15 12:26:51 +01:00
parent 51b1b352ac
commit 0071e25992
2 changed files with 26 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cmd/root.go
View File

@ -1,10 +1,11 @@
package cmd package cmd
import ( import (
"github.com/k8sgpt-ai/k8sgpt/cmd/serve"
"os" "os"
"path/filepath" "path/filepath"
"github.com/k8sgpt-ai/k8sgpt/cmd/serve"
"github.com/k8sgpt-ai/k8sgpt/cmd/analyze" "github.com/k8sgpt-ai/k8sgpt/cmd/analyze"
"github.com/k8sgpt-ai/k8sgpt/cmd/auth" "github.com/k8sgpt-ai/k8sgpt/cmd/auth"
"github.com/k8sgpt-ai/k8sgpt/cmd/filters" "github.com/k8sgpt-ai/k8sgpt/cmd/filters"
@ -81,6 +82,7 @@ func initConfig() {
viper.Set("kubecontext", kubecontext) viper.Set("kubecontext", kubecontext)
viper.Set("kubeconfig", kubeconfig) viper.Set("kubeconfig", kubeconfig)
viper.SetEnvPrefix("K8SGPT")
viper.AutomaticEnv() // read in environment variables that match viper.AutomaticEnv() // read in environment variables that match
// If a config file is found, read it in. // If a config file is found, read it in.

22
cmd/serve/serve.go
View File

@ -28,19 +28,33 @@ var ServeCmd = &cobra.Command{
color.Red("Error: %v", err) color.Red("Error: %v", err)
os.Exit(1) os.Exit(1)
} }
var aiProvider *ai.AIProvider
if len(configAI.Providers) == 0 { if len(configAI.Providers) == 0 {
// Check for env injection
backend = os.Getenv("K8SGPT_BACKEND")
password := os.Getenv("K8SGPT_PASSWORD")
model := os.Getenv("K8SGPT_MODEL")
// If the envs are set, alocate in place to the aiProvider
// else exit with error
if backend != "" || password != "" || model != "" {
aiProvider = &ai.AIProvider{
Name: backend,
Password: password,
Model: model,
}
} else {
color.Red("Error: AI provider not specified in configuration. Please run k8sgpt auth") color.Red("Error: AI provider not specified in configuration. Please run k8sgpt auth")
os.Exit(1) os.Exit(1)
} }
}
var aiProvider ai.AIProvider if aiProvider == nil {
for _, provider := range configAI.Providers { for _, provider := range configAI.Providers {
if backend == provider.Name { if backend == provider.Name {
aiProvider = provider aiProvider = &provider
break break
} }
} }
}
if aiProvider.Name == "" { if aiProvider.Name == "" {
color.Red("Error: AI provider %s not specified in configuration. Please run k8sgpt auth", backend) color.Red("Error: AI provider %s not specified in configuration. Please run k8sgpt auth", backend)