phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-06 02:32:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: increase integration test timeout

Browse source 
To accomodate the slow CI :)

Signed-off-by: Sébastien Han <seb@redhat.com>
This commit is contained in:
Sébastien Han 2025-03-31 22:12:07 +02:00
parent 5da4839c5a
commit 014e3ad280
No known key found for this signature in database
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/integration/inference/test_text_inference.py
View file

@ -275,6 +275,7 @@ def test_text_chat_completion_first_token_profiling(client_with_models, text_mod
model_id=text_model_id, model_id=text_model_id,
messages=messages, messages=messages,
stream=False, stream=False,
timeout=120, # Increase timeout to 2 minutes for large conversation history
) )
message_content = response.completion_message.content.lower().strip() message_content = response.completion_message.content.lower().strip()
assert len(message_content) > 0 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, model_id=text_model_id,
messages=[{"role": "user", "content": question}], messages=[{"role": "user", "content": question}],
stream=True, 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] streamed_content = [str(chunk.event.delta.text.lower().strip()) for chunk in response]
assert len(streamed_content) > 0 assert len(streamed_content) > 0