From 48363ec8616f6cb67a60d08796d77d5d33d4946b Mon Sep 17 00:00:00 2001 From: sabaimran Date: Tue, 1 Aug 2023 09:25:52 -0700 Subject: [PATCH] Add additional check for chat_messages length in UT --- tests/test_conversation_utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_conversation_utils.py b/tests/test_conversation_utils.py index 8fc9e127..3cfddc90 100644 --- a/tests/test_conversation_utils.py +++ b/tests/test_conversation_utils.py @@ -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