فهرست منبع

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

CaIon 7 ماه پیش
والد
کامیت
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 {
 	switch effort {
 	case "high":
 	case "high":
-		return maxBudget * 80 / 100
+		maxBudget = maxBudget * 80 / 100
 	case "medium":
 	case "medium":
-		return maxBudget * 50 / 100
+		maxBudget = maxBudget * 50 / 100
 	case "low":
 	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) {
 func ThinkingAdaptor(geminiRequest *dto.GeminiChatRequest, info *relaycommon.RelayInfo, oaiRequest ...dto.GeneralOpenAIRequest) {