o i>@sdZddlZddlmZddlmZmZddlZddlm Z ddl m Z m Z m Z mZddlmZddlmZdd lmZd d lmZmZmZd d lmZmZmZmZmZmZee Z!d e"de"fddZ#GdddeZ$dgZ%dS)z6FastMCP server implementation for OpenAPI integration.N)Counter)AnyLiteral) SchemaPath) HTTPRoute$extract_output_schema_from_responsesformat_simple_descriptionparse_openapi_to_http_routes)RequestDirector)FastMCP) get_logger)OpenAPIResourceOpenAPIResourceTemplate OpenAPITool)DEFAULT_ROUTE_MAPPINGS ComponentFnMCPTypeRouteMap RouteMapFn_determine_route_typetextreturncCs@|sdStdd|}tdd|}tdd|}|d}|S)z Convert text to a URL-friendly slug format that only contains lowercase letters, uppercase letters, numbers, and underscores. z [\s\-\.]+_z [^a-zA-Z0-9_]z_+)resubstrip)rslugrp/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/experimental/server/openapi/server.py_slugify&s r!cseZdZdZ       ddeeefdejdedBde e dBde dBde dBd eeefdBd e edBd edBd effd d Z ddedeeefdBdefddZdededdefddZdeded e efddZdeded e efddZdeded e efddZZS) FastMCPOpenAPIa FastMCP server implementation that creates components from an OpenAPI schema. This class parses an OpenAPI specification and creates appropriate FastMCP components (Tools, Resources, ResourceTemplates) based on route mappings. Example: ```python from fastmcp.server.openapi import FastMCPOpenAPI, RouteMap, MCPType import httpx # Define custom route mappings custom_mappings = [ # Map all user-related endpoints to ResourceTemplate RouteMap( methods=["GET", "POST", "PATCH"], pattern=r".*/users/.*", mcp_type=MCPType.RESOURCE_TEMPLATE ), # Map all analytics endpoints to Tool RouteMap( methods=["GET"], pattern=r".*/analytics/.*", mcp_type=MCPType.TOOL ), ] # Create server with custom mappings and route mapper server = FastMCPOpenAPI( openapi_spec=spec, client=httpx.AsyncClient(), name="API Server", route_maps=custom_mappings, ) ``` N openapi_specclientname route_maps route_map_fnmcp_component_fn mcp_namestagstimeoutsettingsc  s tjdd|pdi| ||_| |_||_ttttd|_zt||_ t |j |_ Wnt yL} zt d| td| | d} ~ wwt|} |pTgt}| D]} t| |}|j}|durz|| |}|dur|}t d| jd| jd |jWn#t y} zt d | jd| jd | d WYd} ~ nd} ~ ww|| |}t| j|jB|ptB}|tjkr|j| ||d qY|tj kr|j!| ||d qY|tj"kr|j#| ||d qY|tj$krt d| jd| jqYt dt%| ddS)a Initialize a FastMCP server from an OpenAPI schema. Args: openapi_spec: OpenAPI schema as a dictionary or file path client: httpx AsyncClient for making HTTP requests name: Optional name for the server route_maps: Optional list of RouteMap objects defining route mappings route_map_fn: Optional callable for advanced route type mapping. Receives (route, mcp_type) and returns MCPType or None. Called on every route, including excluded ones. mcp_component_fn: Optional callable for component customization. Receives (route, component) and can modify the component in-place. Called on every created component. mcp_names: Optional dictionary mapping operationId to desired component names. If an operationId is not in the dictionary, falls back to using the operationId up to the first double underscore. If no operationId exists, falls back to slugified summary or path-based naming. All names are truncated to 56 characters maximum. tags: Optional set of tags to add to all components. Components always receive any tags from the route. timeout: Optional timeout (in seconds) for all requests **settings: Additional settings for FastMCP r%zOpenAPI FastMCPtoolresourceresource_templatepromptz&Failed to initialize RequestDirector: zInvalid OpenAPI specification: NzRoute  z* mapping customized by route_map_fn: type=zError in route_map_fn for : z. Using default values.)r*zExcluding route: z$Created FastMCP OpenAPI server with z routesr)&super__init___client_timeout_mcp_component_fnr _used_namesr from_dict_specr _director Exceptionloggererror ValueErrorr rrmcp_typedebugmethodpathr%warning_generate_default_namesetr*mcp_tagsrTOOL_create_openapi_toolRESOURCE_create_openapi_resourceRESOURCE_TEMPLATE_create_openapi_templateEXCLUDElen)selfr#r$r%r&r'r(r)r*r+r,e http_routesroute route_map route_typeresultcomponent_name route_tags __class__rr r5csf%         zFastMCPOpenAPI.__init__rT mcp_names_maprcCsvd}|pi}|jr|j|vr||j}n|jdd}n |jp(|jd|j}t|}t|dkr9|dd}|S)zEGenerate a default name from the route using the configured strategy.r__rr8N) operation_idsplitsummaryrCrDr!rP)rQrTr\r%rrr rFs    z%FastMCPOpenAPI._generate_default_namecomponent_typer-c Csn|j||d7<|j||dkr|S|d|j||}td|d|ddd|d|S) a4 Ensure the name is unique within its component type by appending numbers if needed. Args: name: The proposed name component_type: The type of component ("tools", "resources", or "templates") Returns: str: A unique name for the component r rzName collision detected: 'z' already exists as a Nz . Using 'z ' instead.)r9r>rB)rQr%rbnew_namerrr _get_unique_nameszFastMCPOpenAPI._get_unique_namec Cs|j}t|j|j|j}||d}|jp!|jp!d|jd|j }t ||j |j d}t |j||j||||t|jp:g|B|jd } |jdurwz||| td|dWntyv} ztd |d | d WYd} ~ nd} ~ ww| j} | |jj| <dS) z?Creates and registers an OpenAPITool with enhanced description.r.z Executes r2base_description parameters request_body) r$rTdirectorr% descriptionrh output_schemar*r+NzTool  customized by component_fnzError in component_fn for tool r3. Using component as-is.)flat_param_schemar responsesresponse_schemasopenapi_versionrerkrarCrDrrhrirr6r<rGr*r7r8r>rBr=rEr% _tool_manager_tools) rQrTr%r*combined_schemarl tool_namergenhanced_descriptionr.rRfinal_tool_namerrr rJsP   z#FastMCPOpenAPI._create_openapi_toolc Cs||d}d|}|jp|jpd|j}t||j|jd}t|j||j |||t |j p.g|B|j d}|j durkz| ||td|dWntyj} ztd |d | d WYd} ~ nd} ~ wwt|j} ||jj| <dS) zCCreates and registers an OpenAPIResource with enhanced description.r/ resource://z Represents rf)r$rTrjurir%rkr*r+Nz Resource rmz#Error in component_fn for resource r3rn)rerkrarDrrhrirr6r<rGr*r7r8r>rBr=rEstrrz_resource_manager _resources) rQrTr%r* resource_name resource_urirgrwr/rRfinal_resource_urirrr rL>s>     z'FastMCPOpenAPI._create_openapi_resourcec CsJ||d}dd|jD}|d|}|r'|dddd|D7}|jp2|jp2d|j}t||j|jd }d d d |jDd d|jDd} t |j ||j |||| t |j p]g|B|jd } |jdurz||| td|dWnty} ztd|d| dWYd} ~ nd} ~ ww| j} | |jj| <dS)zKCreates and registers an OpenAPIResourceTemplate with enhanced description.r0cSsg|] }|jdkr|jqSrD)locationr%.0prrr zsz;FastMCPOpenAPI._create_openapi_template..ry/css|] }d|dVqdS){}Nrrrrr sz:FastMCPOpenAPI._create_openapi_template..z Template for rfobjectcSsbi|]-}|jdkr|jit|jtr|jni|jr,t|jtr'd|jvs,d|jiniqS)rDrk)rr% isinstanceschema_dictcopyrkrrrr s     z;FastMCPOpenAPI._create_openapi_template..cSs"g|] }|jdkr|jr|jqSr)rrequiredr%rrrr rs)type propertiesr) r$rTrj uri_templater%rkrhr*r+Nz Template rmz#Error in component_fn for template r3rn)rerhsortjoinrkrarDrrirr6r<rGr*r7r8r>rBr=rErr| _templates) rQrTr%r* template_name path_paramsuri_template_strrgrwtemplate_params_schematemplaterRfinal_template_urirrr rNpsX     z'FastMCPOpenAPI._create_openapi_template)NNNNNNN)N)__name__ __module__ __qualname____doc__rr{rhttpx AsyncClientlistrrrrGfloatr5rrFrrerJrLrN __classcell__rrrZr r"=s)     g   > 2r")&rr collectionsrtypingrrrjsonschema_pathr&fastmcp.experimental.utilities.openapirrrr /fastmcp.experimental.utilities.openapi.directorr fastmcp.server.serverr fastmcp.utilities.loggingr componentsrrrroutingrrrrrrrr>r{r!r"__all__rrrr s(