From 0004c1022dba930adaf6ad7ee0ace7e4e24ad42e Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Thu, 1 Feb 2024 18:11:00 +0800
Subject: [PATCH] fix: support AI Gateway, close #52

---
 controller/channel-test.go | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/controller/channel-test.go b/controller/channel-test.go
index 5feaa98..b20bbeb 100644
--- a/controller/channel-test.go
+++ b/controller/channel-test.go
@@ -43,14 +43,10 @@ func testChannel(channel *model.Channel, request ChatRequest) (err error, openai
 	default:
 		request.Model = "gpt-3.5-turbo"
 	}
-	requestURL := common.ChannelBaseURLs[channel.Type]
+	requestURL := getFullRequestURL(channel.GetBaseURL(), "/v1/chat/completions", channel.Type)
+
 	if channel.Type == common.ChannelTypeAzure {
 		requestURL = getFullRequestURL(channel.GetBaseURL(), fmt.Sprintf("/openai/deployments/%s/chat/completions?api-version=2023-03-15-preview", request.Model), channel.Type)
-	} else {
-		if channel.GetBaseURL() != "" {
-			requestURL = channel.GetBaseURL()
-		}
-		requestURL += "/v1/chat/completions"
 	}
 
 	jsonData, err := json.Marshal(request)