o
    *i                     @   s   d Z ddlmZmZ ddlmZ ddlmZmZ e	Z
ee
ged f Zeg ed f ZeG dd dZeG dd	 d	ZeeB dB ZeG d
d dZdS )z
Message wrapper with metadata support.

This module defines a wrapper type that combines JSONRPCMessage with metadata
to support transport-specific features like resumability.
    )	AwaitableCallable)	dataclass)JSONRPCMessage	RequestIdNc                   @   s@   e Zd ZU dZdZedB ed< dZeege	d f dB ed< dS )ClientMessageMetadataz%Metadata specific to client messages.Nresumption_tokenon_resumption_token_update)
__name__
__module____qualname____doc__r   ResumptionToken__annotations__r	   r   r    r   r   O/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/mcp/shared/message.pyr      s   
 "r   c                   @   sR   e Zd ZU dZdZedB ed< dZedB ed< dZ	e
dB ed< dZe
dB ed< dS )ServerMessageMetadataz%Metadata specific to server messages.Nrelated_request_idrequest_contextclose_sse_streamclose_standalone_sse_stream)r
   r   r   r   r   r   r   r   objectr   CloseSSEStreamCallbackr   r   r   r   r   r      s   
 r   c                   @   s&   e Zd ZU dZeed< dZeed< dS )SessionMessagezAA message with specific metadata for transport-specific features.messageNmetadata)r
   r   r   r   r   r   r   MessageMetadatar   r   r   r   r   -   s   
 r   )r   collections.abcr   r   dataclassesr   	mcp.typesr   r   strr   ResumptionTokenUpdateCallbackr   r   r   r   r   r   r   r   r   <module>   s    