o hh@sddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z ddlmZmZmZmZmZddlmZmZdd lmZmZmZmZdd lmZdd lm Z m!Z!dd l"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*m+Z+ddl,m-Z-ddl.m.Z.ddl/m0Z0ddl1m2Z2m3Z3m4Z4edZ5erddl6m7Z7Gdddej8Z9Gddde9ej8Z:Gddde+Z;Gddde:ZGd#d$d$e+Z?Gd%d&d&e:Z@dS)') annotationsN) Awaitable)AbstractAsyncContextManagerAsyncExitStack) timedelta)Path) TYPE_CHECKINGAnyCallableLiteralTypeVar)MemoryObjectReceiveStreamMemoryObjectSendStream) ClientSessionStdioServerParametersTool stdio_client) sse_client)GetSessionIdCallbackstreamablehttp_client)SessionMessage)CallToolResultGetPromptResultInitializeResultListPromptsResult) NotRequired TypedDict) UserError)logger)RunContextWrapper) ToolFilterToolFilterContextToolFilterStaticT) AgentBasec@seZdZdZd$d%ddZejddZeejd&d d Z ejd dZ ej  d'd(ddZ ejd)ddZ ejd*ddZ ej d+d,d"d#ZdS)- MCPServerz.Base class for Model Context Protocol servers.Fuse_structured_contentboolcCs ||_dS)a Args: use_structured_content: Whether to use `tool_result.structured_content` when calling an MCP tool.Defaults to False for backwards compatibility - most MCP servers still include the structured content in the `tool_result.content`, and using it by default will cause duplicate content. You can set this to True if you know the server will not duplicate the structured content in the `tool_result.content`. Nr()selfr(r,R/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/mcp/server.py__init__"s zMCPServer.__init__cdS)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. Nr,r+r,r,r-connect-szMCPServer.connectreturnstrcCdS)A readable name for the server.Nr,r0r,r,r-name5szMCPServer.namecr/)zwCleanup the server. For example, this might mean closing a subprocess or closing a network connection. Nr,r0r,r,r-cleanup;zMCPServer.cleanupN run_contextRunContextWrapper[Any] | NoneagentAgentBase | None list[MCPTool]cr/)'List the tools available on the server.Nr,)r+r9r;r,r,r- list_toolsBszMCPServer.list_tools tool_name argumentsdict[str, Any] | Nonercr/)Invoke a tool on the server.Nr,r+r@rAr,r,r- call_toolKszMCPServer.call_toolrcr/))List the prompts available on the server.Nr,r0r,r,r- list_promptsPr8zMCPServer.list_promptsr6rcr/)&Get a specific prompt from the server.Nr,r+r6rAr,r,r- get_promptWr8zMCPServer.get_prompt)F)r(r)r2r3NNr9r:r;r<r2r=r@r3rArBr2rr2rNr6r3rArBr2r)__name__ __module__ __qualname____doc__r.abcabstractmethodr1propertyr6r7r?rErGrJr,r,r,r-r's*     r'cseZdZdZ    dFdGfdd ZdHddZdIddZdHd d!Zej dJd#d$Z d%d&Z d'd(Z d)d*Z dKd.d/Zd0d1Z  dLdMd4d5ZdNd;d<ZdOd>d?Z dPdQdBdCZdDdEZZS)R_MCPServerWithClientSessionzUBase class for MCP servers that use a `ClientSession` to communicate with the server.NFr?cache_tools_listr)client_session_timeout_seconds float | None tool_filterr"r(max_retry_attemptsintretry_backoff_seconds_basefloatcsZtj|dd|_t|_t|_||_d|_ ||_ ||_ ||_ 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. tool_filter: The tool filter to use for filtering tools. use_structured_content: Whether to use `tool_result.structured_content` when calling an MCP tool. Defaults to False for backwards compatibility - most MCP servers still include the structured content in the `tool_result.content`, and using it by default will cause duplicate content. You can set this to True if you know the server will not duplicate the structured content in the `tool_result.content`. max_retry_attempts: Number of times to retry failed list_tools/call_tool calls. Defaults to no retries. retry_backoff_seconds_base: The base delay, in seconds, used for exponential backoff between retries. r*NT)superr.sessionr exit_stackasyncioLock _cleanup_lockr[server_initialize_resultr\r_ra _cache_dirty _tools_listr^)r+r[r\r^r(r_ra __class__r,r-r.bs  z$_MCPServerWithClientSession.__init__toolsr=r9RunContextWrapper[Any]r;r&r2cs>|jdur|St|jtr|||jS||||IdHS)z+Apply the tool filter to the list of tools.N)r^ isinstancedict_apply_static_tool_filter_apply_dynamic_tool_filter)r+rnr9r;r,r,r-_apply_tool_filters   z._MCPServerWithClientSession._apply_tool_filter static_filterr$csL|}d|vr|dfdd|D}d|vr$|dfdd|D}|S)z=Apply static tool filtering based on allowlist and blocklist.allowed_tool_namescsg|] }|jvr|qSr,r6.0t) allowed_namesr,r- zI_MCPServerWithClientSession._apply_static_tool_filter..blocked_tool_namescsg|] }|jvr|qSr,rwrx) blocked_namesr,r-r|r}r,)r+rnrufiltered_toolsr,)r{rr-rrsz5_MCPServerWithClientSession._apply_static_tool_filterc st|js td|j}t|||jd}g}|D]?}z|||}t|r,|IdH} n|} | r5||WqtyX} zt d|jd|jd| WYd} ~ qd} ~ ww|S)z>Apply dynamic tool filtering using a callable filter function.z2Tool filter must be callable for dynamic filtering)r9r; server_nameNz$Error applying tool filter to tool 'z ' on server 'z': ) callabler^ ValueErrorr#r6inspect isawaitableappend Exceptionrerror) r+rnr9r;tool_filter_funcfilter_contextrtoolresultshould_includeer,r,r-rss6      z6_MCPServerWithClientSession._apply_dynamic_tool_filterAbstractAsyncContextManager[tuple[MemoryObjectReceiveStream[SessionMessage | Exception], MemoryObjectSendStream[SessionMessage], GetSessionIdCallback | None]]cCr4)"Create the streams for the server.Nr,r0r,r,r-create_streamss z*_MCPServerWithClientSession.create_streamscs|IdH|SrP)r1r0r,r,r- __aenter__sz&_MCPServerWithClientSession.__aenter__cs|IdHdSrP)r7)r+exc_type exc_value tracebackr,r,r- __aexit__sz%_MCPServerWithClientSession.__aexit__cCs d|_dS)zInvalidate the tools cache.TN)rjr0r,r,r-invalidate_tools_caches z2_MCPServerWithClientSession.invalidate_tools_cachefuncCallable[[], Awaitable[T]]r%csld} z|IdHWSty4|d7}|jdkr!||jkr!|jd|d}t|IdHYnwq)NrTr!r)rr_rarfsleep)r+rattemptsbackoffr,r,r-_run_with_retriess z-_MCPServerWithClientSession._run_with_retriesc sz6|j|IdH}|^}}}|jt|||jr!t|jdndIdH}|IdH}||_||_WdSt yS}zt d|| IdHd}~ww)zConnect to the server.N)secondszError initializing MCP server: ) reenter_async_contextrrr\r initializerirdrrrr7)r+ transportreadwrite_rd server_resultrr,r,r-r1s,    z#_MCPServerWithClientSession.connectr:r<cs|jstd|jdusJ|jr|js|jr|j}n|fddIdH}|j|_d|_|j}|}|jdurQ|dusC|durGtd||||IdH}|S)r>=Server not initialized. Make sure you call `connect()` first.NcsSrP)r?r,rdr,r--sz8_MCPServerWithClientSession.list_tools..Fz=run_context and agent are required for dynamic tool filtering) rdrr[rjrkrrnr^rt)r+r9r;rnrrr,rr-r?s"  z&_MCPServerWithClientSession.list_toolsr@r3rArBrcs>|jstd|jdusJ|fddIdHS)rCrNcs SrP)rEr,rArdr@r,r-rAs z7_MCPServerWithClientSession.call_tool..)rdrrrDr,rr-rE:s  z%_MCPServerWithClientSession.call_toolrcs |jstd|jIdHS)rFrN)rdrrGr0r,r,r-rGCsz(_MCPServerWithClientSession.list_promptsr6rcs$|jstd|j||IdHS)rHrN)rdrrJrIr,r,r-rJLsz&_MCPServerWithClientSession.get_promptc s|j4IdH;z*z |jIdHWnty.}ztd|WYd}~nd}~wwWd|_nd|_wWdIdHdS1IdHsIwYdS)zCleanup the server.NzError cleaning up server: )rhreacloserrrrd)r+rr,r,r-r7Us.z#_MCPServerWithClientSession.cleanup)NFrrZ) r[r)r\r]r^r"r(r)r_r`rarb)rnr=r9ror;r&r2r=)rnr=rur$r2r=r2r)rrr2r%rLrMrNrOrPrQ)rRrSrTrUr.rtrrrsrVrWrrrrrr1r?rErGrJr7 __classcell__r,r,rlr-rY_s2 /  *        rYc@sLeZdZUdZded< ded< ded< ded < d ed < d ed <dS)MCPServerStdioParamszkMirrors `mcp.client.stdio.StdioServerParameters`, but lets you pass params without another import. r3commandzNotRequired[list[str]]argsNotRequired[dict[str, str]]envzNotRequired[str | Path]cwdzNotRequired[str]encodingz3NotRequired[Literal['strict', 'ignore', 'replace']]encoding_error_handlerNrRrSrTrU__annotations__r,r,r,r-r`s rcJeZdZdZ       dd fdd Zd!ddZed"ddZZS)#MCPServerStdiozMCP server implementation that uses the stdio transport. See the [spec] (https://spec.modelcontextprotocol.io/specification/2024-11-05/basic/transports/#stdio) for details. FNrrZparamsrr[r)r6 str | Noner\r]r^r"r(r_r`rarbc slt||||||t|d|dg|d|d|dd|ddd |_|p2d |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. tool_filter: The tool filter to use for filtering tools. use_structured_content: Whether to use `tool_result.structured_content` when calling an MCP tool. Defaults to False for backwards compatibility - most MCP servers still include the structured content in the `tool_result.content`, and using it by default will cause duplicate content. You can set this to True if you know the server will not duplicate the structured content in the `tool_result.content`. max_retry_attempts: Number of times to retry failed list_tools/call_tool calls. Defaults to no retries. retry_backoff_seconds_base: The base delay, in seconds, for exponential backoff between retries. rrrrrzutf-8rstrict)rrrrrrzstdio: N)rcr.rgetrr_name r+rr[r6r\r^r(r_rarlr,r-r.s"&     zMCPServerStdio.__init__r2rcCs t|jS)r)rrr0r,r,r-rs zMCPServerStdio.create_streamsr3cC|jSr5rr0r,r,r-r6zMCPServerStdio.nameFNrNFrrZ)rrr[r)r6rr\r]r^r"r(r)r_r`rarbrrK rRrSrTrUr.rrXr6rr,r,rlr-r}s : rc@s8eZdZUdZded< ded< ded< ded<d S) MCPServerSseParamsz1Mirrors the params in`mcp.client.sse.sse_client`.r3urlrheaderszNotRequired[float]timeoutsse_read_timeoutNrr,r,r,r-rs rcr)# MCPServerSsezMCP server implementation that uses the HTTP with SSE transport. See the [spec] (https://spec.modelcontextprotocol.io/specification/2024-11-05/basic/transports/#http-with-sse) for details. FNrrrZrrr[r)r6rr\r]r^r"r(r_r`rarbc 6t||||||||_|pd|jd|_dS)amCreate a new MCP server based on the HTTP with SSE transport. Args: params: The params that configure the server. This includes the URL of the server, the headers to send to the server, the timeout for the HTTP request, and the timeout for the SSE connection. 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 URL. client_session_timeout_seconds: the read timeout passed to the MCP ClientSession. tool_filter: The tool filter to use for filtering tools. use_structured_content: Whether to use `tool_result.structured_content` when calling an MCP tool. Defaults to False for backwards compatibility - most MCP servers still include the structured content in the `tool_result.content`, and using it by default will cause duplicate content. You can set this to True if you know the server will not duplicate the structured content in the `tool_result.content`. max_retry_attempts: Number of times to retry failed list_tools/call_tool calls. Defaults to no retries. retry_backoff_seconds_base: The base delay, in seconds, for exponential backoff between retries. zsse: rNrcr.rrrrlr,r-r.s( zMCPServerSse.__init__r2rcCs4t|jd|jdd|jdd|jdddS) rrrNrrr,)rrrr)rrrr0r,r,r-rs     zMCPServerSse.create_streamsr3cCrrrr0r,r,r-r6*rzMCPServerSse.namer)rrr[r)r6rr\r]r^r"r(r)r_r`rarbrrKrr,r,rlr-rs 4rc@sBeZdZUdZded< ded< ded< ded< d ed <d S) MCPServerStreamableHttpParamszHMirrors the params in`mcp.client.streamable_http.streamablehttp_client`.r3rrrzNotRequired[timedelta | float]rrzNotRequired[bool]terminate_on_closeNrr,r,r,r-r0s rcr)#MCPServerStreamableHttpzMCP server implementation that uses the Streamable HTTP transport. See the [spec] (https://modelcontextprotocol.io/specification/2025-03-26/basic/transports#streamable-http) for details. FNrrrZrrr[r)r6rr\r]r^r"r(r_r`rarbc r)aCreate a new MCP server based on the Streamable HTTP transport. Args: params: The params that configure the server. This includes the URL of the server, the headers to send to the server, the timeout for the HTTP request, and the timeout for the Streamable HTTP connection and whether we need to terminate on close. 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 URL. client_session_timeout_seconds: the read timeout passed to the MCP ClientSession. tool_filter: The tool filter to use for filtering tools. use_structured_content: Whether to use `tool_result.structured_content` when calling an MCP tool. Defaults to False for backwards compatibility - most MCP servers still include the structured content in the `tool_result.content`, and using it by default will cause duplicate content. You can set this to True if you know the server will not duplicate the structured content in the `tool_result.content`. max_retry_attempts: Number of times to retry failed list_tools/call_tool calls. Defaults to no retries. retry_backoff_seconds_base: The base delay, in seconds, for exponential backoff between retries. zstreamable_http: rNrrrlr,r-r.Is) z MCPServerStreamableHttp.__init__r2rc Cs@t|jd|jdd|jdd|jdd|jdd d S) rrrNrrrrrT)rrrrr)rrrr0r,r,r-r~s     z&MCPServerStreamableHttp.create_streamsr3cCrrrr0r,r,r-r6rzMCPServerStreamableHttp.namer)rrr[r)r6rr\r]r^r"r(r)r_r`rarbrrKrr,r,rlr-rCs 5r)A __future__rrVrfrcollections.abcr contextlibrrdatetimerpathlibrtypingrr r r r anyio.streams.memoryr rmcprrrMCPToolrmcp.client.ssermcp.client.streamable_httprrmcp.shared.messager mcp.typesrrrrtyping_extensionsrr exceptionsrrr9r utilr"r#r$r%r;r&ABCr'rYrrrrrrr,r,r,r-sB          @RQ