phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-28 04:04:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_completion.py): fix test to use mock response - prevent hang

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-07 07:05:32 -07:00
parent 48e743a81a
commit a4c20294e0
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_completion.py
View file

@ -2538,6 +2538,7 @@ def test_replicate_custom_prompt_dict():
"content": "what is yc write 1 paragraph", "content": "what is yc write 1 paragraph",
} }
], ],
mock_response="hello world",
repetition_penalty=0.1, repetition_penalty=0.1,
num_retries=3, num_retries=3,
) )