Jeremiah Lowin commited on
Commit
e2f3152
·
1 Parent(s): db65850

Fix missing proxy arg

Browse files
Files changed (1) hide show
  1. src/fastmcp/server/proxy.py +11 -4
src/fastmcp/server/proxy.py CHANGED
@@ -220,10 +220,14 @@ class FastMCPProxy(FastMCP):
220
 
221
  return list(resources.values())
222
 
223
- async def _list_resource_templates(self) -> list[ResourceTemplate]:
 
 
224
  templates = {
225
  template.uri_template: template
226
- for template in await super()._list_resource_templates()
 
 
227
  }
228
 
229
  async with self.client:
@@ -244,8 +248,11 @@ class FastMCPProxy(FastMCP):
244
 
245
  return list(templates.values())
246
 
247
- async def _list_prompts(self) -> list[Prompt]:
248
- prompts = {prompt.name: prompt for prompt in await super()._list_prompts()}
 
 
 
249
 
250
  async with self.client:
251
  try:
 
220
 
221
  return list(resources.values())
222
 
223
+ async def _list_resource_templates(
224
+ self, apply_middleware: bool = True
225
+ ) -> list[ResourceTemplate]:
226
  templates = {
227
  template.uri_template: template
228
+ for template in await super()._list_resource_templates(
229
+ apply_middleware=apply_middleware
230
+ )
231
  }
232
 
233
  async with self.client:
 
248
 
249
  return list(templates.values())
250
 
251
+ async def _list_prompts(self, apply_middleware: bool = True) -> list[Prompt]:
252
+ prompts = {
253
+ prompt.name: prompt
254
+ for prompt in await super()._list_prompts(apply_middleware=apply_middleware)
255
+ }
256
 
257
  async with self.client:
258
  try: