From f35fff4f6a24ea20daee1f4f870d1a4a10b3445d Mon Sep 17 00:00:00 2001 From: Migatu Date: Fri, 20 Oct 2023 19:22:08 +0200 Subject: [PATCH] Fix --- bot.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/bot.py b/bot.py index 9b1de6c..96b2a60 100644 --- a/bot.py +++ b/bot.py @@ -335,7 +335,7 @@ async def on_message(message): # *=========================================== Define Functions -def num_tokens_from_string(string, model): +def num_tokens_from_string(message, model): """ The function `num_tokens_from_string` returns the number of tokens in a given string. @@ -347,10 +347,9 @@ def num_tokens_from_string(string, model): tokens_per_name = 1 chat_gpt_encoding = tiktoken.encoding_for_model(model) - num_tokens = len(chat_gpt_encoding.encode(string)) num_tokens = 0 num_tokens += tokens_per_message - for keys, values in string.items(): + for keys, values in message.items(): num_tokens += len(chat_gpt_encoding.encode(values)) if keys == "role": num_tokens += tokens_per_name