o i'* @sdZddlmZmZddlmZmZmZddlm Z ddl m Z ddl m Z ddlmZer4ddlmZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZde defde jde je jBfddZde defde jde je jBfddZde defde jdBde j e jBfddZ!de defde j"de j#e jBfdd Z$de defde j%d!e j&de j'e jBfd"d#Z(de defde j)d!e j&de j'e jBfd$d%Z*eGd&d'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) ClientSessionc@8eZdZdZdedefdejdejej BfddZ dS) GetTaskHandlerFnTzrHandler for tasks/get requests from server. WARNING: This is experimental and may change without notice. contextr paramsreturncdSNselfr rrrY/home/ubuntu/.local/lib/python3.10/site-packages/mcp/client/experimental/task_handlers.py__call__"zGetTaskHandlerFnT.__call__N) __name__ __module__ __qualname____doc__rrtypesGetTaskRequestParams GetTaskResult ErrorDatarrrrrr   r c@r ) GetTaskResultHandlerFnTzuHandler for tasks/result requests from server. WARNING: This is experimental and may change without notice. r r rrcrrrrrrrr/rz GetTaskResultHandlerFnT.__call__N) rrrrrrrGetTaskPayloadRequestParamsGetTaskPayloadResultrrrrrrr!)r r!c@s<eZdZdZdedefdejdBdejej BfddZ dS) ListTasksHandlerFnTzsHandler for tasks/list requests from server. WARNING: This is experimental and may change without notice. r r rNrcrrrrrrrr<rzListTasksHandlerFnT.__call__) rrrrrrrPaginatedRequestParamsListTasksResultrrrrrrr$6s  r$c@r ) CancelTaskHandlerFnTzuHandler for tasks/cancel requests from server. WARNING: This is experimental and may change without notice. r r rrcrrrrrrrrIrzCancelTaskHandlerFnT.__call__N) rrrrrrrCancelTaskRequestParamsCancelTaskResultrrrrrrr'Cr r'c @>eZdZdZdedefdejdejdej ej BfddZ d S) 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. r r r task_metadatarcrrrrr rr,rrrrZz!TaskAugmentedSamplingFnT.__call__N) rrrrrrrCreateMessageRequestParams TaskMetadataCreateTaskResultrrrrrrr+P  r+c @r*) 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. r r rr,rcrrrr-rrrrlr.z$TaskAugmentedElicitationFnT.__call__N) rrrrrrrElicitRequestParamsr0r1rrrrrrr3br2r3r r rrctjtjddS)Nztasks/get not supportedcodemessagerrMETHOD_NOT_FOUNDr rrrrdefault_get_task_handlert r<cr5)Nztasks/result not supportedr6r9r;rrrdefault_get_task_result_handler~r=r>cr5)Nztasks/list not supportedr6r9r;rrrdefault_list_tasks_handlerr=r?cr5)Nztasks/cancel not supportedr6r9r;rrrdefault_cancel_task_handlerr=r@r,cr5)Nz%Task-augmented sampling not supportedr6rrINVALID_REQUESTr rr,rrrdefault_task_augmented_sampling rDcr5)Nz(Task-augmented elicitation not supportedr6rArCrrr"default_task_augmented_elicitationrErFc@seZdZUdZeedZeed<ee dZ e ed<ee dZ eed<eedZeed<eedZeed<eedZeed<d ejd Bfd d Zed ejd efddZdede fde!ejej"fd d fddZ#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_elicitationrNcCs|jtu}|jtu}|jtu}|jtu}t||||gsdSd}|s$|r?t j |r0t j t dnd|r;t j t dndd}t j|rGt nd|rQt |dSd|dS)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)rKr?rLr@rMrDrNrFanyrClientTasksRequestsCapabilityTasksSamplingCapabilityTasksCreateMessageCapabilityTasksElicitationCapability TasksCreateElicitationCapabilityClientTasksCapabilityTasksListCapabilityTasksCancelCapability)rhas_list has_cancel has_samplinghas_elicitationrequests_capabilityrrrbuild_capabilitys0     z)ExperimentalTaskHandlers.build_capabilityrequestcCs t|jtjtjBtjBtjBS)z5Check if this handler handles the given request type.) isinstancerootrGetTaskRequestGetTaskPayloadRequestListTasksRequestCancelTaskRequest)rerrrhandles_requestsz(ExperimentalTaskHandlers.handles_requestctxr respondercs\ttjtjB}|jjtjdr1d}|||IdH}||}| |IdHdStj drWd}| ||IdH}||}| |IdHdStj dr}d}| ||IdH}||}| |IdHdStjdrd}|||IdH}||}| |IdHdS tdt|jj)zHandle a task-related request from the server. Call handles_request() first to check if this handler can handle the request. )rrNzUnhandled request type: )rr ClientResultrrergrhrIvalidate_pythonrespondrirJrjrKrkrL ValueErrortype)rrmrnclient_response_typerresponseclient_responserrrhandle_requests6     z'ExperimentalTaskHandlers.handle_request)$rrrrrr<rIr __annotations__r>rJr!r?rKr$r@rLr'rDrMr+rFrNr3rr\rd staticmethod ServerRequestboolrlrrr rorwrrrrrGs$ $ rG).r dataclassesrrtypingrrrpydanticr mcp.typesrmcp.shared.contextrmcp.shared.sessionr mcp.client.sessionr r r!r$r'r+r3rrrr<r"r#r>r%r&r?r(r)r@r/r0r1rDr4rFrG(default_task_augmented_sampling_callback+default_task_augmented_elicitation_callbackrrrrs                            n