diff --git a/python/packages/azurefunctions/tests/integration_tests/test_03_reliable_streaming.py b/python/packages/azurefunctions/tests/integration_tests/test_03_reliable_streaming.py index c99198b602..b8c64263cb 100644 --- a/python/packages/azurefunctions/tests/integration_tests/test_03_reliable_streaming.py +++ b/python/packages/azurefunctions/tests/integration_tests/test_03_reliable_streaming.py @@ -55,12 +55,11 @@ class TestSampleReliableStreaming: # Wait a moment for the agent to start writing to Redis time.sleep(2) - # Stream response from Redis with shorter timeout - # Note: We use text/plain to avoid SSE parsing complexity + # Stream response from Redis with longer timeout to account for LLM latency stream_response = requests.get( f"{self.stream_url}/{thread_id}", headers={"Accept": "text/plain"}, - timeout=30, # Shorter timeout for test + timeout=60, ) assert stream_response.status_code == 200 @@ -82,7 +81,7 @@ class TestSampleReliableStreaming: stream_response = requests.get( f"{self.stream_url}/{thread_id}", headers={"Accept": "text/event-stream"}, - timeout=30, # Shorter timeout + timeout=60, ) assert stream_response.status_code == 200 content_type = stream_response.headers.get("content-type", "")