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

feat: add claude-3.7-sonnet 128k output support

Browse Source
This commit is contained in:
wozulong 2025-02-26 12:03:16 +08:00
parent d870c1099f
commit 5f1c5945f8
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -48,7 +48,7 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, info *relaycommon.RelayInfo, re
var claudeReq *claude.ClaudeRequest
var err error
claudeReq, err = claude.RequestOpenAI2ClaudeMessage(*request)
claudeReq, err = claude.RequestOpenAI2ClaudeMessage(c, *request)
c.Set("request_model", request.Model)
c.Set("converted_request", claudeReq)

4
relay/channel/claude/relay-claude.go
View File

@ -65,7 +65,9 @@ func RequestOpenAI2ClaudeMessage(c *gin.Context, textRequest dto.GeneralOpenAIRe
if strings.HasSuffix(textRequest.Model, "-thinking") {
textRequest.Model = strings.TrimSuffix(textRequest.Model, "-thinking")
if textRequest.MaxTokens < 2048 {
if textRequest.MaxTokens == 0 {
textRequest.MaxTokens = 8192
} else if textRequest.MaxTokens < 2048 {
textRequest.MaxTokens = 2048
} else if textRequest.MaxTokens > 64000 {
c.Request.Header.Set("anthropic-beta", "output-128k-2025-02-19")

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

@ -122,7 +122,7 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, info *relaycommon.RelayInfo, re
return nil, errors.New("request is nil")
}
if a.RequestMode == RequestModeClaude {
claudeReq, err := claude.RequestOpenAI2ClaudeMessage(*request)
claudeReq, err := claude.RequestOpenAI2ClaudeMessage(c, *request)
if err != nil {
return nil, err
}