xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-27 16:04:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: 加载usage时输出

Browse Source
This commit is contained in:
Rock Chin
2022-12-15 18:14:17 +08:00
parent cc11840dc7
commit e85422bf4e
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/openai/keymgr.py
+2 -1
View File
@@ -72,7 +72,7 @@ class KeysManager:
md5 = hashlib.md5(self.using_key.encode('utf-8')).hexdigest() md5 = hashlib.md5(self.using_key.encode('utf-8')).hexdigest()
if md5 not in self.usage: if md5 not in self.usage:
self.usage[md5] = 0 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: if self.usage[md5] >= self.api_key_usage_threshold:
switch_result, key_name = self.auto_switch() switch_result, key_name = self.auto_switch()
@@ -97,3 +97,4 @@ class KeysManager:
def load_usage(self): def load_usage(self):
self.usage = pkg.database.manager.get_inst().load_api_key_usage() self.usage = pkg.database.manager.get_inst().load_api_key_usage()
logging.debug("load usage:" + str(self.usage)) logging.debug("load usage:" + str(self.usage))
print("load usage:" + str(self.usage))