i/4FddlmZddlZddlmZmZddlmZddlm Z m Z ddl Z ddl Z ddlmZmZddl mZddlmZmZmZdd lmZdd lmZdd lmZdd lmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'e rddl(m)Z)e'e*Z+GddeZ,GddeeefZ-y)) annotationsN) AwaitableCallable)AsyncExitStack) TYPE_CHECKINGAny)MemoryObjectReceiveStreamMemoryObjectSendStream)McpError)LifespanResultTNotificationOptionsRequestT)Server)InitializationOptions) ServerSession) stdio_server)SessionMessage)RequestResponder)AnyUrl)UI_EXTENSION_ID) get_logger)FastMCPcNeZdZdZdfd ZeddZddZ d fd ZxZ S) MiddlewareServerSessionzMServerSession that routes initialization requests through FastMCP middleware.cft||i|tj||_d|_yN)super__init__weakrefref _fastmcp_ref_subscription_task_groupselffastmcpargskwargs __class__s h/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/low_level.pyrz MiddlewareServerSession.__init__'s. $)&)29++g2F@D%c@|j}| td|SzGet the FastMCP instance.z'FastMCP instance is no longer availabler! RuntimeErrorr$r%s r)r%zMiddlewareServerSession.fastmcp-(##% ?HI Ir*c|j}|y|j}|y|jxsi}|jd}|sy||vS)zCheck if the connected client supports a given MCP extension. Inspects the ``extensions`` extra field on ``ClientCapabilities`` sent by the client during initialization. F extensions)_client_params capabilities model_extraget)r$ extension_id client_paramscapsextrasr2s r)client_supports_extensionz1MiddlewareServerSession.client_supports_extension5sZ ++  )) <!!'R,2JJ|,D z))r*c Kddl}ddlm}tjj t jjrd j d fd }|_ d fd }|jjjj4d{}|jj ddd | } jj||d{cdddd{St( Ud{S7r7.7 #t$ri}j s<5j|j"d{7dddn##1swYnxYwt$j'd | Yd}~nd}~wwxYwdddd{7#1d{7swYxYw7w)z Override the _received_request method to route special requests through FastMCP middleware. Handles initialization requests and SEP-1686 task methods. rN)MiddlewareContextc4K||d{S7wr)responsecaptured_responseoriginal_responds r)capturing_respondzDMiddlewareServerSession._received_request..capturing_respondcs %-!-h7777s cKtt d{:tjt j jr jSy7Awr)rr_received_request isinstancerootmcptypesInitializeResult)ctxr(rA responderr$s r)call_original_handlerzHMiddlewareServerSession._received_request..call_original_handlerlsX3TLYWWW$0Z%**CII,F,F6-111 XsAAAA)r%sessionclientrequest initialize)messagesourcetypemethodfastmcp_contextznReceived McpError but responder is already completed. Cannot send error response as response was already sent.)exc_info)r@mcp.types.ServerResultreturnNone)rKr=rYz!mcp.types.InitializeResult | None)fastmcp.server.context$fastmcp.server.middleware.middlewarer=rFrPrGrHrIInitializeRequestrespondservercontextContextr%_run_middlewarer _completederrorloggerwarningrrE) r$rLr%r=rCrM fastmcp_ctx mw_contexterArBr(s `` @@r)rEz)MiddlewareServerSession._received_requestHs &J i'',,cii.I.I J@D (00  80 8 8!2I  & 2  ~~--55 d6.%--22#"'$/  !%!=!="$9">W.y999?  %//&"+"3"3AGG"<<<'YYW%&' !>:sB%G,D-G0"F,D%2D!3D%6 GD#GGG!D%#G% F.F<E*E E*! F*E3 /F F,FF, G%F(&G,F>2F5 3F>:G)r%rrYr)r7strrYbool)rLzARequestResponder[mcp.types.ClientRequest, mcp.types.ServerResult]) __name__ __module__ __qualname____doc__rpropertyr%r;rE __classcell__r(s@r)rr$s:WE *&M:TM:M:r*rceZdZdfd Zed dZ d d fd Z d fd Z d ddZ ddZ ddZ xZ S)LowLevelServerc~t||i|tj||_t ddd|_y)NT)prompts_changedresources_changed tools_changed)rrrr r!r notification_optionsr#s r)rzLowLevelServer.__init__s> $)&)29++g2F%8 "% !r*c@|j}| td|Sr,r-r/s r)r%zLowLevelServer.fastmcpr0r*c D| |j}t|d||d|S)N)rzexperimental_capabilitiesr?)rzrcreate_initialization_options)r$rzr}r'r(s r)r~z,LowLevelServer.create_initialization_optionss> '#'#<#< w4 !5&?   r*cddlm}t| ||xsi}||_t |ddxsi}i|t ii|_|S)a=Override to set capabilities.tasks as a first-class field per SEP-1686. This ensures task capabilities appear in capabilities.tasks instead of capabilities.experimental.tasks, which is required by the MCP spec and enables proper task detection by clients like VS Code Copilot 1.107+. r)get_task_capabilitiesr2N)!fastmcp.server.tasks.capabilitiesrrget_capabilitiestasksgetattrrr2)r$rzr}rr4existing_extensionsr(s r)rzLowLevelServer.get_capabilitiessh Lw/ % + 34  L, 5 ; #O%8"N/2"N r*c FKt4d{}|j|j|d{}|jt|j||||d{}t j 4d{} | |_|j23d{} | j|j| |||)777a7F7,6dddd{7n#1d{7swYnxYwdddd{7y#1d{7swYyxYww)zN Overrides the run method to use the MiddlewareServerSession. N) stateless) renter_async_contextlifespanrr%anyiocreate_task_groupr"incoming_messages start_soon_handle_message) r$ read_stream write_streaminitialization_optionsraise_exceptionsrstacklifespan_contextrNtgrRs r)runzLowLevelServer.runs"##u%*%>%>t}}T?R%SS !55'LL *' G..00B350%,%>%>'MM,,(( #$S1%> 10000$####sD!CD!$D C 0D *C +D CD  C%C#C $C'!C%D! D  D D CC% D C!D %C7 +C.,C7 3D : D!DD! DD DD!c dfd }|S)a Decorator for registering a read_resource handler with CreateTaskResult support. The MCP SDK's read_resource decorator does not support returning CreateTaskResult for background task execution. This decorator wraps the result in ServerResult. This decorator can be removed once the MCP SDK adds native CreateTaskResult support for resources. cl dfd }|jtjj<S)NcK|jjd{}tjj |S7#wr)paramsurirHrI ServerResultreqresultfuncs r)handlerz@LowLevelServer.read_resource..decorator..handler#s8 $CJJNN33yy--f554s A A$A )rzmcp.types.ReadResourceRequestrYrX)request_handlersrHrIReadResourceRequestrrr$s` r) decoratorz/LowLevelServer.read_resource..decorators; 62 6' 6 DKD ! !#))"?"? @Kr*)rXCallable[[AnyUrl], Awaitable[mcp.types.ReadResourceResult | mcp.types.CreateTaskResult]]rYrr?r$rs` r) read_resourcezLowLevelServer.read_resource$0     $r*c dfd }|S)a Decorator for registering a get_prompt handler with CreateTaskResult support. The MCP SDK's get_prompt decorator does not support returning CreateTaskResult for background task execution. This decorator wraps the result in ServerResult. This decorator can be removed once the MCP SDK adds native CreateTaskResult support for prompts. cl dfd }|jtjj<S)NcK|jj|jjd{}tjj |S7#wr)rname argumentsrHrIrrs r)rz=LowLevelServer.get_prompt..decorator..handlerOsE $CJJOOSZZ5I5IJJyy--f55Ks5AA$A)rzmcp.types.GetPromptRequestrYrX)rrHrIGetPromptRequestrs` r)rz,LowLevelServer.get_prompt..decoratorFs; 6/ 6' 6 AHD ! !#))"<"< =Kr*)riCallable[[str, dict[str, Any] | None], Awaitable[mcp.types.GetPromptResult | mcp.types.CreateTaskResult]]rYrr?rs` r) get_promptzLowLevelServer.get_prompt.rr*)r%rr&rr'rrj)NN)rzzNotificationOptions | Noner}z dict[str, dict[str, Any]] | Noner'rrYr)rzr r}zdict[str, dict[str, Any]]rYzmcp.types.ServerCapabilities)FF) rz5MemoryObjectReceiveStream[SessionMessage | Exception]rz&MemoryObjectSendStream[SessionMessage]rrrrlrrl)rYzCallable[[Callable[[AnyUrl], Awaitable[mcp.types.ReadResourceResult | mcp.types.CreateTaskResult]]], Callable[[AnyUrl], Awaitable[mcp.types.ReadResourceResult | mcp.types.CreateTaskResult]]])rYzCallable[[Callable[[str, dict[str, Any] | None], Awaitable[mcp.types.GetPromptResult | mcp.types.CreateTaskResult]]], Callable[[str, dict[str, Any] | None], Awaitable[mcp.types.GetPromptResult | mcp.types.CreateTaskResult]]]) rmrnrorrqr%r~rrrrrrrss@r)rurus  <@FJ  8  $D       1 $=  & N"' "J"="!6 "  "  "H* *X* *r*ru). __future__rrcollections.abcrr contextlibrtypingrrr mcp.typesrHanyio.streams.memoryr r r mcp.server.lowlevel.serverr r rr_Servermcp.server.modelsrmcp.server.sessionrmcp.server.stdiormcp.shared.messagermcp.shared.sessionrpydanticrfastmcp.server.appsrfastmcp.utilities.loggingrfastmcp.server.serverrrmrerrur?r*r)rs"/%% R 4,9-//0- H q:mq:h@W_h67@r*