kKj3dZddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl mZddlZddlmZmZmZmZmZddlmZmZdd lmZer dd lmZmZmZmZ dd Z dd Z!Gd deZ"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)_install_hints)ClientTransport SSETransportStdioTransportStreamableHttpTransportct|}|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 c/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/mcp_config.pyinfer_transport_type_from_urlr#6sT c(C >>& !=.//#J ??D yy"D)c ddlm}|jDcic]%\}}|t||r|n|j |'c}}Scc}}w)Nr)ToolTransformConfig)fastmcp.tools.tool_transformr&items isinstancemodel_validate)toolsr&nameconfigs r"_coerce_tool_transform_configsr.Js]@ "KKM  *LD& f1 2 / / 7 8*   s*AceZdZUdZeeZded< eddZded <edd Z ded <e d e ddZ d dfd Z ddZxZS)_TransformingMCPServerMixinzAA mixin that enables wrapping an MCP Server with tool transforms.default_factorydict[str, Any]r+Nz!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+r6Nr7zFAt least one of 'tools', 'include_tags', or 'exclude_tags' is required)r)dictboolgetr)clsvalues has_tools has_include has_excludes r"%_require_at_least_one_transform_fieldzA_TransformingMCPServerMixin._require_at_least_one_transform_fieldese fd #VZZ01I **^4D@K **^4D@K  \ r$c ddlm}ddlm}ddlm}tdt |-}|||}||| } |j| j|jd |j| j|j |j r*| j#|t%|j | |fS#t $r$}t tjd|d}~wwxYw) zKTurn the transforming server into a FastMCP proxy and return its transport.r)Client) create_proxy) ToolTransformDMCP configs that use FastMCP-specific tool transforms or tag filtersNr) transportr,)r,T)tagsonly)rK)fastmcprFfastmcp.serverrGfastmcp.server.transformsrH ImportErrorr full_packager super to_transportr6enabler7disabler+ add_transformr.) self server_name client_namerFrGrHexcrJclientwrapped_mcp_server __class__s r"#_to_server_and_underlying_transportz?_TransformingMCPServerMixin._to_server_and_underlying_transportzs  & 3 ?*EG,@,BC )+>)&{C    (  % %4+<+<4 % H    (  & &D,=,= & > ::  , ,79$)7%L/ %*7%L/ (## $*#'"&-- - % -@ Sr$r0ceZdZUdZded<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. rcommandr1z list[str]argsr3envstdiozLiteral['stdio']rJNzLiteral['stdio'] | Nonetyperecwd int | Nonetimeoutz bool | None keep_aliver5icondict[str, Any] | Noneauthenticationallowextracddlm}||j|j|j|j |j S)Nr)r)rorprqrtrw)rbrrorprqrtrw)rWrs r"rSzStdioMCPServer.to_transports5<LL   r$)rdr)rfrgrhrirjrlistrpr<rqrJrsrtrvrwr5rxrzr model_configrSr$r"rnrns LD1D)15C5#*I)$(D !(CGZ   #K"D*-1N)0G,L  r$rnceZdZdZy)TransformingStdioMCPServerz$A Stdio server with tool transforms.Nrfrgrhrirr$r"rrs.r$rceZdZUdZded<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'] | NonerJr1zdict[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_timeoutrurvrer5rxryrzr{T)r}arbitrary_types_allowedcHddlm}m}|jt |j }n |j}|dk(r4||j |j |j|jS||j |j |j|jS)Nr)rrr)rrr) rbrrrJr#r rrr)rWrrrJs r"rSzRemoteMCPServer.to_transports >> !5dhh?II   YY!%!6!6  + YY!%!6!6  r$)rdz&StreamableHttpTransport | SSETransport)rfrgrhrirjrJrr<rrrrvr5rxrzr rrSrr$r"rrs HCGI?F#D9G^9    AE=DGZ#K"D*-1N)0tLr$rceZdZdZy)TransformingRemoteMCPServerz%A Remote server with tool transforms.Nrrr$r"rrs/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. r1zdict[str, MCPServerTypes] mcpServersr{r|r8r9cZd|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)ror N)r)r<).0vs r" z1MCPConfig.wrap_servers_at_root..6s4(A1d#Fa)E5A:F(s&()anyr@)r?r@ has_serverss r"wrap_servers_at_rootzMCPConfig.wrap_servers_at_root0s> v %K$f-- r$c"||j|<yz,Add or update a server in the configuration.NrrWr,servers r" add_serverzMCPConfig.add_server?s &r$c$|j|S)z/Parse MCP configuration from dictionary format.)r*)r?r-s r" from_dictzMCPConfig.from_dictCs!!&))r$c&|jdS)z>Convert MCPConfig to dictionary format, preserving all fields.T exclude_none) model_dump)rWs r"to_dictzMCPConfig.to_dictHsD11r$c|jjdd|j|jddy)z!Write configuration to JSON file.T)parentsexist_ok)indentutf-8encodingN)parentmkdir write_textmodel_dump_json)rW file_paths r" write_to_filezMCPConfig.write_to_fileLs<td;T111;gNr$c|jr3|jdjx}r|j|St d|)z"Load configuration from JSON file.rrz&No MCP servers defined in the config: )exists read_textstripmodel_validate_jsonr)r?rcontents r" from_filezMCPConfig.from_fileQsV     **G*<BBD DG D**73 3A)MNNr$Nrc)r,rrMCPServerTypesrdNone)r-r3rdr)rdr3)rrrdr)rrrdr)rfrgrhrirr<rrjr rrrkrrrrrrrr$r"rr$s|-2$,GJ)GG,L(# $ '**2O OOr$rc@eZdZUdZeeZded<eddZ y)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. r1z"dict[str, CanonicalMCPServerTypes]rc"||j|<yrrrs r"rzCanonicalMCPConfig.add_serveres!'r$N)r,rrCanonicalMCPServerTypesrdr) rfrgrhrirr<rrjrrrr$r"rr\s+ 6;45PJ2P ''r$rcBtj|}|jj|x}rJ|j }|j d}|j i||}|j ||n|j |||j|y)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)rrrr>rr*rr)rrX server_configr-existing_server existing_dictnew_dict merged_configs r"update_config_filerks  +F!++// <<<'224  +++>&445R 5R5RS +}5+}5 #r$)r z str | AnyUrlrdzLiteral['http', 'sse'])r+r3rdr3)rrrXrrrrdr)-ri __future__rdatetimerpathlibrtypingrrrrr urllib.parser httpxpydanticr r r rrtyping_extensionsrrrMrrbrrrrr#r.r0rnrrrTransformingMCPServerTypesrrrrrrr$r"rs0# ??! -" (PS)PSf* Y* Z/!