refactor: Update prepare_messages_for_api tests to reflect changes in message handling

This commit is contained in:
2025-08-28 15:31:45 +07:00
parent 7785a8f26e
commit 1cb47f1d25

View File

@@ -216,31 +216,31 @@ class TestOpenAIUtils(unittest.TestCase):
self.assertEqual(untrimmed, short_text)
def test_prepare_messages_for_api(self):
# Test empty messages
# Test empty messages - should return empty list (no system message added)
empty_result = prepare_messages_for_api([])
self.assertEqual(len(empty_result), 1) # Should have system message
self.assertEqual(empty_result[0]["role"], "system") # Verify it's a system message
self.assertEqual(len(empty_result), 0) # Should be empty, no system message added
# Test regular messages
# Test regular messages - should return messages as-is (no system message added)
messages = [
{"role": "user", "content": "Hello"},
{"role": "assistant", "content": "Hi there!"},
{"role": "user", "content": "How are you?"}
]
result = prepare_messages_for_api(messages)
self.assertEqual(len(result), 4) # Should have system message + 3 original messages
self.assertEqual(len(result), 3) # Should have 3 original messages only
self.assertEqual(result[0]["role"], "user")
self.assertEqual(result[0]["content"], "Hello")
# Test with null content
# Test with null content - should filter out null content messages
messages_with_null = [
{"role": "user", "content": None},
{"role": "assistant", "content": "Response"}
]
result_fixed = prepare_messages_for_api(messages_with_null)
self.assertEqual(len(result_fixed), 2) # Should have system message + 1 valid message
# Verify the content is correct (system message + only the assistant message)
self.assertEqual(result_fixed[0]["role"], "system")
self.assertEqual(result_fixed[1]["role"], "assistant")
self.assertEqual(result_fixed[1]["content"], "Response")
self.assertEqual(len(result_fixed), 1) # Should have only 1 valid message (null filtered out)
# Verify the content is correct (only the assistant message)
self.assertEqual(result_fixed[0]["role"], "assistant")
self.assertEqual(result_fixed[0]["content"], "Response")
class TestCodeUtils(unittest.TestCase):
"""Test code utility functions"""