injectlib/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2025-11-25 19:37:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #521 from RockChinQ/fix-usage-not-reported

Browse Source 
[Fix] text的使用量未上报
This commit is contained in:
Junyan Qin
2023-07-31 14:31:48 +08:00
committed by GitHub
parent e26664aa51 45d1c89e45
commit 161b694f71
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/openai/manager.py
View File

@@ -52,6 +52,13 @@ class OpenAIInteract:
# 请求接口 # 请求接口
for resp in request: for resp in request:
if resp['usage']['total_tokens'] > 0:
self.audit_mgr.report_text_model_usage(
model,
resp['usage']['total_tokens']
)
yield resp yield resp
def request_image(self, prompt) -> dict: def request_image(self, prompt) -> dict: