oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #1012 from tbphp/vertex_thinking_support

Browse Source 
feat: support thinking suffix for vertex gemini channel
This commit is contained in:
Calcium-Ion
2025-05-04 17:11:27 +08:00
committed by GitHub
parent e097d5a538 efc9d200b1
commit 9bf90c3baf
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -12,6 +12,7 @@ import (
"one-api/relay/channel/claude" "one-api/relay/channel/claude"
"one-api/relay/channel/gemini" "one-api/relay/channel/gemini"
"one-api/relay/channel/openai" "one-api/relay/channel/openai"
"one-api/setting/model_setting"
relaycommon "one-api/relay/common" relaycommon "one-api/relay/common"
"strings" "strings"
) )
@@ -77,6 +78,15 @@ func (a *Adaptor) GetRequestURL(info *relaycommon.RelayInfo) (string, error) {
a.AccountCredentials = *adc a.AccountCredentials = *adc
suffix := "" suffix := ""
if a.RequestMode == RequestModeGemini { if a.RequestMode == RequestModeGemini {
if model_setting.GetGeminiSettings().ThinkingAdapterEnabled {
// suffix -thinking and -nothinking
if strings.HasSuffix(info.OriginModelName, "-thinking") {
info.UpstreamModelName = strings.TrimSuffix(info.UpstreamModelName, "-thinking")
} else if strings.HasSuffix(info.OriginModelName, "-nothinking") {
info.UpstreamModelName = strings.TrimSuffix(info.UpstreamModelName, "-nothinking")
}
}
if info.IsStream { if info.IsStream {
suffix = "streamGenerateContent?alt=sse" suffix = "streamGenerateContent?alt=sse"
} else { } else {