
    "<i                     2   U d dl Z d dlmZmZ d dlmZ d dlZd dlmZ d dl	m
Z
mZ d dlmZmZ d dlmZ d dlmZ d d	lmZ g d
Zeee         eee
ef         geez  eeez           z  f         Zeed<   ee         ee         z  Zeed<   dee         defdZdS )    N)	AwaitableCallable)	TypeAlias)CreateMessageResult)ClientSessionSamplingFnT)LifespanContextTRequestContext)CreateMessageRequestParams)SamplingMessage)ServerSamplingHandler)SamplingHandlerr   SamplingParamsClientSamplingHandlerr   sampling_handlerreturnc                      dt           t          t          f         dt          dt          t
          j        j        z  f fd}|S )Ncontextparamsr   c                   K   	  |j         ||           }t          j        |          r| d {V }t          |t                    r1t          ddt          j                            d|                    }|S # t          $ rG}t          j        
                    t          j        j        t	          |                    cY d }~S d }~ww xY w)N	assistantzfastmcp-clienttext)typer   )rolemodelcontent)codemessage)messagesinspectisawaitable
isinstancestrr   mcptypesTextContent	Exception	ErrorDataINTERNAL_ERROR)r   r   resulter   s       /Users/kimhansen/Desktop/03 Workspace/ceo-agents/chl-effectiveness/mcp-servers/whoop/.venv/lib/python3.11/site-packages/fastmcp/client/sampling.py_sampling_handlerz3create_sampling_callback.<locals>._sampling_handler"   s      	%%fovwGGF"6** &%&#&& ,$*I11vF1KK  
 M 	 	 	9&&Y-A '        	s   A5A; ;
C<CCC)r
   r   r	   r   r   r$   r%   r(   )r   r-   s   ` r,   create_sampling_callbackr.      sX    /? ?@ 
sy2	2     ,     )r    collections.abcr   r   typingr   	mcp.typesr$   r   mcp.client.sessionr   r   mcp.shared.contextr	   r
   r   r   r   fastmcp.server.sampling.handlerr   __all__listr#   r   __annotations__r   r.    r/   r,   <module>r:      s    / / / / / / / /           # # # # # # 9 9 9 9 9 9 9 9 ? ? ? ? ? ? ? ? B B B B B B % % % % % % A A A A A A
B
B
B $,_}&667
 
	#0C*C DDF$ y    *+.CDT.UU    
+,<=     r/   