mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 18:24:20 +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])
|
||||
|
||||
|
||||
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():
|
||||
# Test with single dictionary message
|
||||
message = {"role": "user", "content": "Hello"}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue