Merge branch 'master' of github.com:zhayujie/chatgpt-on-wechat

zhayujie · Oct 19, 2024 · 0773174 · 0773174
2 parents 70e007d + fcc4d02
commit 0773174
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bot/chatgpt/chat_gpt_session.py b/bot/chatgpt/chat_gpt_session.py
@@ -57,7 +57,7 @@ def calc_tokens(self):
 def num_tokens_from_messages(messages, model):
     """Returns the number of tokens used by a list of messages."""
 
-    if model in ["wenxin", "xunfei", const.GEMINI]:
+    if model in ["wenxin", "xunfei"] or model.startswith(const.GEMINI):
         return num_tokens_by_character(messages)
 
     import tiktoken