i.dZddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z ddl Z ddl mZejeZe rddlmZdZd Zd Zd Z dd Z dd Z ddZy)aYBackground task elicitation support (SEP-1686). This module provides elicitation capabilities for background tasks running in Docket workers. Unlike regular MCP requests, background tasks don't have an active request context, so elicitation requires special handling: 1. Set task status to "input_required" via Redis 2. Send notifications/tasks/status with elicitation metadata 3. Wait for client to send input via tasks/sendInput 4. Resume task execution with the provided input This uses the public MCP SDK APIs where possible, with minimal use of internal APIs for background task coordination. ) annotationsN)datetimetimezone) TYPE_CHECKINGAnycast) ServerSession)FastMCPz2fastmcp:task:{session_id}:{task_id}:elicit:requestz3fastmcp:task:{session_id}:{task_id}:elicit:responsez1fastmcp:task:{session_id}:{task_id}:elicit:statusic  K|j}| tdttj}ddlm}|}| |j} nt|dd} | tdtj| |} tj| |} tj| |} |||d} |j4d{}|j|j| t!j"| t$ d{|j|j| d t$ d{dddd{t'j(t*j,j/}d |d |||t$d zdd|d ||||ddid}ddlm} || ||d{t$} |j4d{}tCtD|jG|j| g|d{}|r|\}}t!jH|}|j;|j| |j| d{t<j>jA|jKdd|jKdcdddd{Sdddd{ |j4d{}|j;|j| |j| |j| d{dddd{t<j>jAddS77Q7%7#1d{7swY)xYw7#t4$r}t6j9d|| |j4d{7}|j;|j| |j| d{7dddd{7n#1d{7swYnxYwn#t4$rYnwxYwt<j>jAddcYd}~Sd}~wwxYw77N777#1d{7swYxYw#t4$r"}t6j9d||Yd}~d}~wwxYw777}#1d{7swYxYw#t4$r"}t6jMd||Yd}~d}~wwxYww)aSend an elicitation request from a background task. This function handles the complexity of eliciting user input when running in a Docket worker context where there's no active MCP request. Args: task_id: The background task ID session: The MCP ServerSession for this task message: The message to display to the user schema: The JSON schema for the expected response fastmcp: The FastMCP server instance Returns: ElicitResult containing the user's response Raises: RuntimeError: If Docket is not available McpError: If the elicitation request fails Nz}Background task elicitation requires Docket. Ensure 'fastmcp[tasks]' is installed and the server has task-enabled components.r)get_task_context_fastmcp_state_prefixzCannot determine session_id for elicitation. This typically means elicit_for_task() was called outside a Docket worker context. session_idtask_id) request_idmessageschemaexwaitingznotifications/tasks/statusinput_requiredi)taskIdstatus statusMessage createdAt lastUpdatedAtttlz$io.modelcontextprotocol/related-task) requestIdrrequestedSchema)rrr elicitation)methodparams_meta)push_notificationzSFailed to queue input_required notification for task %s, cancelling elicitation: %scancelactioncontent)timeoutr'acceptr(z@BLPOP failed for task %s elicitation, falling back to cancel: %szIFailed to clean up elicitation keys for task %s (will expire via TTL): %s)'_docket RuntimeErrorstruuiduuid4fastmcp.server.dependenciesr rgetattrELICIT_REQUEST_KEYformatELICIT_RESPONSE_KEYELICIT_STATUS_KEYredissetkeyjsondumpsELICIT_TTL_SECONDSrnowrutc isoformat"fastmcp.server.tasks.notificationsr$ Exceptionloggerwarningdeletemcptypes ElicitResultrrblpoploadsgetdebug)rsessionrrfastmcpdocketrr task_contextr request_key response_key status_keyelicit_requestr6 timestampnotification_dictr$emax_wait_secondsresult_key response_dataresponse cleanup_errors p/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/tasks/elicitation.pyelicit_for_taskr]*s4__F ~ _  TZZ\"J=#%L!,, W&=tD  e  %++z7+SK&--W-UL"))Z)QJ!N ||~~ii JJ{ # JJ~ &!   ii JJz " !   ~( X\\*446I.&$"&%,   3!*!(!+&'-  5  4EE ,=vFFF.*! <<>>U  ZZ -.,F&,#m::m4llJJ{+JJz* yy--#<<(;$LL3./">>>>H  <<>>U,, ;' <( :&  "> 99 ! !4 ! @@}  ~~~d G E  a    ||~~llJJ{+JJz*&~~~~    yy%%Xt%DD#E2"#">>>>6   N     " ">>>   W     sPC S<MS<>M*M!.M*?M$M* S<M'AS<+ N8N9N=S<RQR5Q+QAQ+%Q"&AQ+) R5Q%6R:S<; RQ(R S R/!S$AR8(R2)R8- S8R59S="S<!M*$M*'S<*M=0M3 1M=8S<N Q Q$P8N;9P=4P 1O42P 7 PPP P PP PQ P+(Q*P++#QQS<QS<RQ+"Q+%R(R+Q>1Q4 2Q>9R R, R'!S<'R,,S</S2R85S8S >S ?S S S9S4.S<4S99S<cK |j|d|dd{}t|||j|j|d{tj d||jy7R7(#t $rW}tjd||t||dd|d{7}|stjd |Yd}~yYd}~yd}~wwxYww) aLRelay elicitation from a background task worker to the client. Called by the notification subscriber when it detects an input_required notification with elicitation metadata. Sends a standard elicitation/create request to the client session, then uses handle_task_input() to push the response to Redis so the blocked worker can resume. Args: session: MCP ServerSession session_id: Session identifier task_id: Background task ID elicitation: Elicitation metadata (message, requestedSchema) fastmcp: FastMCP server instance rr)rrN)rrr'r(rLz4Relayed elicitation response for task %s (action=%s)z+Failed to relay elicitation for task %s: %sr%zGFailed to push cancel response for task %s (worker may block until TTL))elicithandle_task_inputr'r(rArJr@rB)rKrrr rLrWrUsuccesss r\relay_elicitationrbs* ~~ *'(9:&   !==NN      B  MM    DgqQ)!     NN/  s\CA7A3+A7 A5 %A72C3A75A77 C*C*B-+C CCCcK|j}|ytj||}tj||}||d}|j 4d{} | j |j |d{} | | jddk7r dddd{y| j|j |tj|d{| j|j |td{| j|j |dtd{dddd{y 7777w7L77#1d{7swYy xYww) a Handle input sent to a background task via tasks/sendInput. This is called when a client sends input in response to an elicitation request from a background task. Args: task_id: The background task ID session_id: The MCP session ID action: The elicitation action ("accept", "decline", "cancel") content: The response content (for "accept" action) fastmcp: The FastMCP server instance Returns: True if the input was successfully stored, False otherwise NFrr&zutf-8r respondedrT)r+r4r3r5r6rIr8decodelpushr9r:expirer;r7) rrr'r(rLrMrPrQrZr6rs r\r`r`"s[,__F ~&--W-UL"))Z)QJH ||~~yyJ!788 >V]]73y@ ~~kk JJ| $ JJx    ll6::l35GHHHii JJz " !   !~, -8 I !~~~, sAE:EE:$E%EE% E:(E)E:.7E%%E&,E%E.E%E!E% E:E#E:E%E:E%E%!E%#E:%E7+E. ,E73E:) rr-rKzServerSession | Nonerr-rdict[str, Any]rLr returnzmcp.types.ElicitResult) rKr rr-rr-r rhrLr riNone) rr-rr-r'r-r(zdict[str, Any] | NonerLr ribool)__doc__ __future__rr9loggingr.rrtypingrrr mcp.typesrDr getLogger__name__rAfastmcp.server.serverr r2r4r5r;r]rbr`r\rvs# #  '++   8 $-JKG}A }A !}A}A  }A  }A  }A@5 555 5  5  5p8 88 8# 8  8  8ru