Przeglądaj źródła

fix: update budget calculation logic in relay-gemini to use clamping function

CaIon 7 miesięcy temu
rodzic
commit
f46cefbd39
1 zmienionych plików z 4 dodań i 4 usunięć
  1. 4 4
      relay/channel/gemini/relay-gemini.go

+ 4 - 4
relay/channel/gemini/relay-gemini.go

@@ -107,13 +107,13 @@ func clampThinkingBudgetByEffort(modelName string, effort string) int {
 	}
 	switch effort {
 	case "high":
-		return maxBudget * 80 / 100
+		maxBudget = maxBudget * 80 / 100
 	case "medium":
-		return maxBudget * 50 / 100
+		maxBudget = maxBudget * 50 / 100
 	case "low":
-		return maxBudget * 20 / 100
+		maxBudget = maxBudget * 20 / 100
 	}
-	return maxBudget * 50 / 100 // 默认medium
+	return clampThinkingBudget(modelName, maxBudget)
 }
 
 func ThinkingAdaptor(geminiRequest *dto.GeminiChatRequest, info *relaycommon.RelayInfo, oaiRequest ...dto.GeneralOpenAIRequest) {