o i,@sldZddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl mZddlZddlmZmZmZmZmZddlmZmZdd lmZdd lmZerddd lmZmZm Z m!Z!dd l"m#Z#d*ddZ$GdddeZ%GdddeZ&Gddde%e&Z'GdddeZ(Gddde%e(Z)e'e)BZ*e&e(BZ+e*e+BZ,GdddeZ-Gdd d e-Z.d+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 SSETransportStdioTransportStreamableHttpTransportFastMCPurl str | AnyUrlreturnLiteral['http', 'sse']cCsBt|}|dstd|t|}|j}td|rdSdS)zB Infer the appropriate transport type from the given URL. httpz Invalid URL: z/sse(/|\?|&|$)sse)str startswith ValueErrorr pathresearch)r parsed_urlr"r&X/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/mcp_config.pyinfer_transport_type_from_url8s  r(csneZdZUdZedZded< edddZded <edd dZded <  ddfdd Z dddZ Z 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_tags server_name str | None client_namer$tuple[FastMCP[Any], ClientTransport]c sfddlm}ddlm}ddlm}t}t||}|||d}|j |||j |j |j d}||fS)z_Turn the Transforming MCPServer into a FastMCP Server and also return the underlying transport.rr)Client)r) transportname)r5backendtool_transformationsr-r.) fastmcprfastmcp.clientr3fastmcp.client.transportsrsuper to_transportras_proxyr*r-r.) selfr/r1rr3rr4clientwrapped_mcp_server __class__r&r'#_to_server_and_underlying_transport\s      z?_TransformingMCPServerMixin._to_server_and_underlying_transportrcCsddlm}||ddS)z2Get the transport for the transforming MCP server.r)FastMCPTransport)mcp)r:rDrC)r>rDr&r&r'r<ws z(_TransformingMCPServerMixin.to_transport)NN)r/r0r1r0rr2)rr) __name__ __module__ __qualname____doc__r r*__annotations__r-r.rCr< __classcell__r&r&rAr'r)Ls r)c@seZdZUdZded<eedZded<eedZ ded<d Z d ed <d Z d ed<d Z ded<d Z ded<d Zded<d Zded<d Zded<eddZdddZd S)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']r4NzLiteral['stdio'] | Nonetyper0cwd int | Nonetimeoutr,icondict[str, Any] | NoneauthenticationallowextrarrcCs$ddlm}||j|j|j|jdS)Nr)r)rMrPrRrU)r:rrMrPrRrU)r>rr&r&r'r<s zStdioMCPServer.to_transport)rr)rFrGrHrIrJr listrPdictrRr4rTrUrWr,rXrZr model_configr<r&r&r&r'rL~s         rLc@eZdZdZdS)TransformingStdioMCPServerz$A Stdio server with tool transforms.NrFrGrHrIr&r&r&r'rbrbc@seZdZUdZded<dZded<eedZded <dZ d ed <dZ d ed <dZ ded<dZ ded<dZ ded<dZded<edddZdddZdS)RemoteMCPServerzMCP server configuration for HTTP/SSE transport. This is the canonical configuration format for MCP servers using remote transports. rrNz0Literal['http', 'streamable-http', 'sse'] | Noner4rNzdict[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_timeoutrVrWr0r,rXrYrZr[T)r]arbitrary_types_allowedr&StreamableHttpTransport | SSETransportcCsdddlm}m}|jdurt|j}n|j}|dkr&||j|j|j|jdS||j|j|j|jdS)Nr)rrr)rfrgrh) r:rrr4r(rrfrgrh)r>rrr4r&r&r'r<s"  zRemoteMCPServer.to_transport)rrj)rFrGrHrIrJr4r r_rfrgrhrWr,rXrZr r`r<r&r&r&r'res       rec@ra)TransformingRemoteMCPServerz%A Remote server with tool transforms.Nrcr&r&r&r'rkrdrkc@seZdZUdZeedZded<eddZ e dde d"d d Z d#ddZ e d$ddZd%ddZd&ddZe d'dd Zd!S)( 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. rNzdict[str, MCPServerTypes] mcpServersr[r\before)modevaluesrQrcCs.d|vrtdd|D}|rd|iS|S)zMIf there's no mcpServers key but there are server configs at root, wrap them.rmcss*|]}t|tod|vpd|vVqdS)rMrN) isinstancer_).0vr&r&r' s  z1MCPConfig.wrap_servers_at_root..)anyrp)clsrp has_serversr&r&r'wrap_servers_at_rootszMCPConfig.wrap_servers_at_rootr5rserverMCPServerTypesNonecC||j|<dSz,Add or update a server in the configuration.Nrmr>r5ryr&r&r' add_serverszMCPConfig.add_serverconfigrcCs ||S)z/Parse MCP configuration from dictionary format.)model_validate)rvrr&r&r' from_dicts zMCPConfig.from_dictcCs |jddS)z>Convert MCPConfig to dictionary format, preserving all fields.T exclude_none) model_dump)r>r&r&r'to_dicts zMCPConfig.to_dict file_pathrcCs&|jjddd||jdddS)z!Write configuration to JSON file.T)parentsexist_ok)indentN)parentmkdir write_textmodel_dump_json)r>rr&r&r' write_to_fileszMCPConfig.write_to_filecCs0|r|}r||Std|)z"Load configuration from JSON file.z&No MCP servers defined in the config: )exists read_textstripmodel_validate_jsonr!)rvrcontentr&r&r' from_file s zMCPConfig.from_fileN)rprQrrQ)r5rryrzrr{)rrQrr)rrQ)rrrr{)rrrr)rFrGrHrIr r_rmrJr r`r classmethodrxrrrrrr&r&r&r'rls      rlc@s2eZdZUdZeedZded<edd d Z d S)CanonicalMCPConfigzCanonical MCP configuration format. This defines the standard configuration format for Model Context Protocol servers. The format is designed to be client-agnostic and extensible for future use cases. rNz"dict[str, CanonicalMCPServerTypes]rmr5rryCanonicalMCPServerTypesrr{cCr|r}r~rr&r&r'r2szCanonicalMCPConfig.add_serverN)r5rryrrr{) rFrGrHrIr r_rmrJrrr&r&r&r'r)s rrrr/r server_configrr{cCsht|}|j|}r'|}|jdd}|i||}|||n|||||dS)zUpdate an MCP configuration file from a server object, preserving existing fields. This is used for updating the mcpServer configurations of third-party tools so we do not worry about transforming server objects here.TrN)rlrrmgetrrrr)rr/rrexisting_server existing_dictnew_dict merged_configr&r&r'update_config_file8s   r)rrrr)rrr/rrrrr{)0rI __future__rdatetimer#pathlibrtypingrrrrr urllib.parser httpxpydanticr r r r rtyping_extensionsrrfastmcp.tools.tool_transformrfastmcp.utilities.typesrr:rrrrfastmcp.server.serverrr(r)rLrbrerkTransformingMCPServerTypesrrzrlrrr&r&r&r's6       2)=6