From b3b3c59aacd60827bb0eb0e1b14c891b2c5b1f46 Mon Sep 17 00:00:00 2001
From: Gk0Wk <nmg_wk@yeah.net>
Date: Tue, 26 Nov 2024 19:23:54 +0800
Subject: [PATCH] Fix

---
 controller/channel-billing.go | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/controller/channel-billing.go b/controller/channel-billing.go
index 3ebbf1c5..9456aa9c 100644
--- a/controller/channel-billing.go
+++ b/controller/channel-billing.go
@@ -109,7 +109,7 @@ type DeepSeekUsageResponse struct {
 		TotalBalance    string `json:"total_balance"`
 		GrantedBalance  string `json:"granted_balance"`
 		ToppedUpBalance string `json:"topped_up_balance"`
-	}
+	} `json:"balance_infos"`
 }
 
 // GetAuthHeader get auth header
@@ -267,7 +267,17 @@ func updateChannelDeepSeekBalance(channel *model.Channel) (float64, error) {
 	if err != nil {
 		return 0, err
 	}
-	balance, err := strconv.ParseFloat(response.BalanceInfos[0].TotalBalance, 64)
+	index := -1
+	for i, balanceInfo := range response.BalanceInfos {
+		if balanceInfo.Currency == "CNY" {
+			index = i
+			break
+		}
+	}
+	if index == -1 {
+		return 0, errors.New("currency CNY not found")
+	}
+	balance, err := strconv.ParseFloat(response.BalanceInfos[index].TotalBalance, 64)
 	if err != nil {
 		return 0, err
 	}