diff --git a/internal/registry/model_definitions.go b/internal/registry/model_definitions.go index 51f984f2..fd4bd428 100644 --- a/internal/registry/model_definitions.go +++ b/internal/registry/model_definitions.go @@ -222,6 +222,7 @@ func GetGeminiModels() []*ModelInfo { InputTokenLimit: 1048576, OutputTokenLimit: 65536, SupportedGenerationMethods: []string{"generateContent", "countTokens", "createCachedContent", "batchGenerateContent"}, + Thinking: &ThinkingSupport{Min: 128, Max: 32768, ZeroAllowed: false, DynamicAllowed: true}, }, } } @@ -301,6 +302,7 @@ func GetGeminiVertexModels() []*ModelInfo { InputTokenLimit: 1048576, OutputTokenLimit: 65536, SupportedGenerationMethods: []string{"generateContent", "countTokens", "createCachedContent", "batchGenerateContent"}, + Thinking: &ThinkingSupport{Min: 128, Max: 32768, ZeroAllowed: false, DynamicAllowed: true}, }, } }