lzt/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #4412 from xyfacai/fix/image-n

Browse Source 
fix(image): only price image model use N ratio
This commit is contained in:
Calcium-Ion
2026-04-24 14:01:56 +08:00
committed by GitHub
parent 65b1654732 69ba18d392
commit ec8f3dceaa
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
relay/image_handler.go
+2
View File
@@ -122,9 +122,11 @@ func ImageHelper(c *gin.Context, info *relaycommon.RelayInfo) (newAPIError *type
// calculation (both price-based and ratio-based paths). // calculation (both price-based and ratio-based paths).
// Adaptors may have already set a more accurate count from the // Adaptors may have already set a more accurate count from the
// upstream response; only set the default when they haven't. // upstream response; only set the default when they haven't.
if info.PriceData.UsePrice { // only price model use N ratio
if _, hasN := info.PriceData.OtherRatios["n"]; !hasN { if _, hasN := info.PriceData.OtherRatios["n"]; !hasN {
info.PriceData.AddOtherRatio("n", float64(imageN)) info.PriceData.AddOtherRatio("n", float64(imageN))
} }
}
if usage.(*dto.Usage).TotalTokens == 0 { if usage.(*dto.Usage).TotalTokens == 0 {
usage.(*dto.Usage).TotalTokens = 1 usage.(*dto.Usage).TotalTokens = 1