@@ -63,27 +63,31 @@ def mount(self, server: MountedServer) -> None:
6363
6464 async def get_resources (self ) -> dict [str , Resource ]:
6565 """Get all registered resources, keyed by URI."""
66- return await self ._load_resources (via_server = False )
66+ return await self ._load_resources (apply_filtering = False )
6767
6868 async def get_resource_templates (self ) -> dict [str , ResourceTemplate ]:
6969 """Get all registered templates, keyed by URI template."""
70- return await self ._load_resource_templates (via_server = False )
70+ return await self ._load_resource_templates (apply_filtering = False )
7171
72- async def _load_resources (self , * , via_server : bool = False ) -> dict [str , Resource ]:
72+ async def _load_resources (
73+ self , * , apply_filtering : bool = False
74+ ) -> dict [str , Resource ]:
7375 """
74- The single, consolidated recursive method for fetching resources. The 'via_server '
76+ The single, consolidated recursive method for fetching resources. The 'apply_filtering '
7577 parameter determines the communication path.
7678
77- - via_server =False: Manager-to-manager path for complete, unfiltered inventory
78- - via_server =True: Server-to-server path for filtered MCP requests
79+ - apply_filtering =False: Manager-to-manager path for complete, unfiltered inventory
80+ - apply_filtering =True: Server-to-server path for filtered MCP requests
7981 """
8082 all_resources : dict [str , Resource ] = {}
8183
8284 for mounted in self ._mounted_servers :
8385 try :
84- if via_server :
86+ if apply_filtering :
8587 # Use the server-to-server filtered path
86- child_resources_list = await mounted .server ._list_resources ()
88+ child_resources_list = (
89+ await mounted .server ._list_resources_middleware ()
90+ )
8791 child_resources = {
8892 resource .key : resource for resource in child_resources_list
8993 }
@@ -123,22 +127,24 @@ async def _load_resources(self, *, via_server: bool = False) -> dict[str, Resour
123127 return all_resources
124128
125129 async def _load_resource_templates (
126- self , * , via_server : bool = False
130+ self , * , apply_filtering : bool = False
127131 ) -> dict [str , ResourceTemplate ]:
128132 """
129- The single, consolidated recursive method for fetching templates. The 'via_server '
133+ The single, consolidated recursive method for fetching templates. The 'apply_filtering '
130134 parameter determines the communication path.
131135
132- - via_server =False: Manager-to-manager path for complete, unfiltered inventory
133- - via_server =True: Server-to-server path for filtered MCP requests
136+ - apply_filtering =False: Manager-to-manager path for complete, unfiltered inventory
137+ - apply_filtering =True: Server-to-server path for filtered MCP requests
134138 """
135139 all_templates : dict [str , ResourceTemplate ] = {}
136140
137141 for mounted in self ._mounted_servers :
138142 try :
139- if via_server :
143+ if apply_filtering :
140144 # Use the server-to-server filtered path
141- child_templates = await mounted .server ._list_resource_templates ()
145+ child_templates = (
146+ await mounted .server ._list_resource_templates_middleware ()
147+ )
142148 else :
143149 # Use the manager-to-manager unfiltered path
144150 child_templates = (
@@ -179,14 +185,14 @@ async def list_resources(self) -> list[Resource]:
179185 """
180186 Lists all resources, applying protocol filtering.
181187 """
182- resources_dict = await self ._load_resources (via_server = True )
188+ resources_dict = await self ._load_resources (apply_filtering = True )
183189 return list (resources_dict .values ())
184190
185191 async def list_resource_templates (self ) -> list [ResourceTemplate ]:
186192 """
187193 Lists all templates, applying protocol filtering.
188194 """
189- templates_dict = await self ._load_resource_templates (via_server = True )
195+ templates_dict = await self ._load_resource_templates (apply_filtering = True )
190196 return list (templates_dict .values ())
191197
192198 def add_resource_or_template_from_fn (
@@ -492,7 +498,7 @@ async def read_resource(self, uri: AnyUrl | str) -> str | bytes:
492498 continue
493499
494500 try :
495- result = await mounted .server ._read_resource (key )
501+ result = await mounted .server ._read_resource_middleware (key )
496502 return result [0 ].content
497503 except NotFoundError :
498504 continue
0 commit comments