From a4c20294e0633ab90e34fea10fdbbc1f0a0fbbad Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 7 Jun 2024 07:05:32 -0700 Subject: [PATCH] test(test_completion.py): fix test to use mock response - prevent hang --- 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..40bc31eaaf 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, )