i ?8dZddlmZddlmZmZmZddlmZm Z m Z m Z ddl Z ddlmZddl mZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZmZmZerddl m!Z!ee"Z#e de Z$e dZ% ddZ& ddZ'GddZ(y)zGMCP protocol handler setup and wire-format handlers for FastMCP Server.) annotations) AwaitableCallableSequence) TYPE_CHECKINGAnyTypeVarcastN)McpError) ContentBlock)AnyUrl) DisabledError NotFoundError)TaskMeta) get_logger)paginate_sequence) VersionSpecparse_version_keyversion_sort_key)FastMCPC)bound PaginateTc i}|D])}|j||gj|+g}|jD]}ttt |t }td|Drwt|Dcgc]}|j|jc}td}|jxsi}|jdi|di|jdid|iii}|j||Scc}w) aDeduplicate components by key, keeping highest version. Groups components by key, selects the highest version from each group, and injects available versions into meta if any component is versioned. Args: components: Sequence of components to deduplicate. key_fn: Function to extract the grouping key from a component. Returns: Deduplicated list with versions injected into meta. )keyc38K|]}|jduywN)version).0cs t/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/mixins/mcp_operations.py z(_dedupe_with_versions..3s7hqyy$hsT)rreversemetafastmcpversions)update) setdefaultappendvaluesr rmaxranysortedrrr$ model_copyget) componentskey_fnby_keyr resultr&highest all_versionsr$s r!_dedupe_with_versionsr6s "$F &)R(//2FMMO!S/?@A 7h7 7!$,FHq 0EHF%L <<%2D((!$"hhy"5$& $) G  g)$* M#Gs D D c | t|dfS t|||S#t$r9}ttj j dt||d}~wwxYw)zApply pagination to items, raising McpError for invalid cursors. If page_size is None, returns all items without pagination. Nicodemessage)listr ValueErrorr mcptypes ErrorDatastr)itemscursor page_sizees r!_apply_paginationrEIseE{D  P  :: Psyy**A*GHaOPs  A 4AA ceZdZdZd dZ d dZ ddZ ddZ ddZ ddZ ddZ dd Z dd Z y )MCPOperationsMixinaMixin providing MCP protocol handler setup and wire-format handlers. Note: Methods registered with SDK decorators (e.g., _list_tools_mcp, _call_tool_mcp) cannot use `self: FastMCP` type hints because the SDK's `get_type_hints()` fails to resolve FastMCP at runtime (it's only available under TYPE_CHECKING). When type hints fail to resolve, the SDK falls back to calling handlers with no arguments. These methods use untyped `self` to avoid this issue. c|jj|j|jj|j|jj |j |j|j|jjtjj<|jj|j|j|jj!|j"|jj%|j&|j)y)aRSet up core MCP protocol handlers. List handlers use SDK decorators that pass the request object to our handler (needed for pagination cursor). The SDK also populates caches like _tool_cache. Exception: list_resource_templates SDK decorator doesn't pass the request, so we register that handler directly. The call_tool decorator is from the SDK (supports CreateTaskResult + validate_input). The read_resource and get_prompt decorators are from LowLevelServer to add CreateTaskResult support until the SDK provides it natively. )validate_inputN) _mcp_server list_tools_list_tools_mcplist_resources_list_resources_mcp list_prompts_list_prompts_mcp_wrap_list_handler_list_resource_templates_mcprequest_handlersr=r>ListResourceTemplatesRequest call_toolstrict_input_validation_call_tool_mcp read_resource_read_resource_mcp get_prompt_get_prompt_mcp_setup_task_protocol_handlers)selfs r!_setup_handlersz"MCPOperationsMixin._setup_handlersds &##%d&:&:;)'')$*B*BC'%%'(>(>?  # #D$E$E F ))#))*P*PQ P""$2N2N"O     )&&()@)@A%##%d&:&:; **,cdfd }|S)z@Wrap a list handler to pass the request and return ServerResult.cnK|d{}tjj|S7#wr)r=r> ServerResult)requestr3handlers r!wrapperz6MCPOperationsMixin._wrap_list_handler..wrappers."7++F99))&1 1,s 53$5)rcrreturnzmcp.types.ServerResult)r]rdres ` r!rQz%MCPOperationsMixin._wrap_list_handlers  2r_cKtd|}tjd|jdt t |j d{d}|Dcgc]}|j|j }}|"|jr|jjnd}t|||j\}}tjj||S7cc}ww)z List all available tools, in the format expected by the low-level MCP server. Supports pagination when list_page_size is configured. r[z] Handler called: list_toolsNc|jSrnamets r!z4MCPOperationsMixin._list_tools_mcp..s QRQWQWr_rk)tools nextCursor)r loggerdebugrlr6r;rK to_mcp_toolparamsrBrE_list_page_sizer=r>ListToolsResult) r]rcserverrptool sdk_toolsrBpage next_cursors r!rLz"MCPOperationsMixin._list_tools_mcpsi& q %ABC%d1B1B1D+D&EGWXBGH%$T%%499%5% H&-%8W^^GNN ! !QU .iAWAWXkyy((t (LL,EHs%A C/C( C/"#C*A$C/*C/cKtd|}tjd|jdt t |j d{d}|Dcgc]'}|jt|j)}}|jr|jjnd}t|||j\}}tjj!||S7cc}ww)z List all available resources, in the format expected by the low-level MCP server. Supports pagination when list_page_size is configured. rriz ] Handler called: list_resourcesNc,t|jSr)r@uri)rs r!roz8MCPOperationsMixin._list_resources_mcp..s 3quu:r_)r) resourcesrq)r rrrsrlr6r;rMto_mcp_resourcer@rrurBrErvr=r>ListResourcesResult) r]rcrxrresource sdk_resourcesrBr{r|s r!rNz&MCPOperationsMixin._list_resources_mcpsi& q %EFG) v,,.. /1E MV LUH $ $X\\): $ ;I  +2..&&d- 66#9#9 kyy,,t ,TT/ s%A C6C/ C6",C1A"C61C6cKtd|}tjd|jdt t |j d{d}|Dcgc]}|j|j }}|jr|jjnd}t|||j\}}tjj||S7cc}ww)z List all available resource templates, in the format expected by the low-level MCP server. Supports pagination when list_page_size is configured. rriz)] Handler called: list_resource_templatesNc|jSr) uri_templaterms r!rozAMCPOperationsMixin._list_resource_templates_mcp..sANNr_) uriTemplate)resourceTemplatesrq)r rrrsrlr6r;list_resource_templatesto_mcp_templaterrurBrErvr=r>ListResourceTemplatesResult) r]rcrx templatestemplate sdk_templatesrBr{r|s r!rRz/MCPOperationsMixin._list_resource_templates_mcpsi& q %NOP) v5577 8:R & %  $ $1F1F $ G%  +2..&&d- 66#9#9 kyy44"{5  8 %A C-C& C-"#C(A"C-(C-cKtd|}tjd|jdt t |j d{d}|Dcgc]}|j|j }}|jr|jjnd}t|||j\}}tjj||S7cc}ww)z List all available prompts, in the format expected by the low-level MCP server. Supports pagination when list_page_size is configured. rriz] Handler called: list_promptsNc|jSrrk)ps r!roz6MCPOperationsMixin._list_prompts_mcp..sr_rk)promptsrq)r rrrsrlr6r;rO to_mcp_promptrurBrErvr=r>ListPromptsResult) r]rcrxrprompt sdk_promptsrBr{r|s r!rPz$MCPOperationsMixin._list_prompts_mcpsi& q %CDE' v**,, -/? MTTG&v+++=G T*1..&&d- !7!7 kyy**4K*PP-TrcRKtd|}tjd|jd|| d}d} |jj }|j r=|j jd}|jdijd}|jjrC|jj}|jd} t| jd  }|r t| nd} |j!||| | d{} t#| t$j&j(r| S| j+S#ttf$rYtwxYw7O#t,$r} t/d || d} ~ wt.$r} t/d || d} ~ wwxYww)a Handle MCP 'callTool' requests. Extracts task metadata from MCP request context and passes it explicitly to call_tool(). The tool's _run() method handles the backgrounding decision, ensuring middleware runs before Docket. Args: key: The name of the tool to call arguments: Arguments to pass to the tool Returns: Tool result or CreateTaskResult for background execution rriz&] Handler called: call_tool %s with %sNT exclude_noner%rttlreqr task_metazUnknown tool: )r rrrsrlrJrequest_contextr$ model_dumpr/ experimentalis_task task_metadatarAttributeError LookupErrorrrU isinstancer=r>CreateTaskResult to_mcp_resultrr) r]r argumentsrx version_strrctx meta_dict mcp_task_metatask_meta_dictrr3rDs r!rWz!MCPOperationsMixin._call_tool_mcps,i&  }B CS)  A'+K)-I ((8888 # 3 3 3 FI"+-- 2">"B"B9"MK##++$'$4$4$B$BM%2%=%=4%=%PN (^-?-?-F GI6Ak[1dG!++Y9,F&#))"<"<= '') )#K0   A. 89q @ A. 89q @ Asp2F'E,B8E2(E,E*)E,F'E,F'E'$E,&E''E,, F$5F F$FF$$F'c Ktd|}tjd|jd| d}d} |jj }|j rB|j jd}|jdxsi}|jd}|jjrC|jj}|jd} t| jd  }|r t| nd} |j!t#|| | d{} t%| t&j(j*r| S| j-|S#ttf$rY}wxYw7P#t.$r<} t1t&j(j3d dt#|| d} ~ wt4$r3} t1t&j(j3d d| | d} ~ wwxYww)aHandle MCP 'readResource' requests. Extracts task metadata from MCP request context and passes it explicitly to read_resource(). The resource's _read() method handles the backgrounding decision, ensuring middleware runs before Docket. Args: uri: The resource URI Returns: ReadResourceResult or CreateTaskResult for background execution rriz"] Handler called: read_resource %sNTrr%rrrrrizResource not found: r8)r rrrsrlrJrr$rr/rrrrrrrrXr@rr=r>rrrr r?r) r]rrxrrrr fastmcp_metarrrr3rDs r!rYz%MCPOperationsMixin._read_resource_mcp%si& q %GH#N$ &*K)-I ((8888 # 3 3 3 FI#,==#;#ArL"."2"29"=K##++$'$4$4$B$BM%2%=%=4%=%PN (^-?-?-F GI6Ak[1dG!//C'Y0F&#))"<"<= '', ,#K0    ##+?C|)L$     ##;OPQs9S#T  sp1G;E9B=E"60E9&E7')E9G;E9!G;"E41E93E44E99 G87F99 G8.G33G88G;c.Ktd|}tjd|jd|| d}d} |jj }|j r=|j jd}|jdijd}|jjrC|jj}|jd} t| jd  }|r t| nd} |j!||| | d{} t#| t$j&j(r| S| j+S#ttf$rYtwxYw7O#t,$r} t/d || d} ~ wt.$rwxYww)aHandle MCP 'getPrompt' requests. Extracts task metadata from MCP request context and passes it explicitly to render_prompt(). The prompt's _render() method handles the backgrounding decision, ensuring middleware runs before Docket. Args: name: The prompt name arguments: Prompt arguments Returns: GetPromptResult or CreateTaskResult for background execution rriz'] Handler called: get_prompt %s with %sNTrr%rrrrrzUnknown prompt: )r rrrsrlrJrr$rr/rrrrrrr render_promptrr=r>rto_mcp_prompt_resultrr) r]rlrrxrrrrrrrr3rDs r!r[z"MCPOperationsMixin._get_prompt_mcp]s i&  }C DdI  '+K)-I ((8888 # 3 3 3 FI"+-- 2">"B"B9"MK##++$'$4$4$B$BM%2%=%=4%=%PN (^-?-?-F GI6Ak[1dG!//iI0F&#))"<"<= ..0 0#K0   D"24( ;rsM"9944*";00:WW- H  Cs K ** * *ZP P PP( P"vvr_