i1dZddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl mZddlZddlmZmZmZmZmZddlmZmZdd lmZdd lmZerdd lmZmZm Z m!Z!dd l"m#Z# dd Z$GddeZ%GddeZ&Gdde%e&Z'GddeZ(Gdde%e(Z)e'e)zZ*e&e(zZ+e*e+zZ,GddeZ-Gdde-Z. ddZ/y)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)FastMCPct|}|jdstd|t|}|j}t j d|ryy)zB Infer the appropriate transport type from the given URL. httpz Invalid URL: z/sse(/|\?|&|$)sse)str startswith ValueErrorr pathresearch)url parsed_urlrs b/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/mcp_config.pyinfer_transport_type_from_urlr%8sT c(C >>& !=.//#J ??D yy"D)ceZdZUdZeeZded< eddZded <edd Z ded <e d e ddZ d dfd 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_tagsbeforemodect|trQt|jd}|jddu}|jddu}|s|s |s t 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+r.Nr/zFAt least one of 'tools', 'include_tags', or 'exclude_tags' is required) isinstancedictboolgetr)clsvalues has_tools has_include has_excludes r$%_require_at_least_one_transform_fieldzA_TransformingMCPServerMixin._require_at_least_one_transform_field\se fd #VZZ01I **^4D@K **^4D@K  \ r&cddlm}ddlm}ddlm}t |}t||}|||}|||}|j|j|jd|j|j|j |jr'dd lm} |j!| |j||fS) z_Turn the Transforming MCPServer into a FastMCP Server and also return the underlying transport.r)Client)r) create_proxy) transportname)rBT)tagsonly)rC) ToolTransform)fastmcp.clientr?fastmcp.client.transportsrfastmcp.serverr@super to_transportr r.enabler/disabler+fastmcp.server.transformsrE add_transform) self server_name client_namer?rr@rAclientwrapped_mcp_serverrE __class__s r$#_to_server_and_underlying_transportz?_TransformingMCPServerMixin._to_server_and_underlying_transportqs *  0%*W%9%; )4 *09;*W)      (  % %4+<+<4 % H    (  & &D,=,= & > :: ?  , ,]4::-F G!9,,r&cBddlm}||jdS)z2Get the transport for the transforming MCP server.r)FastMCPTransport)mcp)rGrWrU)rOrWs r$rJz(_TransformingMCPServerMixin.to_transports>D$L$L$Nq$QRRr&r9dict[str, Any]returnrZ)NN)rP str | NonerQr\r[z$tuple[FastMCP[Any], ClientTransport])r[r)__name__ __module__ __qualname____doc__rr5r+__annotations__r.r/r classmethodr=rUrJ __classcell__)rTs@r$r(r(LsK,1$,GE )G9$)7%L/ %*7%L/ (## $*#'"&!-!- !- . !-FSr&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<edZddZy )StdioMCPServerzMCP server configuration for stdio transport. This is the canonical configuration format for MCP servers using stdio transport. rcommandr)z list[str]argsrZenvstdiozLiteral['stdio']rANzLiteral['stdio'] | Nonetyper\cwd int | Nonetimeoutz bool | None keep_aliver-icondict[str, Any] | Noneauthenticationallowextracddlm}||j|j|j|j |j S)Nr)r)rfrgrhrkrn)rGrrfrgrhrkrn)rOrs r$rJzStdioMCPServer.to_transports5<LL   r&)r[r)r]r^r_r`rarlistrgr5rhrArjrkrmrnr-rorqr model_configrJr&r$reres LD1D)15C5#*I)$(D !(CGZ   #K"D*-1N)0G,L  r&receZdZdZy)TransformingStdioMCPServerz$A Stdio server with tool transforms.Nr]r^r_r`rxr&r$rzrzs.r&rzceZdZUdZded<dZded<eeZded <dZ d ed <dZ d ed <dZ ded<dZ ded<dZ ded<dZded<eddZddZy)RemoteMCPServerzMCP server configuration for HTTP/SSE transport. This is the canonical configuration format for MCP servers using remote transports. rr"Nz0Literal['http', 'streamable-http', 'sse'] | NonerAr)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_timeoutrlrmr\r-rorprqrrT)rtarbitrary_types_allowedcHddlm}m}|jt |j }n |j}|dk(r4||j |j |j|jS||j |j |j|jS)Nr)rrr)r~rr) rGrrrAr%r"r~rr)rOrrrAs r$rJzRemoteMCPServer.to_transportsS >> !5dhh?II   YY!%!6!6  + YY!%!6!6  r&)r[z&StreamableHttpTransport | SSETransport)r]r^r_r`rarArr5r~rrrmr-rorqr rwrJrxr&r$r}r}s HCGI?F#D9G^9    AE=DGZ#K"D*-1N)0tLr&r}ceZdZdZy)TransformingRemoteMCPServerz%A Remote server with tool transforms.Nr{rxr&r$rr s/r&rceZdZUdZeeZded<edZ e de dd Z dd Z e dd Zdd Zdd Ze ddZy) 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] mcpServersrrrsr0r1cZd|vr&td|jD}|rd|iS|S)zMIf there's no mcpServers key but there are server configs at root, wrap them.rc3TK|] }t|txr d|vxsd|v"yw)rfr"N)r4r5).0vs r$ z1MCPConfig.wrap_servers_at_root..&s4(A1d#Fa)E5A:F(s&()anyr9)r8r9 has_serverss r$wrap_servers_at_rootzMCPConfig.wrap_servers_at_root s> v %K$f-- r&c"||j|<yz,Add or update a server in the configuration.NrrOrBservers r$ add_serverzMCPConfig.add_server/s &r&c$|j|S)z/Parse MCP configuration from dictionary format.)model_validate)r8configs r$ from_dictzMCPConfig.from_dict3s!!&))r&c&|jdS)z>Convert MCPConfig to dictionary format, preserving all fields.T exclude_none) model_dump)rOs r$to_dictzMCPConfig.to_dict8sD11r&c|jjdd|j|jdy)z!Write configuration to JSON file.T)parentsexist_ok)indentN)parentmkdir write_textmodel_dump_json)rO file_paths r$ write_to_filezMCPConfig.write_to_file<s7td;T111;&445R 5R5RS +}5+}5 #r&)r"z str | AnyUrlr[zLiteral['http', 'sse'])rrrPrrrr[r)0r` __future__rdatetimer pathlibrtypingrrrrr urllib.parser httpxpydanticr r r rrtyping_extensionsrrfastmcp.tools.tool_transformrfastmcp.utilities.typesrrGrrrrfastmcp.server.serverrr%r(rerzr}rTransformingMCPServerTypesrrrrrrxr&r$rs0# ??! -<4 . (LS"2LS^* Y* Z/!