Fix error when locating config.yaml

stulzq · May 12, 2024 · 481108d · 481108d
1 parent e85eed3
commit 481108d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/azure/init.go b/azure/init.go
@@ -7,6 +7,7 @@ import (
 	"github.com/stulzq/azure-openai-proxy/util"
 	"log"
 	"net/url"
+	"os"
 	"path/filepath"
 	"strings"
 )
@@ -101,10 +102,15 @@ func InitFromConfigFile() error {
 		configFile = filepath.Join(util.GetWorkdir(), configFile)
 	}
 
+	if _, err := os.Stat(configFile); os.IsNotExist(err) {
+		log.Println("config.yaml does not exist, attempting to read configuration from environment variables")
+		return InitFromEnvironmentVariables(viper.GetString(constant.ENV_AZURE_OPENAI_API_VER), viper.GetString(constant.ENV_AZURE_OPENAI_ENDPOINT), viper.GetString(constant.ENV_AZURE_OPENAI_MODEL_MAPPER))
+	}
+
 	viper.SetConfigType("yaml")
 	viper.SetConfigFile(configFile)
 	if err := viper.ReadInConfig(); err != nil {
-		log.Printf("read config file error: %+v\n", err)
+		log.Printf("Failed to read config file: %s. Please check if the file exists or consider using environment variables for configuration.\n", err)
 		return err
 	}