jfdZddlmZddlmZmZmZmZdededefdZ dedefd Z dedefd Z d ed zdd fd Z d ed zdd fdZ d S)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"|jdS|jdS|jj2|jjdS|jjj|jjjdS|jj2|jjdS|jjj|jjjdSdS)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 NTF)requests elicitationcreatesampling createMessage)rr s XC:\Projects\DSCons\.venv\Lib\site-packages\mcp\shared\experimental\tasks\capabilities.pycheck_tasks_capabilityrs t u$0 ? & .5   ( / ;*19u!- ? # +5   % 3 ?'5=u 4capsc|jdS|jjdS|jjjdS|jjjjduS)zACheck if capabilities include task-augmented elicitation support.NF)tasksr r rrs rhas_task_augmented_elicitationr:sH zu z"u z&.u :  * 1 ==rc|jdS|jjdS|jjjdS|jjjjduS)z>Check if capabilities include task-augmented sampling support.NF)rr rrrs rhas_task_augmented_samplingrEsH zu z"u z#+u :  ' 5T AAr client_capsNcn|t|s#tttddS)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!PsJ"@"M"M $L      rcn|t|s#tttddS)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#csJ"=k"J"J $I      r)__doc__mcp.shared.exceptionsr mcp.typesrrrrboolrrrr!r#rrr)s5+*****### !# ####L>);>>>>>B&8BTBBBB 4F4M RV    & 1Cd1J t      r