Mason Daugherty
2025-10-30 16:07:58 -04:00
committed by GitHub
parent e00b7233cf
commit dc5b7dace8
3 changed files with 3 additions and 0 deletions

View File

@@ -1003,6 +1003,7 @@ def test_audio_input_modality() -> None:
assert "audio" in output.additional_kwargs
@pytest.mark.flaky(retries=3, delay=1)
def test_prediction_tokens() -> None:
code = dedent(
"""

View File

@@ -88,6 +88,7 @@ class TestOpenAIStandard(ChatModelIntegrationTests):
# For now, we test with filename in OpenAI-specific tests
return False
@pytest.mark.flaky(retries=3, delay=1)
def test_openai_pdf_inputs(self, model: BaseChatModel) -> None:
"""Test that the model can process PDF inputs."""
url = "https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf"

View File

@@ -54,6 +54,7 @@ class TestOpenAIResponses(TestOpenAIStandard):
input_ = "What was the 3rd highest building in 2000?"
return _invoke(llm, input_, stream)
@pytest.mark.flaky(retries=3, delay=1)
def test_openai_pdf_inputs(self, model: BaseChatModel) -> None:
"""Test that the model can process PDF inputs."""
super().test_openai_pdf_inputs(model)