From b240833c0cb0ac89eec9600e9749ab86952f715c Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Fri, 13 Mar 2026 00:09:08 -0700
Subject: [PATCH] [Fix] Add flaky marker to test_pdf_handling integration test

The test makes real API calls to OpenAI's o1 model which can crash
pytest-xdist workers in CI, especially the async variant.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 tests/llm_translation/base_llm_unit_tests.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/llm_translation/base_llm_unit_tests.py b/tests/llm_translation/base_llm_unit_tests.py
index d65735a6200..12df71839b2 100644
--- a/tests/llm_translation/base_llm_unit_tests.py
+++ b/tests/llm_translation/base_llm_unit_tests.py
@@ -347,6 +347,7 @@ class BaseLLMChatTest(ABC):
         assert response is not None
         print(f"response={response}")
 
+    @pytest.mark.flaky(retries=3, delay=1)
     @pytest.mark.parametrize("sync_mode", [True, False])
     @pytest.mark.asyncio
     async def test_pdf_handling(self, pdf_messages, sync_mode):