Jelajahi Sumber

fix: 修正nothinking判断逻辑,确保仅当预算为零时返回true

RedwindA 9 bulan lalu
induk
melakukan
e06d0ba0e4
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  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
 }