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

feat: support thinking suffix for vertex gemini channel

Browse Source
This commit is contained in:
tbphp
2025-04-29 13:30:03 +08:00
parent fe37718259
commit efc9d200b1
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 {