修复负数情况

denzhe22 · Mar 9, 2023 · fb1d490 · fb1d490
1 parent 41f3d4c
commit fb1d490
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/code/services/gpt3.go b/code/services/gpt3.go
@@ -177,6 +177,10 @@ func (gpt *ChatGPT) getApiKey() (string, error) {
 	}
 	totalWeight := 0
 	for _, weight := range gpt.apiKeyWeights {
+		if weight <= 0 {
+			// 如果权重为0或负数，则永久禁用该apikey
+			continue
+		}
 		totalWeight += weight
 	}
 	if totalWeight == 0 {
@@ -185,6 +189,10 @@ func (gpt *ChatGPT) getApiKey() (string, error) {
 	}
 	randNum := rand.Intn(totalWeight)
 	for i, weight := range gpt.apiKeyWeights {
+		if weight <= 0 {
+			// 如果权重为0或负数，则跳过该apikey
+			continue
+		}
 		if randNum < weight {
 			gpt.currentApiKeyIndex = i
 			break