From d892bfc278f763479a4d39cd482e3c24d81ee474 Mon Sep 17 00:00:00 2001
From: feitianbubu <feitianbubu@qq.com>
Date: Wed, 16 Jul 2025 16:53:29 +0800
Subject: [PATCH] fix: usage cost(any) to support claude and gemini

---
 dto/openai_response.go | 2 +-
 service/quota.go       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/dto/openai_response.go b/dto/openai_response.go
index 64601427..7af17b82 100644
--- a/dto/openai_response.go
+++ b/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 {
diff --git a/service/quota.go b/service/quota.go
index 7a6177de..27546ed4 100644
--- a/service/quota.go
+++ b/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)