jO!dZddlmZddlmZmZmZddlmZddl 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"||_d|_dS)N)_sessiontask_session_scope)selfrs VC:\Projects\DSCons\.venv\Lib\site-packages\mcp\server\experimental\session_features.py__init__z*ExperimentalServerSessionFeatures.__init__)s /3task_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_task2sq]//   4E>cg *;777"m88  # 8 '(7"/C888    "          $+~~g..       ! /))'53EFFFFFFFFFs C) r0 system_promptinclude_context temperaturestop_sequencesmetadatamodel_preferencestools tool_choicemessages max_tokensr@rArBrCrDrErFrGc 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/) rH maxTokens systemPromptincludeContextrB stopSequencesrDmodelPreferencesrF toolChoicer4r)rr5r6r rrr r!r"CreateMessageRequestCreateMessageRequestParamsr9r:r4rr.r+CreateMessageResult)rrHrIr0r@rArBrCrDrErFrGr<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 ToolChoicerSrTrrrrs5  3333  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)rXcollections.abcrtypingrrr mcp.typesr!mcp.server.validationrr*mcp.shared.experimental.tasks.capabilitiesr r %mcp.shared.experimental.tasks.pollingr mcp.server.sessionr Resultr rrerrrns*)))))..........\\\\\\\\FEEEEE1000000 ')5< 0 0 0GNGNGNGNGNGNGNGNGNGNr