o *i @sdZddlmZddlmZmZmZmZdededefddZ d edefd d Z d edefd d Z dedBddfddZ dedBddfddZ dS)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 ErrorDatarequiredclientreturncCs|jdurdS|jdurdS|jjdur,|jjdurdS|jjjdur,|jjjdur,dS|jjdurJ|jjdur:dS|jjjdurJ|jjjdurJdSdS)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)rrrg/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/mcp/shared/experimental/tasks/capabilities.pycheck_tasks_capabilitys"      rcapscC@|jdurdS|jjdurdS|jjjdurdS|jjjjduS)zACheck if capabilities include task-augmented elicitation support.NF)tasksr r r rrrrhas_task_augmented_elicitation:  rcCr)z>Check if capabilities include task-augmented sampling support.NF)rr r rrrrrhas_task_augmented_samplingErr client_capsNcC$|dust|stttdddS)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)rrrrrrrr"require_task_augmented_elicitationP rcCr)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)rrrrrrrrrequire_task_augmented_samplingcr r!)__doc__mcp.shared.exceptionsr mcp.typesrrrrboolrrrrr!rrrrs  &