)`iYdZddlmZddlZddlmZddlmZddlm Z m Z ddl m Z ddl mZmZdd lmZdd lmZmZdd lmZdd lmZdd lmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0GddZ1dS)z ServerTaskContext - Server-integrated task context with elicitation and sampling. This wraps the pure TaskContext and adds server-specific functionality: - Elicitation (task.elicit()) - Sampling (task.create_message()) - Status notifications )AnyN)TaskResultHandler) ServerSession)validate_sampling_tools!validate_tool_use_result_messages)McpError)"require_task_augmented_elicitationrequire_task_augmented_sampling) TaskContext) QueuedMessageTaskMessageQueue)Resolver) TaskStore)INVALID_REQUESTTASK_STATUS_INPUT_REQUIREDTASK_STATUS_WORKINGClientCapabilitiesCreateMessageResultCreateTaskResultElicitationCapabilityElicitRequestedSchema ElicitResult ErrorDataIncludeContextModelPreferences RequestIdResultSamplingCapabilitySamplingMessageServerNotificationTask TaskMetadataTaskStatusNotificationTaskStatusNotificationParamsTool ToolChoiceceZdZdZdddedededededzf d Z e d e fd Z e d efd Z e d efd Zd2dZddde ded dfdZdddeded dfdZddde ded dfdZd2dZd2dZd2dZde ded efdZde de de d efdZddddddddd d!eed"ed#e dzd$edzd%e dzd&ee dzd'e!e e"fdzd(e#dzd)ee$dzd*e%dzd e&fd+Z'd,d-de ded.ed efd/Z(d,ddddddddd0 d!eed"ed.ed#e dzd$edzd%e dzd&ee dzd'e!e e"fdzd(e#dzd)ee$dzd*e%dzd e&fd1Z)dS)3ServerTaskContextaE Server-integrated task context with elicitation and sampling. This wraps a pure TaskContext and adds server-specific functionality: - elicit() for sending elicitation requests to the client - create_message() for sampling requests - Status notifications via the session Example: async def my_task_work(task: ServerTaskContext) -> CallToolResult: await task.update_status("Starting...") result = await task.elicit( message="Continue?", requestedSchema={"type": "object", "properties": {"ok": {"type": "boolean"}}} ) if result.content.get("ok"): return CallToolResult(content=[TextContent(text="Done!")]) else: return CallToolResult(content=[TextContent(text="Cancelled")]) N)handlertaskstoresessionqueuer)cjt|||_||_||_||_||_dS)aH Create a ServerTaskContext. Args: task: The Task object store: The task store session: The server session queue: The message queue for elicitation/sampling handler: The result handler for response routing (required for elicit/create_message) )r*r+N)r _ctx_session_queue_handler_store)selfr*r+r,r-r)s x/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/server/experimental/task_context.py__init__zServerTaskContext.__init__Ms8& T777     returnc|jjS)zThe task identifier.)r/task_idr4s r5r:zServerTaskContext.task_idhsy  r7c|jjS)zThe current task state.)r/r*r;s r5r*zServerTaskContext.taskmsy~r7c|jjS)z(Whether cancellation has been requested.)r/ is_cancelledr;s r5r>zServerTaskContext.is_cancelledrsy%%r7c8|jdS)z"Request cancellation of this task.N)r/request_cancellationr;s r5r@z&ServerTaskContext.request_cancellationws &&(((((r7T)notifymessagerAcK|j|d{V|r|d{VdSdS)z Update the task's status message. Args: message: The new status message notify: Whether to send a notification to the client N)r/ update_status_send_notification)r4rBrAs r5rDzServerTaskContext.update_status}sli%%g.........  ,))++ + + + + + + + + + , ,r7resultcK|j|d{V|r|d{VdSdS)z Mark the task as completed with the given result. Args: result: The task result notify: Whether to send a notification to the client N)r/completerE)r4rFrAs r5rHzServerTaskContext.completesli  (((((((((  ,))++ + + + + + + + + + , ,r7errorcK|j|d{V|r|d{VdSdS)z Mark the task as failed with an error message. Args: error: The error message notify: Whether to send a notification to the client N)r/failrE)r4rIrAs r5rKzServerTaskContext.failsjinnU#########  ,))++ + + + + + + + + + , ,r7cK|jj}|jt t t |j|j|j |j |j |j |j d{VdS)z.Send a task status notification to the client.)taskIdstatus statusMessage createdAt lastUpdatedAtttl pollInterval)paramsN)r/r*r0send_notificationr r#r$rMrNrOrPrQrRrS)r4r*s r5rEz$ServerTaskContext._send_notificationsy~m-- &7#{#{&*&8"&.&*&8 H%)%6                r7c|jtts#t t t ddS)z)Check if the client supports elicitation.) elicitationz.Client does not support elicitation capabilitycoderBN)r0check_client_capabilityrrrrrr;s r5_check_elicitation_capabilityz/ServerTaskContext._check_elicitation_capabilitysa}445GTiTkTk5l5l5lmm (L   r7c|jtts#t t t ddS)z&Check if the client supports sampling.)samplingz+Client does not support sampling capabilityrXN)r0rZrrrrrr;s r5_check_sampling_capabilityz,ServerTaskContext._check_sampling_capabilitysa}445GQcQeQe5f5f5fgg (I   r7requestedSchemacK||jtd|j|jt d{V|j|||j}|j }t}||jj |<td|||}|j |j|d{V |d{V}|j|jt d{Vt#j|S#t'j$r.|j|jt d{VwxYw)a Send an elicitation request via the task message queue. This method: 1. Checks client capability 2. Updates task status to "input_required" 3. Queues the elicitation request 4. Waits for the response (delivered via tasks/result round-trip) 5. Updates task status back to "working" 6. Returns the result Args: message: The message to present to the user requestedSchema: Schema defining the expected response structure Returns: The client's response Raises: McpError: If client doesn't support elicitation capability NzEhandler is required for elicit(). Pass handler= to ServerTaskContext.rN)rBr_related_task_idrequesttyperBresolveroriginal_request_id)r[r2 RuntimeErrorr3 update_taskr:rr0_build_elicit_form_requestidr_pending_requestsr r1enqueuewaitrrmodel_validateanyioget_cancelled_exc_class)r4rBr_rc request_idrfqueued response_datas r5elicitzServerTaskContext.elicits4 **,,, = fgg gk%%dl;U%VVVVVVVVV-::+ L;   !( -5ZZ6> ' 3 *     k!!$,777777777 "*--//111111M+))$,?R)SS S S S S S S S.}== =,..   +))$,?R)SS S S S S S S S   sAD33AE7urlelicitation_idcK||jtd|j|jt d{V|j||||j}|j }t}||jj |<td|||}|j |j|d{V |d{V}|j|jt d{Vt#j|S#t'j$r.|j|jt d{VwxYw)a Send a URL mode elicitation request via the task message queue. This directs the user to an external URL for out-of-band interactions like OAuth flows, credential collection, or payment processing. This method: 1. Checks client capability 2. Updates task status to "input_required" 3. Queues the elicitation request 4. Waits for the response (delivered via tasks/result round-trip) 5. Updates task status back to "working" 6. Returns the result Args: message: Human-readable explanation of why the interaction is needed url: The URL the user should navigate to elicitation_id: Unique identifier for tracking this elicitation Returns: The client's response indicating acceptance, decline, or cancellation Raises: McpError: If client doesn't support elicitation capability RuntimeError: If handler is not configured NzIhandler is required for elicit_url(). Pass handler= to ServerTaskContext.ra)rBrvrwrbrcrd)r[r2rhr3rir:rr0_build_elicit_url_requestrkrrlr r1rmrnrrrorprq) r4rBrvrwrcrrrfrsrts r5 elicit_urlzServerTaskContext.elicit_url s@ **,,, = jkk kk%%dl;U%VVVVVVVVV-99) L :   !( -5ZZ6> ' 3 *     k!!$,777777777 "*--//111111M+))$,?R)SS S S S S S S S.}== =,..   +))$,?R)SS S S S S S S S  sAD44AE8) system_promptinclude_context temperaturestop_sequencesmetadatamodel_preferencestools tool_choicemessages max_tokensr{r|r}r~rrrrc K||jjr|jjjnd} t | | | t ||jtd|j |j td{V|j ||||||||| | |j  } | j } t}||jj| <t!d| || }|j|j |d{V |d{V}|j |j t(d{Vt+j|S#t/j$r.|j |j t(d{VwxYw)a Send a sampling request via the task message queue. This method: 1. Checks client capability 2. Updates task status to "input_required" 3. Queues the sampling request 4. Waits for the response (delivered via tasks/result round-trip) 5. Updates task status back to "working" 6. Returns the result Args: messages: The conversation messages for sampling max_tokens: Maximum tokens in the response system_prompt: Optional system prompt include_context: Context inclusion strategy temperature: Sampling temperature stop_sequences: Stop sequences metadata: Additional metadata model_preferences: Model selection preferences tools: Optional list of tools the LLM can use during sampling tool_choice: Optional control over tool usage behavior Returns: The sampling result from the client Raises: McpError: If client doesn't support sampling capability or tools ValueError: If tool_use or tool_result message structure is invalid NzMhandler is required for create_message(). Pass handler= to ServerTaskContext.ra) rrr{r|r}r~rrrrrbrcrd)r^r0 client_params capabilitiesrrr2rhr3rir:r_build_create_message_requestrkrrlr r1rmrnrrrorprq)r4rrr{r|r}r~rrrr client_capsrcrrrfrsrts r5create_messagez ServerTaskContext.create_messagePsX '')))BF-B]gdm1>>cg  UK@@@)(333 = noo ok%%dl;U%VVVVVVVVV-==!'+#)/# L>   !( -5ZZ6> ' 3 *     k!!$,777777777 "*--//111111M+))$,?R)SS S S S S S S S&5mDD D,..   +))$,?R)SS S S S S S S S   s AE::AF>i`rRrRcK|jjr|jjjnd}t||jt d|j|jtd{V|j |||jt|}|j }t}||jj|<td|||}|j|j|d{V |d{V} t'j| } | jj} |jj| 23d{V} 6|jj| t4d{V} |j|jt6d{V| S#t9j$r.|j|jt6d{VwxYw)aJ Send a task-augmented elicitation via the queue, then poll client. This is for use inside a task-augmented tool call when you want the client to handle the elicitation as its own task. The elicitation request is queued and delivered when the client calls tasks/result. After the client responds with CreateTaskResult, we poll the client's task until complete. Args: message: The message to present to the user requestedSchema: Schema defining the expected response structure ttl: Task time-to-live in milliseconds for the client's task Returns: The client's elicitation response Raises: McpError: If client doesn't support task-augmented elicitation RuntimeError: If handler is not configured Nz(handler is required for elicit_as_task()rar)rBr_rbr*rcrd)r0rrr r2rhr3rir:rrjr"rkrrlr r1rmrnrror*rM experimental poll_taskget_task_resultrrrprq)r4rBr_rRrrcrrrfrsrt create_resultclient_task_id_rFs r5elicit_as_taskz ServerTaskContext.elicit_as_tasksu6CG-B]gdm1>>cg *;777 = IJJ Jk%%dl;U%VVVVVVVVV-::+ L#&&& ;   !( -5ZZ6> ' 3 *     k!!$,777777777 "*--//111111M,;MJJM*/6N =5??OO       aP =5EEF +))$,?R)SS S S S S S S SM,..   +))$,?R)SS S S S S S S S  sAF=E#!AF==AH) rRr{r|r}r~rrrrc XK|jjr|jjjnd} t| t | | | t ||jtd|j |j td{V|j |||||||| | | |j t| } | j}t}||jj|<t#d| ||}|j|j |d{V |d{V}t+j|}|jj}|jj|23d{V}6|jj|t8d{V}|j |j t:d{V|S#t=j$r.|j |j t:d{VwxYw)a< Send a task-augmented sampling request via the queue, then poll client. This is for use inside a task-augmented tool call when you want the client to handle the sampling as its own task. The request is queued and delivered when the client calls tasks/result. After the client responds with CreateTaskResult, we poll the client's task until complete. Args: messages: The conversation messages for sampling max_tokens: Maximum tokens in the response ttl: Task time-to-live in milliseconds for the client's task system_prompt: Optional system prompt include_context: Context inclusion strategy temperature: Sampling temperature stop_sequences: Stop sequences metadata: Additional metadata model_preferences: Model selection preferences tools: Optional list of tools the LLM can use during sampling tool_choice: Optional control over tool usage behavior Returns: The sampling result from the client Raises: McpError: If client doesn't support task-augmented sampling or tools ValueError: If tool_use or tool_result message structure is invalid RuntimeError: If handler is not configured Nz0handler is required for create_message_as_task()rar) rrr{r|r}r~rrrrrbr*rcrd) r0rrr rrr2rhr3rir:rrr"rkrrlr r1rmrnrror*rMrrrrrrprq)r4rrrRr{r|r}r~rrrrrrcrrrfrsrtrrrrFs r5create_message_as_taskz(ServerTaskContext.create_message_as_tasksXCG-B]gdm1>>cg ' 444 UK@@@)(333 = QRR Rk%%dl;U%VVVVVVVVV-==!'+#)/# L#&&&>   !( -5ZZ6> ' 3 *     k!!$,777777777 "*--//111111M,;MJJM*/6N =5??OO       aP =5EE#F +))$,?R)SS S S S S S S SM,..   +))$,?R)SS S S S S S S S  s*AG%F  AG%%AH))r8N)*__name__ __module__ __qualname____doc__r!rrr rr6propertystrr:r*boolr>r@rDrrHrKrEr[r^rrrurzlistrintrfloatdictrrr%r&rrrrr7r5r(r(5s<-1     #T)6!!!!X!dX&d&&&X&)))) CG , , ,3 ,4 ,4 , , , ,@D , , ,V , , , , , ,8< , , , , , , , , ,    *??/?  ????BCCC C  CCCCT%)15$(+/*.59#')-\\\'\ \ Tz \ ($. \T\\S D(\sCx.4'\,d2\DzD \ $&\ \\\\F LLLL/L  L  LLLLf$(15$(+/*.59#')-hhh'h h  h Tz h($.hT\hS D(hsCx.4'h,d2hDzD h $&h hhhhhhr7r()2rtypingrrp+mcp.server.experimental.task_result_handlerrmcp.server.sessionrmcp.server.validationrrmcp.shared.exceptionsr*mcp.shared.experimental.tasks.capabilitiesr r %mcp.shared.experimental.tasks.contextr +mcp.shared.experimental.tasks.message_queuer r &mcp.shared.experimental.tasks.resolverr#mcp.shared.experimental.tasks.storer mcp.typesrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r(rr7r5rs( IIIIII,,,,,,\\\\\\\\******>=====WWWWWWWW;;;;;;9999996oooooooooor7