diff --git a/tests/utilities/test_events.py b/tests/utilities/test_events.py index ba3ada98a..e1f621fbb 100644 --- a/tests/utilities/test_events.py +++ b/tests/utilities/test_events.py @@ -1,3 +1,4 @@ +import os from datetime import datetime from unittest.mock import Mock, patch @@ -49,6 +50,11 @@ from crewai.utilities.events.tool_usage_events import ( ToolUsageErrorEvent, ) +# Skip streaming tests when running in CI/CD environments +skip_streaming_in_ci = pytest.mark.skipif( + os.getenv("CI") is not None, reason="Skipping streaming tests in CI/CD environments" +) + base_agent = Agent( role="base_agent", llm="gpt-4o-mini", @@ -618,6 +624,7 @@ def test_llm_emits_call_failed_event(): assert received_events[0].error == error_message +@skip_streaming_in_ci @pytest.mark.vcr(filter_headers=["authorization"]) def test_llm_emits_stream_chunk_events(): """Test that LLM emits stream chunk events when streaming is enabled.""" @@ -642,6 +649,7 @@ def test_llm_emits_stream_chunk_events(): assert "".join(received_chunks) == response +@skip_streaming_in_ci @pytest.mark.vcr(filter_headers=["authorization"]) def test_llm_no_stream_chunks_when_streaming_disabled(): """Test that LLM doesn't emit stream chunk events when streaming is disabled."""