Small updates from PR feedback

This commit is contained in:
William Easton 2025-08-13 17:58:14 -05:00
commit 3c8ea1207d
No known key found for this signature in database
3 changed files with 8 additions and 6 deletions

View file

@ -879,7 +879,7 @@ class MCPConfigTransport(ClientTransport):
# if there's exactly one server, create a client for that server
elif len(self.config.mcpServers) == 1:
self.transport = list(self.config.mcpServers.values())[0].to_transport()
self._underlying_transport = self.transport
self._underlying_transports.append(self.transport)
# otherwise create a composite client
else:

View file

@ -90,10 +90,10 @@ class _TransformingMCPServerMixin(FastMCPBaseModel):
description="The tags to exclude in the proxy.",
)
def _to_server_and_transport(
def _to_server_and_underlying_transport(
self,
) -> tuple[FastMCP[Any], ClientTransport]:
"""Get the server and transport for the server."""
"""Turn the Transforming MCPServer into a FastMCP Server and also return the underlying transport."""
from fastmcp import FastMCP
transport: ClientTransport = super().to_transport() # pyright: ignore[reportUnknownMemberType, reportAttributeAccessIssue, reportUnknownVariableType]
@ -108,8 +108,10 @@ class _TransformingMCPServerMixin(FastMCPBaseModel):
return wrapped_mcp_server, transport
def to_transport(self) -> ClientTransport:
"""Get the transport for the server."""
return self._to_server_and_transport()[1]
"""Get the transport for the transforming MCP server."""
from fastmcp.client.transports import FastMCPTransport
return FastMCPTransport(mcp=self._to_server_and_underlying_transport()[0])
class StdioMCPServer(BaseModel):

View file

@ -32,7 +32,7 @@ def mcp_server_type_to_servers_and_transports(
transport: ClientTransport
if isinstance(mcp_server, TransformingRemoteMCPServer | TransformingStdioMCPServer):
server, transport = mcp_server._to_server_and_transport()
server, transport = mcp_server._to_server_and_underlying_transport()
else:
transport = mcp_server.to_transport()
server = FastMCP.as_proxy(backend=transport)