test: add more unit testing

This commit is contained in:
Krrish Dholakia 2025-04-22 12:46:45 -07:00
parent 74a109ee13
commit f76e42cd1a

View file

@ -11,6 +11,7 @@ sys.path.insert(
from litellm.litellm_core_utils.prompt_templates.common_utils import (
get_format_from_file_id,
handle_any_messages_to_chat_completion_str_messages_conversion,
update_messages_with_model_file_ids,
)
@ -64,3 +65,50 @@ def test_update_messages_with_model_file_ids():
],
}
]
def test_handle_any_messages_to_chat_completion_str_messages_conversion_list():
# Test with list of messages
messages = [
{"role": "user", "content": "Hello"},
{"role": "assistant", "content": "Hi there"},
]
result = handle_any_messages_to_chat_completion_str_messages_conversion(messages)
assert len(result) == 2
assert result[0]["input"] == json.dumps(messages[0])
assert result[1]["input"] == json.dumps(messages[1])
def test_handle_any_messages_to_chat_completion_str_messages_conversion_dict():
# Test with single dictionary message
message = {"role": "user", "content": "Hello"}
result = handle_any_messages_to_chat_completion_str_messages_conversion(message)
assert len(result) == 1
assert result[0]["input"] == json.dumps(message)
def test_handle_any_messages_to_chat_completion_str_messages_conversion_str():
# Test with string message
message = "Hello"
result = handle_any_messages_to_chat_completion_str_messages_conversion(message)
assert len(result) == 1
assert result[0]["input"] == message
def test_handle_any_messages_to_chat_completion_str_messages_conversion_other():
# Test with non-string/dict/list type
message = 123
result = handle_any_messages_to_chat_completion_str_messages_conversion(message)
assert len(result) == 1
assert result[0]["input"] == "123"
def test_handle_any_messages_to_chat_completion_str_messages_conversion_complex():
# Test with complex nested structure
message = {
"role": "user",
"content": {"text": "Hello", "metadata": {"timestamp": "2024-01-01"}},
}
result = handle_any_messages_to_chat_completion_str_messages_conversion(message)
assert len(result) == 1
assert result[0]["input"] == json.dumps(message)