o iۂ@sUdZddlmZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z ddlmZmZmZddlmZmZmZmZddlZddlZddlmZmZddlmZdd lmZddl m!Z!dd l"m#Z#dd l$m%Z%dd l&m'Z'dd l(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1m2Z2ddl3m4Z4m5Z5ddl6m7Z7ddl8m9Z9e:e;Ze?e@efZAdeBd<e e!jCZDdeBd<eEeDeAfZFdeBd<eGdZHdeBd<GdddZIed-d$d%ZJGd&d'd'ee=e>fZKd.d+d,ZLdS)/ay MCP Server Module This module provides a framework for creating an MCP (Model Context Protocol) server. It allows you to easily define and handle various types of requests and notifications in an asynchronous manner. Usage: 1. Create a Server instance: server = Server("your_server_name") 2. Define request handlers using decorators: @server.list_prompts() async def handle_list_prompts(request: types.ListPromptsRequest) -> types.ListPromptsResult: # Implementation @server.get_prompt() async def handle_get_prompt( name: str, arguments: dict[str, str] | None ) -> types.GetPromptResult: # Implementation @server.list_tools() async def handle_list_tools(request: types.ListToolsRequest) -> types.ListToolsResult: # Implementation @server.call_tool() async def handle_call_tool( name: str, arguments: dict | None ) -> list[types.TextContent | types.ImageContent | types.EmbeddedResource]: # Implementation @server.list_resource_templates() async def handle_list_resource_templates() -> list[types.ResourceTemplate]: # Implementation 3. Define notification handlers if needed: @server.progress_notification() async def handle_progress( progress_token: str | int, progress: float, total: float | None, message: str | None ) -> None: # Implementation 4. Run the server: async def main(): async with mcp.server.stdio.stdio_server() as (read_stream, write_stream): await server.run( read_stream, write_stream, InitializationOptions( server_name="your_server_name", server_version="your_version", capabilities=server.get_capabilities( notification_options=NotificationOptions(), experimental_capabilities={}, ), ), ) asyncio.run(main()) The Server class provides methods to register handlers for various MCP requests and notifications. It automatically manages the request context and handles incoming messages from the client. ) annotationsN) AsyncIterator AwaitableCallableIterable)AbstractAsyncContextManagerAsyncExitStackasynccontextmanager)AnyGeneric TypeAliascast)MemoryObjectReceiveStreamMemoryObjectSendStream)AnyUrl)TypeVar) Experimental)ExperimentalHandlers)create_call_wrapper)ReadResourceContents)InitializationOptions) ServerSession)RequestContext)McpErrorUrlElicitationRequiredError)ServerMessageMetadataSessionMessage)RequestResponder)validate_and_warn_tool_nameLifespanResultT)defaultRequestTr StructuredContentUnstructuredContentCombinationContent request_ctxz?contextvars.ContextVar[RequestContext[ServerSession, Any, Any]]c@seZdZ   d d ddZdS) NotificationOptionsFprompts_changedboolresources_changed tools_changedcCs||_||_||_dSN)r'r)r*)selfr'r)r*r-Q/home/app/Keep/.python/lib/python3.10/site-packages/mcp/server/lowlevel/server.py__init__qs zNotificationOptions.__init__N)FFF)r'r(r)r(r*r()__name__ __module__ __qualname__r/r-r-r-r.r&ps r&_!Server[LifespanResultT, RequestT]returnAsyncIterator[dict[str, Any]]cCs iVdS)zDefault lifespan context manager that does nothing. Args: server: The server instance this lifespan is managing Returns: An empty context object Nr-)r3r-r-r.lifespan|s r7c@seZdZddddefd`d d Z  dadbddZdcddZeddddZededd Z d!d"Z d#d$Z d%d&Z d'd(Z d)d*Zd+d,Zd-d.Zd/d0Zd1d2Zdfd5d6Zdgd9d:Zd;d<dhd?d@ZdAdBZdCdDZ E EdidjdMdNZ EdkdldUdVZdmdZd[Zdnd^d_ZdS)oServerNnamestrversion str | None instructions website_urliconslist[types.Icon] | Noner7[Callable[[Server[LifespanResultT, RequestT]], AbstractAsyncContextManager[LifespanResultT]]cCsR||_||_||_||_||_||_tjti|_ i|_ i|_ d|_ t d|dS)NzInitializing server %r)r9r;r=r>r?r7typesZ PingRequest _ping_handlerrequest_handlersnotification_handlers _tool_cache_experimental_handlersloggerdebug)r,r9r;r=r>r?r7r-r-r.r/s zServer.__init__notification_optionsNotificationOptions | Noneexperimental_capabilities dict[str, dict[str, Any]] | Noner5rcCsHd dd}t|j|jr|jn|d||pt|pi|j|j|jdS) z8Create initialization options from this server instance.packager:r5cSs,z ddlm}||WStyYdSw)Nr)r;unknown)importlib.metadatar; Exception)rNr;r-r-r. pkg_versions   z9Server.create_initialization_options..pkg_versionZmcp)Z server_nameZserver_version capabilitiesr=r>r?N)rNr:r5r:)rr9r;get_capabilitiesr&r=r>r?)r,rJrLrRr-r-r.create_initialization_optionss  z$Server.create_initialization_optionsr&dict[str, dict[str, Any]]types.ServerCapabilitiesc Csd}d}d}d}d}tj|jvrtj|jd}tj|jvr%tjd|jd}tj|jvr2tj |j d}tj |jvr.decorator..handler)rlrm)rHrIrrBr_rDrirurfrsr. decorator   z&Server.list_prompts..decorator)rirjr-r,rxr-rfr. list_promptszServer.list_promptscrh)NriHCallable[[str, dict[str, str] | None], Awaitable[types.GetPromptResult]]c(tddfdd }|jtj<S)Nz(Registering handler for GetPromptRequestrltypes.GetPromptRequestcs$|jj|jjIdH}t|Sr+)paramsr9 argumentsrBrp)rlZ prompt_getrir-r.rus z5Server.get_prompt..decorator..handler)rlr)rHrIrDrBZGetPromptRequestrwrfrr.rxs  z$Server.get_prompt..decorator)rir}r-rzr-rfr. get_prompts zServer.get_promptcrh)Nri|Callable[[], Awaitable[list[types.Resource]]] | Callable[[types.ListResourcesRequest], Awaitable[types.ListResourcesResult]]crk)Nz,Registering handler for ListResourcesRequestrltypes.ListResourcesRequestcrn)N)r[)rorBZListResourcesResultrprqrsr-r.ru1rvz9Server.list_resources..decorator..handler)rlr)rHrIrrBr`rDrwrfrsr.rx)ryz(Server.list_resources..decorator)rirr-rzr-rfr.list_resources(r|zServer.list_resourcescrh)Nri5Callable[[], Awaitable[list[types.ResourceTemplate]]]cr~)Nz4Registering handler for ListResourceTemplatesRequestr3r cs IdH}ttj|dS)N)ZresourceTemplates)rBrpZListResourceTemplatesResult)r3Z templatesrr-r.ruCs zBServer.list_resource_templates..decorator..handler)r3r )rHrIrDrBZListResourceTemplatesRequestrwrfrr.rx@  z1Server.list_resource_templates..decorator)rirr-rzr-rfr.list_resource_templates? zServer.list_resource_templatescrh)NriKCallable[[AnyUrl], Awaitable[str | bytes | Iterable[ReadResourceContents]]]cr~)Nz+Registering handler for ReadResourceRequestrltypes.ReadResourceRequestcsjjIdH}ddfdd |td rn td r$nn}tjd td d |d}n&td rR}fd d|D}t tj |dS t dt |t tj |gdS)Ndata str | bytes mime_typer<metadict[str, Any] | Nonecs|durd|ini}|tdr$}tjdjj||pdd|StdrB}tjdjjt| |p.decorator..handler..create_contentr-zdReturning str or bytes from read_resource is deprecated. Use Iterable[ReadResourceContents] instead.) stacklevelc s$g|]}|j|jt|ddqS)rN)contentrgetattr).0Z content_item)rr-r. rs zLServer.read_resource..decorator..handler..)contentsz+Unexpected return type from read_resource: r+)rrrr<rr) rrr:rwarningswarnDeprecationWarningrrBrpZReadResourceResult ValueErrortype)rlrrrrrZ contents_listr)rrlr.ruRs:.   z8Server.read_resource..decorator..handler)rlr)rHrIrDrBZReadResourceRequestrwrfrr.rxMs  4z'Server.read_resource..decorator)rirr-rzr-rfr. read_resourceLs.decorator..handler)rlr)rHrIrDrBrbrwrfrr.rxrz+Server.set_logging_level..decorator)rirr-rzr-rfr.set_logging_levelrzServer.set_logging_levelcrh)Nri#Callable[[AnyUrl], Awaitable[None]]cr~)Nz(Registering handler for SubscribeRequestrltypes.SubscribeRequestcrr+rrrBrprrrr-r.rurz=Server.subscribe_resource..decorator..handler)rlr)rHrIrDrBZSubscribeRequestrwrfrr.rxrz,Server.subscribe_resource..decoratorrirr-rzr-rfr.subscribe_resourcerzServer.subscribe_resourcecrh)Nrircr~)Nz*Registering handler for UnsubscribeRequestrltypes.UnsubscribeRequestcrr+rrrr-r.rurz?Server.unsubscribe_resource..decorator..handler)rlr)rHrIrDrBZUnsubscribeRequestrwrfrr.rxrz.Server.unsubscribe_resource..decoratorrr-rzr-rfr.unsubscribe_resourcerzServer.unsubscribe_resourcecrh)NripCallable[[], Awaitable[list[types.Tool]]] | Callable[[types.ListToolsRequest], Awaitable[types.ListToolsResult]]cs6tdt|tjdfdd }|jtj<|S)Nz(Registering handler for ListToolsRequestrltypes.ListToolsRequestcs|IdH}t|tjr$|jD] }t|j|j|j<qt|Sj|D] }t|j|j|j<q+ttj|dS)N)r\) rorBZListToolsResultr\rr9rFrpclear)rlrrtool)r,rtr-r.rus      z5Server.list_tools..decorator..handler)rlr)rHrIrrBrarDrwrfrsr.rxs   z$Server.list_tools..decorator)rirr-rzr-rfr. list_toolsszServer.list_tools error_messagetypes.ServerResultcCs ttjtjd|dgddS)z3Create a ServerResult with an error CallToolResult.rrrT)risError)rBrpCallToolResult TextContent)r,rr-r-r._make_error_results zServer._make_error_result tool_nametypes.Tool | Nonecs^||jvrtj|jvrtd||jtjdIdH|j|}|dur-td||S)z|Get tool definition from cache, refreshing if necessary. Returns the Tool object if found, None otherwise. z(Tool cache miss for %s, refreshing cacheNz5Tool '%s' not listed, no validation will be performed)rFrBrarDrHrIrewarning)r,rrr-r-r._get_cached_tool_definitions     z"Server._get_cached_tool_definitionT)validate_inputrr(csdfdd }|S)aRegister a tool call handler. Args: validate_input: If True, validates input against inputSchema. Default is True. The handler validates input against inputSchema (if validate_input=True), calls the tool function, and builds a CallToolResult with the results: - Unstructured content (iterable of ContentBlock): returned in content - Structured content (dict): returned in structuredContent, serialized JSON text returned in content - Both: returned in content and structuredContent If outputSchema is defined, validates structuredContent or errors if missing. riCallable[..., Awaitable[UnstructuredContent | StructuredContent | CombinationContent | types.CallToolResult | types.CreateTaskResult]]cs,tddfdd }|jtj<S)Nz'Registering handler for CallToolRequestrltypes.CallToolRequestc sz|jj}|jjp i}|IdH}r@|r@z tj||jdWntjy?}zd|j WYd}~WSd}~ww||IdH}t |t j rTt |WSt |t jr`t |WSt |trst|dkrstt|\}}n1t |trtt|}t jdtj|dddg}nt|drtt|}d}n dt|jWS|r|jdur|durd WSz tj||jdWntjy}zd |j WYd}~WSd}~wwt t j t||d d WStyty }z t |WYd}~Sd}~ww) N)instanceZschemazInput validation error: rr)indentr__iter__z"Unexpected return type from tool: zOOutput validation error: outputSchema defined but no structured output returnedzOutput validation error: F)rZstructuredContentr)!rr9rr jsonschemavalidateZ inputSchemaZValidationErrorrmessagerorBrrpZCreateTaskResulttuplelenr r$dictr"rjsondumpshasattrr#rr0Z outputSchemalistrrQr:)rlrrreresultsZunstructured_contentZmaybe_structured_content)rir,rr-r.ru sd            z4Server.call_tool..decorator..handler)rlr)rHrIrDrBZCallToolRequestrwr,rrr.rxs  Az#Server.call_tool..decoratorN)rirr-)r,rrxr-rr. call_toolsRzServer.call_toolcrh)NriGCallable[[str | int, float, float | None, str | None], Awaitable[None]]cr~)Nz,Registering handler for ProgressNotificationrltypes.ProgressNotificationcs*|jj|jj|jj|jjIdHdSr+)rZ progressTokenprogresstotalrrrr-r.ruUsz@Server.progress_notification..decorator..handler)rlr)rHrIrErBZProgressNotificationrwrfrr.rxPs  z/Server.progress_notification..decorator)rirr-rzr-rfr.progress_notificationOszServer.progress_notificationcsdfdd }|S)z7Provides completions for prompts and resource templatesriCallable[[types.PromptReference | types.ResourceTemplateReference, types.CompletionArgument, types.CompletionContext | None], Awaitable[types.Completion | None]]cr~)Nz'Registering handler for CompleteRequestrltypes.CompleteRequestcsP|jj|jj|jjIdH}ttj|dur|dStjgddddS)N)valuesrZhasMore) completion)rrefargumentcontextrBrpZCompleteResultZ Completion)rlrrr-r.ruqsz5Server.completion..decorator..handler)rlr)rHrIrDrBrcrwrfrr.rxes  z$Server.completion..decoratorN)rirr-rzr-rfr.rbszServer.completionF read_stream5MemoryObjectReceiveStream[SessionMessage | Exception] write_stream&MemoryObjectSendStream[SessionMessage]initialization_optionsraise_exceptions statelessc st4IdHz}|||IdH}|t||||dIdH}|jr)|jjnd} | dur>| ||| IdHt 4IdH&} |j 2z3dHW} t d| | |j| |||qJ6WdIdHn 1IdHsrwYWdIdHdS1IdHswYdS)N)rzReceived message: %s)renter_async_contextr7rrG task_supportZconfigure_sessionrunanyioZcreate_task_groupZincoming_messagesrHrIZ start_soon_handle_message) r,rrrrrstacklifespan_contextsessionrtgrr-r-r.rs8    *.z Server.runr`RequestResponder[types.ClientRequest, types.ServerResult] | types.ClientNotification | Exceptionrrrrc s2tjdd}|tdrAdtjdr@d}}|||||||IdHWdnA1s:wYn8tjdrYd}||IdHntdrwt d||j dd d d IdH|rv|n|D] } t d | j j| jqzWddS1swYdS) NT)recordrequestr)rootr-z Received exception from stream: errorzInternal Server Errorzmcp.server.exception_handler)rrrHzWarning: %s: %s)rcatch_warningsrrBZ ClientRequest_handle_requestZClientNotification_handle_notificationrQrHrZsend_log_messageinfocategoryr0r) r,rrrrwrlZ respondernotifyrr-r-r.rs20   "zServer._handle_message9RequestResponder[types.ClientRequest, types.ServerResult]rltypes.ClientRequestTypec stdt|j|jt|}rtdt|jd}zzbd}d} d} |jdur>t|jt r>|jj }|jj } |jj } |j rE|j jnd} |jrN|jjnd} d} t|drc|jdurct|jdd} tt|j|j||t| | || d|| | d}||IdH}WnRty}z |j}WYd}~nBd}~wtytd|jYW|durt|dSdSty}z|r|t j!dt"|dd }WYd}~nd}~wwW|durt|n |durt|ww|#|IdHn|#t j!t j$d d IdHtd dS) NzProcessing request of type %szDispatching request of type %srtask) task_metadataZ_client_capabilities_sessionZ _task_support)rclose_sse_streamclose_standalone_sse_streamz4Request %s cancelled - duplicate response suppressedr)coderrzMethod not found)r rz Response sent)%rHrrr0rDrerIZmessage_metadatarorrgr r Z client_paramsrSrGrrrrr%setrZ request_idZ request_metarrrrZget_cancelled_exc_classresetrQrBZ ErrorDatar:ZrespondMETHOD_NOT_FOUND)r,rrlrrrrutokenZ request_dataZclose_sse_stream_cbZclose_standalone_sse_stream_cbZclient_capabilitiesrrresponseerrr-r-r.rs     zServer._handle_requestrr cs`|jt|}r.tdt|jz ||IdHWdSty-tdYdSwdS)Nz#Dispatching notification of type %sz*Uncaught exception in notification handler)rErerrHrIr0rQ exception)r,rrur-r-r.rs zServer._handle_notification) r9r:r;r<r=r<r>r<r?r@r7rA)NN)rJrKrLrMr5r)rJr&rLrVr5rW)r5rd)r5r)rr:r5r)rr:r5r)rr()FF) rrrrrrrr(rr()F)rrrrrrrr() rrrlrrrrrrr()rr )r0r1r2r7r/rUrTpropertyrgr^r{rrrrrrrrrrrrrrrrrr-r-r-r.r8sH  .    ?   ! c' 2 Pr8rtypes.PingRequestrcsttSr+)rBrprrr-r-r.rC"srC)r3r4r5r6)rrr5r)M__doc__ __future__rZ _annotationsr contextvarsrr]rcollections.abcrrrr contextlibrrr typingr r r r rrZanyio.streams.memoryrrZpydanticrZtyping_extensionsrZ mcp.typesrBZ'mcp.server.experimental.request_contextrZ mcp.server.lowlevel.experimentalrZ#mcp.server.lowlevel.func_inspectionrZ mcp.server.lowlevel.helper_typesrZmcp.server.modelsrZmcp.server.sessionrZmcp.shared.contextrZmcp.shared.exceptionsrrZmcp.shared.messagerrZmcp.shared.sessionrZmcp.shared.tool_name_validationr getLoggerr0rHrr!rr:r"__annotations__Z ContentBlockr#rr$ ContextVarr%r&r7r8rCr-r-r-r.sX C