Răsfoiți Sursa

fix: fix claude channel test

CaIon 1 an în urmă
părinte
comite
26ef906c61
2 a modificat fișierele cu 4 adăugiri și 4 ștergeri
  1. 4 2
      controller/channel-test.go
  2. 0 2
      relay/channel/claude/adaptor.go

+ 4 - 2
controller/channel-test.go

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

+ 0 - 2
relay/channel/claude/adaptor.go

@@ -6,7 +6,6 @@ import (
 	"github.com/gin-gonic/gin"
 	"io"
 	"net/http"
-	"one-api/common"
 	"one-api/dto"
 	"one-api/relay/channel"
 	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) {
-	common.SysLog(fmt.Sprintf("Request mode: %d", a.RequestMode))
 	if request == nil {
 		return nil, errors.New("request is nil")
 	}