phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test: mark flaky test

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-22 00:09:16 +05:30
parent 1a6831e091
commit 48834c9b63
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

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