o i @sFUddlZddlmZmZddlmZddlZddlZddlm Z ddl m Z ddl m Z mZeeeejjBeeejjBBZeed<eee e fgefeee e fgeefBZeed<d ed eejjfd d Zd eeBd e fddZd ed e fddZdeee e fgefeee e fgeefBd e fddZdS)N) AwaitableCallable) TypeAlias) ClientSession) ListRootsFnT)LifespanContextTRequestContext RootsList RootsHandlerrootsreturncCsg}|D]9}t|tjjr||qt|tjr$|tjj|dqt|tr7|tjjt|dqtd||S)N)urizInvalid root: ) isinstancemcptypesRootappendpydanticFileUrlstr ValueError)r roots_listrrZ/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/client/roots.pyconvert_roots_lists   rhandlercCs2t|tr t|St|rt|Std|)NzInvalid roots handler: )rlist!_create_roots_callback_from_rootsinspect isfunction_create_roots_callback_from_fnr)rrrrcreate_roots_callback!s  r"cs.tdtttfdtjjffdd }|S)Ncontextr cstjjdS)Nr )rrListRootsResult)r#r$rr_roots_callback1sz:_create_roots_callback_from_roots.._roots_callback)rrrrrrr%)r r&rr$rr,s rfncs.dtttfdtjjtjjBffdd }|S)Nr#r c spz|}t|r|IdH}tjjt|dWSty7}ztjjtjjt |dWYd}~Sd}~ww)Nr$)codemessage) r isawaitablerrr%r Exception ErrorDataINTERNAL_ERRORr)r#r er'rrr&=s  z7_create_roots_callback_from_fn.._roots_callback)rrrrrr%r,)r'r&rr/rr!9s  r!)rcollections.abcrrtypingr mcp.typesrrrmcp.client.sessionrmcp.shared.contextrrrrrrr __annotations__r rr"rr!rrrrs>    ,