Update to v.1.2.0 with tool filtering.

Author: neverinfamous

src/postgres_mcp/server.py

@@ -44,7 +44,6 @@
 from .statistics import StatisticalTools
 from .text import TextProcessingTools
 from .tool_filtering import filter_tools_from_server
-from .tool_filtering import is_tool_enabled
 from .top_queries import TopQueriesCalc
 from .vector import VectorTools
 

src/postgres_mcp/tool_filtering.py

@@ -206,7 +206,7 @@ def is_tool_enabled(name: str) -> bool:
     return name in included
 
 
-def filter_tools(tools: list["types.Tool"]) -> list["types.Tool"]:
+def filter_tools(tools: list[types.Tool]) -> list[types.Tool]:
     """Filter a list of tools based on environment configuration.
 
     Args:
@@ -263,8 +263,4 @@ def filter_tools_from_server(fastmcp_server: Any) -> None:
                 del tools_dict[tool_name]
                 logger.debug(f"Removed disabled tool: {tool_name}")
 
-            logger.info(
-                f"Tool filtering applied: {len(tools_dict)} tools remaining "
-                f"({len(tools_to_remove)} removed)"
-            )
-
+            logger.info(f"Tool filtering applied: {len(tools_dict)} tools remaining ({len(tools_to_remove)} removed)")

tests/unit/test_tool_filtering.py

@@ -13,16 +13,14 @@
 
 import pytest
 
-from postgres_mcp.tool_filtering import (
-    ALL_TOOLS,
-    TOOL_GROUPS,
-    clear_cache,
-    filter_tools,
-    get_all_tool_names,
-    get_available_groups,
-    get_included_tools,
-    is_tool_enabled,
-)
+from postgres_mcp.tool_filtering import ALL_TOOLS
+from postgres_mcp.tool_filtering import TOOL_GROUPS
+from postgres_mcp.tool_filtering import clear_cache
+from postgres_mcp.tool_filtering import filter_tools
+from postgres_mcp.tool_filtering import get_all_tool_names
+from postgres_mcp.tool_filtering import get_available_groups
+from postgres_mcp.tool_filtering import get_included_tools
+from postgres_mcp.tool_filtering import is_tool_enabled
 
 
 class TestToolFilteringBase:
@@ -194,9 +192,7 @@ def test_disable_multiple_groups(self, groups: list[str]) -> None:
         result = get_included_tools()
 
         for group in groups:
-            assert not (
-                result & TOOL_GROUPS[group]
-            ), f"{group} tools should be excluded"
+            assert not (result & TOOL_GROUPS[group]), f"{group} tools should be excluded"
 
     def test_disable_all_groups_results_in_empty_set(self) -> None:
         """Disabling all groups should result in empty set."""
@@ -478,4 +474,3 @@ def test_cicd_mode(self) -> None:
 
 if __name__ == "__main__":
     pytest.main([__file__, "-v"])
-