Spaces:
Running
Running
Jeremiah Lowin commited on
Commit ·
e2f3152
1
Parent(s): db65850
Fix missing proxy arg
Browse files- 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(
|
|
|
|
|
|
|
| 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 = {
|
|
|
|
|
|
|
|
|
|
| 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:
|