)`i1dZddlmZddlmZmZmZmZmZm Z dedzde fdZ dedzde edzd e dzddfd Z d e eddfd ZdS) z Shared validation functions for server requests. This module provides validation logic for sampling and elicitation requests that is shared across normal and task-augmented code paths. )McpError)INVALID_PARAMSClientCapabilities ErrorDataSamplingMessageTool ToolChoice client_capsNreturnc<|dS|jdS|jjdSdS)z Check if the client supports sampling tools capability. Args: client_caps: The client's declared capabilities Returns: True if client supports sampling.tools, False otherwise NFT)samplingtools)r s i/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/server/validation.pycheck_sampling_tools_capabilityrs3u#u!)u 4r tool_choicecv||4t|s#tttddSdS)a\ Validate that the client supports sampling tools if tools are being used. Args: client_caps: The client's declared capabilities tools: The tools list, if provided tool_choice: The tool choice setting, if provided Raises: McpError: If tools/tool_choice are provided but client doesn't support them Nz1Client does not support sampling tools capability)codemessage)rrrr)r rrs rvalidate_sampling_toolsr&s[  K3.{;; 'O   43rmessagesc|sdS|dj}td|D}t|dkr |djnd}|otd|D}|rJtd|Drtd|td |std |r/|r/d |D}d |D}||krtd dSdSdS)a Validate tool_use/tool_result message structure per SEP-1577. This validation ensures: 1. Messages with tool_result content contain ONLY tool_result content 2. tool_result messages are preceded by a message with tool_use 3. tool_result IDs match the tool_use IDs from the previous message See: https://github.com/modelcontextprotocol/modelcontextprotocol/issues/1577 Args: messages: The list of sampling messages to validate Raises: ValueError: If the message structure is invalid Nc3,K|]}|jdkVdS tool_resultNtype.0cs r z4validate_tool_use_result_messages..Us)IIq16]2IIIIIIrc3,K|]}|jdkVdS)tool_useNrrs rr"z4validate_tool_use_result_messages..Xs)4d4daQVz5I4d4d4d4d4d4drc3,K|]}|jdkVdSrrrs rr"z4validate_tool_use_result_messages..]s)==1qv&======rzHThe last message must contain only tool_result content if any is presentz;tool_result requires a previous message containing tool_usezDtool_result blocks do not match any tool_use in the previous messagec2h|]}|jdk |jS)r&)ridrs r z4validate_tool_use_result_messages..es&OOO!&J:N:N:N:N:Nrc2h|]}|jdk |jS)r)r toolUseIdrs rr*z4validate_tool_use_result_messages..fs&XXX1-@W@W1;@W@W@WrzPids of tool_result blocks and tool_use blocks from previous message do not match)content_as_listanylen ValueError)r last_contenthas_tool_resultsprevious_contenthas_previous_tool_use tool_use_idstool_result_idss r!validate_tool_use_result_messagesr7@sY" B</LIILIIIII7:8}}7I7Ix|33t,d4d4dSc4d4d4d1d1de == === = = ighh h  #Z[[ [$ ecdd dq!1qOO&6OOO XX XXX ? * *opp p qqqq + *r)__doc__mcp.shared.exceptionsr mcp.typesrrrrrr boolrlistrr7rrr>s+*****1Cd1Jt&#d* : d" 4(q_0E(q$(q(q(q(q(q(qr