From e7c9003ccf1f3b76019a1a04874a1d8ecdbe2473 Mon Sep 17 00:00:00 2001
From: mrhaoji <mrhaoji@gmail.com>
Date: Mon, 24 Jul 2023 18:53:33 +0800
Subject: [PATCH] fix: baseURL not working in APITypePaLM

---
 controller/relay-text.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/controller/relay-text.go b/controller/relay-text.go
index 8dfdf6e1..961d9517 100644
--- a/controller/relay-text.go
+++ b/controller/relay-text.go
@@ -142,6 +142,9 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
 		fullRequestURL = "https://generativelanguage.googleapis.com/v1beta2/models/chat-bison-001:generateMessage"
 		apiKey := c.Request.Header.Get("Authorization")
 		apiKey = strings.TrimPrefix(apiKey, "Bearer ")
+		if baseURL != "" {
+			fullRequestURL = strings.Replace(fullRequestURL, "https://generativelanguage.googleapis.com", baseURL, 1)
+		}
 		fullRequestURL += "?key=" + apiKey
 	case APITypeZhipu:
 		method := "invoke"