xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-21 23:46:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

♻️ refactor: change test logic

Browse Source
This commit is contained in:
Edward
2023-05-26 20:40:00 +08:00
parent 708b254563
commit cee2253313
5 changed files with 27 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
controller/channel-test.go
View File

@@ -5,31 +5,21 @@ import (
"encoding/json"
"errors"
"fmt"
"github.com/gin-gonic/gin"
"net/http"
"one-api/common"
"one-api/model"
"strconv"
"sync"
"time"
"github.com/gin-gonic/gin"
)
func testChannel(channel *model.Channel, request *ChatRequest) error {
if request.Model == "" {
request.Model = "gpt-3.5-turbo"
if channel.Type == common.ChannelTypeAzure {
request.Model = "gpt-35-turbo"
}
}
requestURL := common.ChannelBaseURLs[channel.Type]
if channel.Type == common.ChannelTypeAzure {
requestURL = fmt.Sprintf("%s/openai/deployments/%s/chat/completions?api-version=2023-03-15-preview", channel.BaseURL, request.Model)
} else {
if channel.Type == common.ChannelTypeCustom {
requestURL = channel.BaseURL
}
requestURL += "/v1/chat/completions"
}
requestURL := common.ServerAddress + "/v1/chat/completions"
jsonData, err := json.Marshal(request)
if err != nil {
@@ -39,11 +29,7 @@ func testChannel(channel *model.Channel, request *ChatRequest) error {
if err != nil {
return err
}
if channel.Type == common.ChannelTypeAzure {
req.Header.Set("api-key", channel.Key)
} else {
req.Header.Set("Authorization", "Bearer "+channel.Key)
}
req.Header.Set("Authorization", fmt.Sprintf("%s-%d", common.ServerToken, channel.Id))
req.Header.Set("Content-Type", "application/json")
client := &http.Client{}
resp, err := client.Do(req)