i dZddlmZddlmZmZmZmZdededefdZ dedefd Z dedefd Z d ed zdd fd Z d ed zdd fdZ y )a Tasks capability checking utilities. This module provides functions for checking and requiring task-related capabilities. All tasks capability logic is centralized here to keep the main session code clean. WARNING: These APIs are experimental and may change without notice. )McpError)INVALID_REQUESTClientCapabilitiesClientTasksCapability ErrorDatarequiredclientreturnc|jy|jy|jjX|jjy|jjj!|jjjy|jjX|jjy|jjj!|jjjyy)a Check if client's tasks capability matches the required capability. Args: required: The capability being checked for client: The client's declared capabilities Returns: True if client has the required capability, False otherwise TF)requests elicitationcreatesampling createMessage)rr s j/root/aria/mcps/aria-brain/venv/lib/python3.12/site-packages/mcp/shared/experimental/tasks/capabilities.pycheck_tasks_capabilityrs  $$0 ?? & & .    ( ( / / ;**119!!- ?? # # +    % % 3 3 ?''55= capsc|jy|jjy|jjjy|jjjjduS)zACheck if capabilities include task-augmented elicitation support.NF)tasksr r rrs rhas_task_augmented_elicitationr:s\ zz zz" zz&&. ::   * * 1 1 ==rc|jy|jjy|jjjy|jjjjduS)z>Check if capabilities include task-augmented sampling support.NF)rr rrrs rhas_task_augmented_samplingrEs\ zz zz" zz##+ ::   ' ' 5 5T AAr client_capsNcR| t|stttdy)a Raise McpError if client doesn't support task-augmented elicitation. Args: client_caps: The client's declared capabilities, or None if not initialized Raises: McpError: If client doesn't support task-augmented elicitation Nz2Client does not support task-augmented elicitationcodemessage)rrrrrs r"require_task_augmented_elicitationr!Ps3"@"M $L   #NrcR| t|stttdy)z Raise McpError if client doesn't support task-augmented sampling. Args: client_caps: The client's declared capabilities, or None if not initialized Raises: McpError: If client doesn't support task-augmented sampling Nz/Client does not support task-augmented samplingr)rrrrr s rrequire_task_augmented_samplingr#cs3"=k"J $I   #Kr)__doc__mcp.shared.exceptionsr mcp.typesrrrrboolrrrr!r#rrr)s+### !# #L>);>>B&8BTB 4F4M RV & 1Cd1J t r