Fix #177: Returning multiple tool results (#222)

* feat: allow lowlevel servers to return a list of resources

The resource/read message in MCP allows of multiple resources
to be returned. However, in the SDK we do not allow this. This
change is such that we allow returning multiple resource in
the lowlevel API if needed. However in FastMCP we stick to
one, since a FastMCP resource defines the mime_type in the decorator
and hence a resource cannot dynamically return different mime_typed resources.
It also is just the better default to only return one resource.
However in the lowlevel API we will allow this.

Strictly speaking this is not a BC break since the new return value
is additive, but if people subclassed server, it will break them.

* feat: lower the type requriements for call_tool to Iterable
This commit is contained in:
David Soria Parra
2025-02-20 21:31:26 +00:00
committed by GitHub
parent 2628e01f4b
commit b1942b31c4
7 changed files with 62 additions and 32 deletions

View File

@@ -3,7 +3,7 @@
import inspect
import json
import re
from collections.abc import AsyncIterator
from collections.abc import AsyncIterator, Iterable
from contextlib import (
AbstractAsyncContextManager,
asynccontextmanager,
@@ -236,7 +236,7 @@ class FastMCP:
for template in templates
]
async def read_resource(self, uri: AnyUrl | str) -> ReadResourceContents:
async def read_resource(self, uri: AnyUrl | str) -> Iterable[ReadResourceContents]:
"""Read a resource by URI."""
resource = await self._resource_manager.get_resource(uri)
@@ -245,7 +245,7 @@ class FastMCP:
try:
content = await resource.read()
return ReadResourceContents(content=content, mime_type=resource.mime_type)
return [ReadResourceContents(content=content, mime_type=resource.mime_type)]
except Exception as e:
logger.error(f"Error reading resource {uri}: {e}")
raise ResourceError(str(e))
@@ -649,7 +649,7 @@ class Context(BaseModel):
progress_token=progress_token, progress=progress, total=total
)
async def read_resource(self, uri: str | AnyUrl) -> ReadResourceContents:
async def read_resource(self, uri: str | AnyUrl) -> Iterable[ReadResourceContents]:
"""Read a resource by URI.
Args: