diff --git a/tests/acp/test_mcp_e2e.py b/tests/acp/test_mcp_e2e.py index 88e89acf..45aed78e 100644 --- a/tests/acp/test_mcp_e2e.py +++ b/tests/acp/test_mcp_e2e.py @@ -124,7 +124,7 @@ class TestMcpRegistrationE2E: mock_conn.request_permission = AsyncMock() acp_agent._conn = mock_conn - def mock_run_conversation(user_message, conversation_history=None, task_id=None): + def mock_run_conversation(user_message, conversation_history=None, task_id=None, **kwargs): """Simulate an agent turn that calls terminal, gets a result, then responds.""" agent = state.agent @@ -213,7 +213,7 @@ class TestMcpRegistrationE2E: mock_conn.request_permission = AsyncMock() acp_agent._conn = mock_conn - def mock_run(user_message, conversation_history=None, task_id=None): + def mock_run(user_message, conversation_history=None, task_id=None, **kwargs): agent = state.agent # Fire two tool calls if agent.tool_progress_callback: