idvdZddlmZddlZddlmZmZddlmZddl m Z m Z m Z ddl Zddl mZddlmZmZdd lmZmZdd lmZdd lmZdd lmZdd lmZddlmZm Z ddl!m"Z"ddl#m$Z$e rddl%m&Z&ddl'm(Z(ddl)m*Z*ddZ+GddeZ,GddeZ-GddeZ.GddeZ/GddeZ0y) alFastMCPProvider for wrapping FastMCP servers as providers. This module provides the `FastMCPProvider` class that wraps a FastMCP server and exposes its components through the Provider interface. It also provides FastMCPProvider* component classes that delegate execution to the wrapped server's middleware, ensuring middleware runs when components are executed. ) annotationsN) AsyncIteratorSequence)asynccontextmanager) TYPE_CHECKINGAnyoverload)AnyUrl)Prompt PromptResult)ResourceResourceResult)ResourceTemplate)Provider)TaskMeta) delegate_span)Tool ToolResult)FastMCPComponent) VersionSpec)Docket) Execution)FastMCPc|}|jD])\}}tjd|dt||}+|S)znExpand a URI template with parameters. Simple implementation that handles {name} style placeholders. z\{z\})itemsresubstr)templateparamsresultkeyvalues y/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/providers/fastmcp_provider.py_expand_uri_templater%&s@ Flln U3se3UV<% MceZdZUdZdZded<dZded< dfd ZeddZ e d dd Z e dd Z d dd Z dd Z dfd Z xZS)FastMCPProviderToolzTool that delegates execution to a wrapped server's middleware. When `run()` is called, this tool invokes the wrapped server's `_call_tool_middleware()` method, ensuring the server's middleware chain is executed. Nr_server str | None_original_namec @t|di|||_||_yNsuper__init__r)r+selfserver original_namekwargs __class__s r$r1zFastMCPProviderTool.__init__A$ "6" +r&c$|||j|j|j|j|j|j|j |j |j|j|j|j S)z=Wrap a Tool to delegate execution to the server's middleware.) r4r5nameversion description parameters output_schematagsr task_configmetatitleicons) r:r;r<r=r>r?rr@get_metarBrC)clsr4tools r$wrapzFastMCPProviderTool.wrapKsr))LL((,,((((****  r&c KywNr.r3 arguments task_metas r$_runzFastMCPProviderTool._run^s c KywrIr.rJs r$rMzFastMCPProviderTool._rune &)rNc6K|jrt|jnd}t|jxsdd|jxsd5|jj |j|||d{cdddS7 #1swYyxYww)zDelegate to child server's call_tool() with task_meta. Passes task_meta through to the child server so it can handle backgrounding appropriately. fn_key is already set by the parent server before calling this method. eqNFastMCPProviderr;rL)r;rrr+r) call_toolr3rKrLr;s r$rMzFastMCPProviderTool._runls37,,+.D     %2'8$:M:M:SQS //##Y90    0AB-B =B >B  B B  BBcK|jrt|jnd}|jj|j||d{}t |t jjr td|S75w)zDelegate to child server's call_tool() without task_meta. This is called when the tool is used within a TransformedTool forwarding function or other contexts where task_meta is not available. rRNr;zB ?B B BBBc6Kt|jxsd|}|jrt|jnd}|jj ||d{}t |tjjr td|S75w)zRead the resource content for background task execution. Reads the resource via the wrapped server and returns the ResourceResult. This method is called by Docket during background task execution. rTrRNr[z3Unexpected CreateTaskResult during Docket execution) r%rr;rr)rr\r]r^r_r`)r3rKr~r;r!s r$readz$FastMCPProviderResourceTemplate.reads,  ' ' -2y 37,,+.D||11,1PP fcii88 9TU U QsAB!B"6Bcy)zANo-op: the child's actual template is registered via get_tasks().Nr.)r3dockets r$register_with_docketz4FastMCPProviderResourceTemplate.register_with_docketsr&)fn_keytask_keyc~K|xs |j}|r||d<|j|fi|di|d{S7w)zSchedule this template for background execution via docket. The child's FunctionResourceTemplate.fn is registered (via get_tasks), and it expects splatted **kwargs, so we splat params here. r"Nr.)r"add)r3rr rrr6 lookup_keys r$ add_to_docketz-FastMCPProviderResourceTemplate.add_to_docketsH'txx $F5M5ZVZZ 5f5?????s 4=;=c@t|d|jdzS)NrU)rfz&fastmcp.delegate.original_uri_template)r0rhrris r$rhz3FastMCPProviderResourceTemplate.get_span_attributess)w*,%66:6Q6Q0   r&)r4rrrr6r)r4rrrrlr)rrr rmrlr rI)rrr rmrLrnrlr)rrr rmrLrrlro)rrr rmrLrprlr)rKrmrlzstr | bytes | ResourceResult)rrrlrn) rrr rmrr*rr*r6rrlrrq)rrrsrtrur)rvrr1rwrGrr rrrrrhrxrys@r$rrDskGS)-J-<< #< <  $4 (  * $BF .;? )) .);C) #)) NR .;J 4,*P"# @@@  @  @@ @&  r&rceZdZdZd fd ZddZ d ddZddZ d ddZddZ d ddZ dd Z d dd Z dd Z edd ZxZS)rUaProvider that wraps a FastMCP server. This provider enables mounting one FastMCP server onto another, exposing the mounted server's tools, resources, and prompts through the parent server. Components returned by this provider are wrapped in FastMCPProvider* classes that delegate execution to the wrapped server's middleware chain. This ensures middleware runs when components are executed. Example: ```python from fastmcp import FastMCP from fastmcp.server.providers import FastMCPProvider main = FastMCP("Main") sub = FastMCP("Sub") @sub.tool def greet(name: str) -> str: return f"Hello, {name}!" # Mount directly - tools accessible by original names main.add_provider(FastMCPProvider(sub)) # Or with namespace from fastmcp.server.transforms import Namespace provider = FastMCPProvider(sub) provider.add_transform(Namespace("sub")) main.add_provider(provider) ``` Note: Normally you would use `FastMCP.mount()` which handles proxy conversion and creates the provider with namespace automatically. c0t|||_y)zeInitialize a FastMCPProvider. Args: server: The FastMCP server to wrap. N)r0r1r4)r3r4r7s r$r1zFastMCPProvider.__init__s  r&cK|jjd{}|Dcgc]"}tj|j|$c}S72cc}ww)aList all tools from the mounted server as FastMCPProviderTools. Runs the mounted server's middleware so filtering/transformation applies. Wraps each tool as a FastMCPProviderTool that delegates execution to the nested server's middleware. N)r4 list_toolsr(rG)r3 raw_toolsts r$ _list_toolszFastMCPProvider._list_tools sL++0022 BKL)Q#((a8)LL3L AAA'AAAcK|jj||d{}|ytj|j|S7'w)aGet a tool by name as a FastMCPProviderTool. Passes the full VersionSpec to the nested server, which handles both exact version matching and range filtering. Uses get_tool to ensure the nested server's transforms are applied. N)r4get_toolr(rG)r3r:r;raw_tools r$ _get_toolzFastMCPProvider._get_toolsF--dG<<  "'' X>>= A A (A cK|jjd{}|Dcgc]"}tj|j|$c}S72cc}ww)aList all resources from the mounted server as FastMCPProviderResources. Runs the mounted server's middleware so filtering/transformation applies. Wraps each resource as a FastMCPProviderResource that delegates reading to the nested server's middleware. N)r4list_resourcesr{rG)r3 raw_resourcesrs r$_list_resourceszFastMCPProvider._list_resources'sL#kk88:: FSTm',,T[[!>sGLL  .33DKKNNMrcK|jjd{}|Dcgc]"}tj|j|$c}S72cc}ww)aList all prompts from the mounted server as FastMCPProviderPrompts. Runs the mounted server's middleware so filtering/transformation applies. Returns FastMCPProviderPrompt instances that delegate rendering to the wrapped server's middleware. N)r4 list_promptsrrG)r3 raw_promptsps r$ _list_promptszFastMCPProvider._list_promptsasL!KK4466 DOPKq%**4;;:KPP7PrcK|jj||d{}|ytj|j|S7'w)a Get a prompt by name as a FastMCPProviderPrompt. Passes the full VersionSpec to the nested server, which handles both exact version matching and range filtering. Uses get_prompt to ensure the nested server's transforms are applied. N)r4 get_promptrrG)r3r:r; raw_prompts r$ _get_promptzFastMCPProvider._get_promptksG ;;11$@@  $))$++zBBArcKt|jjd{}|Dcgc]}t|ts|}}|Dcgc]}t|t s|}}|Dcgc]}t|t s|}}|Dcgc]}t|ts|}}|jD]f}|j|d{}|j|d{}|j|d{}|j|d{}hg||||Dcgc]}|jjr| c}S70cc}wcc}wcc}wcc}w777l7Ucc}ww)aKReturn task-eligible components from the mounted server. Returns the child's ACTUAL components (not wrapped) so their actual functions get registered with Docket. Gets components with child server's transforms applied, then applies this provider's transforms for correct registration keys. N)listr4 get_tasksr\rr rr transformsrrrrr@supports_tasks)r3 componentsctools resources templatesprompts transforms r$rzFastMCPProvider.get_tasks}sz 5 5 778 '>Jq*Q*=J> *F 1jH.EQ F *N 1jrs# 3*//7?7202/92*- b $b JD hD N_ F_ DL &6L hVhVr&