Fix remaining mock LLM objects missing supports_stop_words attribute

Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
Devin AI
2025-06-04 07:19:13 +00:00
parent bb72804e74
commit 755bb185c0
2 changed files with 2 additions and 0 deletions

View File

@@ -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(

View File

@@ -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(