Skip to content

Commit a06e590

Browse files
authored
fix: gemini vertexai image thoughts need disable and enabled by default (#467)
1 parent cb67639 commit a06e590

2 files changed

Lines changed: 7 additions & 5 deletions

File tree

core/relay/adaptor/gemini/openai.go

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -125,12 +125,14 @@ func buildGenerationConfig(
125125
// https://ai.google.dev/gemini-api/docs/thinking
126126
if config.ThinkingConfig == nil &&
127127
(strings.Contains(meta.ActualModel, "-2.5") ||
128-
strings.Contains(meta.ActualModel, "-3")) &&
128+
strings.Contains(meta.ActualModel, "-3")) {
129129
// disable vertexai image model include thoughts
130130
// because error call gemini-3-pro-image-preview model
131-
(meta.Channel.Type == model.ChannelTypeVertexAI && !strings.Contains(meta.ActualModel, "image")) {
132-
config.ThinkingConfig = &relaymodel.GeminiThinkingConfig{
133-
IncludeThoughts: true,
131+
if meta.Channel.Type != model.ChannelTypeVertexAI ||
132+
!strings.Contains(meta.ActualModel, "image") {
133+
config.ThinkingConfig = &relaymodel.GeminiThinkingConfig{
134+
IncludeThoughts: true,
135+
}
134136
}
135137
}
136138

core/relay/adaptor/streamlake/adaptor.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,9 @@ import (
1010
"github.com/labring/aiproxy/core/relay/adaptor"
1111
"github.com/labring/aiproxy/core/relay/adaptor/anthropic"
1212
"github.com/labring/aiproxy/core/relay/adaptor/openai"
13-
relaymodel "github.com/labring/aiproxy/core/relay/model"
1413
"github.com/labring/aiproxy/core/relay/meta"
1514
"github.com/labring/aiproxy/core/relay/mode"
15+
relaymodel "github.com/labring/aiproxy/core/relay/model"
1616
"github.com/labring/aiproxy/core/relay/utils"
1717
)
1818

0 commit comments

Comments
 (0)