diff --git a/tests/local_testing/test_text_completion.py b/tests/local_testing/test_text_completion.py index 19588a9720..cf78b4b3d6 100644 --- a/tests/local_testing/test_text_completion.py +++ b/tests/local_testing/test_text_completion.py @@ -3969,14 +3969,15 @@ def test_completion_hf_prompt_array(): def test_text_completion_stream(): try: - response = text_completion( - model="huggingface/mistralai/Mistral-7B-v0.1", - prompt="good morning", - stream=True, - max_tokens=10, - ) - for chunk in response: - print(f"chunk: {chunk}") + for _ in range(2): # check if closed client used + response = text_completion( + model="huggingface/mistralai/Mistral-7B-v0.1", + prompt="good morning", + stream=True, + max_tokens=10, + ) + for chunk in response: + print(f"chunk: {chunk}") except Exception as e: pytest.fail(f"GOT exception for HF In streaming{e}")