From 755bb185c0b20fc465d07469aad337f0fe404282 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 4 Jun 2025 07:19:13 +0000 Subject: [PATCH] Fix remaining mock LLM objects missing supports_stop_words attribute MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: João --- tests/test_streaming.py | 1 + tests/test_streaming_integration.py | 1 + 2 files changed, 2 insertions(+) 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(