^sj( dZddlmZddlmZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZmZGd d eZGd d ee j(e j*e j,e j.e j0fZy) a ServerSession Module This module provides the ServerSession class, which manages communication between the server and client in the MCP (Model Context Protocol) framework. It is most commonly used in MCP servers to interact with the client. Common usage pattern: ``` server = Server(name) @server.call_tool() async def handle_tool_call(ctx: RequestContext, arguments: dict[str, Any]) -> Any: # Check client capabilities before proceeding if ctx.session.check_client_capability( types.ClientCapabilities(experimental={"advanced_tools": True}) ): # Perform advanced tool operations result = await perform_advanced_tool_operation(arguments) else: # Fall back to basic tool operations result = await perform_basic_tool_operation(arguments) return result @server.list_prompts() async def handle_list_prompts(ctx: RequestContext) -> list[types.Prompt]: # Access session for any necessary checks or operations if ctx.session.client_params: # Customize prompts based on client initialization parameters return generate_custom_prompts(ctx.session.client_params) else: return default_prompts ``` The ServerSession class is typically used internally by the Server class and should not be instantiated directly by users of the MCP framework. )Enum)AnyN)MemoryObjectReceiveStreamMemoryObjectSendStream)AnyUrl)InitializationOptions) BaseSessionRequestResponderceZdZdZdZdZy)InitializationStateN)__name__ __module__ __qualname__NotInitialized Initializing InitializedE/root/.hermes/venv/lib/python3.12/site-packages/mcp/server/session.pyr r 8sNLKrr ceZdZUejZeed<dZejdzed<de eje zde ejdeddffd Zedejdzfd Zd ej$defd Zd eej,ej.ffd Zdej2ddfdZ d)dej6dededzddfdZdeddfdZ dddddddde!ejDde#dedzdejHdzde%dzde!edzde&eefdzdejNdzdejPfdZ)dejTfd Z+dejXfd!Z- d)d"ee#zd#e%d$e%dzddfd%Z.d*d&Z/d*d'Z0d*d(Z1xZ2S)+ ServerSession _initializedN_client_params read_stream write_stream init_optionsreturnct|||tjtjt j |_||_yN) super__init__types ClientRequestClientNotificationr r_initialization_state _init_options)selfrrr __class__s rr$zServerSession.__init__Js?  u':':EsV)<(J(JL%J;?NE11D8? *.u/C/Ci/OP *-U-A-AB *, *  *#u<rs|%N R3 $ b           "b r