.h/ddlmZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZmZddlmZmZmZmZdd lmZdd lmZmZdd lmZmZd d lm Z d dl!m!Z!Gddej"Z#Gdde#ej"Z$GddeZ%Gdde$Z&GddeZ'Gdde$Z(dS)) annotationsN)AbstractAsyncContextManagerAsyncExitStack) timedelta)Path)AnyLiteral)MemoryObjectReceiveStreamMemoryObjectSendStream) ClientSessionStdioServerParametersTool stdio_client) sse_client)CallToolResultJSONRPCMessage) NotRequired TypedDict) UserError)loggerceZdZdZejdZeejddZejdZ ejddZ ejdd Z dS) MCPServerz.Base class for Model Context Protocol servers.c KdS)zConnect to the server. For example, this might mean spawning a subprocess or opening a network connection. The server is expected to remain connected until `cleanup()` is called. Nselfs N/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/mcp/server.pyconnectzMCPServer.connects returnstrcdS)A readable name for the server.Nrrs rnamezMCPServer.names  r c KdS)zwCleanup the server. For example, this might mean closing a subprocess or closing a network connection. Nrrs rcleanupzMCPServer.cleanup%s r list[MCPTool]c KdS)'List the tools available on the server.Nrrs r list_toolszMCPServer.list_tools,  r tool_name argumentsdict[str, Any] | Nonerc KdS)Invoke a tool on the server.Nrrr-r.s r call_toolzMCPServer.call_tool1r,r Nr!r"r!r(r-r"r.r/r!r) __name__ __module__ __qualname____doc__abcabstractmethodrpropertyr%r'r+r3rr rrrs88      X                 r rcjeZdZdZddZejdd Zd Zd Z d Z d Z ddZ ddZ dZdS)_MCPServerWithClientSessionzUBase class for MCP servers that use a `ClientSession` to communicate with the server.cache_tools_listboolclient_session_timeout_seconds float | Nonecd|_t|_tj|_||_||_d|_d|_ dS)a Args: cache_tools_list: Whether to cache the tools list. If `True`, the tools list will be cached and only fetched from the server once. If `False`, the tools list will be fetched from the server on each call to `list_tools()`. The cache can be invalidated by calling `invalidate_tools_cache()`. You should set this to `True` if you know the server will not change its tools list, because it can drastically improve latency (by avoiding a round-trip to the server every time). client_session_timeout_seconds: the read timeout passed to the MCP ClientSession. NT) sessionr exit_stackasyncioLock _cleanup_lockr@rB _cache_dirty _tools_list)rr@rBs r__init__z$_MCPServerWithClientSession.__init__:sN.2 *8*:*:+2<>> 0.L+!15r r!AbstractAsyncContextManager[tuple[MemoryObjectReceiveStream[JSONRPCMessage | Exception], MemoryObjectSendStream[JSONRPCMessage]]]cdS)"Create the streams for the server.Nrrs rcreate_streamsz*_MCPServerWithClientSession.create_streamsQs  r c>K|d{V|SN)rrs r __aenter__z&_MCPServerWithClientSession.__aenter__]s+llnn r c>K|d{VdSrR)r')rexc_type exc_value tracebacks r __aexit__z%_MCPServerWithClientSession.__aexit__as,llnnr cd|_dS)zInvalidate the tools cache.TN)rJrs rinvalidate_tools_cachez2_MCPServerWithClientSession.invalidate_tools_cacheds r c K |j|d{V}|\}}|jt|||jrt |jndd{V}|d{V||_dS#t$r7}tj d|| d{Vd}~wwxYw)zConnect to the server.N)secondszError initializing MCP server: ) rFenter_async_contextrPr rBr initializerE Exceptionrerrorr')r transportreadwriterEes rrz#_MCPServerWithClientSession.connecths< "oAA$BUBUBWBWXXXXXXXXI#KD% O??:Id&IJJJJ G$$&& & & & & & & &"DLLL    L>1>> ? ? ?,,..  sB#B)) C*32C%%C*r(cK|jstd|jr|js|jr|jSd|_|jd{Vj|_|jS)r*=Server not initialized. Make sure you call `connect()` first.FN)rErr@rJrKr+toolsrs rr+z&_MCPServerWithClientSession.list_tools}s| ][\\ \   $): $t?O $# #"#',"9"9";";;;;;;;Br r-r"r.r/rctK|jstd|j||d{VS)r1rfN)rErr3r2s rr3z%_MCPServerWithClientSession.call_toolsJ| ][\\ \\++IyAAAAAAAAAr c6K|j4d{V |jd{Vn.#t$r!}t jd|Yd}~nd}~wwxYwd|_n #d|_wxYw dddd{VdS#1d{VswxYwYdS)zCleanup the server.NzError cleaning up server: )rIrFacloser_rr`rE)rrds rr'z#_MCPServerWithClientSession.cleanupsc% $ $ $ $ $ $ $ $ $o,,.......... ? ? ? =!==>>>>>>>> ? $ t #####  $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $sHB2A( AAA(AA( B( A11B BBN)r@rArBrCr!rMr5r6)r7r8r9r:rLr;r<rPrSrXrZrr+r3r'rr rr?r?7s__6666.       !!!*    BBBB$$$$$r r?cZeZdZUdZded< ded< ded< ded < d ed < d ed <dS)MCPServerStdioParamszkMirrors `mcp.client.stdio.StdioServerParameters`, but lets you pass params without another import. r"commandzNotRequired[list[str]]argsNotRequired[dict[str, str]]envzNotRequired[str | Path]cwdzNotRequired[str]encodingz3NotRequired[Literal['strict', 'ignore', 'replace']]encoding_error_handlerNr7r8r9r:__annotations__rr rrmrmszLLLU    *%$$$<    AdOOOOr rmcLeZdZdZ ddfd ZddZeddZxZS)MCPServerStdiozMCP server implementation that uses the stdio transport. See the [spec] (https://spec.modelcontextprotocol.io/specification/2024-11-05/basic/transports/#stdio) for details. FNparamsrmr@rAr% str | NonerBrCc |t||t|d|dg|d|d|dd|dd |_|pd |jj|_d S) aCreate a new MCP server based on the stdio transport. Args: params: The params that configure the server. This includes the command to run to start the server, the args to pass to the command, the environment variables to set for the server, the working directory to use when spawning the process, and the text encoding used when sending/receiving messages to the server. cache_tools_list: Whether to cache the tools list. If `True`, the tools list will be cached and only fetched from the server once. If `False`, the tools list will be fetched from the server on each call to `list_tools()`. The cache can be invalidated by calling `invalidate_tools_cache()`. You should set this to `True` if you know the server will not change its tools list, because it can drastically improve latency (by avoiding a round-trip to the server every time). name: A readable name for the server. If not provided, we'll create one from the command. client_session_timeout_seconds: the read timeout passed to the MCP ClientSession. rnrorqrrrszutf-8rtstrict)rnrorqrrrsrtzstdio: N)superrLr getrzrn_namerrzr@r%rB __class__s rrLzMCPServerStdio.__init__s0 )+IJJJ+9%FB'' 5!! 5!!ZZ G44#)::.F#Q#Q     <    Xr r)) __future__rr;rG contextlibrrdatetimerpathlibrtypingrr anyio.streams.memoryr r mcpr r rMCPToolrmcp.client.sser mcp.typesrrtyping_extensionsrr exceptionsrrABCrr?rmrxrrrr rrs@"""""" BBBBBBBBRRRRRRRRSSSSSSSSSSSS%%%%%%4444444444444444""""""          Fe$e$e$e$e$)SWe$e$e$P9:999990999x Q Q Q Q Q Q Q Q 88888.88888r