mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
Fix remaining mock LLM objects missing supports_stop_words attribute
Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
@@ -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(
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user