+ wiYRt^RIHt^RIt^RIHt^RIHt^RIH t H t ^RI H t ^RI HtHt^RIHt^R IHtHt^R IHt^R IHt^R IHtHtHtHtHtHtH t H!t!H"t"H#t#H$t$H%t%H&t&H't'H(t(H)t)H*t*H+t+H,t,H-t-H.t.H/t/H0t0!R R4t1R#)z ServerTaskContext - Server-integrated task context with elicitation and sampling. This wraps the pure TaskContext and adds server-specific functionality: - Elicitation (task.elicit()) - Sampling (task.create_message()) - Status notifications )AnyN)TaskResultHandler) ServerSession)validate_sampling_tools!validate_tool_use_result_messages)McpError)"require_task_augmented_elicitationrequire_task_augmented_sampling) TaskContext) QueuedMessageTaskMessageQueue)Resolver) TaskStore)INVALID_REQUESTTASK_STATUS_INPUT_REQUIREDTASK_STATUS_WORKINGClientCapabilitiesCreateMessageResultCreateTaskResultElicitationCapabilityElicitRequestedSchema ElicitResult ErrorDataIncludeContextModelPreferences RequestIdResultSamplingCapabilitySamplingMessageServerNotificationTask TaskMetadataTaskStatusNotificationTaskStatusNotificationParamsTool ToolChoiceca]tRt^5toRtRR/V3RlRllt]V3RlRl4t]V3RlR l4t]V3R lR l4t V3R lR lt RR/V3RlRllt RR/V3RlRllt RR/V3RlRllt V3RlRltV3RlRltV3RlRltV3RlRltV3RlRltR RR!RR"RR#RR$RR%RR&RR'R/V3R(lR)lltR*R+/V3R,lR-lltR*R+R RR!RR"RR#RR$RR%RR&RR'R/ V3R.lR/lltR0tVtR#)1ServerTaskContexta Server-integrated task context with elicitation and sampling. This wraps a pure TaskContext and adds server-specific functionality: - elicit() for sending elicitation requests to the client - create_message() for sampling requests - Status notifications via the session Example: async def my_task_work(task: ServerTaskContext) -> CallToolResult: await task.update_status("Starting...") result = await task.elicit( message="Continue?", requestedSchema={"type": "object", "properties": {"ok": {"type": "boolean"}}} ) if result.content.get("ok"): return CallToolResult(content=[TextContent(text="Done!")]) else: return CallToolResult(content=[TextContent(text="Cancelled")]) handlerNc F<V^8dQhRS[RS[RS[RS[RS[R,/#)taskstoresessionqueuer(N)r rrr r)format __classdict__s"X/home/ubuntu/.local/lib/python3.14/site-packages/mcp/server/experimental/task_context.py __annotate__ServerTaskContext.__annotate__MsD     #T)c X\WR7VnW0nW@nWPnW nR#)a Create a ServerTaskContext. Args: task: The Task object store: The task store session: The server session queue: The message queue for elicitation/sampling handler: The result handler for response routing (required for elicit/create_message) )r+r,N)r _ctx_session_queue_handler_store)selfr+r,r-r.r(s&$$$$$r1__init__ServerTaskContext.__init__Ms$& T7     r4c <V^8dQhRS[/#r*return)str)r/r0s"r1r2r3is!!!r4c .VPP#)zThe task identifier.)r6task_idr;s&r1rCServerTaskContext.task_idhsyy   r4c <V^8dQhRS[/#r?)r )r/r0s"r1r2r3nsdr4c .VPP#)zThe current task state.)r6r+rDs&r1r+ServerTaskContext.taskmsyy~~r4c <V^8dQhRS[/#r?)bool)r/r0s"r1r2r3ss&&d&r4c .VPP#)z(Whether cancellation has been requested.)r6 is_cancelledrDs&r1rLServerTaskContext.is_cancelledrsyy%%%r4c<V^8dQhRR/#r*r@N)r/r0s"r1r2r3ws))d)r4c :VPP4R#)z"Request cancellation of this task.N)r6request_cancellationrDs&r1rR&ServerTaskContext.request_cancellationws &&(r4notifyTc*<V^8dQhRS[RS[RR/#)r*messagerTr@NrArJ)r/r0s"r1r2r3}s" , ,3 ,4 ,4 ,r4c "VPPV4GRjxL V'dVP4GRjxL R#R#L(L 5i)z Update the task's status message. Args: message: The new status message notify: Whether to send a notification to the client N)r6 update_status_send_notification)r;rVrTs&&$r1rYServerTaskContext.update_status}sAii%%g... ))+ + +  / +&AA  AAA  A Ac*<V^8dQhRS[RS[RR/#)r*resultrTr@N)rrJ)r/r0s"r1r2r3s" , ,V , , ,r4c "VPPV4GRjxL V'dVP4GRjxL R#R#L(L 5i)z Mark the task as completed with the given result. Args: result: The task result notify: Whether to send a notification to the client N)r6completerZ)r;r^rTs&&$r1r`ServerTaskContext.completesAii  ((( ))+ + +  ) +r\c*<V^8dQhRS[RS[RR/#)r*errorrTr@NrW)r/r0s"r1r2r3s" , , , , ,r4c "VPPV4GRjxL V'dVP4GRjxL R#R#L(L 5i)z Mark the task as failed with an error message. Args: error: The error message notify: Whether to send a notification to the client N)r6failrZ)r;rcrTs&&$r1reServerTaskContext.fails?iinnU### ))+ + +  $ +r\c<V^8dQhRR/#rOrP)r/r0s"r1r2r3s  $ r4c V"VPPpVPP\ \ \ VPVPVPVPVPVPVPR7R744GRjxL R#L5i)z.Send a task status notification to the client.)taskIdstatus statusMessage createdAt lastUpdatedAtttl pollInterval)paramsN)r6r+r7send_notificationrr"r#rirjrkrlrmrnro)r;r+s& r1rZ$ServerTaskContext._send_notifications}yy~~mm-- &7#{{#{{&*&8&8"&..&*&8&8 HH%)%6%6     sBB) B'!B)c<V^8dQhRR/#rOrP)r/r0s"r1r2r3str4c VPP\\4R74'g\ \ \ RR74hR#)z)Check if the client supports elicitation.) elicitationz.Client does not support elicitation capabilitycoderVN)r7check_client_capabilityrrrrrrDs&r1_check_elicitation_capability/ServerTaskContext._check_elicitation_capabilitysB}}445GTiTk5lmm(L nr4c<V^8dQhRR/#rOrP)r/r0s"r1r2r3sDr4c VPP\\4R74'g\ \ \ RR74hR#)z&Check if the client supports sampling.)samplingz+Client does not support sampling capabilityrvN)r7rxrrrrrrDs&r1_check_sampling_capability,ServerTaskContext._check_sampling_capabilitysB}}445GQcQe5fgg(I hr4c,<V^8dQhRS[RS[RS[/#)r*rVrequestedSchemar@)rArr)r/r0s"r1r2r3s)???/?  ?r4c b"VP4VPf \R4hVPP VP \ R7GRjxL VPPVVVP R7pVPp\4pWPPPV&\RVVVR7pVPPVP V4GRjxL VP4GRjxL pVPP VP \ R7GRjxL \"P$!V4#LLhLQL \&P(!4d7TPP TP \ R7GRjxL hi;i5i)a Send an elicitation request via the task message queue. This method: 1. Checks client capability 2. Updates task status to "input_required" 3. Queues the elicitation request 4. Waits for the response (delivered via tasks/result round-trip) 5. Updates task status back to "working" 6. Returns the result Args: message: The message to present to the user requestedSchema: Schema defining the expected response structure Returns: The client's response Raises: McpError: If client doesn't support elicitation capability NzEhandler is required for elicit(). Pass handler= to ServerTaskContext.rj)rVrrelated_task_idrequesttyperVresolveroriginal_request_id)ryr9 RuntimeErrorr: update_taskrCrr7_build_elicit_form_requestidr _pending_requestsr r8enqueuewaitrrmodel_validateanyioget_cancelled_exc_class)r;rVrr request_idrqueued response_datas&&& r1elicitServerTaskContext.elicitsf4 **, == fg gkk%%dll;U%VVV--::+ LL;  !( -5Z6> '' 3 *   kk!!$,,777 "*--/1M++))$,,?R)S S S..}= =3 W& 82 S,,. ++))$,,?R)S S S   snAF/EBF/-E.F/3EE2E9E:EF/F/EEAF,$F'%F,,F/c2<V^8dQhRS[RS[RS[RS[/#)r*rVurlelicitation_idr@)rAr)r/r0s"r1r2r3 s9CCCC C  Cr4c d"VP4VPf \R4hVPP VP \ R7GRjxL VPPVVVVP R7pVPp\4pW`PPV&\RVVVR7pVPPVP V4GRjxL VP4GRjxL pVPP VP \ R7GRjxL \"P$!V4#LLhLQL \&P(!4d7TPP TP \ R7GRjxL hi;i5i)aL Send a URL mode elicitation request via the task message queue. This directs the user to an external URL for out-of-band interactions like OAuth flows, credential collection, or payment processing. This method: 1. Checks client capability 2. Updates task status to "input_required" 3. Queues the elicitation request 4. Waits for the response (delivered via tasks/result round-trip) 5. Updates task status back to "working" 6. Returns the result Args: message: Human-readable explanation of why the interaction is needed url: The URL the user should navigate to elicitation_id: Unique identifier for tracking this elicitation Returns: The client's response indicating acceptance, decline, or cancellation Raises: McpError: If client doesn't support elicitation capability RuntimeError: If handler is not configured NzIhandler is required for elicit_url(). Pass handler= to ServerTaskContext.r)rVrrrrr)ryr9rr:rrCrr7_build_elicit_url_requestrr rr r8rrrrrrr) r;rVrrrrrrrs &&&& r1 elicit_urlServerTaskContext.elicit_url sh@ **, == jk kkk%%dll;U%VVV--99) LL :  !( -5Z6> '' 3 *   kk!!$,,777 "*--/1M++))$,,?R)S S S..}= =5 W( 82 S,,. ++))$,,?R)S S S  snAF0EBF0.E/F04EE2E:E;EF0F0EEAF-%F(&F--F0 system_promptinclude_context temperaturestop_sequencesmetadatamodel_preferencestools tool_choicec<V^8dQhRS[S[,RS[RS[R,RS[R,RS[R,RS[S[,R,RS[S[S[3,R,R S[R,R S[S[ ,R,R S[ R,R S[ / #) r*messages max_tokensrNrrrrrrrr@ listrintrArfloatdictrrr$r%r)r/r0s"r1r2r3Ps\\'\ \ Tz \ ($. \T\\S D(\sCx.4'\,d2\DzD \ $&\ \r4c  "VP4VPP'd!VPPPMRp \ WV 4\ V4VP f \R4hVPPVP\R7GRjxL VPPVVVVVVVVV V VPR7 p V Pp \4pWP PV &\!RV VV R7pVP"P%VPV4GRjxL VP'4GRjxL pVPPVP\(R7GRjxL \*P,!V4#ELLiLRL! \.P0!4d7TPPTP\(R7GRjxL hi;i5i)a Send a sampling request via the task message queue. This method: 1. Checks client capability 2. Updates task status to "input_required" 3. Queues the sampling request 4. Waits for the response (delivered via tasks/result round-trip) 5. Updates task status back to "working" 6. Returns the result Args: messages: The conversation messages for sampling max_tokens: Maximum tokens in the response system_prompt: Optional system prompt include_context: Context inclusion strategy temperature: Sampling temperature stop_sequences: Stop sequences metadata: Additional metadata model_preferences: Model selection preferences tools: Optional list of tools the LLM can use during sampling tool_choice: Optional control over tool usage behavior Returns: The sampling result from the client Raises: McpError: If client doesn't support sampling capability or tools ValueError: If tool_use or tool_result message structure is invalid NzMhandler is required for create_message(). Pass handler= to ServerTaskContext.r) rrrrrrrrrrrrr)r~r7 client_params capabilitiesrrr9rr:rrCr_build_create_message_requestrr rr r8rrrrrrr)r;rrrrrrrrrr client_capsrrrrrs&&$$$$$$$$$ r1create_message ServerTaskContext.create_messagePsX '')BF--B]B]B]dmm11>>cg  K@)(3 == no okk%%dll;U%VVV--==!'+#)/# LL>  !( -5Z6> '' 3 *   kk!!$,,777 "*--/1M++))$,,?R)S S S&55mD DC W6 82 S,,. ++))$,,?R)S S S   snB-H /F10BH  F4 H F:#F6$2F:F8F:0H 4H 6F:8F::AH HH  H rni`c2<V^8dQhRS[RS[RS[RS[/#)r*rVrrnr@)rArrr)r/r0s"r1r2r3s9LLL/L  L  Lr4c "VPP'd!VPPPMRp\V4VPf \ R4hVP PVP\R7GRjxL VPPVVVP\VR7R7pVPp\4pWpPPV&\RVVVR7pVP P#VPV4GRjxL VP%4GRjxL p \&P(!V 4p V P*P,p VPP.P1V 4RjxL p K ELL}LfLDTPP.P3T \44GRjxL p TP PTP\6R7GRjxL T # \8P:!4d7TP PTP\6R7GRjxL hi;i5i)a Send a task-augmented elicitation via the queue, then poll client. This is for use inside a task-augmented tool call when you want the client to handle the elicitation as its own task. The elicitation request is queued and delivered when the client calls tasks/result. After the client responds with CreateTaskResult, we poll the client's task until complete. Args: message: The message to present to the user requestedSchema: Schema defining the expected response structure ttl: Task time-to-live in milliseconds for the client's task Returns: The client's elicitation response Raises: McpError: If client doesn't support task-augmented elicitation RuntimeError: If handler is not configured Nz(handler is required for elicit_as_task()rrn)rVrrr+rr)r7rrrr9rr:rrCrrr!rr rr r8rrrrr+ri experimental poll_taskget_task_resultrrrr)r;rVrrnrrrrrr create_resultclient_task_id_r^s&&&$ r1elicit_as_task ServerTaskContext.elicit_as_tasks6CG--B]B]B]dmm11>>cg *;7 == IJ Jkk%%dll;U%VVV--::+ LL#& ;  !( -5Z6> '' 3 *   kk!!$,,777 "*--/1M,;;MJM*//66N ==55??O  a; W& 82  O ==55EEF ++))$,,?R)S S SM,,. ++))$,,?R)S S S  sBI2F,BI21F/2I27H F1 AH F5$F3%F5(H,I2/I21H3F55.H#G&$3HHHI2AI/'I*(I//I2c<V^8dQhRS[S[,RS[RS[RS[R,RS[R,RS[R,RS[S[,R,R S[S[S[3,R,R S[R,R S[S[ ,R,R S[ R,R S[ / #)r*rrrnrNrrrrrrrr@r)r/r0s"r1r2r3shh'h h  h Tz h($.hT\hS D(hsCx.4'h,d2hDzD h $&h hr4c  &"VPP'd!VPPPMRp \V 4\ WV 4\ V4VP f \R4hVPPVP\R7GRjxL VPPVVVVVVVV V V VP\VR7R7 p V Pp\4pWP P V&\#RV VVR7pVP$P'VPV4GRjxL VP)4GRjxL p\*P,!V4pVP.P0pVPP2P5V4RjxL pK EL!L}LfLDTPP2P7T\84GRjxL pTPPTP\:R7GRjxL T# \<P>!4d7TPPTP\:R7GRjxL hi;i5i)a| Send a task-augmented sampling request via the queue, then poll client. This is for use inside a task-augmented tool call when you want the client to handle the sampling as its own task. The request is queued and delivered when the client calls tasks/result. After the client responds with CreateTaskResult, we poll the client's task until complete. Args: messages: The conversation messages for sampling max_tokens: Maximum tokens in the response ttl: Task time-to-live in milliseconds for the client's task system_prompt: Optional system prompt include_context: Context inclusion strategy temperature: Sampling temperature stop_sequences: Stop sequences metadata: Additional metadata model_preferences: Model selection preferences tools: Optional list of tools the LLM can use during sampling tool_choice: Optional control over tool usage behavior Returns: The sampling result from the client Raises: McpError: If client doesn't support task-augmented sampling or tools ValueError: If tool_use or tool_result message structure is invalid RuntimeError: If handler is not configured Nz0handler is required for create_message_as_task()rr) rrrrrrrrrrrr+rr) r7rrr rrr9rr:rrCrrr!rr rr r8rrrrr+rirrrrrrr)r;rrrnrrrrrrrrrrrrrrrrrr^s&&$$$$$$$$$$ r1create_message_as_task(ServerTaskContext.create_message_as_tasks.XCG--B]B]B]dmm11>>cg ' 4 K@)(3 == QR Rkk%%dll;U%VVV--==!'+#)/# LL#&>  !( -5Z6> '' 3 *   kk!!$,,777 "*--/1M,;;MJM*//66N ==55??O  aM W8 82  O ==55EE#F ++))$,,?R)S S SM,,. ++))$,,?R)S S S  sB(J*G +B%JGJH>)G*AH>?GGGH> JJH>G.H>H3H>6H97H>=J>AJJ JJ)r6r9r8r7r:)__name__ __module__ __qualname____firstlineno____doc__r<propertyrCr+rLrRrYr`rerZryr~rrrrr__static_attributes____classdictcell__)r0s@r1r'r'5s.-16!!&&)) ,$ , , ,t , , ,t , ,  *??BCCJ\ %) \ 26 \%)\,0\+/\6:\$(\*.\\|L  LL\h  h %) h26h%)h,0h+/h6:h$(h*.hhhr4r')2rtypingrr+mcp.server.experimental.task_result_handlerrmcp.server.sessionrmcp.server.validationrrmcp.shared.exceptionsr*mcp.shared.experimental.tasks.capabilitiesrr %mcp.shared.experimental.tasks.contextr +mcp.shared.experimental.tasks.message_queuer r &mcp.shared.experimental.tasks.resolverr #mcp.shared.experimental.tasks.storer mcp.typesrrrrrrrrrrrrrrrrrr r!r"r#r$r%r'rPr4r1rs` I,\*>W;96oor4