test: mark flaky test

This commit is contained in:
Krrish Dholakia 2024-11-22 00:09:16 +05:30
parent 1a6831e091
commit 48834c9b63

View file

@ -92,6 +92,7 @@ class BaseLLMChatTest(ABC):
# relevant issue: https://github.com/BerriAI/litellm/issues/6741
assert response.choices[0].message.content is not None
@pytest.mark.flaky(retries=6, delay=1)
def test_json_response_pydantic_obj(self):
litellm.set_verbose = True
from pydantic import BaseModel