o i@sUddlZddlmZmZddlmZddlZddlmZddl m Z m Z ddl m Z mZddlmZddlmZdd lmZgd Zeeeeee e fgeeBeeeBBfZeed <ee ee BZeed <d ee de fddZdS)N) AwaitableCallable) TypeAlias)CreateMessageResult) ClientSession SamplingFnT)LifespanContextTRequestContext)CreateMessageRequestParams)SamplingMessage)ServerSamplingHandler)SamplingHandlerr SamplingParamsClientSamplingHandlerr sampling_handlerreturncs.dtttfdtdttjjBffdd }|S)Ncontextparamsrc sz&|j||}t|r|IdH}t|tr%tddtjjd|dd}|WSt yE}ztjj tjj t|dWYd}~Sd}~ww)N assistantzfastmcp-clienttext)typer)rolemodelcontent)codemessage) messagesinspect isawaitable isinstancestrrmcptypes TextContent Exception ErrorDataINTERNAL_ERROR)rrresulter]/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/client/sampling.py_sampling_handler"s&   z3create_sampling_callback.._sampling_handler)r rrrrr!r"r%)rr,r*r)r+create_sampling_callbacks  r-)rcollections.abcrrtypingr mcp.typesr!rmcp.client.sessionrrmcp.shared.contextrr r rr fastmcp.server.sampling.handlerr __all__listr r__annotations__r r-r*r*r*r+s4