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

2025-11-26 03:44:58 +08:00 · 2023-02-05 15:12:54 +08:00
parent 64c587c17d
commit b7ae1fa516
2 changed files with 5 additions and 3 deletions
--- a/pkg/openai/manager.py
+++ b/pkg/openai/manager.py
@@ -38,8 +38,10 @@ class OpenAIInteract:
            **config.completion_api_params
        )

+        logging.debug("OpenAI response: %s", response)
+
        self.audit_mgr.report_text_model_usage(config.completion_api_params['model'],
-                                               prompt + response['choices'][0]['text'])
+                                               response['usage']['total_tokens'])

        return response