
    "<i                         U d dl mZ d dlZd dlmZ d dlmZ eej        j	        ej        j
        f         ej        j        z  ez  Zeed<   eZeed<    G d d          ZdS )	    )	TypeAliasN)MessageHandlerFnT)RequestResponderMessageMessageHandlerTc                   ~   e Zd ZdZdeej        j        ej        j        f         ej        j	        z  e
z  ddfdZdeddfdZdeddfdZdeej        j        ej        j        f         ddfdZdej        j        ddfd	Zdej        j        ddfd
Zdej        j        ddfdZdej        j	        ddfdZde
ddfdZdej        j        ddfdZdej        j        ddfdZdej        j        ddfdZdej        j        ddfdZdej        j         ddfdZ!dej        j"        ddfdZ#dej        j$        ddfdZ%dS )MessageHandlerz
    This class is used to handle MCP messages sent to the client. It is used to handle all messages,
    requests, notifications, and exceptions. Users can override any of the hooks
    messagereturnNc                 <   K   |                      |           d {V S N)dispatchselfr
   s     /Users/kimhansen/Desktop/03 Workspace/ceo-agents/chl-effectiveness/mcp-servers/whoop/.venv/lib/python3.11/site-packages/fastmcp/client/messages.py__call__zMessageHandler.__call__   s,       ]]7+++++++++    c                 0  K   |                      |           d {V  |xt          d x\    |                     |           d {V  |j        j        xt
          j        j        d x*\    |                     |j        j                   d {V  d S  xt
          j        j	        d x*\    | 
                    |j        j                   d {V  d S  t
          j        j        d x)\   |                     |j        j                   d {V  d S  d S  xt
          j        j        d x\    |                     |           d {V  |j        xt
          j        j        d x%\    |                     |j                   d {V  d S  xt
          j        j        d x%\    |                     |j                   d {V  d S  xt
          j        j        d x%\    |                     |j                   d {V  d S  xt
          j        j        d x%\    |                     |j                   d {V  d S  xt
          j        j        d x%\    |                     |j                   d {V  d S  xt
          j        j        d x%\    |                     |j                   d {V  d S  t
          j        j        d x$\   |                     |j                   d {V  d S  d S  t:          d x\   |                     |           d {V  d S  d S )N )
on_messager   
on_requestrequestrootmcptypesPingRequeston_pingListRootsRequeston_list_rootsCreateMessageRequeston_create_messageServerNotificationon_notificationCancelledNotificationon_cancelledProgressNotificationon_progressLoggingMessageNotificationon_logging_messageToolListChangedNotificationon_tool_list_changedResourceListChangedNotificationon_resource_list_changedPromptListChangedNotificationon_prompt_list_changedResourceUpdatedNotificationon_resource_updated	Exceptionon_exceptionr   s     r   r   zMessageHandler.dispatch   s2     oog&&&&&&&&&#!###### oog......... o*0.000000"ll7?+?@@@@@@@@@@@ 153555555"001EFFFFFFFFFFF 6799999"44W_5IJJJJJJJJJJJ :99 $  0-//////**7333333333 l:8::::::"//=========== ;97999999"..w|<<<<<<<<<<< :?=??????"55glCCCCCCCCCCC @@>@@@@@@"77EEEEEEEEEEE ADBDDDDDD";;GLIIIIIIIIIII EB@BBBBBB"99',GGGGGGGGGGG C>@@@@@"66w|DDDDDDDDDDD A@@% 0* ''00000000000 r   c                 
   K   d S r   r   r   s     r   r   zMessageHandler.on_messageL         r   c                 
   K   d S r   r   r   s     r   r   zMessageHandler.on_requestO          	r   c                 
   K   d S r   r   r   s     r   r   zMessageHandler.on_pingT   r5   r   c                 
   K   d S r   r   r   s     r   r   zMessageHandler.on_list_rootsW   r5   r   c                 
   K   d S r   r   r   s     r   r!   z MessageHandler.on_create_messageZ   r5   r   c                 
   K   d S r   r   r   s     r   r#   zMessageHandler.on_notification]   r5   r   c                 
   K   d S r   r   r   s     r   r3   zMessageHandler.on_exception`   r5   r   c                 
   K   d S r   r   r   s     r   r'   zMessageHandler.on_progressc   r5   r   c                 
   K   d S r   r   r   s     r   r)   z!MessageHandler.on_logging_messagef   r7   r   c                 
   K   d S r   r   r   s     r   r+   z#MessageHandler.on_tool_list_changedk   r7   r   c                 
   K   d S r   r   r   s     r   r-   z'MessageHandler.on_resource_list_changedp   r7   r   c                 
   K   d S r   r   r   s     r   r/   z%MessageHandler.on_prompt_list_changedu   r7   r   c                 
   K   d S r   r   r   s     r   r1   z"MessageHandler.on_resource_updatedz   r7   r   c                 
   K   d S r   r   r   s     r   r%   zMessageHandler.on_cancelled   r5   r   )&__name__
__module____qualname____doc__r   r   r   ServerRequestClientResultr"   r2   r   r   r   r   r   r   r   r   r   r    r!   r#   r3   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r$   r%   r   r   r   r	   r	      s        
,!#)"939;Q"QR
)
&'
,
 
, , , ,,1g ,1$ ,1 ,1 ,1 ,1\ D    '	(?AW(WX	   
SY%: t    39+E $    sy/M RV    SY-I d    )     )G D    y;	   
y<	   
y@	   
y>	   
y<	   
#)*I d      r   r	   )typingr   	mcp.typesr   mcp.client.sessionr   mcp.shared.sessionr   r   rH   rI   r"   r2   r   __annotations__r   r	   r   r   r   <module>rO      s               0 0 0 0 0 0 / / / / / / SY,ci.DDE	i"#     / . . .p p p p p p p p p pr   