diff --git a/tests/test_streaming.py b/tests/test_streaming.py index d0ee7543e..926cd90e9 100644 --- a/tests/test_streaming.py +++ b/tests/test_streaming.py @@ -187,6 +187,7 @@ def test_streaming_integration_with_llm(): with patch('crewai.llm.LLM') as mock_llm_class: mock_llm = Mock() mock_llm.call.return_value = "Here's a joke: Why did the robot cross the road? To get to the other side!" + mock_llm.supports_stop_words = True mock_llm_class.return_value = mock_llm agent = Agent( diff --git a/tests/test_streaming_integration.py b/tests/test_streaming_integration.py index f5c65ae47..044d84392 100644 --- a/tests/test_streaming_integration.py +++ b/tests/test_streaming_integration.py @@ -96,6 +96,7 @@ def test_streaming_with_multiple_agents(): with patch('crewai.llm.LLM') as mock_llm_class: mock_llm = Mock() mock_llm.call.return_value = "Agent response" + mock_llm.supports_stop_words = True mock_llm_class.return_value = mock_llm agent1 = Agent(