)`idZddlmZddlmZmZmZddlmZddl m Z m Z ddl m Z mZddlmZerddlmZed ej ZGd d ZdS) u8 Experimental server session features for server→client task operations. This module provides the server-side equivalent of ExperimentalClientFeatures, allowing the server to send task-augmented requests to the client and poll for results. WARNING: These APIs are experimental and may change without notice. ) AsyncIterator) TYPE_CHECKINGAnyTypeVarN)validate_sampling_tools!validate_tool_use_result_messages)"require_task_augmented_elicitationrequire_task_augmented_sampling)poll_until_terminal) ServerSessionResultT)boundceZdZdZddZdedejfdZded e e de fd Z dede ejfd Z d d dedejdedejfdZd ddddddddd deejdedededzdejdzdedzdeedzdeeefdzdejdzdeejdzdejdzdejfdZdS)!ExperimentalServerSessionFeaturesu Experimental server session features for server→client task operations. This provides the server-side equivalent of ExperimentalClientFeatures, allowing the server to send task-augmented requests to the client and poll for results. WARNING: These APIs are experimental and may change without notice. Access via session.experimental: result = await session.experimental.elicit_as_task(...) sessionr returnNc||_dS)N)_session)selfrs |/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/server/experimental/session_features.py__init__z*ExperimentalServerSessionFeatures.__init__)s  task_idc K|jtjtjtj|tjd{VS)z Send tasks/get to the client to get task status. Args: task_id: The task identifier Returns: GetTaskResult containing the task status taskIdparamsN)r send_requesttypes ServerRequestGetTaskRequestGetTaskRequestParams GetTaskResult)rrs rget_taskz*ExperimentalServerSessionFeatures.get_task,sq]//   4E>cg *;777"m88  # 8 '(7"/C888    "          $+~~g..       ! /))'53EFFFFFFFFFs C) r/ system_promptinclude_context temperaturestop_sequencesmetadatamodel_preferencestools tool_choicemessages max_tokensr?r@rArBrCrDrErFc BK|jjr|jjjnd} t| t | | | t ||jtjtj tj |||||||| | | tj | tj d{V} | j j}||23d{V}6||tjd{VS)a5 Send a task-augmented sampling request and poll until complete. The client will create a local task, process the sampling request asynchronously, and return the result when ready. Args: messages: The conversation messages for sampling max_tokens: Maximum tokens in the response ttl: Task time-to-live in milliseconds 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 Nr.) rG maxTokens systemPromptincludeContextrA stopSequencesrCmodelPreferencesrE toolChoicer3r)rr4r5r rrrr r!CreateMessageRequestCreateMessageRequestParamsr8r9r3rr-r*CreateMessageResult)rrGrHr/r?r@rArBrCrDrErFr;r<rr=s rcreate_message_as_taskz8ExperimentalServerSessionFeatures.create_message_as_tasksrRCG-B]gdm1>>cg ' 444 UK@@@)(333"m88  * ;!)",%2'6$/&4!)):##."/C888     "  "%        * $+~~g..       ! /))'53LMMMMMMMMMs2C:)rr rN)__name__ __module__ __qualname____doc__rstrr r$r%typer r*rr-ElicitRequestedSchemaintr:r>listSamplingMessageIncludeContextfloatdictrModelPreferencesTool ToolChoicerRrSrrrrs5        c  e.A       ']       (s}U=P/Q. 1G1G1G1G41G  1G   1G1G1G1Gp$(7;$(+/*.;?)-/3HNHNHNu,-HN HN  HN Tz HN-4HNT\HNS D(HNsCx.4'HN!1D8HNEJ$&HN%,HN  "HNHNHNHNHNHNrr)rWcollections.abcrtypingrrr mcp.typesr mcp.server.validationrr*mcp.shared.experimental.tasks.capabilitiesr r %mcp.shared.experimental.tasks.pollingr mcp.server.sessionr Resultr rrdrrrms*)))))..........\\\\\\\\FEEEEE1000000 ')5< 0 0 0ANANANANANANANANANANr