perf: keymgr对已储存的结果取整

2026-06-02 03:55:55 +00:00 · 2022-12-17 18:14:57 +08:00
parent 816bfc4e47
commit 8c588c5c1d
1 changed files with 2 additions and 0 deletions
--- 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()