~hUddlZddlmZmZddlmZddlZddlmZm Z ddl m Z ddl m Z mZddlmZddlmZgd ZGd d e Zeeeeeee fgee zeee zzfZeed <d ede fdZy)N) AwaitableCallable) TypeAlias) ClientSessionCreateMessageResult) SamplingFnT)LifespanContextTRequestContext)CreateMessageRequestParams)SamplingMessage)r SamplingParams MessageResultSamplingHandlerceZdZUdZej j ed<ej jej jzed<dZ e ed<y)r assistantrolecontentz client-modelmodelN) __name__ __module__ __qualname__rmcptypesRole__annotations__ TextContent ImageContentrstrS/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/fastmcp/client/sampling.pyrrs=&D#))..& YY " "SYY%;%; ;;E3r rrsampling_handlerreturncdtttfdtdtt j jzffd }|S)Ncontextparamsr#cK |j||}tj|r |d{}t|tr+t t jjd|}|S7A#t$rL}t jjt jjt |cYd}~Sd}~wwxYww)Ntext)typer()r)codemessage) messagesinspect isawaitable isinstancerrrrr Exception ErrorDataINTERNAL_ERROR)r%r&resulter"s r!_sampling_handlerz3create_sampling_callback.._sampling_handler s %foovwGF""6*%&#&&II11vF1KM & 99&&YY--A'  sFC.A7A5AA74C5A77 C ACC CC  C)r rr r rrrr1)r"r5s` r!create_sampling_callbackr6sD /? ?@ syy22 2( r )r-collections.abcrrtypingr mcp.typesrrrmcp.client.sessionrmcp.shared.contextr r r r r __all__rlistrrrr6rr r!r>s/2*?B% S ' & _}&667   #0C*C DD F;r