kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-20 21:57:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: hello plugin support system prompt

Browse Source
This commit is contained in:
zhayujie
2023-11-29 17:41:14 +08:00
parent 6e0d2f9437
commit 865e4b5349
4 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
bot/linkai/link_ai_bot.py
View File

@@ -370,12 +370,9 @@ class LinkAISessionManager(SessionManager):
class LinkAISession(ChatGPTSession):
def calc_tokens(self):
try:
cur_tokens = super().calc_tokens()
except Exception as e:
logger.debug("Exception when counting tokens precisely for query: {}".format(e))
cur_tokens = len(str(self.messages))
return cur_tokens
if not self.messages:
return 0
return len(str(self.messages))
def discard_exceeding(self, max_tokens, cur_tokens=None):
cur_tokens = self.calc_tokens()