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

perf: 根据测试修改keymgr统计的文字量比例

Browse Source
This commit is contained in:
Rock Chin
2022-12-17 17:21:20 +08:00
parent 595c907566
commit a2d34b53de
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/openai/keymgr.py
+5 -1
View File
@@ -72,7 +72,11 @@ 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))
# 经测算得出的理论与实际的偏差比例
salt_rate = 0.94
self.usage[md5] += int((len(new_content.encode('utf-8')) - len(new_content)) / 2 + len(new_content))*salt_rate
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()