mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 03:04:13 +00:00
test: add unit test
This commit is contained in:
parent
f76e42cd1a
commit
19c3beed93
1 changed files with 13 additions and 0 deletions
|
@ -79,6 +79,19 @@ def test_handle_any_messages_to_chat_completion_str_messages_conversion_list():
|
||||||
assert result[1]["input"] == json.dumps(messages[1])
|
assert result[1]["input"] == json.dumps(messages[1])
|
||||||
|
|
||||||
|
|
||||||
|
def test_handle_any_messages_to_chat_completion_str_messages_conversion_list_infinite_loop():
|
||||||
|
# Test that list handling doesn't cause infinite recursion
|
||||||
|
messages = [
|
||||||
|
{"role": "user", "content": "Hello"},
|
||||||
|
{"role": "assistant", "content": "Hi there"},
|
||||||
|
]
|
||||||
|
# This should complete without stack overflow
|
||||||
|
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():
|
def test_handle_any_messages_to_chat_completion_str_messages_conversion_dict():
|
||||||
# Test with single dictionary message
|
# Test with single dictionary message
|
||||||
message = {"role": "user", "content": "Hello"}
|
message = {"role": "user", "content": "Hello"}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue