fix

Calcium-Ion · Feb 28, 2025 · 1529504 · 1529504
1 parent d6fd50e
commit 1529504
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/relay/helper/price.go b/relay/helper/price.go
@@ -26,7 +26,8 @@ func ModelPriceHelper(c *gin.Context, info *relaycommon.RelayInfo, promptTokens
 		if maxTokens != 0 {
 			preConsumedTokens = promptTokens + maxTokens
 		}
-		modelRatio, success := common.GetModelRatio(info.OriginModelName)
+		var success bool
+		modelRatio, success = common.GetModelRatio(info.OriginModelName)
 		if !success {
 			return PriceData{}, fmt.Errorf("model %s ratio not found", info.OriginModelName)
 		}