klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-08 05:39:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
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) 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]) 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) < 26
assert len(prompt) > 1 assert len(prompt) > 1
assert prompt[0] != copy_big_chat_message assert prompt[0] != copy_big_chat_message