diff --git a/tests/unit_tests/client/test_runner_utils.py b/tests/unit_tests/client/test_runner_utils.py index 4487657e3b215..2c0ccd4403376 100644 --- a/tests/unit_tests/client/test_runner_utils.py +++ b/tests/unit_tests/client/test_runner_utils.py @@ -176,12 +176,17 @@ async def mock_arun_chain( {"result": f"Result for example {example.id}"} for _ in range(n_repetitions) ] + def mock_create_session(*args: Any, **kwargs: Any) -> None: + pass + with mock.patch.object( LangChainPlusClient, "read_dataset", new=mock_read_dataset ), mock.patch.object( LangChainPlusClient, "list_examples", new=mock_list_examples ), mock.patch( "langchain.client.runner_utils._arun_llm_or_chain", new=mock_arun_chain + ), mock.patch.object( + LangChainPlusClient, "create_session", new=mock_create_session ): client = LangChainPlusClient(api_url="http://localhost:1984", api_key="123") chain = mock.MagicMock()