Răsfoiți Sursa

fix: usage cost(any) to support claude and gemini

feitianbubu 5 luni în urmă
părinte
comite
d892bfc278
2 a modificat fișierele cu 2 adăugiri și 2 ștergeri
  1. 1 1
      dto/openai_response.go
  2. 1 1
      service/quota.go

+ 1 - 1
dto/openai_response.go

@@ -182,7 +182,7 @@ type Usage struct {
 	OutputTokens           int                `json:"output_tokens"`
 	InputTokensDetails     *InputTokenDetails `json:"input_tokens_details"`
 	// OpenRouter Params
-	Cost float64 `json:"cost,omitempty"`
+	Cost any `json:"cost,omitempty"`
 }
 
 type InputTokenDetails struct {

+ 1 - 1
service/quota.go

@@ -326,7 +326,7 @@ func CalcOpenRouterCacheCreateTokens(usage dto.Usage, priceData helper.PriceData
 	promptCacheReadPrice := quotaPrice * priceData.CacheRatio
 	completionPrice := quotaPrice * priceData.CompletionRatio
 
-	cost := usage.Cost
+	cost, _ := usage.Cost.(float64)
 	totalPromptTokens := float64(usage.PromptTokens)
 	completionTokens := float64(usage.CompletionTokens)
 	promptCacheReadTokens := float64(usage.PromptTokensDetails.CachedTokens)