File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments