From cb4ee5e86a70d53e6fd8636ad983a7b73ca44d11 Mon Sep 17 00:00:00 2001 From: RandyZhang Date: Thu, 15 May 2025 16:12:56 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Denable=E5=BC=80=E5=90=AF?= =?UTF-8?q?=E7=9A=84=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- relay/adaptor/vertexai/gemini/adapter.go | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/relay/adaptor/vertexai/gemini/adapter.go b/relay/adaptor/vertexai/gemini/adapter.go index abdd13c7..3e91e844 100644 --- a/relay/adaptor/vertexai/gemini/adapter.go +++ b/relay/adaptor/vertexai/gemini/adapter.go @@ -41,16 +41,16 @@ func (a *Adaptor) parseGeminiChatGenerationThinking(model string) (string, *gemi _modelName := parts[0] if len(parts) >= 2 { modelOptions, err := url.ParseQuery(parts[1]) - if err != nil && modelOptions != nil { + if err == nil && modelOptions != nil { modelName = _modelName - enableThinking := modelOptions.Has("thinking") - if enableThinking { - thinkingConfig.IncludeThoughts = true + hasThinkingFlag := modelOptions.Has("thinking") + if hasThinkingFlag { + thinkingConfig.IncludeThoughts = modelOptions.Get("thinking") == "1" } thinkingBudget := modelOptions.Get("thinking_budget") if thinkingBudget != "" { thinkingBudgetInt, err := strconv.Atoi(thinkingBudget) - if err != nil { + if err == nil { thinkingConfig.ThinkingBudget = thinkingBudgetInt } }