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

[Fix] text的使用量未上报
This commit is contained in:
Junyan Qin 2023-07-31 14:31:48 +08:00 committed by GitHub
commit 161b694f71
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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: