Austin/QChatGPT
1
0
Fork 0
mirror of https://github.com/RockChinQ/QChatGPT.git synced 2024-11-16 03:32:33 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: text的使用量未上报

Browse Source
This commit is contained in:
RockChinQ 2023-07-31 14:28:48 +08:00
parent e26664aa51
commit 45d1c89e45
1 changed files with 7 additions and 0 deletions
Show Stats 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:
if resp['usage']['total_tokens'] > 0:
self.audit_mgr.report_text_model_usage(
model,
resp['usage']['total_tokens']
)
yield resp
def request_image(self, prompt) -> dict: