From ddc82774b04e70830f8a784cc5a259de2074adeb Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 6 Jun 2024 22:17:30 -0700 Subject: [PATCH] test(test_completion.py): fix replicate test which was hanging --- litellm/tests/test_completion.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 47c55ca4f3..9d1fa33b1e 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -2538,6 +2538,7 @@ def test_replicate_custom_prompt_dict(): "content": "what is yc write 1 paragraph", } ], + mock_response="Hello world", repetition_penalty=0.1, num_retries=3, )