""" '39   (    "F$%@@@ UV=S U U   $*#6G  # # $$SY%F%F%H%HII   Mc0eZdZdZd fd Zd fd ZxZS) TaskNotificationHandlerzEMessageHandler that routes task status notifications to Task objects.clientrczttj||_dSN)super__init__weakrefref _client_ref)selfr@ __class__s r;rDz TaskNotificationHandler.__init__\s1 07 F0C0Cr=messagerrNonec,Kt|tjjrJt|jt r0|}|r||jt |d{VdS)z7Dispatch messages, including task status notifications.N) isinstancer&r'ServerNotificationrootrrG _handle_task_status_notificationrCdispatch)rHrJr@rIs r;rQz TaskNotificationHandler.dispatch`s gsy; < < J',(>?? J))++J;;GLIIIggw'''''''''''r=)r@r)rJrrrK)__name__ __module__ __qualname____doc__rDrQ __classcell__rIs@r;r?r?YsgOODDDDDD((((((((((r=r? TaskResultTceZdZdZ d"d#d Zd$d Zed%d Zed&dZd'dZ d(dZ d)dZ e j d*dZdddd+dZd$d Zd!ZdS),Taska Abstract base class for MCP background tasks (SEP-1686). Provides a uniform API whether the server accepts background execution or executes synchronously (graceful degradation per SEP-1686). Subclasses: - ToolTask: For tool calls (result type: CallToolResult) - PromptTask: For prompts (future, result type: GetPromptResult) - ResourceTask: For resources (future, result type: ReadResourceResult) Nr@rtask_idstrimmediate_resultTaskResultT | Nonecz||_||_||_|du|_d|_d|_g|_d|_dS)z Create a Task wrapper. Args: client: The FastMCP client task_id: The task identifier immediate_result: If server executed synchronously, the immediate result N)_client_task_id_immediate_result _is_immediate _status_cache _status_event_status_callbacks_cached_result)rHr@r[r]s r;rDz Task.__init__{sV  !1-T94837  37r=rrKcv|jrdS |jj}dS#t$r}td|d}~wwxYw)zValidate that client context is still active. Raises: RuntimeError: If accessed outside client context (unless immediate) NzoCannot access task results outside client context. Task futures must be used within 'async with client:' block.)rcr`rr5)rH_es r;_check_client_connectedzTask._check_client_connectedsc    F  $AAA   O  s  838c|jS)zGet the task ID.)rarHs r;r[z Task.task_ids }r=boolc|jS)zCheck if server executed the task immediately. Returns: True if server executed synchronously (graceful degradation or no task support) False if server accepted background execution )rcrms r;returned_immediatelyzTask.returned_immediatelys !!r=statusrc<||_|j|j|jD]l} ||}t j|rt j|7#t$r)}t d|dYd}~ed}~wwxYwdS)aProcess incoming notifications/tasks/status (internal). Called by Client when a notification is received for this task. Updates cache, triggers events, and invokes user callbacks. Args: status: Task status from notification NzTask callback error: T)exc_info) rdresetrfinspect isawaitableasyncio create_task Exceptionloggerwarning)rHrqcallbackr:rjs r;_handle_status_notificationz Task._handle_status_notifications$   )   " " $ $ $. K KH K!&))&v..0'/// K K K:q::TJJJJJJJJ K  K Ks3A&& B0BBr|1Callable[[GetTaskResult], None | Awaitable[None]]c:|j|dS)a>Register callback for status change notifications. The callback will be invoked when a notifications/tasks/status is received for this task (optional server feature per SEP-1686 lines 436-444). Supports both sync and async callbacks (auto-detected). Args: callback: Function to call with GetTaskResult when status changes. Can return None (sync) or Awaitable[None] (async). Example: >>> task = await client.call_tool("slow_operation", {}, task=True) >>> >>> def on_update(status: GetTaskResult): ... print(f"Task {status.taskId} is now {status.status}") >>> >>> task.on_status_change(on_update) >>> result = await task # Callback fires when status changes N)rfappend)rHr|s r;on_status_changezTask.on_status_changes!0 %%h/////r=c.K||jr8tjtj}t |jd||ddS|j |j}|S|j |jd{V|_|jS)zGet current task status. If server executed immediately, returns synthetic completed status. Otherwise queries the server for current status. completedNi)taskIdrq createdAt lastUpdatedAtttl pollInterval) rkrcrnowrutcrrardr`get_task_status)rHrcacheds r;rqz Task.statuss $$&&&   ,x|,,C }"!!     )'FM$(<#?#? #N#NNNNNNN!!r=rXc KdS)zWait for and return the task result. Must be implemented by subclasses to return the appropriate result type. Nrms r;r:z Task.results r=gr@)statetimeoutr str | Nonerfloatc K||jr|d{VS|jt j|_t j}hd}d} |jr'|jj}| ||vr|jSn ||kr|jSt j|z }||kr td|j d|pdd|d ||z } t j |j t|| d{V|j n?#tj$r-|j|j d{V|_YnwxYw ) aWait for task to reach a specific state or complete. Uses event-based waiting when notifications are available (fast), with fallback to polling (reliable). Optimally wakes up immediately on status changes when server sends notifications/tasks/status. Args: state: Desired state ('submitted', 'working', 'completed', 'failed'). If None, waits for any terminal state (completed/failed) timeout: Maximum time to wait in seconds Returns: GetTaskResult: Final task status Raises: TimeoutError: If desired state not reached within timeout N>failed cancelledrg?TzTask z did not reach zterminal statez within s)r)rkrcrqrerwEventtimerd TimeoutErrorrawait_forwaitminclearr`r) rHrrstartterminal_states poll_intervalcurrentelapsed remainings r;rz Task.waits( $$&&&   '&&&&&& &   %!(D  >>>  W! .,3=/11#112%%--ikkE)G'!!"gDMgg%:SCSgg]dggg ')I W&&++--s=)7T7T"((****' W W W+/<+G+G +V+V%V%V%V%V%V%V""" W3 Ws!AD<<9E87E8cK|jrdS||j|jd{Vd|_dS)aOCancel this task, transitioning it to cancelled state. Sends a tasks/cancel protocol request. The server will attempt to halt execution and move the task to cancelled state. Note: If server executed immediately (graceful degradation), this is a no-op as there's no server-side task to cancel. N)rcrkr` cancel_taskrardrms r;cancelz Task.cancelOsb    F $$&&&l&&t}555555555!r=cN|S)z!Allow 'await task' to get result.)r: __await__rms r;rzTask.__await__`s{{}}&&(((r=rB)r@rr[r\r]r^)rrK)rr\)rrn)rqrrrK)r|r~rrK)rr)rrX)rrrrrr)rRrSrTrUrDrkpropertyr[rpr}rrqabcabstractmethodr:rrrrr=r;rZrZnsC   04 777776"X"""X"KKKK400004""""<     &*E=W=W=W=W=W=W~""""")))))r=rZc0eZdZdZ ddfd Zdd ZxZS)ToolTaskaM Represents a tool call that may execute in background or immediately. Provides a uniform API whether the server accepts background execution or executes synchronously (graceful degradation per SEP-1686). Usage: task = await client.call_tool_as_task("analyze", args) # Check status status = await task.status() # Wait for completion await task.wait() # Get result (waits if needed) result = await task.result() # Returns CallToolResult # Or just await the task directly result = await task Nr@rr[r\ tool_namer]CallToolResult | Nonec\t|||||_dS)a Create a ToolTask wrapper. Args: client: The FastMCP client task_id: The task identifier tool_name: Name of the tool being executed immediate_result: If server executed synchronously, the immediate result N)rCrD _tool_name)rHr@r[rr]rIs r;rDzToolTask.__init__|s- *:;;;#r=rrcXK|j|jS|jr|jJ|j}ny||d{V|j|jd{V}t|trMtj j |}|j|j|dd{V}nt|tj j r)|j|j|dd{V}n|t!|drjt!|drZtj |j|j|j}|j|j|dd{V}n|}||_|S)a#Wait for and return the tool result. If server executed immediately, returns the immediate result. Otherwise waits for background task to complete and retrieves result. Returns: CallToolResult: The parsed tool result (same as call_tool returns) NT)raise_on_errorcontentstructured_content)rstructuredContent_meta)rgrcrbrkrr`get_task_resultrarMdictr&r'rmodel_validate_parse_call_tool_resultrhasattrrrmeta)rHr: raw_result mcp_results r;r:zToolTask.results(   *& &  ( ()555+FF  ( ( * * *))++        $|;;DMJJJJJJJJJ*d++ ( Y5DDZPP #|CCOZ D  J (@AA (#|CCOZ D   :y11 (g 477 ("%!9!9 * 2*4*G(o":""J $(<#G#GD$H$$FF (F% r=rB)r@rr[r\rr\r]r)rrrRrSrTrUrDr:rVrWs@r;rrese637 $$$$$$$&99999999r=rrc0eZdZdZ ddfd Zdd ZxZS) PromptTaskac Represents a prompt call that may execute in background or immediately. Provides a uniform API whether the server accepts background execution or executes synchronously (graceful degradation per SEP-1686). Usage: task = await client.get_prompt_as_task("analyze", args) result = await task # Returns GetPromptResult Nr@rr[r\ prompt_namer] mcp.types.GetPromptResult | Nonec\t|||||_dS)a Create a PromptTask wrapper. Args: client: The FastMCP client task_id: The task identifier prompt_name: Name of the prompt being executed immediate_result: If server executed synchronously, the immediate result N)rCrD _prompt_name)rHr@r[rr]rIs r;rDzPromptTask.__init__s0 *:;;;'r=rmcp.types.GetPromptResultcRK|j|jS|jr|jJ|j}nw||d{V|j|jd{V}tj j |}||_|S)a#Wait for and return the prompt result. If server executed immediately, returns the immediate result. Otherwise waits for background task to complete and retrieves result. Returns: GetPromptResult: The prompt result with messages and description N) rgrcrbrkrr`rrar&r'GetPromptResultr)rHr:rs r;r:zPromptTask.results   *& &   J)555+FF  ( ( * * *))++        $|;;DMJJJJJJJJJY.==jIIF% r=rB)r@rr[r\rr\r]r)rrrrWs@r;rrsf   >B (((((((&r=rc0eZdZdZ ddfd Zdd ZxZS) ResourceTaskaw Represents a resource read that may execute in background or immediately. Provides a uniform API whether the server accepts background execution or executes synchronously (graceful degradation per SEP-1686). Usage: task = await client.read_resource_as_task("file://data.txt") contents = await task # Returns list[ReadResourceContents] Nr@rr[r\urir]Llist[mcp.types.TextResourceContents | mcp.types.BlobResourceContents] | Nonec\t|||||_dS)a Create a ResourceTask wrapper. Args: client: The FastMCP client task_id: The task identifier uri: URI of the resource being read immediate_result: If server executed synchronously, the immediate result N)rCrD_uri)rHr@r[rr]rIs r;rDzResourceTask.__init__s-& *:;;; r=rElist[mcp.types.TextResourceContents | mcp.types.BlobResourceContents]c6K|j|jS|jr|jJ|j}nh||d{V|j|jd{V}t|tj j rt|j }nt|trd|vrg}|dD]}t|trtd|vr8|tj j|S|tj j||||}nt|tr|n|g}||_|S)aWait for and return the resource contents. If server executed immediately, returns the immediate result. Otherwise waits for background task to complete and retrieves result. Returns: list[ReadResourceContents]: The resource contents Ncontentsblob)rgrcrbrkrr`rrarMr&r'ReadResourceResultlistrrrBlobResourceContentsrTextResourceContents)rHr:rparsed_contentsitems r;r:zResourceTask.result0s   *& &  # V)555+FF  ( ( * * *))++        $|;;DMJJJJJJJJJ*ci&BCC Vj122J-- V* 2J2J"$&z2 5 5D!$-- 5!T>>+22 # > M Md S S,22 # > M Md S S(..t4444((2*d'C'CU*% r=rB)r@rr[r\rr\r]r)rrrrWs@r;rr se  &,66666666r=r)rr rr)2rU __future__rrrwrurrEcollections.abcrrrrtypingrr r mcp.typesr&r mcp.client.sessionr r rrrrfastmcp.client.messagesrrfastmcp.utilities.loggingrrRrzfastmcp.client.clientrrr<r?rXABCrZrr'rrrrrrrr=r;rs##""""""  ////////''''''''2222222222 <;;;;;;;;;;;;;;;000000 H  =<<<<<<<<1111h(((((n((($gm$$ t)t)t)t)t)37GK(t)t)t)nccccct$%cccL>>>>>ci/0>>>BZZZZZci,sy/MM NOZZZZZr=