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

fix: 修复错误的使用量计算方式

Browse Source
This commit is contained in:
Rock Chin
2023-02-05 15:12:54 +08:00
parent 64c587c17d
commit b7ae1fa516
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/audit/gatherer.py
+2 -2
View File
@@ -43,7 +43,7 @@ class DataGatherer:
def get_usage(self, key_md5): def get_usage(self, key_md5):
return self.usage[key_md5] if key_md5 in self.usage else {} return self.usage[key_md5] if key_md5 in self.usage else {}
def report_text_model_usage(self, model, text): def report_text_model_usage(self, model, total_tokens):
key_md5 = pkg.utils.context.get_openai_manager().key_mgr.get_using_key_md5() key_md5 = pkg.utils.context.get_openai_manager().key_mgr.get_using_key_md5()
if key_md5 not in self.usage: if key_md5 not in self.usage:
@@ -55,7 +55,7 @@ class DataGatherer:
if model not in self.usage[key_md5]["text"]: if model not in self.usage[key_md5]["text"]:
self.usage[key_md5]["text"][model] = 0 self.usage[key_md5]["text"][model] = 0
length = int((len(text.encode('utf-8')) - len(text)) / 2 + len(text)) length = total_tokens
self.usage[key_md5]["text"][model] += length self.usage[key_md5]["text"][model] += length
self.dump_to_db() self.dump_to_db()
pkg/openai/manager.py
+3 -1
View File
@@ -38,8 +38,10 @@ class OpenAIInteract:
**config.completion_api_params **config.completion_api_params
) )
logging.debug("OpenAI response: %s", response)
self.audit_mgr.report_text_model_usage(config.completion_api_params['model'], self.audit_mgr.report_text_model_usage(config.completion_api_params['model'],
prompt + response['choices'][0]['text']) response['usage']['total_tokens'])
return response return response