KidZddlmZddlmZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZdd lmZerdd lmZdd l mZdd lmZeeZ dd Z ddZ ddZy)zTask subscription helpers for sending MCP notifications (SEP-1686). Subscribes to Docket execution state changes and sends notifications/tasks/status to clients when their tasks change state. ) annotations)suppress)datetimetimezone) TYPE_CHECKING)ExecutionState)TaskStatusNotificationTaskStatusNotificationParams)DOCKET_TO_MCP_STATE) get_logger)Docket) Execution) ServerSessionc K |j|d{}|tjd|y|j23d{}|ddk(rt |||||dd{-|ddk(s6t |||||d{O7~7L7+7 6y#t $r%}tjd|d |Yd}~yd}~wwxYww) aSubscribe to Docket execution events and send MCP notifications. Per SEP-1686 lines 436-444, servers MAY send notifications/tasks/status when task state changes. This is an optional optimization that reduces client polling frequency. Args: task_id: Client-visible task ID (server-generated UUID) task_key: Internal Docket execution key (includes session, type, component) session: MCP ServerSession for sending notifications docket: Docket instance for subscribing to execution events NzNo execution found for task typestate)sessiontask_idtask_keydocketrprogress)rrrr executionz%subscribe_to_task_updates failed for z: ) get_executionloggerwarning subscribe_send_status_notification_send_progress_notification Exceptionerror)rrrrreventes n/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/server/tasks/subscriptions.pysubscribe_to_task_updatesr$s$M ..x88   NN9'C D %..0  %V}'/##%!. v*,1##%!' %9  1( M B!BB!BB!B!B! C! C*C C CCcKtj|d}ddlm}||}|d}|j d|d|d} |j 4d{} | j| d{} dddd{ r| j d n0tjtjj} d} |tjk(rd } n+|tjk(rd } n|tjk(rd } ||| tjtjjd dd}| r| |d<t!t#j$|}t't(5|j+|d{dddy7E7/7"#1d{7swY3xYw7,#1swYyxYww)aSend notifications/tasks/status to client. Per SEP-1686 line 454: notification SHOULD NOT include related-task metadata (taskId is already in params). Args: session: MCP ServerSession task_id: Client-visible task ID task_key: Internal task key (for metadata lookup) docket: Docket instance state: Docket execution state (enum) failedrparse_task_key session_id fastmcp:task:: :created_atNutf-8zTask completed successfullyz Task failedzTask cancelled`)taskIdstatus createdAt lastUpdatedAtttl pollInterval statusMessageparams)r getfastmcp.server.tasks.keysr(keyredisdecodernowrutc isoformatr COMPLETEDFAILED CANCELLEDr r model_validaterrsend_notification)rrrrr mcp_statusr( key_partsr)created_at_keyr<created_at_bytes created_atstatus_message params_dict notifications r#rrLs(%((9J9x(I<(JZZ- |1WI[ QRN||~;;!&>!::;;  ( \\(,, ' 1 1 3N (((6 .'' '& .** *)!hll3==? K'5 O$*+::;GL ) 6'' 55566I;:;;;;J 666sAG FG F&1F 2F&6 G F#C8G :F>F<F> G  F&#G &F9,F/ -F94G <F>>GG cK|jd{|jr|jjsytj|j d}ddlm}||}|d}|jd|d|d} |j4d{} | j | d{} dddd{ r| jd n0tjtjj} ||| tjtjjd d |jjd } t!t#j$|  }t't(5|j+|d{dddy7777#1d{7swYxYw7,#1swYyxYww)aSend notifications/tasks/status when progress updates. Args: session: MCP ServerSession task_id: Client-visible task ID task_key: Internal task key docket: Docket instance execution: Execution object with current progress Nr&rr'r)r*r+r,r-r.r/)r0r1r2r3r4r5r6r7)syncrmessager r9rr:r(r;r<r=rr>rr?r@r r rDrrrE)rrrrrrFr(rGr)rHr<rIrJrLrMs r#rrs" ..    Y%7%7%?%?%(((CJ9x(I<(JZZ- |1WI[ QRN||~;;!&>!::;;  ( \\(,, ' 1 1 3!hll3==?"++33K*+::;GL ) 6'' 55566Q ;:;;;;2 666sGF-BGF0G!F76F37F7; GF5CG GG  G$ G0G3F75G7G =G >G G GGGN) rstrrrQrrrr returnNone) rrrrQrrQrr rrrRrS) rrrrQrrQrr rrrRrS)__doc__ __future__r contextlibrrrtypingrdocket.executionr mcp.typesr r fastmcp.server.tasks.protocolr fastmcp.utilities.loggingr rr rmcp.server.sessionr__name__rr$rrr#r`s #' +J=0*0 H .M .M.M.M  .M  .MbB6 B6 B6B6  B6  B6  B6J:6 :6 :6:6  :6  :6  :6r_