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

test: add retry on flaky test

Browse source
This commit is contained in:
Krrish Dholakia 2024-12-02 21:06:13 -08:00
parent 2fda90742f
commit aaa8c7caf0
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/llm_translation/base_llm_unit_tests.py
View file

@ -138,6 +138,7 @@ class BaseLLMChatTest(ABC):
except litellm.InternalServerError: except litellm.InternalServerError:
pytest.skip("Model is overloaded") pytest.skip("Model is overloaded")
@pytest.mark.flaky(retries=6, delay=1)
def test_json_response_format_stream(self): def test_json_response_format_stream(self):
""" """
Test that the JSON response format with streaming is supported by the LLM API Test that the JSON response format with streaming is supported by the LLM API