ソースを参照

Merge pull request #1405 from RedwindA/fix/gemini-nothinking-handler

fix: improve gemini nothinking handler
IcedTangerine 5 ヶ月 前
コミット
db9b333930
1 ファイル変更1 行追加1 行削除
  1. 1 1
      relay/gemini_handler.go

+ 1 - 1
relay/gemini_handler.go

@@ -80,7 +80,7 @@ func getGeminiInputTokens(req *gemini.GeminiChatRequest, info *relaycommon.Relay
 
 func isNoThinkingRequest(req *gemini.GeminiChatRequest) bool {
 	if req.GenerationConfig.ThinkingConfig != nil && req.GenerationConfig.ThinkingConfig.ThinkingBudget != nil {
-		return *req.GenerationConfig.ThinkingConfig.ThinkingBudget <= 0
+		return *req.GenerationConfig.ThinkingConfig.ThinkingBudget == 0
 	}
 	return false
 }