From 8c588c5c1dfd3d812ae43cad3aa6465723802687 Mon Sep 17 00:00:00 2001
From: Rock Chin <1010553892@qq.com>
Date: Sat, 17 Dec 2022 18:14:57 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20keymgr=E5=AF=B9=E5=B7=B2=E5=82=A8?=
 =?UTF-8?q?=E5=AD=98=E7=9A=84=E7=BB=93=E6=9E=9C=E5=8F=96=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/openai/keymgr.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pkg/openai/keymgr.py b/pkg/openai/keymgr.py
index 965a022d..5947c0a1 100644
--- a/pkg/openai/keymgr.py
+++ b/pkg/openai/keymgr.py
@@ -78,6 +78,8 @@ class KeysManager:
 
         self.usage[md5] += round(int((len(new_content.encode('utf-8')) - len(new_content)) / 2 + len(new_content))*salt_rate)
 
+        self.usage[md5] = int(self.usage[md5])
+
         if self.usage[md5] >= self.api_key_usage_threshold:
             switch_result, key_name = self.auto_switch()