xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-18 00:16:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix claude channel test

Browse Source
This commit is contained in:
CaIon 2024-03-08 21:38:43 +08:00
parent 655dfe0d09
commit 26ef906c61
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
controller/channel-test.go
View File

@ -52,6 +52,7 @@ func testChannel(channel *model.Channel, testModel string) (err error, openaiErr
case common.ChannelTypeAli: case common.ChannelTypeAli:
c.Set("plugin", channel.Other) c.Set("plugin", channel.Other)
} }
meta := relaycommon.GenRelayInfo(c) meta := relaycommon.GenRelayInfo(c)
apiType := constant.ChannelType2APIType(channel.Type) apiType := constant.ChannelType2APIType(channel.Type)
adaptor := relay.GetAdaptor(apiType) adaptor := relay.GetAdaptor(apiType)
@ -60,13 +61,14 @@ func testChannel(channel *model.Channel, testModel string) (err error, openaiErr
} }
if testModel == "" { if testModel == "" {
testModel = adaptor.GetModelList()[0] testModel = adaptor.GetModelList()[0]
meta.UpstreamModelName = testModel
} }
request := buildTestRequest() request := buildTestRequest()
request.Model = testModel
meta.UpstreamModelName = testModel
adaptor.Init(meta, *request) adaptor.Init(meta, *request)
request.Model = testModel
meta.UpstreamModelName = testModel
convertedRequest, err := adaptor.ConvertRequest(c, constant.RelayModeChatCompletions, request) convertedRequest, err := adaptor.ConvertRequest(c, constant.RelayModeChatCompletions, request)
if err != nil { if err != nil {
return err, nil return err, nil

2
relay/channel/claude/adaptor.go
View File

@ -6,7 +6,6 @@ import (
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"io" "io"
"net/http" "net/http"
"one-api/common"
"one-api/dto" "one-api/dto"
"one-api/relay/channel" "one-api/relay/channel"
relaycommon "one-api/relay/common" relaycommon "one-api/relay/common"
@ -50,7 +49,6 @@ func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, info *re
} }
func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *dto.GeneralOpenAIRequest) (any, error) { func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *dto.GeneralOpenAIRequest) (any, error) {
common.SysLog(fmt.Sprintf("Request mode: %d", a.RequestMode))
if request == nil { if request == nil {
return nil, errors.New("request is nil") return nil, errors.New("request is nil")
} }