$j3<dZddlmZddlmZmZmZddlmZm Z m Z m Z ddl Z ddlmZddl mZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZmZerddlm Z ee!Z"e dZ#ddZ$GddZ%dS)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) VersionSpecdedupe_with_versions)FastMCP PaginateTitemsSequence[PaginateT]cursor str | None page_size int | Nonereturn"tuple[list[PaginateT], str | None]c |t|dfS t|||S#t$rA}ttjdt||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)rrres l/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/server/mixins/mcp_operations.py_apply_paginationr+sE{{D  P  ::: PPPsy**A*GGHHaOPs& A1???  # #D$E F F )#)*PQ P""$2N"OO      )&&(()@AAA%##%%d&:;;;,**,,T-HIII **,,,,,handlerCallable[..., Awaitable[Any]]0Callable[..., Awaitable[mcp.types.ServerResult]]cdfd }|S)z@Wrap a list handler to pass the request and return ServerResult.requestrrmcp.types.ServerResultchK|d{V}tj|SN)r%r& ServerResult)rMresultrIs r*wrapperz6MCPOperationsMixin._wrap_list_handler..wrapper\s>"77++++++++F9))&11 1rH)rMrrrN)r.rIrSs ` r*r9z%MCPOperationsMixin._wrap_list_handlerWs(  2 2 2 2 2 2rHrMmcp.types.ListToolsRequestmcp.types.ListToolsResultcKtd|}td|jdt t |d{Vd}d|D}||jr |jjnd}t|||j \}}tj ||S)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|jSrPnamets r*z4MCPOperationsMixin._list_tools_mcp..ns PQPVrHcDg|]}||jSrZ) to_mcp_toolr[).0tools r* z6MCPOperationsMixin._list_tools_mcp..os+HHH$T%%49%55HHHrH)tools nextCursor)r loggerdebugr[rr#r3paramsrr+_list_page_sizer%r&ListToolsResult)r.rMserverre sdk_toolsrpage next_cursors r*r4z"MCPOperationsMixin._list_tools_mcpbsi&& BBBBCCC$T0A0A0C0C*C*C*C*C*C*C%D%DFVFVWWHH%HHH &-%8W^%8GN ! !QU .iAWXXky((t (LLLrHmcp.types.ListResourcesRequestmcp.types.ListResourcesResultcKtd|}td|jdt t |d{Vd}d|D}|jr |jjnd}t|||j \}}tj ||S)z List all available resources, in the format expected by the low-level MCP server. Supports pagination when list_page_size is configured. rrXz ] Handler called: list_resourcesNc*t|jSrP)r(uri)rs r*r^z8MCPOperationsMixin._list_resources_mcp..s3qu::rHc^g|]*}|t|j+S))rt)to_mcp_resourcer(rt)rbresources r*rdz:MCPOperationsMixin._list_resources_mcp..s@   @HH $ $X\):): $ ; ;   rH) resourcesrf)r rgrhr[rr#r5rirr+rjr%r&ListResourcesResult)r.rMrlry sdk_resourcesrrnros r*r6z&MCPOperationsMixin._list_resources_mcpxsi&& FFFFGGG( v,,........ / /1E1E    LU   +2.B&&d- 66#9  ky,,t ,TTTrH&mcp.types.ListResourceTemplatesRequest%mcp.types.ListResourceTemplatesResultcKtd|}td|jdt t |d{Vd}d|D}|jr |jjnd}t|||j \}}tj ||S)z List all available resource templates, in the format expected by the low-level MCP server. Supports pagination when list_page_size is configured. rrXz)] Handler called: list_resource_templatesNc|jSrP) uri_templater\s r*r^zAMCPOperationsMixin._list_resource_templates_mcp..sANrHcDg|]}||jS)) uriTemplate)to_mcp_templater)rbtemplates r*rdzCMCPOperationsMixin._list_resource_templates_mcp..s<     $ $1F $ G G   rH)resourceTemplatesrf)r rgrhr[rr#list_resource_templatesrirr+rjr%r&ListResourceTemplatesResult)r.rMrl templates sdk_templatesrrnros r*r:z/MCPOperationsMixin._list_resource_templates_mcpsi&& OOOOPPP( v5577777777 8 8:R:R    %   +2.B&&d- 66#9  ky44"{5   rHmcp.types.ListPromptsRequestmcp.types.ListPromptsResultcKtd|}td|jdt t |d{Vd}d|D}|jr |jjnd}t|||j \}}tj ||S)z List all available prompts, in the format expected by the low-level MCP server. Supports pagination when list_page_size is configured. rrXz] Handler called: list_promptsNc|jSrPrZ)ps r*r^z6MCPOperationsMixin._list_prompts_mcp..srHcDg|]}||jSr`) to_mcp_promptr[)rbprompts r*rdz8MCPOperationsMixin._list_prompts_mcp..s+TTT&v+++==TTTrH)promptsrf)r rgrhr[rr#r7rirr+rjr%r&ListPromptsResult)r.rMrlr sdk_promptsrrnros r*r8z$MCPOperationsMixin._list_prompts_mcpsi&& DDDDEEE& v**,,,,,,,, - -/?/?  UTGTTT *1.B&&d- !7  ky**4K*PPPrHkeyr( argumentsdict[str, Any]vlist[ContentBlock] | tuple[list[ContentBlock], dict[str, Any]] | mcp.types.CallToolResult | mcp.types.CreateTaskResultcVKtd|}td|jd|| d}d} |jj}|jrD|jd}|did}|j j rE|j j }|d} t| d  }n#ttf$rYnwxYw|rt| nd} |||| | d{V} t#| t$jjr| S| S#t,$r} t/d || d} ~ wt.$r} t/d || d} ~ wwxYw)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 rrXz&] Handler called: call_tool %s with %sNT exclude_nonefastmcpversionttlreqr task_metazUnknown tool: )r rgrhr[r2request_contextmeta model_dumpget experimentalis_task task_metadatarAttributeError LookupErrorrr= isinstancer%r&CreateTaskResult to_mcp_resultrr) r.rrrl version_strrctx meta_dict mcp_task_metatask_meta_dictrrRr)s r*r?z!MCPOperationsMixin._call_tool_mcps,i&& C C C CS)    A'+K)-I (88N # 3 3 3 F FI"+-- 2">">"B"B9"M"MK#+H$'$4$BM%2%=%=4%=%P%PN (^-?-?-F-F G G GI"K0    6AJk[1111dG!++Y9,F&#)"<==  '')) ) A A A 8 8 899q @ A A A 8 8 899q @ AsNE&B(C'&E&'C;8E&:C;;AE&E&& F(0F F(F##F(rt AnyUrl | str9mcp.types.ReadResourceResult | mcp.types.CreateTaskResultc  Ktd|}td|jd| d}d} |jj}|jrG|jd}|dpi}|d}|j j rE|j j }|d} t| d  }n#ttf$rYnwxYw|rt| nd} |t#|| | d{V} t%| t&jjr| S| |S#t.$rD} t1t&jd dt#|| d} ~ wt4$r7} t1t&jd d| | d} ~ wwxYw)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 rrXz"] Handler called: read_resource %sNTrrrrrrrizResource not found: r )r rgrhr[r2rrrrrrrrrrrr@r(rr%r&rrrr r'r) r.rtrlrrrr fastmcp_metarrrrRr)s r*rAz%MCPOperationsMixin._read_resource_mcps^i&& HHHH#NNN$ &*K)-I (88> # 3 3 3 F FI#,==#;#;#ArL"."2"29"="=K#+H$'$4$BM%2%=%=4%=%P%PN (^-?-?-F-F G G GI"K0    6AJk[1111dG!//C'Y0F&#)"<==  '',, ,    ##)LC)L)L$       ##9SPQ9S9S#TT  sNE5B+C)(E5)C=:E5<C==A"E5 E55 H??F>> H 2G==Hr[dict[str, Any] | None6mcp.types.GetPromptResult | mcp.types.CreateTaskResultc*Ktd|}td|jd|| d}d} |jj}|jrD|jd}|did}|j j rE|j j }|d} t| d  }n#ttf$rYnwxYw|rt| nd} |||| | d{V} t#| t$jjr| S| S#t,$r} t/d || d} ~ wt.$rwxYw)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 rrXz'] Handler called: get_prompt %s with %sNTrrrrrrrzUnknown prompt: )r rgrhr[r2rrrrrrrrrrr render_promptrr%r&rto_mcp_prompt_resultrr) r.r[rrlrrrrrrrrRr)s r*rCz"MCPOperationsMixin._get_prompt_mcp0s i&& D D D DdI    '+K)-I (88N # 3 3 3 F FI"+-- 2">">"B"B9"M"MK#+H$'$4$BM%2%=%=4%=%P%PN (^-?-?-F-F G G GI"K0    6AJk[1111dG!//iI0F&#)"<==  ..00 0 D D D ;4 ; ;<rsMM""""""9999999999444444444444******"""""";;;;;;;;000000000000::::::HHHHHHHH.------ H   GK PPPP"IIIIIIIIIIrH