From 8aa1dd24ac18c20afa7c2d760addbf44c418784e Mon Sep 17 00:00:00 2001
From: Rock Chin <1010553892@qq.com>
Date: Mon, 2 Jan 2023 20:42:08 +0800
Subject: [PATCH] =?UTF-8?q?debug:=20=E6=B7=BB=E5=8A=A0=E4=B8=8A=E6=8A=A5?=
 =?UTF-8?q?=E8=B4=B9=E7=94=A8=E7=9A=84debug=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/openai/keymgr.py  | 2 ++
 pkg/openai/pricing.py | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/pkg/openai/keymgr.py b/pkg/openai/keymgr.py
index cf9898aa..a9d95243 100644
--- a/pkg/openai/keymgr.py
+++ b/pkg/openai/keymgr.py
@@ -78,6 +78,8 @@ class KeysManager:
         return self.fee[md5]
 
     def report_fee(self, fee: float) -> bool:
+        logging.debug("report fee:" + str(fee))
+
         md5 = hashlib.md5(self.using_key.encode('utf-8')).hexdigest()
         if md5 not in self.fee:
             self.fee[md5] = 0
diff --git a/pkg/openai/pricing.py b/pkg/openai/pricing.py
index 1040a04c..26455bb9 100644
--- a/pkg/openai/pricing.py
+++ b/pkg/openai/pricing.py
@@ -1,3 +1,5 @@
+import logging
+
 pricing = {
     "base": {  # 文字模型单位是1000字符
         "text-davinci-003": 0.02,
@@ -13,9 +15,11 @@ pricing = {
 def language_base_price(model, text):
     salt_rate = 0.93
     length = ((len(text.encode('utf-8')) - len(text)) / 2 + len(text)) * salt_rate
+    logging.debug("text length: %d" % length)
 
     return pricing["base"][model] * length / 1000
 
 
 def image_price(size):
+    logging.debug("image size: %s" % size)
     return pricing["image"][size]