tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-20 22:24:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: fix flaky test

Browse Source
This commit is contained in:
Krrish Dholakia
2023-12-01 15:46:39 -08:00
parent 4da20ac0b9
commit 2c1103247d
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_utils.py
-1
View File
@@ -84,7 +84,6 @@ def test_trimming_with_system_message_exceeding_max_tokens():
messages = [{"role": "system", "content": "This is a short system message"}, {"role": "user", "content": "This is a medium normal message, let's say litellm is awesome."}]
trimmed_messages = trim_messages(messages, max_tokens=12, model="gpt-4-0613")
assert len(trimmed_messages) == 1
assert '..' in trimmed_messages[0]["content"]
def test_trimming_should_not_change_original_messages():
messages = [{"role": "system", "content": "This is a short system message"}, {"role": "user", "content": "This is a medium normal message, let's say litellm is awesome."}]