From 2a83ae3de131ce5eb19a031a428a5f0da0abc63f Mon Sep 17 00:00:00 2001 From: vasilije Date: Sun, 18 Jan 2026 12:47:39 +0000 Subject: [PATCH] added fixes to cli --- cognee/tests/unit/api/v1/config/test_config_set.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/cognee/tests/unit/api/v1/config/test_config_set.py b/cognee/tests/unit/api/v1/config/test_config_set.py index 40918f9fe..b2e04f810 100644 --- a/cognee/tests/unit/api/v1/config/test_config_set.py +++ b/cognee/tests/unit/api/v1/config/test_config_set.py @@ -156,11 +156,14 @@ class TestConfigSet: """Test setting system root directory""" test_dir = "/tmp/test" - with patch("cognee.api.v1.config.config.get_base_config") as mock_get_base_config, \ - patch("cognee.api.v1.config.config.get_relational_config") as mock_get_relational_config, \ - patch("cognee.api.v1.config.config.get_graph_config") as mock_get_graph_config, \ - patch("cognee.api.v1.config.config.get_vectordb_config") as mock_get_vectordb_config: - + with ( + patch("cognee.api.v1.config.config.get_base_config") as mock_get_base_config, + patch( + "cognee.api.v1.config.config.get_relational_config" + ) as mock_get_relational_config, + patch("cognee.api.v1.config.config.get_graph_config") as mock_get_graph_config, + patch("cognee.api.v1.config.config.get_vectordb_config") as mock_get_vectordb_config, + ): mock_base_config = MagicMock() mock_base_config.system_root_directory = "" mock_get_base_config.return_value = mock_base_config