o id@sddlmZddlZddlZddlmZmZddlmZddl m Z m Z m Z ddl mZddlZddlmZddlmZdd lmZmZdd lmZdd lmZmZmZmZdd lmZddlZdd l m!Z!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,m-Z-m.Z.ddl/m0Z0ddl1m2Z2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9m:Z:ddl;mZ>ddl?m@Z@ddlAmBZBddlCmDZDmEZEddlFmGZGddlHmIZIddlJmKZKdd lLmMZMe rddlNm>Z>eMeOZPege!fegee!fBZQGd!d"d"ZRGd#d$d$eGeRZSGd%d&d&e) annotationsN) AwaitableCallable)Path) TYPE_CHECKINGAnycastquote) ServerSession) ClientSession)LifespanContextTRequestContext)McpError)METHOD_NOT_FOUNDBlobResourceContentsGetPromptResultTextResourceContents)AnyUrl)ClientFastMCP1Server) ElicitResult) LogMessage) RootsList)ClientTransportT) NotFoundError ResourceError ToolError) MCPConfig)Prompt PromptMessage)PromptArgument) PromptManager)ResourceResourceTemplate)ResourceManager)Context) get_context)FastMCP)Tool ToolResult) ToolManager)apply_transformations_to_tools)MirroredComponent) get_loggerc@s$eZdZUdZded<d ddZdS) ProxyManagerMixinzHA mixin for proxy managers to provide a unified client retrieval method.ClientFactoryTclient_factoryreturnrcs"|}t|r|IdH}|S)zwYYdSw)zFRenders a prompt, trying local/mounted first, then proxy if not found.N)rD render_promptrr9 get_prompt)r5rOrar6rgrGr7r8rs 6z ProxyPromptManager.render_promptrh)r2r)r2rN)rOr`rarr2r) r:r;r<r=rErrrrir7r7rGr8rs rcs>eZdZdZdfdd Zedd d Z ddddZZS)rPzH A Tool that represents and executes a tool on a remote server. r6rr@rc rArBrDrE_clientr5r6r@rGr7r8rErIzProxyTool.__init__mcp_toolmcp.types.Toolr2c Cs<|||j|j|j|j|j|j|jpididgdd S)z@Factory method to create a ProxyTool from a raw MCP tool schema._fastmcptagsT) r6rO description parametersr output_schemametar _mirrored)rOr inputSchemar outputSchemarget)clsr6rr7r7r8rQ szProxyTool.from_mcp_toolNrarbcontextContext | Noner*c s|j4IdH|jj|j|dIdH}WdIdHn 1IdHs%wY|jr:tttjj|j dj t |j |j dS)z6Executes the tool by making a call through the client.N)rOrarrc) r call_tool_mcprOisErrorrrmcptypes TextContentrdrr*structuredContent)r5rarrgr7r7r8runs(z ProxyTool.runr6rr@r)r6rrrr2rPr)rarbrrr2r*) r:r;r<r=rE classmethodrQrrir7r7rGr8rPs rPcsVeZdZUdZded<dZded<dddfd d ZedddZdddZ Z S)rozO A Resource that represents and reads a resource from a remote server. rrNstr | bytes | None_value)rr6c s"tjdi|||_||_dSrB)rDrErr)r5r6rr@rGr7r8rE5s zProxyResource.__init__ mcp_resourcemcp.types.Resourcer2c Cs<|||j|j|j|jp d|j|jpididgddS)zHFactory method to create a ProxyResource from a raw MCP resource schema. text/plainrrT)r6rnrOr mime_typerrr)rnrOrmimeTyperr)rr6rr7r7r8rp@szProxyResource.from_mcp_resourcerc s|jdur |jS|j4IdH|j|jIdH}WdIdHn 1IdHs+wYt|dtr<|djSt|dtrH|djSt dt |d)z1Read the resource content from the remote server.Nrr) rrrrnrrrrrrr)r5rgr7r7r8readSs (  zProxyResource.read)r6rrr)r6rrrr2ro)r2r) r:r;r<r=r>rrErrprrir7r7rGr8ro-s   rocs>eZdZdZdfdd Zedd d Z ddddZZS)ryza A ResourceTemplate that represents and creates resources from a remote server template. r6rr@rc rArBrrrGr7r8rEgrIzProxyTemplate.__init__ mcp_templatemcp.types.ResourceTemplater2c Cs>|||j|j|j|jp di|j|jpididgdd S)zHFactory method to create a ProxyTemplate from a raw MCP template schema.rrrT) r6 uri_templaterOrrrrrr)rxrOrrrr)rr6rr7r7r8rzkszProxyTemplate.from_mcp_templateNrnr`paramsrbrrroc s|jjd idd|D}|j4IdH|j|IdH}WdIdHn 1IdHs2wYt|dtrD|dj}nt|dtrQ|dj }n t dt |dt |j||j |j|dj|j|jpnididg|dS) zACreate a resource from the template by calling the remote server.cSsi|] \}}|t|ddqS))safer ).0kvr7r7r8 sz1ProxyTemplate.create_resource..Nrrrr)r6rnrOrrrrrr7)rformatitemsrrrrrrrrrrorOrrrr)r5rnrrparameterized_urirgvaluer7r7r8create_resource|s, (  zProxyTemplate.create_resourcer)r6rrrr2ryr)rnr`rrbrrr2ro) r:r;r<r=rErrzrrir7r7rGr8rybs rycsDeZdZUdZded<dfdd Zedd d ZdddZZ S)rzM A Prompt that represents and renders a prompt from a remote server. rrr6c rArBrrrGr7r8rErIzProxyPrompt.__init__ mcp_promptmcp.types.Promptr2c CsFdd|jpgD}|||j|j||j|jpididgddS)zDFactory method to create a ProxyPrompt from a raw MCP prompt schema.cSs$g|]}t|j|j|jp ddqS)F)rOrrequired)r!rOrr)rargr7r7r8 sz/ProxyPrompt.from_mcp_prompt..rrT)r6rOrrarrr)rarOrrr)rr6rrar7r7r8rszProxyPrompt.from_mcp_promptrarblist[PromptMessage]c s\|j4IdH|j|j|IdH}WdIdH|jS1IdHs&wY|jS)z6Render the prompt by making a call through the client.N)rrrOmessages)r5rargr7r7r8renderszProxyPrompt.render)r6r)r6rrrr2r)rarbr2r) r:r;r<r=r>rErrrrir7r7rGr8rs  rcs,eZdZdZ d ddd fdd ZZS) FastMCPProxyz A FastMCP server that acts as a proxy to a remote MCP-compliant server. It uses specialized managers that fulfill requests via a client factory. Nr1r6 Client | Noner1ClientFactoryT | Nonec stjd i|dur|durtddur/tjjr%tjdtddfdd}||_ n |dur7||_ ntdt |j |j j d |_ t |j d |_t|j d |_dS) a, Initializes the proxy server. FastMCPProxy requires explicit session management via client_factory. Use FastMCP.as_proxy() for convenience with automatic session strategy. Args: client: [DEPRECATED] A Client instance. Use client_factory instead for explicit session management. When provided, a client_factory will be automatically created that provides session isolation for backwards compatibility. client_factory: A callable that returns a Client instance when called. This gives you full control over session creation and reuse. Can be either a synchronous or asynchronous function. **kwargs: Additional settings for the FastMCP server. Nz1Cannot specify both 'client' and 'client_factory'zPassing 'client' to FastMCPProxy is deprecated. Use 'client_factory' instead for explicit session management. For automatic session strategy, use FastMCP.as_proxy().) stacklevelcsSr)newr7r6r7r8deprecated_client_factorysz8FastMCPProxy.__init__..deprecated_client_factoryzMust specify 'client_factory')r1rLrr7)rDrE ValueErrorfastmcpsettingsdeprecation_warningswarningswarnDeprecationWarningr1r? _tool_managerrLrj_resource_managerr_prompt_manager)r5r6r1r@rrGrr8rEs. zFastMCPProxy.__init__r)r6rr1r)r:r;r<r=rErir7r7rGr8rs rr/RequestContext[ClientSession, LifespanContextT]r2rcst}|IdHS)z A handler that forwards the list roots request from the remote server to the proxy's connected clients and relays the response back to the remote server. N)r' list_roots)rctxr7r7r8default_proxy_roots_handler srcsZeZdZdZd#fdd Zed$ddZed%ddZed&ddZed'd!d"Z Z S)( ProxyClientz A proxy client that forwards advanced interactions between a remote MCP server and the proxy's connected clients. Supports forwarding roots, sampling, elicitation, logging, and progress. transportcClientTransportT | FastMCP[Any] | FastMCP1Server | AnyUrl | Path | MCPConfig | dict[str, Any] | strc sd|vr ||d<d|vrt|d<d|vrtj|d<d|vr$tj|d<d|vr-tj|d<d|vr6tj|d<tjdi|d|iBdS) NrOrootssampling_handlerelicitation_handler log_handlerprogress_handlerrr7) generate_namerrdefault_sampling_handlerdefault_elicitation_handlerdefault_log_handlerdefault_progress_handlerrDrE)r5rr@rGr7r8rEs     zProxyClient.__init__rlist[mcp.types.SamplingMessage]r$mcp.types.CreateMessageRequestParamsrrr2mcp.types.CreateMessageResultcs^t}|jt||j|j|j|jdIdH}t|tj j tj j Br&t dtj j dd|dS)z A handler that forwards the sampling request from the remote server to the proxy's connected clients and relays the response back to the remote server. ) system_prompt temperature max_tokensmodel_preferencesNzContent is not supported assistantzfastmcp-client)rolemodelrd)r'sampler[ systemPromptr maxTokensmodelPreferencesrrr ResourceLinkEmbeddedResource RuntimeErrorCreateMessageResult)rrrrrrdr7r7r8r5s   z$ProxyClient.default_sampling_handlermessager` response_typermcp.types.ElicitRequestParamsrcs4t}|jj||j|jdIdH}t|j|jdS)z A handler that forwards the elicitation request from the remote server to the proxy's connected clients and relays the response back to the remote server. )r requestedSchemarelated_request_idN)actionrd)r'sessionelicitr  request_idrrrd)rr r rrrrgr7r7r8rOs  z'ProxyClient.default_elicitation_handlerrNonecs@t}|jd}|jd}|j||j|j|dIdHdS)zw A handler that forwards the log notification from the remote server to the proxy's connected clients. msgextra)level logger_namerN)r'datarlogrlogger)rr rrrr7r7r8rbs    zProxyClient.default_log_handlerprogressfloattotal float | None str | Nonecs t}||||IdHdS)z| A handler that forwards the progress notification from the remote server to the proxy's connected clients. N)r'report_progress)rrrr rr7r7r8rls z$ProxyClient.default_progress_handler)rr)rrrrrrr2r) r r`r rrr rrr2r)r rr2r)rrrrr rr2r) r:r;r<r=rErrrrrrir7r7rGr8rs    rcs>eZdZdZdfdd Zdd d Zd d ZdddZZS)StatefulProxyClienta A proxy client that provides a stateful client factory for the proxy server. The stateful proxy client bound its copy to the server session. And it will be disconnected when the session is exited. This is useful to proxy a stateful mcp server such as the Playwright MCP server. Note that it is essential to ensure that the proxy server itself is also stateful. argsrr@cstj|i|i|_dSr)rDrE_caches)r5r!r@rGr7r8rErIzStatefulProxyClient.__init__r2rcsdS)z The stateful proxy client will be forced disconnected when the session is exited. So we do nothing here. Nr7)r5exc_type exc_value tracebackr7r7r8 __aexit__szStatefulProxyClient.__aexit__cs6|jr|j\}}|jddIdH|jsdSdS)zE Clear all cached clients and force disconnect them. TforceN)r"popitem _disconnect)r5_cacher7r7r8clears zStatefulProxyClient.clearClient[ClientTransportT]csdtjjddur0tdj<fdd}j|S)z Create a new stateful proxy client instance with the same configuration. Use this method as the client factory for stateful proxy server. Nz created for cs4jtdjddIdHdS)Nz will be disconnectTr')r"poprdebugr*r7 proxy_clientr5rr7r8_on_session_exits z:StatefulProxyClient.new_stateful.._on_session_exit) r'rr"rrrr0 _exit_stackpush_async_callback)r5r3r7r1r8 new_statefuls  z StatefulProxyClient.new_stateful)r!rr@r)r2r)r2r.) r:r;r<r=rEr&r-r6rir7r7rGr8r zs  r )rrr2r)^ __future__rr3rcollections.abcrrpathlibrtypingrrr urllib.parser mcp.typesrr mcp.client.sessionr mcp.shared.contextr rmcp.shared.exceptionsrrrrrpydantic.networksrrfastmcp.client.clientrrfastmcp.client.elicitationrfastmcp.client.loggingrfastmcp.client.rootsrfastmcp.client.transportsrfastmcp.exceptionsrrrfastmcp.mcp_configrfastmcp.promptsrr fastmcp.prompts.promptr!fastmcp.prompts.prompt_managerr"fastmcp.resourcesr#r$"fastmcp.resources.resource_managerr%fastmcp.server.contextr&fastmcp.server.dependenciesr'fastmcp.server.serverr(fastmcp.tools.toolr)r*fastmcp.tools.tool_managerr+fastmcp.tools.tool_transformr,fastmcp.utilities.componentsr-fastmcp.utilities.loggingr.fastmcp.serverr:rr0r/r?rjrrProryrrrrr r7r7r7r8sf                        6S4+5=) C e