Ki,dZddlmZddlZddlZddlmZddlmZm 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 SSETransportStdioTransportStreamableHttpTransportFastMCPct|}|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_urlr s ^/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/mcp_config.pyinfer_transport_type_from_urlr&8sT c(C >>& !=.//#J ??D yy"D)ceZdZUdZedZded< eddZded <edd Zded < d dfd Z dd Z xZ S)_TransformingMCPServerMixinzAA mixin that enables wrapping an MCP Server with tool transforms..zdict[str, ToolTransformConfig]toolsNz!The tags to include in the proxy.)default descriptionzset[str] | None include_tagsz!The tags to exclude in the proxy. exclude_tagscddlm}ddlm}ddlm}t |}t||}|||}|j|||j|j|j}||fS)z_Turn the Transforming MCPServer into a FastMCP Server and also return the underlying transport.rr)Client)r) transportname)r2backendtool_transformationsr-r.) fastmcprfastmcp.clientr0fastmcp.client.transportsrsuper to_transportr as_proxyr*r-r.) self server_name client_namerr0rr1clientwrapped_mcp_server __class__s r%#_to_server_and_underlying_transportz?_TransformingMCPServerMixin._to_server_and_underlying_transport\sv $) &+W%9%; )4 *09;*W-W--!%****  "9,,r'cBddlm}||jdS)z2Get the transport for the transforming MCP server.r)FastMCPTransport)mcp)r7rCrA)r;rCs r%r9z(_TransformingMCPServerMixin.to_transportws>D$L$L$Nq$QRRr')NN)r< str | Noner=rEreturnz$tuple[FastMCP[Any], ClientTransport])rFr) __name__ __module__ __qualname____doc__rr*__annotations__r-r.rAr9 __classcell__)r@s@r%r)r)LswK,1#JE )69$)7%L/ %*7%L/#'"&-- - . -6Sr'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<edZddZy )StdioMCPServerzMCP server configuration for stdio transport. This is the canonical configuration format for MCP servers using stdio transport. rcommanddefault_factoryz list[str]argsdict[str, Any]envstdiozLiteral['stdio']r1NzLiteral['stdio'] | NonetyperEcwd int | Nonetimeoutr,icondict[str, Any] | Noneauthenticationallowextracvddlm}||j|j|j|j S)Nr)r)rOrRrTrW)r7rrOrRrTrW)r;rs r%r9zStdioMCPServer.to_transports.<LL   r')rFr)rGrHrIrJrKrlistrRdictrTr1rVrWrYr,rZr\r model_configr9r'r%rNrN~s LD1D)15C5#*I)$(D !(CGZ#K"D*-1N)0G,L r'rNceZdZdZy)TransformingStdioMCPServerz$A Stdio server with tool transforms.NrGrHrIrJrdr'r%rfrfs.r'rfceZdZUdZded<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'] | Noner1rPzdict[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_timeoutrXrYrEr,rZr[r\r]T)r_arbitrary_types_allowedcHddlm}m}|jt |j }n |j}|dk(r4||j |j |j|jS||j |j |j|jS)Nr)rrr)rjrkrl) r7rrr1r&r#rjrkrl)r;rrr1s r%r9zRemoteMCPServer.to_transportsS >> !5dhh?II   YY!%!6!6  + YY!%!6!6  r')rFz&StreamableHttpTransport | SSETransport)rGrHrIrJrKr1rrbrjrkrlrYr,rZr\r rcr9rdr'r%riris HCGI?F#D9G^9    AE=DGZ#K"D*-1N)0tLr'riceZdZdZy)TransformingRemoteMCPServerz%A Remote server with tool transforms.Nrgrdr'r%rprps/r'rpceZdZUdZeeZded<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. rPzdict[str, MCPServerTypes] mcpServersr]r^before)modecZd|vr&td|jD}|rd|iS|S)zMIf there's no mcpServers key but there are server configs at root, wrap them.rsc3TK|] }t|txr d|vxsd|v"yw)rOr#N) isinstancerb).0vs r% z1MCPConfig.wrap_servers_at_root..s41d#Fa)E5A:Fs&()anyvalues)clsr} has_serverss r%wrap_servers_at_rootzMCPConfig.wrap_servers_at_roots> v %K$f-- r'c"||j|<yz,Add or update a server in the configuration.Nrsr;r2servers r% add_serverzMCPConfig.add_servers &r'c$|j|S)z/Parse MCP configuration from dictionary format.)model_validate)r~configs r% from_dictzMCPConfig.from_dicts!!&))r'c&|jdS)z>Convert MCPConfig to dictionary format, preserving all fields.T exclude_none) model_dump)r;s r%to_dictzMCPConfig.to_dictsD11r'c|jjdd|j|jdy)z!Write configuration to JSON file.T)parentsexist_ok)indentN)parentmkdir write_textmodel_dump_json)r; file_paths r% write_to_filezMCPConfig.write_to_files7td;T111;&445R 5R5RS +}5+}5 #r')r#z str | AnyUrlrFzLiteral['http', 'sse'])rrr<rrrrFr)0rJ __future__rdatetimer!pathlibrtypingrrrrr urllib.parser httpxpydanticr r r rrtyping_extensionsrrfastmcp.tools.tool_transformrfastmcp.utilities.typesrr7rrrrfastmcp.server.serverrr&r)rNrfrirpTransformingMCPServerTypesrrrrrrrdr'r%rs0# ??! -<4 . (/S"2/Sd& Y& R/!