fix: text的使用量未上报

2025-11-26 03:44:58 +08:00 · 2023-07-31 14:28:48 +08:00
parent e26664aa51
commit 45d1c89e45
1 changed files with 7 additions and 0 deletions
--- a/pkg/openai/manager.py
+++ b/pkg/openai/manager.py
@@ -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: