瀏覽代碼

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

CaIon 8 月之前
父節點
當前提交
f46cefbd39
共有 1 個文件被更改,包括 4 次插入4 次删除
  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) {