Skip to content

Commit 86d7d76

Browse files
committed
Update test mocks for mini agents to reference the correct class path
1 parent 12e59a5 commit 86d7d76

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/praisonai/tests/unit/agent/test_mini_agents_sequential.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@ def __init__(self, content):
2626
return MockResponse("84")
2727
return MockResponse("mock response")
2828

29-
@patch('praisonaiagents.llms.PraisonAIModel.chat', side_effect=mock_completion)
30-
@patch('praisonaiagents.llms.PraisonAIModel.stream_chat', side_effect=mock_completion)
29+
@patch('praisonaiagents.agent.Agent.llm.chat', side_effect=mock_completion)
30+
@patch('praisonaiagents.agent.Agent.llm.stream_chat', side_effect=mock_completion)
3131
def test_mini_agents_sequential_data_passing(mock_stream, mock_chat):
3232
"""Test that output from previous task is passed to next task in Mini Agents"""
3333

0 commit comments

Comments
 (0)