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

Parallelize list_* calls in Provider.get_tasks() (#2731)

Browse source
This commit is contained in:
Jeremiah Lowin 2025-12-25 12:16:44 -05:00 • committed by GitHub
commit 9f72096c67
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

35
src/fastmcp/server/providers/base.py
View file

@ -250,30 +250,19 @@ class Provider:
Used by the server during startup to register functions with Docket.
"""
from fastmcp.prompts.prompt import Prompt
from fastmcp.resources.resource import Resource
from fastmcp.resources.template import ResourceTemplate
from fastmcp.tools.tool import Tool
# Fetch all component types in parallel
tools, resources, templates, prompts = await gather(
self.list_tools(),
self.list_resources(),
self.list_resource_templates(),
self.list_prompts(),
)
components: list[FastMCPComponent] = []
for t in await self.list_tools():
if isinstance(t, Tool) and t.task_config.supports_tasks():
components.append(t)
for r in await self.list_resources():
if isinstance(r, Resource) and r.task_config.supports_tasks():
components.append(r)
for t in await self.list_resource_templates():
if isinstance(t, ResourceTemplate) and t.task_config.supports_tasks():
components.append(t)
for p in await self.list_prompts():
if isinstance(p, Prompt) and p.task_config.supports_tasks():
components.append(p)
return components
return [
c
for c in [*tools, *resources, *templates, *prompts]
if c.task_config.supports_tasks()
]
# -------------------------------------------------------------------------
# Lifecycle methods