From 481108dece355e824b0378703c09e99847e893bf Mon Sep 17 00:00:00 2001
From: Peter Dave Hello <hsu@peterdavehello.org>
Date: Sun, 12 May 2024 14:56:32 +0800
Subject: [PATCH] Fix error when locating config.yaml

---
 azure/init.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/azure/init.go b/azure/init.go
index b128ee4..d394e19 100644
--- 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
 	}