)`i'* dZddlmZmZddlmZmZmZddlm Z ddl m Z ddl m Z ddlmZerddlmZGd d eZGd d eZGd deZGddeZGddeZGddeZde defde jde je jzfdZde defde jde je jzfdZde defde jdzde j e jzfdZ!de defde j"de j#e jzfdZ$de defde j%de j&de j'e jzfdZ(de defde j)de j&de j'e jzfdZ*eGd d!Z+e(Z,e*Z-dS)"a Experimental task handler protocols for server -> client requests. This module provides Protocol types and default handlers for when servers send task-related requests to clients (the reverse of normal client -> server flow). WARNING: These APIs are experimental and may change without notice. Use cases: - Server sends task-augmented sampling/elicitation request to client - Client creates a local task, spawns background work, returns CreateTaskResult - Server polls client's task status via tasks/get, tasks/result, etc. ) dataclassfield) TYPE_CHECKINGAnyProtocol) TypeAdapterN)RequestContext)RequestResponder) ClientSessioncZeZdZdZdedefdejdejej zfdZ dS)GetTaskHandlerFnTzrHandler for tasks/get requests from server. WARNING: This is experimental and may change without notice. contextr paramsreturnc KdSNselfrrs y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/client/experimental/task_handlers.py__call__zGetTaskHandlerFnT.__call__"s 14N) __name__ __module__ __qualname____doc__r rtypesGetTaskRequestParams GetTaskResult ErrorDatarrrrr r se 4 454*4  u . 444444rr cZeZdZdZdedefdejdejej zfdZ dS)GetTaskResultHandlerFnTzuHandler for tasks/result requests from server. WARNING: This is experimental and may change without notice. rr rrc KdSrrrs rrz GetTaskResultHandlerFnT.__call__/s 8;srN) rrrrr rrGetTaskPayloadRequestParamsGetTaskPayloadResultr rrrrr"r")se ; 45;1;  #eo 5 ;;;;;;rr"c`eZdZdZdedefdejdzdejej zfdZ dS)ListTasksHandlerFnTzsHandler for tasks/list requests from server. WARNING: This is experimental and may change without notice. rr rNrc KdSrrrs rrzListTasksHandlerFnT.__call__<s 36#r) rrrrr rrPaginatedRequestParamsListTasksResultr rrrrr'r'6sj 6 456,t36   0 666666rr'cZeZdZdZdedefdejdejej zfdZ dS)CancelTaskHandlerFnTzuHandler for tasks/cancel requests from server. WARNING: This is experimental and may change without notice. rr rrc KdSrrrs rrzCancelTaskHandlerFnT.__call__Is 473rN) rrrrr rrCancelTaskRequestParamsCancelTaskResultr rrrrr,r,Cse 7 457-7  %/ 1 777777rr,c heZdZdZdedefdejdejdej ej zfdZ dS) TaskAugmentedSamplingFnTaRHandler for task-augmented sampling/createMessage requests from server. When server sends a CreateMessageRequest with task field, this callback is invoked. The callback should create a task, spawn background work, and return CreateTaskResult immediately. WARNING: This is experimental and may change without notice. rr r task_metadatarc KdSrrrrrr2s rrz!TaskAugmentedSamplingFnT.__call__Z 473rN) rrrrr rrCreateMessageRequestParams TaskMetadataCreateTaskResultr rrrrr1r1Psr7 45707) 7  %/ 1 777777rr1c heZdZdZdedefdejdejdej ej zfdZ dS) TaskAugmentedElicitationFnTaHHandler for task-augmented elicitation/create requests from server. When server sends an ElicitRequest with task field, this callback is invoked. The callback should create a task, spawn background work, and return CreateTaskResult immediately. WARNING: This is experimental and may change without notice. rr rr2rc KdSrrr4s rrz$TaskAugmentedElicitationFnT.__call__lr5rN) rrrrr rrElicitRequestParamsr7r8r rrrrr:r:bsr7 457)7) 7  %/ 1 777777rr:rr rrcFKtjtjdS)Nztasks/get not supportedcodemessagerr METHOD_NOT_FOUNDrrs rdefault_get_task_handlerrDts* ?  #)   rcFKtjtjdS)Nztasks/result not supportedr>rArCs rdefault_get_task_result_handlerrF~* ?  #,   rcFKtjtjdS)Nztasks/list not supportedr>rArCs rdefault_list_tasks_handlerrIs* ?  #*   rcFKtjtjdS)Nztasks/cancel not supportedr>rArCs rdefault_cancel_task_handlerrKrGrr2cFKtjtjdS)Nz%Task-augmented sampling not supportedr>rr INVALID_REQUESTrrr2s rdefault_task_augmented_samplingrPs* ?  "7   rcFKtjtjdS)Nz(Task-augmented elicitation not supportedr>rMrOs r"default_task_augmented_elicitationrRs* ?  ":   rcneZdZUdZeeZeed<ee Z e ed<ee Z eed<eeZeed<eeZeed<eeZeed<d ejd zfd Zed ejd efd Zdede fde!ejej"fd d fdZ#d S)ExperimentalTaskHandlersaBContainer for experimental task handlers. Groups all task-related handlers that handle server -> client requests. This includes both pure task requests (get, list, cancel, result) and task-augmented request handlers (sampling, elicitation with task field). WARNING: These APIs are experimental and may change without notice. Example: handlers = ExperimentalTaskHandlers( get_task=my_get_task_handler, list_tasks=my_list_tasks_handler, ) session = ClientSession(..., experimental_task_handlers=handlers) )defaultget_taskget_task_result list_tasks cancel_taskaugmented_samplingaugmented_elicitationrNc|jtu}|jtu}|jt u}|jtu}t||||gsdSd}|s|rftj |r&tj tj nd|r&tj tjnd}tj|rtjnd|rtjnd|S)a4Build ClientTasksCapability from the configured handlers. Returns a capability object that reflects which handlers are configured (i.e., not using the default "not supported" handlers). Returns: ClientTasksCapability if any handlers are provided, None otherwise N) createMessage)create)sampling elicitation)listcancelrequests)rXrIrYrKrZrPr[rRanyrClientTasksRequestsCapabilityTasksSamplingCapabilityTasksCreateMessageCapabilityTasksElicitationCapability TasksCreateElicitationCapabilityClientTasksCapabilityTasksListCapabilityTasksCancelCapability)rhas_list has_cancel has_samplinghas_elicitationrequests_capabilitys rbuild_capabilityz)ExperimentalTaskHandlers.build_capabilitys%?*DD%-HH .6UU 4<^^Hj,HII 4KO  ? "'"E6UEgEiEijjjj"EH5.000D(    rrequestct|jtjtjztjztjzS)z5Check if this handler handles the given request type.) isinstancerootrGetTaskRequestGetTaskPayloadRequestListTasksRequestCancelTaskRequest)rss rhandles_requestz(ExperimentalTaskHandlers.handles_requests: L  5#> >AW WZ_Zq q   rctxr respondercKttjtjz}|jjxtjdxR\}|||d{V}||}| |d{VdSxtj dxR\}| ||d{V}||}| |d{VdSxtj dxR\}| ||d{V}||}| |d{VdStjdxQ\}|||d{V}||}| |d{VdS t!dt#|jj)zHandle a task-related request from the server. Call handles_request() first to check if this handler can handle the request. )rNzUnhandled request type: )rr ClientResultr rsrvrwrVvalidate_pythonrespondrxrWryrXrzrY ValueErrortype)rr|r}client_response_typerresponseclient_responses rhandle_requestz'ExperimentalTaskHandlers.handle_requeststS^   0S S $4%44444VV!%sF!;!;;;;;;;"6"F"Fx"P"P''888888888885 <,;;;;;FF!%!5!5c6!B!BBBBBBB"6"F"Fx"P"P''88888888888< 7'66666vv!%f!=!======="6"F"Fx"P"P''888888888887 (77777!%!1!1#v!>!>>>>>>>"6"F"Fx"P"P''888888888888  !ZDARAWGi9j9j9j6jjj" %"="D" " " " H !4    \ #\ OS0 1#\$E$79K$KL#\  #\#\#\#\#\#\rrT).r dataclassesrrtypingrrrpydanticr mcp.typesrmcp.shared.contextr mcp.shared.sessionr mcp.client.sessionr r r"r'r,r1r:rrr rDr$r%rFr)r*rIr.r/rKr6r7r8rPr<rRrT(default_task_augmented_sampling_callback+default_task_augmented_elicitation_callbackrrrrs  )(((((((////////// ------//////1000000 4 4 4 4 4 4 4 4 ; ; ; ; ;h ; ; ; 6 6 6 6 6( 6 6 6 7 7 7 7 78 7 7 777777x777$77777(777$ OS0 1  & 5?* OS0 1  - %/1 OS0 1  (4 / U_, OS0 1  ) eo- OS0 1  ,% eo-  OS0 1  %% eo-  j\j\j\j\j\j\j\ j\\,K(.P+++r