diff --git a/tests/llm_translation/base_llm_unit_tests.py b/tests/llm_translation/base_llm_unit_tests.py index bcdbe9e95..7e5113dfe 100644 --- a/tests/llm_translation/base_llm_unit_tests.py +++ b/tests/llm_translation/base_llm_unit_tests.py @@ -82,6 +82,10 @@ class BaseLLMChatTest(ABC): print(response) + # OpenAI guarantees that the JSON schema is returned in the content + # relevant issue: https://github.com/BerriAI/litellm/issues/6741 + assert response.choices[0].message.content is not None + @pytest.fixture def pdf_messages(self): import base64