sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add additional check for chat_messages length in UT

Browse source
This commit is contained in:
sabaimran 2023-08-01 09:25:52 -07:00
parent 3a5d93d673
commit 48363ec861
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_conversation_utils.py
View file

@ -56,7 +56,10 @@ class TestTruncateMessage:
prompt = utils.truncate_messages(chat_messages, self.max_prompt_size, self.model_name)
tokens = sum([len(self.encoder.encode(message.content)) for message in prompt])
# The original object has been modified. Verify certain properties
# The original object has been modified. Verify certain properties.
assert len(prompt) == (
len(chat_messages) + 1
) # Because the system_prompt is popped off from the chat_messages lsit
assert len(prompt) < 26
assert len(prompt) > 1
assert prompt[0] != copy_big_chat_message