kKj2dZddlmZddlmZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZdd lmZmZdd lmZdd lmZerdd lmZdd l mZddlmZeeZ d ddZ d ddZ d ddZ y)a Task 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. This module requires fastmcp[tasks] (pydocket). It is only imported when docket is available. ) annotations)suppress)datetimetimezone) TYPE_CHECKING)ExecutionState)TaskStatusNotificationTaskStatusNotificationParams)DEFAULT_TTL_MS)parse_task_keytask_redis_prefix)DOCKET_TO_MCP_STATE) get_logger)Docket) Execution) ServerSessionc 0K |j|d{}|tjd|ytjtj tj h}|j23d{}|ddk(r.t|d}t||||||d{||vs=y|ddk(sHt||||||d{b77_727 6y#t$r'} tjd|d | d Yd} ~ yd} ~ wwxYww) aoSubscribe 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 poll_interval_ms: Poll interval in milliseconds to include in notifications NzNo execution found for task typestate)sessiontask_idtask_keydocketrpoll_interval_msprogress)rrrr executionrzSubscription task failed for z: T)exc_info) get_executionloggerwarningr COMPLETEDFAILED CANCELLED subscribe_send_status_notification_send_progress_notification Exception) rrrrrrterminal_stateseventres s/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/server/tasks/subscriptions.pysubscribe_to_task_updatesr,sB('V ..x88   NN9'C D   $ $  ! !  $ $  %..0 %V}'&uW~6/##%!%5 O+v*,1##%!'%5 99 #14 V6wir!EPTUUVsDC#CC#D?C#7C!;C<C!?+C#*C+C#4C#5D6C#?C#CC#C!C#C#!C#"D# D,D DDDcKt}|j|d}t|}|d} |jt | d|d} |j 4d{} | j| d{} dddd{ r| j dn0tjtjj} d}|tjk(rd}n+|tjk(rd}n|tjk(rd }||| tjtjjt |d }|r||d <t#t%j&| }t)t*5|j-|d{dddy7I737&#1d{7swY7xYw7,#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) poll_interval_ms: Poll interval in milliseconds failed task_scope: :created_atNutf-8zTask completed successfullyz Task failedzTask cancelled)taskIdstatus createdAt lastUpdatedAtttl pollInterval statusMessageparams)rgetr keyr redisdecodernowrutc isoformatrr!r"r#r r r model_validaterr'send_notification)rrrrrr state_map mcp_status key_partsr/created_at_keyr>created_at_bytes created_atstatus_message params_dict notifications r+r%r%]s,$Iuh/Jx(I<(JZZ#4Z#@"A7); WXN||~~!&>!::~  ( \\(,, ' 1 1 3N (((6 .'' '& .** *)!hll3==?( K'5 O$*+::;GL ) '' 555  I:~~~J 6  sAGF( G#F18F+9F1= GF. Cr?rr@rrArBr r r rCrr'rD)rrrrrrrErFrGr/rHr>rIrJrLrMs r+r&r&s& ..    Y%7%7%?%?$Iy9Jx(I<(JZZ#4Z#@"A7); WXN||~~!&>!::~  ( \\(,, ' 1 1 3!hll3==?("++33K*+::;GL ) '' 555  O:~~~2 6  sG'F8B G'$F;%G'(G=F>>G G' GCG'G*G+G/ G';G'>GG'G G GG'GG$ G'N)i) rstrrrQrrrrrintreturnNone)rrrrQrrQrrrrrrRrSrT)rrrrQrrQrrrrrrRrSrT)!__doc__ __future__r contextlibrrrtypingrdocket.executionr mcp.typesr r fastmcp.server.tasks.configr fastmcp.server.tasks.keysr r fastmcp.server.tasks.requestsrfastmcp.utilities.loggingrrrrmcp.server.sessionr__name__rr,r%r&r+rcs+#' +J6G=0*0 H ! ;V ;V;V;V  ;V  ;V  ;VH! C6 C6 C6C6  C6  C6  C6 C6X! ;6 ;6 ;6;6  ;6  ;6  ;6 ;6rb