From 014e3ad2808601b71d03002d11c6a48ae8ed6eb1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?S=C3=A9bastien=20Han?= Date: Mon, 31 Mar 2025 22:12:07 +0200 Subject: [PATCH] fix: increase integration test timeout MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit To accomodate the slow CI :) Signed-off-by: Sébastien Han --- tests/integration/inference/test_text_inference.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/integration/inference/test_text_inference.py b/tests/integration/inference/test_text_inference.py index f558254e5..c1eec35f0 100644 --- a/tests/integration/inference/test_text_inference.py +++ b/tests/integration/inference/test_text_inference.py @@ -275,6 +275,7 @@ def test_text_chat_completion_first_token_profiling(client_with_models, text_mod model_id=text_model_id, messages=messages, stream=False, + timeout=120, # Increase timeout to 2 minutes for large conversation history ) message_content = response.completion_message.content.lower().strip() assert len(message_content) > 0 @@ -301,6 +302,7 @@ def test_text_chat_completion_streaming(client_with_models, text_model_id, test_ model_id=text_model_id, messages=[{"role": "user", "content": question}], stream=True, + timeout=120, # Increase timeout to 2 minutes for large conversation history ) streamed_content = [str(chunk.event.delta.text.lower().strip()) for chunk in response] assert len(streamed_content) > 0