From 3d4d1c734a88f808ac15d95a29520b1fa99e19f4 Mon Sep 17 00:00:00 2001
From: zwssunny <zwssunny@qq.com>
Date: Sun, 5 Mar 2023 09:43:59 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=BC=9A=E8=AF=9D=E8=B6=85?=
 =?UTF-8?q?=E9=95=BF=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bot/chatgpt/chat_gpt_bot.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/bot/chatgpt/chat_gpt_bot.py b/bot/chatgpt/chat_gpt_bot.py
index cbfc736..00588d0 100644
--- a/bot/chatgpt/chat_gpt_bot.py
+++ b/bot/chatgpt/chat_gpt_bot.py
@@ -6,6 +6,7 @@ from common.log import logger
 from common.expired_dict import ExpiredDict
 import openai
 import time
+import json
 
 if conf().get('expires_in_seconds'):
     user_session = ExpiredDict(conf().get('expires_in_seconds'))
@@ -139,7 +140,7 @@ class Session(object):
         if not max_tokens:
             # default 3000
             max_tokens = 1000
-            
+
         session = user_session.get(user_id)
         if session:
             # append conversation
@@ -156,7 +157,7 @@ class Session(object):
         for i in range(len(session)-1, -1, -1):
         # count tokens of conversation list
             history_conv = session[i]
-            tokens=history_conv.split()
+            tokens=json.dumps(history_conv).split()
             count += len(tokens)
             count_list.append(count)