From e85422bf4e2d849473ebd0a5f173d866bbb95086 Mon Sep 17 00:00:00 2001 From: Rock Chin <1010553892@qq.com> Date: Thu, 15 Dec 2022 18:14:17 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=8A=A0=E8=BD=BDusage=E6=97=B6?= =?UTF-8?q?=E8=BE=93=E5=87=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/openai/keymgr.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pkg/openai/keymgr.py b/pkg/openai/keymgr.py index 2c724a0b..8163999c 100644 --- a/pkg/openai/keymgr.py +++ b/pkg/openai/keymgr.py @@ -72,7 +72,7 @@ class KeysManager: md5 = hashlib.md5(self.using_key.encode('utf-8')).hexdigest() if md5 not in self.usage: self.usage[md5] = 0 - self.usage[md5] += int((len(new_content.encode('utf-8')) - len(new_content))/2 + len(new_content)) + self.usage[md5] += int((len(new_content.encode('utf-8')) - len(new_content)) / 2 + len(new_content)) if self.usage[md5] >= self.api_key_usage_threshold: switch_result, key_name = self.auto_switch() @@ -96,4 +96,5 @@ class KeysManager: def load_usage(self): self.usage = pkg.database.manager.get_inst().load_api_key_usage() - logging.debug("load usage:" + str(self.usage)) \ No newline at end of file + logging.debug("load usage:" + str(self.usage)) + print("load usage:" + str(self.usage))