kKjRdZddlmZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z ddl mZmZmZddlZddlmZmZddlmZmZdd lmZdd lmZeeZerdd lmZm Z Gd d eZ!edZ"GddejFee"Z$Gdde$dZ%Gdde$ejLjNZ(Gdde$e)ejLjTejLjVzZ,y)zSEP-1686 client Task classes.) annotationsN) AwaitableCallable)datetimetimezone) TYPE_CHECKINGGenericTypeVar) GetTaskResultTaskStatusNotification)MessageMessageHandler) ToolError) get_logger)CallToolResultClientc0eZdZdZdfd Zdfd ZxZS)TaskNotificationHandlerzEMessageHandler that routes task status notifications to Task objects.cVt|tj||_yN)super__init__weakrefref _client_ref)selfclient __class__s e/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/client/tasks.pyrz TaskNotificationHandler.__init__s 07 F0CcKt|tjjrGt|jt r-|j }|r|j|jt|%|d{y7w)z7Dispatch messages, including task status notifications.N) isinstancemcptypesServerNotificationrootr r _handle_task_status_notificationrdispatch)rmessagerrs rr(z TaskNotificationHandler.dispatch"s` gsyy;; <',,(>?))+;;GLLIgw'''sA>B BB )rr)r)r returnNone)__name__ __module__ __qualname____doc__rr( __classcell__rs@rrrsOD((r r TaskResultTceZdZdZ d ddZddZeddZeddZddZ ddZ dd Z e jdd Zdd d  dd ZdddZddZdZy)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) Ncx||_||_||_|du|_d|_d|_g|_d|_y)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)rrtask_idimmediate_results rrz Task.__init__=sP  !1-T94837  37r c|jry |jj}y#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.)r9r6session RuntimeError)r_es r_check_client_connectedzTask._check_client_connectedXsL      $$A O  s& A ;Ac|jS)zGet the task ID.)r7rs rr>z Task.task_idis}}r c|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 )r9rGs rreturned_immediatelyzTask.returned_immediatelyns!!!r cF||_|j|jj|jD]5} ||}t j |rt j|7y#t$r$}tjd|dYd}~ad}~wwxYw)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) r:r;setr<inspect isawaitableasyncio create_task Exceptionloggerwarning)rstatuscallbackresultrDs r_handle_status_notificationz Task._handle_status_notificationxs$    )    " " $..H K!&)&&v.''/ /  K!6qc:TJJ Ks2A33 B <BB c:|jj|y)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)r<append)rrUs ron_status_changezTask.on_status_changes0 %%h/r crK|j|jr>tjtj }t |jd||ddS|j|j}|S|jj|jd{|_|jS7w)zGet current task status. If server executed immediately, returns synthetic completed status. Otherwise queries the server for current status. completedNi)taskIdrT createdAt lastUpdatedAtttl pollInterval) rEr9rnowrutcr r7r:r6get_task_status)rrbcacheds rrTz Task.statuss $$&   ,,x||,C }}"!!      )''FM$(<<#?#? #NN!!!OsBB7B5 B7c Kyw)zWait for and return the task result. Must be implemented by subclasses to return the appropriate result type. NrGs rrVz Task.results sr@)statetimeoutc ^K|j|jr|jd{S|jt j |_t j }dh}d} |jr9|jj}|||vr|jS||k(r |jSt j |z }||k\r#td|jd|xsdd|d||z } t j|jjt|| d{|jj7*7"#tj$r6|jj|jd{7|_YNwxYww) 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 ('working', 'input_required', 'completed', 'failed', 'cancelled'). If None, waits until the task exits the 'working' state (completed, failed, cancelled, input_required, etc.) timeout: Maximum time to wait in seconds Returns: GetTaskResult: Final task status Raises: TimeoutError: If desired state not reached within timeout Nworkingg?zTask z did not reach zterminal statez within srj)rEr9rTr;rOEventtimer: TimeoutErrorr7wait_forwaitminclearr6rd) rrirjstartin_progress_states poll_intervalcurrentelapsed remainings rrsz Task.waits( $$&   & &    %!(D  '[ !!,,33=&88#111%---iikE)G'!"DMM?/%:SCS9TT\]d\eefg ')I W&&&&++-s=)7T""((*1'>'' W+/<<+G+G +V%V%V" WsR0F-EC F-failed cancelledr\rnN)rsrTr:)rrjterminal_statesrTs r_wait_terminalzTask._wait_terminalsi ?yyy11mm?2"&D 99W955F mm?2  2 6s'A%A!.A% A# A%A%#A%cK|jry|j|jj|jd{d|_y7 w)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)r9rEr6 cancel_taskr7r:rGs rcancelz Task.cancel sI     $$&ll&&t}}555! 6sAAA Ac>|jjS)z!Allow 'await task' to get result.)rV __await__rGs rrzTask.__await__1s{{}&&((r r)rrr>strr?zTaskResultT | None)r*r+)r*r)r*bool)rTr r*r+)rUz1Callable[[GetTaskResult], None | Awaitable[None]]r*r+)r*r )r*r2)riz str | Nonerjfloatr*r )rh)rjrr*r )r,r-r.r/rrEpropertyr>rIrWrZrTabcabstractmethodrVrsrrrrgr rr4r40s  04 777- 76"""K40C0 04"<   &*E=W"=W49=W =W~ "")r r4cDeZdZdZ d dfd ZddZxZS)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 cDt||||||_||_y)aj 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 raise_on_error: Whether task.result() should raise ToolError on errors N)rr _tool_name_raise_on_error)rrr> tool_namer?raise_on_errorrs rrzToolTask.__init__Ms&$ *:;#-r cXK|j |jS|jr|jJ|j}|jrF|jr9|j rUt |j dtjjr$|j dj}t|d|jd}t||j|jd{|jj!|j"d{}t |t$rdtjj&j)|}|jj+|j||jd{}nt |tjj&r;|jj+|j||jd{}nt-|drt-|dr{tjj'|j |j.|j0}|jj+|j||jd{}n|}||_|S77y7 77w) 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) NrzTool 'z' returned an error)rcontentstructured_content)rstructuredContent_meta)r=r9r8is_errorrrr"r#r$ TextContenttextrrrErr6get_task_resultr7dictrmodel_validate_parse_call_tool_resulthasattrrmeta)rrVmsg raw_result mcp_results rrVzToolTask.resultcsR    *&& &   ))5 55++F4#7#7>>jNN1%syy'<'<'!..+00C n$#4??"33FGCn$  ( ( *%%' ' ' $||;;DMMJJJ*d+ YY55DDZP #||CCOO#'#7#7 D  J (@(@A#||CCOO#'#7#7 D :y1g 47"%!9!9 * 2 2*4*G*G(oo":"J $(<<#G#G"'+';';$H$F(F% U (K s]C8J*:J;,J*'J (A2J*J#AJ*9J&:BJ* J( J* J*#J*&J*(J*)NT) rrr>rrrr?zCallToolResult | Nonerr)r*rr,r-r.r/rrVr0r1s@rrr6sK637# ... . 0 .  .,Gr rrc>eZdZdZ d dfd ZddZxZS) 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 c6t||||||_y)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)rr _prompt_name)rrr> prompt_namer?rs rrzPromptTask.__init__s *:;'r cK|j |jS|jr|jJ|j}n~|j|j d{|j j |jd{}tjjj|}||_|S7c78w)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) r=r9r8rErr6rr7r#r$GetPromptResultr)rrVrs rrVzPromptTask.results    *&& &   ))5 55++F  ( ( *%%' ' ' $||;;DMMJJJYY..==jIF%  (Ks$A#C %C &,C C 7C  C r)rrr>rrrr?z mcp.types.GetPromptResult | None)r*zmcp.types.GetPromptResultrr1s@rrrs>  >B ((( ( ; (&r rcBeZdZdZ d dfd Z ddZxZS) 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] c6t||||||_y)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)rr_uri)rrr>urir?rs rrzResourceTask.__init__s& *:; r cdK|j |jS|jr|jJ|j}n^|j|j d{|j j |jd{}t|tjjrt|j}nt|trd|vrg}|dD]}t|trvd|vr9|jtjj j#|P|jtjj$j#||j||}nt|tr|n|g}||_|S7D7w)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)r=r9r8rErr6rr7r"r#r$ReadResourceResultlistrrrYBlobResourceContentsrTextResourceContents)rrVrparsed_contentsitems rrVzResourceTask.resultsu    *&& &   ))5 55++F  ( ( *%%' ' ' $||;;DMMJJJ*cii&B&BCj112J-* 2J"$&z2D!$-!T>+22 # > > M Md S,22 # > > M Md S(..t43)(2*d'C*% ? (Ks%A$F0&F*',F0F-DF0-F0r)rrr>rrrr?zLlist[mcp.types.TextResourceContents | mcp.types.BlobResourceContents] | None)r*zElist[mcp.types.TextResourceContents | mcp.types.BlobResourceContents]rr1s@rrrsE &   ,6 N6r r)-r/ __future__rrrOrMrprcollections.abcrrrrtypingrr r mcp.typesr#r r fastmcp.client.messagesr rfastmcp.exceptionsrfastmcp.utilities.loggingrr,rRfastmcp.client.clientrrrr2ABCr4rr$rrrrrrrgr rrs#"  /'22;;(0 H <(n($m$ C)377GK(C)Ltt$%tn>cii//0>BZcii,,syy/M/MM NOZr