Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: fix the shared field is modified

Browse Source
This commit is contained in:
JustSong 2023-06-18 00:20:06 +08:00
parent d97640374c
commit 1cb1f727c0
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
controller/channel-test.go
View File

@ -14,12 +14,12 @@ import (
"time" "time"
) )
func testChannel(channel *model.Channel, request *ChatRequest) error { func testChannel(channel *model.Channel, request ChatRequest) error {
if request.Model == "" { switch channel.Type {
case common.ChannelTypeAzure:
request.Model = "gpt-35-turbo"
default:
request.Model = "gpt-3.5-turbo" request.Model = "gpt-3.5-turbo"
if channel.Type == common.ChannelTypeAzure {
request.Model = "gpt-35-turbo"
}
} }
requestURL := common.ChannelBaseURLs[channel.Type] requestURL := common.ChannelBaseURLs[channel.Type]
if channel.Type == common.ChannelTypeAzure { if channel.Type == common.ChannelTypeAzure {
@ -97,7 +97,7 @@ func TestChannel(c *gin.Context) {
} }
testRequest := buildTestRequest(c) testRequest := buildTestRequest(c)
tik := time.Now() tik := time.Now()
err = testChannel(channel, testRequest) err = testChannel(channel, *testRequest)
tok := time.Now() tok := time.Now()
milliseconds := tok.Sub(tik).Milliseconds() milliseconds := tok.Sub(tik).Milliseconds()
go channel.UpdateResponseTime(milliseconds) go channel.UpdateResponseTime(milliseconds)
@ -165,7 +165,7 @@ func testAllChannels(c *gin.Context) error {
continue continue
} }
tik := time.Now() tik := time.Now()
err := testChannel(channel, testRequest) err := testChannel(channel, *testRequest)
tok := time.Now() tok := time.Now()
milliseconds := tok.Sub(tik).Milliseconds() milliseconds := tok.Sub(tik).Milliseconds()
if err != nil || milliseconds > disableThreshold { if err != nil || milliseconds > disableThreshold {