From 5c35be42fbc19ad021b07033559738ffbb6f297a Mon Sep 17 00:00:00 2001 From: cthomas Date: Mon, 13 Oct 2025 16:13:31 -0700 Subject: [PATCH] fix: increase delay for responses api to fix flake (#5391) --- tests/integration_test_send_message_v2.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/integration_test_send_message_v2.py b/tests/integration_test_send_message_v2.py index 3a756f4b..c56daffb 100644 --- a/tests/integration_test_send_message_v2.py +++ b/tests/integration_test_send_message_v2.py @@ -293,8 +293,8 @@ async def accumulate_chunks(chunks: List[Any], verify_token_streaming: bool = Fa return [m for m in messages if m is not None] -async def cancel_run_after_delay(client: AsyncLetta, agent_id: str): - await asyncio.sleep(0.5) +async def cancel_run_after_delay(client: AsyncLetta, agent_id: str, delay: float = 0.5): + await asyncio.sleep(delay) await client.agents.messages.cancel(agent_id=agent_id) @@ -625,7 +625,8 @@ async def test_tool_call( agent_state = await client.agents.modify(agent_id=agent_state.id, llm_config=llm_config) if cancellation == "with_cancellation": - _cancellation_task = asyncio.create_task(cancel_run_after_delay(client, agent_state.id)) + delay = 5 if llm_config.model == "gpt-5" else 0.5 # increase delay for responses api + _cancellation_task = asyncio.create_task(cancel_run_after_delay(client, agent_state.id, delay=delay)) if send_type == "step": response = await client.agents.messages.create(