Просмотр исходного кода

Merge branch 'main' of github.com:xqx121/new-api into xqx121-main

creamlike1024 8 месяцев назад
Родитель
Сommit
5b54624cd5
1 измененных файлов с 3 добавлено и 0 удалено
  1. 3 0
      relay/channel/gemini/relay-gemini-native.go

+ 3 - 0
relay/channel/gemini/relay-gemini-native.go

@@ -64,6 +64,9 @@ func GeminiTextGenerationHandler(c *gin.Context, resp *http.Response, info *rela
 			usage.PromptTokensDetails.TextTokens = detail.TokenCount
 			usage.PromptTokensDetails.TextTokens = detail.TokenCount
 		}
 		}
 	}
 	}
+	
+	// 计算最终使用量
+	usage.CompletionTokens = usage.TotalTokens - usage.PromptTokens
 
 
 	// 直接返回 Gemini 原生格式的 JSON 响应
 	// 直接返回 Gemini 原生格式的 JSON 响应
 	jsonResponse, err := json.Marshal(geminiResponse)
 	jsonResponse, err := json.Marshal(geminiResponse)