fix test_json_response_format

This commit is contained in:
Ishaan Jaff 2024-11-14 10:58:00 -08:00
parent a866bdb01e
commit eaf5723b94

View file

@ -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