o c۷i1@sdZddlmZddlmZmZmZmZmZm Z dedBde fddZ dedBd e edBd e dBddfd d Z d e eddfddZdS)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_capsNreturncCs.|durdS|jdur dS|jjdurdSdS)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 r K/home/ubuntu/vllm_env/lib/python3.10/site-packages/mcp/server/validation.pycheck_sampling_tools_capabilitys   rr tool_choicecCs0|dus|durt|stttdddSdS)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 r rr r rvalidate_sampling_tools&srmessagescCs|sdS|dj}tdd|D}t|dkr|djnd}|o)tdd|D}|rGtdd|Dr9td |durAtd |sGtd |ra|rcd d |D}dd |D}||kretddSdSdS)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 Ncs|]}|jdkVqdS tool_resultNtype.0cr r r Uz4validate_tool_use_result_messages..csr)tool_useNrrr r rrXrcss|]}|jdkVqdSrrrr r rr]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 messagecSh|] }|jdkr|jqS)r")ridrr r r ez4validate_tool_use_result_messages..cSr#)r)r toolUseIdrr r rr%fr&zPids 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_idsr r r!validate_tool_use_result_messages@s( r2)__doc__mcp.shared.exceptionsr mcp.typesrrrrrrboolrlistrr2r r r rs