o c۷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-P/home/ubuntu/vllm_env/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?r7types 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)rOr;r-r-r. pkg_versions   z9Server.create_initialization_options..pkg_versionmcp) server_nameserver_version capabilitiesr=r>r?N)rOr:r5r:)rr9r;get_capabilitiesr&r=r>r?)r,rKrMrSr-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)r}r~)rIrJrrBrdrErzrrwrr. decorator   z&Server.list_prompts..decorator)rzr{r-r,rr-rwr. list_promptszServer.list_promptscry)NrzHCallable[[str, dict[str, str] | None], Awaitable[types.GetPromptResult]]c(tddfdd }|jtj<S)Nz(Registering handler for GetPromptRequestr}types.GetPromptRequestcs$|jj|jjIdH}t|Sr+)paramsr9 argumentsrBr)r} prompt_getrzr-r.rs z5Server.get_prompt..decorator..handler)r}r)rIrJrErBGetPromptRequestrrwrr.rs  z$Server.get_prompt..decorator)rzrr-rr-rwr. get_prompts zServer.get_promptcry)Nrz|Callable[[], Awaitable[list[types.Resource]]] | Callable[[types.ListResourcesRequest], Awaitable[types.ListResourcesResult]]cr|)Nz,Registering handler for ListResourcesRequestr}types.ListResourcesRequestcr)N)r_)rrBListResourcesResultrrrr-r.r1rz9Server.list_resources..decorator..handler)r}r)rIrJrrBrfrErrwrr.r)rz(Server.list_resources..decorator)rzrr-rr-rwr.list_resources(rzServer.list_resourcescry)Nrz5Callable[[], Awaitable[list[types.ResourceTemplate]]]cr)Nz4Registering handler for ListResourceTemplatesRequestr3r cs IdH}ttj|dS)N)resourceTemplates)rBrListResourceTemplatesResult)r3 templatesrr-r.rCs zBServer.list_resource_templates..decorator..handler)r3r )rIrJrErBListResourceTemplatesRequestrrwrr.r@  z1Server.list_resource_templates..decorator)rzrr-rr-rwr.list_resource_templates? zServer.list_resource_templatescry)NrzKCallable[[AnyUrl], Awaitable[str | bytes | Iterable[ReadResourceContents]]]cr)Nz+Registering handler for ReadResourceRequestr}types.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).0 content_item)rr-r. rs zLServer.read_resource..decorator..handler..)contentsz+Unexpected return type from read_resource: r+)rrrr<rr) rrr:rwarningswarnDeprecationWarningrrBrReadResourceResult ValueErrortype)r}rrrr contents_listr)rr}r.rRs:.   z8Server.read_resource..decorator..handler)r}r)rIrJrErBReadResourceRequestrrwrr.rMs  4z'Server.read_resource..decorator)rzrr-rr-rwr. read_resourceLs.decorator..handler)r}r)rIrJrErBrjrrwrr.rrz+Server.set_logging_level..decorator)rzrr-rr-rwr.set_logging_levelrzServer.set_logging_levelcry)Nrz#Callable[[AnyUrl], Awaitable[None]]cr)Nz(Registering handler for SubscribeRequestr}types.SubscribeRequestcrr+rrrBrrrrr-r.rrz=Server.subscribe_resource..decorator..handler)r}r)rIrJrErBSubscribeRequestrrwrr.rrz,Server.subscribe_resource..decoratorrzrr-rr-rwr.subscribe_resourcerzServer.subscribe_resourcecry)Nrzrcr)Nz*Registering handler for UnsubscribeRequestr}types.UnsubscribeRequestcrr+rrrr-r.rrz?Server.unsubscribe_resource..decorator..handler)r}r)rIrJrErBUnsubscribeRequestrrwrr.rrz.Server.unsubscribe_resource..decoratorrr-rr-rwr.unsubscribe_resourcerzServer.unsubscribe_resourcecry)NrzpCallable[[], Awaitable[list[types.Tool]]] | Callable[[types.ListToolsRequest], Awaitable[types.ListToolsResult]]cs6tdt|tjdfdd }|jtj<|S)Nz(Registering handler for ListToolsRequestr}types.ListToolsRequestcs|IdH}t|tjr$|jD] }t|j|j|j<qt|Sj|D] }t|j|j|j<q+ttj|dS)N)r`) rrBListToolsResultr`rr9rGrclear)r}rtool)r,rr-r.rs      z5Server.list_tools..decorator..handler)r}r)rIrJrrBrhrErrwrr.rs   z$Server.list_tools..decorator)rzrr-rr-rwr. list_toolsszServer.list_tools error_messagetypes.ServerResultcCs ttjtjd|dgddS)z3Create a ServerResult with an error CallToolResult.rrrT)risError)rBrCallToolResult 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)rGrBrhrErIrJrvwarning)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. rzCallable[..., Awaitable[UnstructuredContent | StructuredContent | CombinationContent | types.CallToolResult | types.CreateTaskResult]]cs,tddfdd }|jtj<S)Nz'Registering handler for CallToolRequestr}types.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)instanceschemazInput validation error: rr)indentr__iter__z"Unexpected return type from tool: zOOutput validation error: outputSchema defined but no structured output returnedzOutput validation error: F)rstructuredContentr)!rr9rr jsonschemavalidate inputSchemaValidationErrorrmessagerrBrrCreateTaskResulttuplelenr r$dictr"rjsondumpshasattrr#rr0 outputSchemalistrrRr:)r}rrreresultsunstructured_contentmaybe_structured_content)rzr,rr-r.r sd            z4Server.call_tool..decorator..handler)r}r)rIrJrErBCallToolRequestrr,rrr.rs  Az#Server.call_tool..decoratorN)rzrr-)r,rrr-rr. call_toolsRzServer.call_toolcry)NrzGCallable[[str | int, float, float | None, str | None], Awaitable[None]]cr)Nz,Registering handler for ProgressNotificationr}types.ProgressNotificationcs*|jj|jj|jj|jjIdHdSr+)r progressTokenprogresstotalrrrr-r.rUsz@Server.progress_notification..decorator..handler)r}r )rIrJrFrBProgressNotificationrrwrr.rPs  z/Server.progress_notification..decorator)rzrr-rr-rwr.progress_notificationOszServer.progress_notificationcsdfdd }|S)z7Provides completions for prompts and resource templatesrzCallable[[types.PromptReference | types.ResourceTemplateReference, types.CompletionArgument, types.CompletionContext | None], Awaitable[types.Completion | None]]cr)Nz'Registering handler for CompleteRequestr}types.CompleteRequestcsP|jj|jj|jjIdH}ttj|dur|dStjgddddS)N)valuesr hasMore) completion)rrefargumentcontextrBrCompleteResult Completion)r}rrr-r.rqsz5Server.completion..decorator..handler)r}r)rIrJrErBrlrrwrr.res  z$Server.completion..decoratorN)rzrr-rr-rwr.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_contextr7rrH task_supportconfigure_sessionrunanyiocreate_task_groupincoming_messagesrIrJ start_soon_handle_message) r,rrrrrstacklifespan_contextsessionr!tgrr-r-r.r#s8    *.z Server.runr`RequestResponder[types.ClientRequest, types.ServerResult] | types.ClientNotification | Exceptionr+rr*rc 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)rrrIzWarning: %s: %s)rcatch_warningsrrB ClientRequest_handle_requestClientNotification_handle_notificationrRrIr2send_log_messageinfocategoryr0r) r,rr+r*rwr} respondernotifyrr-r-r.r(s20   "zServer._handle_message9RequestResponder[types.ClientRequest, types.ServerResult]r}types.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_metadata_client_capabilities_session _task_support)r0close_sse_streamclose_standalone_sse_streamz4Request %s cancelled - duplicate response suppressedr)coderrzMethod not found)rGrz Response sent)%rIr9rr0rErvrJmessage_metadatarrrxrErF client_paramsrWrHr!rrrr%setr request_id request_metarrr2r$get_cancelled_exc_classresetrRrB ErrorDatar:respondMETHOD_NOT_FOUND)r,rr}r+r*rrtoken request_dataclose_sse_stream_cbclose_standalone_sse_stream_cbclient_capabilitiesr!rAresponseerrr-r-r.r5s     zServer._handle_requestr=r cs`|jt|}r.tdt|jz ||IdHWdSty-tdYdSwdS)Nz#Dispatching notification of type %sz*Uncaught exception in notification handler)rFrvrrIrJr0rR exception)r,r=rr-r-r.r7s zServer._handle_notification) r9r:r;r<r=r<r>r<r?r@r7rA)NN)rKrLrMrNr5r)rKr&rMrZr5r[)r5ru)r5r)rr:r5r)rr:r5r)rr()FF) rrrrrrrr(rr()F)rr-r+rr*rrr() rr>r}r?r+rr*rrr()r=r )r0r1r2r7r/rYrXpropertyrxrbrrrrrrrrrrrrrrr#r(r5r7r-r-r-r.r8sH  .    ?   ! c' 2 Pr8r0types.PingRequestrcsttSr+)rBrrr/r-r-r.rD"srD)r3r4r5r6)r0r[r5r)M__doc__ __future__r _annotationsr contextvarsrrarcollections.abcrrrr contextlibrrr typingr r r r r$ranyio.streams.memoryrrpydanticrtyping_extensionsr mcp.typesrB'mcp.server.experimental.request_contextr mcp.server.lowlevel.experimentalr#mcp.server.lowlevel.func_inspectionr mcp.server.lowlevel.helper_typesrmcp.server.modelsrmcp.server.sessionrmcp.shared.contextrmcp.shared.exceptionsrrmcp.shared.messagerrmcp.shared.sessionrmcp.shared.tool_name_validationr getLoggerr0rIrr!rr:r"__annotations__ ContentBlockr#rr$ ContextVarr%r&r7r8rDr-r-r-r.sX C