#j1dZddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl mZddlZddlmZmZmZmZmZddlmZmZdd lmZdd lmZerdd lmZmZm Z m!Z!dd l"m#Z#d(dZ$GddeZ%GddeZ&Gdde%e&Z'GddeZ(Gdde%e(Z)e'e)zZ*e&e(zZ+e*e+zZ,GddeZ-Gdde-Z.d)d'Z/dS)*aCanonical MCP Configuration Format. This module defines the standard configuration format for Model Context Protocol (MCP) servers. It provides a client-agnostic, extensible format that can be used across all MCP implementations. The configuration format supports both stdio and remote (HTTP/SSE) transports, with comprehensive field definitions for server metadata, authentication, and execution parameters. Example configuration: ```json { "mcpServers": { "my-server": { "command": "npx", "args": ["-y", "@my/mcp-server"], "env": {"API_KEY": "secret"}, "timeout": 30000, "description": "My MCP server" } } } ``` ) annotationsN)Path) TYPE_CHECKING AnnotatedAnyLiteralcast)urlparse)AnyUrl BaseModel ConfigDictFieldmodel_validator)Selfoverride)ToolTransformConfig)FastMCPBaseModel)ClientTransport SSETransportStdioTransportStreamableHttpTransport)FastMCPurl str | AnyUrlreturnLiteral['http', 'sse']ct|}|dstd|t|}|j}t jd|rdSdS)zB Infer the appropriate transport type from the given URL. httpz Invalid URL: z/sse(/|\?|&|$)sse)str startswith ValueErrorr pathresearch)r parsed_urlr#s Z/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/mcp_config.pyinfer_transport_type_from_urlr(8sm c((C >>& ! !0...///#J ?D y"D))uvceZdZUdZeeZded< eddZded <edd Z ded <e d e ddZ ddfd Z ddZxZS)_TransformingMCPServerMixinzAA mixin that enables wrapping an MCP Server with tool transforms.default_factoryzdict[str, ToolTransformConfig]toolsNz!The tags to include in the proxy.)default descriptionzset[str] | None include_tagsz!The tags to exclude in the proxy. exclude_tagsbeforemodevaluesdict[str, Any]rct|tret|d}|ddu}|ddu}|s|s|st d|S)aReject if none of the transforming fields are set. This ensures that plain server configs (without tools, include_tags, or exclude_tags) fall through to the base server types during union validation, avoiding unnecessary proxy wrapping. r.r1Nr2zFAt least one of 'tools', 'include_tags', or 'exclude_tags' is required) isinstancedictboolgetr")clsr6 has_tools has_include has_excludes r'%_require_at_least_one_transform_fieldzA_TransformingMCPServerMixin._require_at_least_one_transform_field\s fd # # VZZ0011I **^44D@K **^44D@K     \ r) server_name str | None client_name$tuple[FastMCP[Any], ClientTransport]cddlm}ddlm}ddlm}t }t||}|||}|||}|j | |j d |j | |j |j r)dd lm} || |j ||fS) z_Turn the Transforming MCPServer into a FastMCP Server and also return the underlying transport.r)Client)r) create_proxy) transportname)rJNT)tagsonly)rK) ToolTransform)fastmcp.clientrGfastmcp.client.transportsrfastmcp.serverrHsuper to_transportr r1enabler2disabler.fastmcp.server.transformsrM add_transform) selfrBrDrGrrHrIclientwrapped_mcp_serverrM __class__s r'#_to_server_and_underlying_transportz?_TransformingMCPServerMixin._to_server_and_underlying_transportqs2 *)))))       0/////%*WW%9%9%;%; )44 *0&9;*W*W*W)\       (  % %4+<4 % H H H   (  & &D,= & > > > : H ? ? ? ? ? ?  , ,]]4:-F-F G G G!9,,r)rcVddlm}||dS)z2Get the transport for the transforming MCP server.r)FastMCPTransport)mcp)rOr]r[)rWr]s r'rRz(_TransformingMCPServerMixin.to_transports9>>>>>>D$L$L$N$Nq$QRRRRr)r6r7rr7)NN)rBrCrDrCrrE)rr)__name__ __module__ __qualname____doc__rr:r.__annotations__r1r2r classmethodrAr[rR __classcell__)rZs@r'r+r+Ls*KK,1E$,G,G,GEGGGG9$)E7%%%L %*E7%%%L _(###[$#*#'"&!-!-!-!-!-!-!-FSSSSSSSSr)r+ceZdZUdZded<eeZded<eeZ ded<d Z d ed <d Z d ed<d Z ded<d Z ded<d Zded<d Zded<d Zded<d Zded<edZddZd S)StdioMCPServerzMCP server configuration for stdio transport. This is the canonical configuration format for MCP servers using stdio transport. r commandr,z list[str]argsr7envstdiozLiteral['stdio']rINzLiteral['stdio'] | NonetyperCcwd int | Nonetimeoutz bool | None keep_aliver0icondict[str, Any] | Noneauthenticationallowextrarrc`ddlm}||j|j|j|j|jS)Nr)r)rirjrkrnrq)rOrrirjrkrnrq)rWrs r'rRzStdioMCPServer.to_transportsF<<<<<<~L     r))rr)r`rarbrcrdrlistrjr:rkrIrmrnrprqr0rrrtr model_configrRr)r'rhrhs* LLLeD111D1111%555C5555#*I))))$(D((((CG  #K""""D-1N0000:G,,,L       r)rhceZdZdZdS)TransformingStdioMCPServerz$A Stdio server with tool transforms.Nr`rarbrcr{r)r'r}r}s....r)r}ceZdZUdZded<dZded<eeZded <dZ d ed <dZ d ed <dZ ded<dZ ded<dZ ded<dZded<eddZddZdS)RemoteMCPServerzMCP server configuration for HTTP/SSE transport. This is the canonical configuration format for MCP servers using remote transports. r rNz0Literal['http', 'streamable-http', 'sse'] | NonerIr,zdict[str, str]headerszAnnotated[str | Literal['oauth'] | httpx.Auth | None, Field(description='Either a string representing a Bearer token, the literal "oauth" to use OAuth authentication, or an httpx.Auth instance for custom authentication.')]authz'datetime.timedelta | int | float | Nonesse_read_timeoutrorprCr0rrrsrtruT)rwarbitrary_types_allowedr&StreamableHttpTransport | SSETransportcddlm}m}|jt |j}n|j}|dkr#||j|j|j|jS||j|j|j|jS)Nr)rrr)rrr) rOrrrIr(rrrr)rWrrrIs r'rRzRemoteMCPServer.to_transportsSSSSSSSS > !5dh??III   < Y!%!6  +* Y!%!6  r))rr)r`rarbrcrdrIrr:rrrrpr0rrrtr rzrRr{r)r'rrs   HHHCGIFFFF#eD999G9999      AEDDDDG#K""""D-1N0000:tLr)rceZdZdZdS)TransformingRemoteMCPServerz%A Remote server with tool transforms.Nr~r{r)r'rr s////r)rceZdZUdZeeZded<edZ e de dd Z ddZ e ddZddZd dZe d!dZdS)" MCPConfiga8A configuration object for MCP Servers that conforms to the canonical MCP configuration format while adding additional fields for enabling FastMCP-specific features like tool transformations and filtering by tags. For an MCPConfig that is strictly canonical, see the `CanonicalMCPConfig` class. r,zdict[str, MCPServerTypes] mcpServersrurvr3r4r6r7rcpd|vr1td|D}|rd|iS|S)zMIf there's no mcpServers key but there are server configs at root, wrap them.rc3PK|]!}t|tod|vpd|vV"dS)rirN)r9r:).0vs r' z1MCPConfig.wrap_servers_at_root..&sR1d##Fa)E5A:r))anyr6)r=r6 has_serverss r'wrap_servers_at_rootzMCPConfig.wrap_servers_at_root s\ v % %K .$f-- r)rJr serverMCPServerTypesNonec||j|<dSz,Add or update a server in the configuration.NrrWrJrs r' add_serverzMCPConfig.add_server/s &r)configrc,||S)z/Parse MCP configuration from dictionary format.)model_validate)r=rs r' from_dictzMCPConfig.from_dict3s!!&)))r)c.|dS)z>Convert MCPConfig to dictionary format, preserving all fields.T exclude_none) model_dump)rWs r'to_dictzMCPConfig.to_dict8sD111r) file_pathrc|jdd||ddS)z!Write configuration to JSON file.T)parentsexist_ok)indentN)parentmkdir write_textmodel_dump_json)rWrs r' write_to_filezMCPConfig.write_to_file<sItd;;;T111;;<<<<>&445R 5R5RSS +}5555+}555 #####r))rrrr)rrrBr rrrr)0rc __future__rdatetimer$pathlibrtypingrrrrr urllib.parser httpxpydanticr r r rrtyping_extensionsrrfastmcp.tools.tool_transformrfastmcp.utilities.typesrrOrrrrfastmcp.server.serverrr(r+rhr}rrTransformingMCPServerTypesrrrrrr{r)r'rs0#""""" ??????????????!!!!!! -,,,,,,,<<<<<<444444. .-----(LSLSLSLSLS"2LSLSLS^* * * * * Y* * * Z/////!