ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-28 02:10:38 +02:00
Code Issues Projects Releases Packages Wiki Activity

Fix missing proxy arg

Browse source
This commit is contained in:
Jeremiah Lowin 2025-06-18 21:57:19 -04:00
commit cafa11d3ad
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/fastmcp/server/proxy.py
View file

@ -220,10 +220,14 @@ class FastMCPProxy(FastMCP):
return list(resources.values())
async def _list_resource_templates(self) -> list[ResourceTemplate]:
async def _list_resource_templates(
self, apply_middleware: bool = True
) -> list[ResourceTemplate]:
templates = {
template.uri_template: template
for template in await super()._list_resource_templates()
for template in await super()._list_resource_templates(
apply_middleware=apply_middleware
)
}
async with self.client:
@ -244,8 +248,11 @@ class FastMCPProxy(FastMCP):
return list(templates.values())
async def _list_prompts(self) -> list[Prompt]:
prompts = {prompt.name: prompt for prompt in await super()._list_prompts()}
async def _list_prompts(self, apply_middleware: bool = True) -> list[Prompt]:
prompts = {
prompt.name: prompt
for prompt in await super()._list_prompts(apply_middleware=apply_middleware)
}
async with self.client:
try: