mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 18:24:20 +00:00
test: add more unit testing
This commit is contained in:
parent
74a109ee13
commit
f76e42cd1a
1 changed files with 48 additions and 0 deletions
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue