defend against no tokenizer

This commit is contained in:
Paul Gauthier 2023-11-14 13:09:50 -08:00
parent 6c77c32d06
commit 4aba307ebd

View file

@ -436,7 +436,11 @@ class Coder:
reminder_tokens = self.main_model.token_count(reminder_message) reminder_tokens = self.main_model.token_count(reminder_message)
cur_tokens = self.main_model.token_count(self.cur_messages) cur_tokens = self.main_model.token_count(self.cur_messages)
total_tokens = messages_tokens + reminder_tokens + cur_tokens if None not in (messages_tokens, reminder_tokens, cur_tokens):
total_tokens = messages_tokens + reminder_tokens + cur_tokens
else:
# add the reminder anyway
total_tokens = 0
# Add the reminder prompt if we still have room to include it. # Add the reminder prompt if we still have room to include it.
if total_tokens < self.main_model.max_context_tokens: if total_tokens < self.main_model.max_context_tokens: