Просмотр исходного кода

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

fix: improve gemini nothinking handler
IcedTangerine 5 месяцев назад
Родитель
Сommit
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 {
 func isNoThinkingRequest(req *gemini.GeminiChatRequest) bool {
 	if req.GenerationConfig.ThinkingConfig != nil && req.GenerationConfig.ThinkingConfig.ThinkingBudget != nil {
 	if req.GenerationConfig.ThinkingConfig != nil && req.GenerationConfig.ThinkingConfig.ThinkingBudget != nil {
-		return *req.GenerationConfig.ThinkingConfig.ThinkingBudget <= 0
+		return *req.GenerationConfig.ThinkingConfig.ThinkingBudget == 0
 	}
 	}
 	return false
 	return false
 }
 }