Add support for OPENAI_URL environment variable

aandrew-me · Oct 11, 2024 · 1e751fb · 1e751fb
1 parent 28bd255
commit 1e751fb
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/providers/openai/openai.go b/providers/openai/openai.go
@@ -31,6 +31,11 @@ func NewRequest(input string, params structs.Params) (*http.Response, error) {
 		apiKey = params.ApiKey
 	}
 
+	url := params.Url
+	if os.Getenv("OPENAI_URL") != "" {
+		url = os.Getenv("OPENAI_URL")
+	}
+
 	temperature := "0.5"
 	if params.Temperature != "" {
 		temperature = params.Temperature
@@ -60,7 +65,7 @@ func NewRequest(input string, params structs.Params) (*http.Response, error) {
 	}
 	`, params.PrevMessages, string(safeInput), model, temperature, top_p))
 
-	req, err := http.NewRequest("POST", params.Url, data)
+	req, err := http.NewRequest("POST", url, data)
 	if err != nil {
 		fmt.Println("\nSome error has occurred.")
 		fmt.Println("Error:", err)