kKj-(dZddlmZddlZddlZddlZddlmZmZddlm Z m Z ddl Z ddl m Z ddlmZmZddlmZdd lmZej*eZe rdd lmZd Zdd Z dd Z ddZ 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_CHECKINGAny) ServerSession)get_task_contextget_task_session_idtask_redis_prefix)push_notification)FastMCPic@t|d|d}|d|d|dfS)zFBuild (request, response, status) Redis keys for a task's elicitation.:z:elicitz:requestz :responsez:statusr ) task_scopetask_idprefixs q/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/server/tasks/elicitation.py _elicit_keysr)s<!*-.ay @FXX 6() 46H HHc  K|j}| tdttj}t }|&|j }t|ddxs t} n tdt||\} } } |||d} |j4d{}|j|j| tj| td{|j|j| dtd{dddd{t!j"t$j&j)}d|d |||td zd d |d |||||d did}| 7t*j-d|t.j0j3ddS t5| ||d{t} |j4d{}|j;|j| g|d{}|r|\}}tj<|}|j9|j| |j| d{t.j0j3|j?dd|j?dcdddd{Sdddd{ |j4d{}|j9|j| |j| |j| d{dddd{t.j0j3ddS77z7N7A#1d{7swYRxYw7#t6$r}t*j-d|| |j4d{7}|j9|j| |j| d{7dddd{7n#1d{7swYnxYwn#t6$rYnwxYwt.j0j3ddcYd}~Sd}~wwxYw7v7N777#1d{7swYxYw#t6$r"}t*j-d||Yd}~d}~wwxYw777}#1d{7swYxYw#t6$r"}t*jAd||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._fastmcp_state_prefixzCannot determine task scope for elicitation. This typically means elicit_for_task() was called outside a Docket worker context.) request_idmessageschemaexwaitingznotifications/tasks/statusinput_requiredi)taskIdstatus statusMessage createdAt lastUpdatedAtttlz$io.modelcontextprotocol/related-task) requestIdrrequestedSchema)r r!r"r elicitation)methodparams_metazLNo session_id available for task %s, cannot deliver elicitation notificationcancelactioncontentzSFailed to queue input_required notification for task %s, cancelling elicitation: %s)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 RuntimeErrorstruuiduuid4r rgetattrr rredissetkeyjsondumpsELICIT_TTL_SECONDSrnowrutc isoformatloggerwarningmcptypes ElicitResultr Exceptiondeleteblpoploadsgetdebug)rsessionrrfastmcpdocketr task_contextr session_id request_key response_key status_keyelicit_requestr8 timestampnotification_dictemax_wait_secondsresult_key response_dataresponse cleanup_errors relicit_for_taskr^/s4__F ~ _  TZZ\"J#%L!,,  G4d ; T?R?T  a  -9W,M)Kz!N ||~~ii JJ{ # JJ~ &!   ii JJz " !   ~( X\\*446I.&$"&%,   3!*!((!+&'-  5  2 Z  yy%%Xt%DDE ,=vFFF.* <<>>U!;;L)*('F &,#m::m4llJJ{+JJz* yy--#<<(;$LL3.)">>>>B  <<>>U,, ;' <( :&  "> 99 ! !4 ! @@  ~~~l G E  a    ||~~llJJ{+JJz*&~~~~    yy%%Xt%DD#E2"">>>>0   N     " ">>>   W     sPBS7 M!S7$>M%"M#.M%MM% S7!M""BS71M>M;M>S7 Q<!Q"Q<%'Q& Q AQ& Q!AQ&$ Q<0Q 1Q<5S76 Q<Q#Q<S R*S AR3#R-$R3( S 3R04S 8"S7M%M%"S7%M8+M. ,M83S7;M>> QQP3N64P84P,O/-P2 P=P>PP P  P PQ P&#Q%P&&#Q Q S7QS7Q<Q&Q& Q<#Q<&Q9,Q/ -Q94Q<< R'R"S7"R''S7*S -R30S 3S9R< :SS S4S/)S7/S44S7cK |j|d|dd{}t|||j|j|d{tj d||jy7R7(#t $rW}tjd||t||dd|d{7}|stjd |Yd}~yYd}~yd}~wwxYww) ahRelay 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 task_scope: Authorization scope for Redis key construction task_id: Background task ID elicitation: Elicitation metadata (message, requestedSchema) fastmcp: FastMCP server instance rr')rr'N)rrr.r/rMz4Relayed 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/rArKrFrB)rLrrr(rMrYrWsuccesss rrelay_elicitationrcs* ~~ *'(9:&   !==NN      B  MM    DgqQ)!     NN/  s\CA7A3+A7 A5 %A72C3A75A77 C*C*B-+C CCCcK|j}|yt||\}}}||d} |j4d{} | j|j |d{} | | j ddk7r dddd{y| j |j |tj| d{| j|j |td{| j|j |dtd{dddd{y7777w7L77#1d{7swYyxYww) 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 task_scope: Authorization scope for Redis key construction 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 NFr-zutf-8r respondedrT) r2rr8rJr:decodelpushr;r<expirer=r9) rrr.r/rMrN_rRrSr\r8r!s rrara$sD,__F ~".z7"CA|ZH ||~~yyJ!788 >V]]73y@ ~~kk JJ| $ JJx    ll6::l35GHHHii JJz " !   !~, -8 I !~~~, s9ED9E$E#D;$E? E D= E7ED?,E4E5.E#E$E( E3E4E;E=E?EEEEE E EE)r str | Nonerr4returnztuple[str, str, str]) rr4rLzServerSession | Nonerr4rdict[str, Any]rMrrkzmcp.types.ElicitResult) rLrrrjrr4r(rlrMrrkNone) rr4rrjr.r4r/zdict[str, Any] | NonerMrrkbool)__doc__ __future__rr;loggingr5rrtypingrr mcp.typesrCrfastmcp.server.tasks.contextr r fastmcp.server.tasks.keysr "fastmcp.server.tasks.notificationsr getLogger__name__rAfastmcp.server.serverrr=rr^rcrarrr{s #  '%N7@   8 $-I zA zA !zAzA  zA  zA  zAz5 555 5  5  5p7 77 7# 7  7  7r