o i\@sdZddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZddlmZmZmZddlZdd lmZdd lmZddlZdd lmZdd lmZmZdd lmZddl m!Z!ddl"m#Z#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/erddl0m1Z1e(e2Z3edZ4d3ddZ5e e*dgdfZ6e e*dgdfZ7Gdddej8Z9Gdddej8Z:e d d!Gd"d#d#Z;e;e9jGd+d,d,e#Z?Gd-d.d.eZ@Gd/d0d0eZAGd1d2d2e!ZBdS)5z6FastMCP server implementation for OpenAPI integration.) annotationsN)Counter)Callable) dataclassfield)Pattern) TYPE_CHECKINGAnyLiteral)ToolAnnotations)AnyUrl) ToolError)ResourceResourceTemplate)get_http_headers)FastMCP)Tool ToolResult)openapi) get_logger) HTTPRoute_combine_schemas$extract_output_schema_from_responsesformat_array_parameterformat_deep_object_parameter!format_description_with_responses)Context)GETPOSTPUTPATCHDELETEOPTIONSHEADtextstrreturncCs@|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)r$slugr-\/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/server/openapi.py_slugify,s r/MCPTypeMCPType | Nonez7OpenAPITool | OpenAPIResource | OpenAPIResourceTemplatec@ eZdZdZdZdZdZdZdS)r0aType of FastMCP component to create from a route. Enum values: TOOL: Convert the route to a callable Tool RESOURCE: Convert the route to a Resource (typically GET endpoints) RESOURCE_TEMPLATE: Convert the route to a ResourceTemplate (typically GET with path params) EXCLUDE: Exclude the route from being converted to any MCP component IGNORE: Deprecated, use EXCLUDE instead TOOLRESOURCERESOURCE_TEMPLATEEXCLUDEN)__name__ __module__ __qualname____doc__r3r4r5r6r-r-r-r.r0Ns  c@r2) RouteTypez Deprecated: Use MCPType instead. This enum is kept for backward compatibility and will be removed in a future version. r3r4r5IGNOREN)r7r8r9r:r3r4r5r<r-r-r-r.r;as r;T)kw_onlyc@seZdZUdZeddZded<eddZded<ed dZd ed <ee d d idZ ded<ed d didZ ded<ee d didZ ded<ddZ d S)RouteMapzAMapping configuration for HTTP routes to FastMCP component types.*)defaultzlist[HttpMethod] | Literal['*']methodsz.*zPattern[str] | strpatternNzRouteType | MCPType | None route_type descriptionz,A set of tags to match. All tags must match.)default_factorymetadataset[str]tagsz(The type of FastMCP component to create.)r@rFr1mcp_typez:A set of tags to apply to the generated FastMCP component.mcp_tagscCs|jdurV|jdurVtjjrtjdtddt|jt r(tjjr(tjdtdd|jt j kr:tjjr:tjdtddt|jt rQ|jj }|dkrJd}t t ||_n|j|_n |jdur_td |jdurj|j|_dSdS) z8Validate and process the route map after initialization.NzThe 'route_type' parameter is deprecated and will be removed in a future version. Use 'mcp_type' instead with the appropriate MCPType value.) stacklevelz_The RouteType class is deprecated and will be removed in a future version. Use MCPType instead.zdRouteType.IGNORE is deprecated and will be removed in a future version. Use MCPType.EXCLUDE instead.r<r6z`mcp_type` must be provided)rIrCfastmcpsettingsdeprecation_warningswarningswarnDeprecationWarning isinstancer;r<namegetattrr0 ValueError)selfroute_type_namer-r-r. __post_init__s@       zRouteMap.__post_init__)r7r8r9r:rrA__annotations__rBrCsetrHrIrJrYr-r-r-r.r>ns&  r>rIrouteopenapi.HTTPRoutemappingslist[RouteMap]c Cs|D]S}|jdks|j|jvrUt|jtr|j|j}nt|j|j}|rU|jr8t |jp/g}|j |s8q|j dus?Jt d|jd|jd|j j|SqttjdS)a' Determines the FastMCP component type based on the route and mappings. Args: route: HTTPRoute object mappings: List of RouteMap objects in priority order Returns: The RouteMap that matches the route, or a catchall "Tool" RouteMap if no match is found. r?NRoute  z matched mapping to r\)rAmethodrSrBrsearchpathr)rHr[issubsetrIloggerdebugrTr>r0r3)r]r_ route_mappattern_matchesroute_tags_setr-r-r._determine_route_types"   rlcsBeZdZdZ     dd fdd Zd!ddZd"ddZZS)# OpenAPIToolz*Tool implementation for OpenAPI endpoints.Nclienthttpx.AsyncClientr]r^rTr%rD parametersdict[str, Any] output_schemadict[str, Any] | NonerHset[str] | Nonetimeout float | NonerToolAnnotations | None serializerCallable[[Any], str] | Nonec s6tj|||||p t| | d||_||_||_dS)N)rTrDrprrrHrrx)super__init__r[_client_route_timeout) rWrnr]rTrDrprrrHrurrx __class__r-r.r{s   zOpenAPITool.__init__r&cCs"d|jd|jjd|jjdS)@Custom representation to prevent recursion errors when printing.zOpenAPITool(name=z , method=, path=))rTr}rcrerWr-r-r.__repr__s"zOpenAPITool.__repr__ argumentsrc$ sfi}t}|jjr)|jjjr)tt|jjj}|jjj|}t|di}|jjD]!}|j }|j d|j }|j |vrG||j f||<||j f||<q-|jj } i} |jjD]8} | j dkr| j d| j }||vr|||dur|||| | j <qY| j |vr|| j dur|| j | | j <qYdd|jjD} | | } | rt d| | D]E\}tfdd |jjDd}|rt|tr|j}|d d k}|rt|d d }| ddt|} q| ddt|} qi}|jjD]} | j dkr| j d| j }d}||}||vr5|dur5|dkr5t|ttBr0t|dks5||}n*|| j }| j |vr_|dur_|dkr_t|ttBrZt|dks_|| j }|dur| jpid}| jdurs| jnd}|dkrt|trt|dkr|rt|| j }||qtd| j dt||| j <qt|tr| jd d krt|dkr|r||| j <qt|| j dd }||| j <q||| j <qi}i}|jjD]H} | j dkr+| j d| j }d}||vr ||dur ||}n| j |vr|| j dur|| j }|dur+t||| j <q||t}||d}|jjr||jjjr|t|jjD]} | j |vrd | j d| j qN | j qNfdd| D}|r||}zX|j!j"|jj#| ||||j$dIdH}|%z/|}d} |j&dur|j&drd|i} n|} n t|tsd|i} n|} t'| dWWStj(yt'|j)dYWSwt*j+y}!z;d |!j,j-d!|!j,j.}"z|!j,}#|"d"|#7}"Wntj(t/fy|!j,j)r|"d"|!j,j)7}"Ynwt/|"|!d}!~!wt*j0y2}!zt/d#|!|!d}!~!ww)$z:Execute the HTTP request based on the route configuration. properties__reNcSs"h|] }|jdkr|jr|jqSrelocationrequiredrT.0pr-r-r. 2s z"OpenAPITool.run..z"Missing required path parameters: c3s|] }|jkr|VqdSN)rTr) param_namer-r. ?sz"OpenAPITool.run..typearrayF)is_query_parameter{}queryr'rformT deepObjectz3deepObject style with explode=false for parameter 'z5' is not standard. Using JSON serialization fallback.headercsi|] \}}|vr||qSr-r-)rkv)params_to_excluder-r. sz#OpenAPITool.run..)rcurlparamsheadersjsonruzx-fastmcp-wrap-resultresult)structured_content)content HTTP error :  - Request error: )1r[r} request_bodycontent_schemanextitergetkeysrprTrrer itemsrSlistschema_rreplacer%dictlenstyleexploderupdatergwarningrdumpslowerraddr|requestrcr~raise_for_statusrrrJSONDecodeErrorr$httpxHTTPStatusErrorresponse status_code reason_phraserV RequestError)$rWr param_mapping body_props content_type body_schemaparam original_name suffixed_namere path_paramsrrequired_path_paramsmissing_params param_value param_infoschemais_arrayformatted_value query_paramssuffixed_value name_value param_style param_explodedeep_obj_paramsropenapi_headers mcp_headers json_data body_paramsrrstructured_outpute error_message error_datar-)rrr.runsJ                                               zOpenAPITool.run)NNNNN)rnror]r^rTr%rDr%rprqrrrsrHrtrurvrrwrxryr&r%)rrqr&r)r7r8r9r:r{rr __classcell__r-r-rr.rms  rmcs>eZdZdZ   ddfdd ZdddZdddZZS)OpenAPIResourcez.Resource implementation for OpenAPI endpoints.application/jsonNrnror]r^urir%rTrD mime_typerHrtrurvc s>|durt}tjt|||||d||_||_||_dS)N)rrTrDrrH)r[rzr{r r|r}r~) rWrnr]rrTrDrrHrurr-r.r{s  zOpenAPIResource.__init__r&cC d|jd|jd|jjdS)rzOpenAPIResource(name=z, uri=rr)rTrr}rerr-r-r.r& zOpenAPIResource.__repr__ str | bytesc shz|jj}t|j}d|vr`d|vr`|d}t|dkr`i}td|}|rK|jddt|d}t |D]\}}||krJ|d|} | ||<q8| D]\}} | d|dt| }qOi} |jj D]&} | j d krt|d | jrt|d | j} | d ur| d kr| | | j<qfi} t}| ||jj|jj|| | |jd Id H}||jdd dvr|}t|WStfdddDr|jWS|jWSt j!y}z;d|j"j#d|j"j$}z|j"}|d|7}Wntj%t&fy|j"jr|d|j"j7}Ynwt&||d }~wt j'y3}zt&d||d }~ww)z2Fetch the resource data by making an HTTP request.rr/z \{([^}]+)\}T)reverserr(Nr')rcrrrruz content-typerc3s|]}|vVqdSrr-)rctrr-r.rlsz'OpenAPIResource.read..)ztext/zapplication/xmlrrrr)(r}rer%rsplitrr)findallsort enumeraterrrprhasattrrTrUrrr|rrcr~rrrrrranyr$rrrrrrrrVr)rWre resource_uripartsr param_matchesexpected_param_countirrrrvaluerrrrrrrr-rr.read*sz                zOpenAPIResource.read)rNN)rnror]r^rr%rTr%rDr%rr%rHrtrurvr)r&r)r7r8r9r:r{rrrr-r-rr.r s  rcs@eZdZdZ  ddfdd ZdddZ d d!ddZZS)"OpenAPIResourceTemplatez7Resource template implementation for OpenAPI endpoints.Nrnror]r^ uri_templater%rTrDrprqrHrtrurvc s:|durt}tj|||||d||_||_||_dS)N)rrTrDrprH)r[rzr{r|r}r~) rWrnr]rrTrDrprHrurr-r.r{s  z OpenAPIResourceTemplate.__init__r&cCr)rzOpenAPIResourceTemplate(name=z, uri_template=rr)rTrr}rerr-r-r.rrz OpenAPIResourceTemplate.__repr__rrcontextContext | Nonerc sxg}|D]\}}||d|qt|j|j||jdd||jp/d|jjdt |jj p6g|j dS)z,Create a resource with the given parameters.=-z Resource for r)rnr]rrTrDrrHru) rappendrr|r}rTjoinrDrer[rHr~)rWrrr uri_partskeyrr-r-r.create_resourcesz'OpenAPIResourceTemplate.create_resource)NN)rnror]r^rr%rTr%rDr%rprqrHrtrurvrr)rr%rrqrrr&r)r7r8r9r:r{rr rr-r-rr.rs  rcsheZdZdZ       d+d,fdd Z d-d.ddZd/d"d#Zd0d%d&Zd0d'd(Zd0d)d*Z Z S)1FastMCPOpenAPIa 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_specrqrnrorT str | None route_mapslist[RouteMap] | None route_map_fnRouteMapFn | Nonemcp_component_fnComponentFn | None mcp_namesdict[str, str] | NonerHrtrurvrNr c  stjdd|pdi| ||_| |_||_ttttd|_t|} d} |p,gt }| D]} t | |}|j dus?J|j }|durz|| |}|durc|}t d| jd| jd|jWn#ty}zt d | jd| jd |d WYd}~nd}~ww|| |}t| j|jB|ptB}|tjkr|j| ||d q1|tjkr|j| ||d q1|tjkr|j| ||d q1|tjkrt d | jd| j| d7} q1t 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 rTzOpenAPI FastMCP)toolresourceresource_templatepromptrNrarbz* mapping customized by route_map_fn: type=zError in route_map_fn for rz. Using default values.)rHzExcluding route: rz$Created FastMCP OpenAPI server with z routesr-)!rzr{r|r~_mcp_component_fnr _used_namesrparse_openapi_to_http_routesDEFAULT_ROUTE_MAPPINGSrlrIrgrhrcrerT Exceptionr_generate_default_namer[rHrJr0r3_create_openapi_toolr4_create_openapi_resourcer5_create_openapi_templater6infor)rWrrnrTrrrrrHrurN http_routes num_excludedr]rirCrrcomponent_name route_tagsrr-r.r{s`%         zFastMCPOpenAPI.__init__r]r^ mcp_names_mapr&r%cCsvd}|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'rrr(8N) operation_idrsummaryrcrer/r)rWr]r*rTr-r-r.r!Es    z%FastMCPOpenAPI._generate_default_namecomponent_type:Literal['tool', 'resource', 'resource_template', 'prompt']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 rr(zName collision detected: 'z' already exists as a Nrz . Using 'z ' instead.)rrgrh)rWrTr.new_namer-r-r._get_unique_name^szFastMCPOpenAPI._get_unique_namerGc Cs.t|}t|j|j|j}||d}|jp"|jp"d|jd|j }t ||j|j |j d}t |j|||||t|jp;g|B|jd} |jdurxz||| td|dWntyw} ztd |d | d WYd} ~ nd} ~ ww| j} | |jj| <td | d |jd|j d|jdS)z?Creates and registers an OpenAPITool with enhanced description.rz Executes rbbase_description responsesrpr)rnr]rTrDrprrrHruNzTool  customized by component_fnzError in component_fn for tool r. Using component as-is.zRegistered TOOL:  ( ) with tags: )rrr4schema_definitionsopenapi_versionr1rDr-rcrerrprrmr|r[rHr~rrgrhr rrT _tool_manager_tools) rWr]rTrHcombined_schemarr tool_namer3enhanced_descriptionrrfinal_tool_namer-r-r.r"|sR      z#FastMCPOpenAPI._create_openapi_toolc Cs||d}d|}|jp|jpd|j}t||j|j|jd}t|j ||||t |j p.g|B|j d}|j durkz| ||td|dWntyj} ztd |d | d WYd} ~ nd} ~ wwt|j} ||jj| <td | d |jd|jd|j dS)zCCreates and registers an OpenAPIResource with enhanced description.r resource://z Represents r2)rnr]rrTrDrHruNz Resource r5z#Error in component_fn for resource rr6zRegistered RESOURCE: r7rbr8)r1rDr-rerr4rprrr|r[rHr~rrgrhr rr%r_resource_manager _resourcesrc) rWr]rTrH resource_namerr3r?rrfinal_resource_urir-r-r.r#sD       z'FastMCPOpenAPI._create_openapi_resourcec Csr||d}dd|jD}|d|}|r'|dddd|D7}|jp2|jp2d|j}t||j|j|j d }d d d |jDd d|jDd} t |j ||||| t |j p]g|B|jd} |jdurz||| td|dWnty} ztd|d| dWYd} ~ nd} ~ ww| j} | |jj| <td| d|jd|jd|j dS)zKCreates and registers an OpenAPIResourceTemplate with enhanced description.rcSsg|] }|jdkr|jqSr)rrTrr-r-r. sz;FastMCPOpenAPI._create_openapi_template..rArcss|] }d|dVqdS)rrNr-rr-r-r.rsz:FastMCPOpenAPI._create_openapi_template..z Template for r2objectcSsbi|]-}|jdkr|jit|jtr|jni|jr,t|jtr'd|jvs,d|jiniqS)rerD)rrTrSrrcopyrDrr-r-r.rs     z;FastMCPOpenAPI._create_openapi_template..cSs"g|] }|jdkr|jr|jqSrrrr-r-r.rFs)rrr)rnr]rrTrDrprHruNz Template r5z#Error in component_fn for template rr6zRegistered TEMPLATE: r7rbr8)r1rprr rDr-rerr4rrr|r[rHr~rrgrhr rrrB _templatesrc) rWr]rTrH template_nameruri_template_strr3r?template_params_schematemplaterfinal_template_urir-r-r.r$s^       z'FastMCPOpenAPI._create_openapi_template)NNNNNNN)rrqrnrorTrrrrrrrrrrHrtrurvrNr r)r]r^r*rr&r%)rTr%r.r/r&r%)r]r^rTr%rHrG) r7r8r9r:r{r!r1r"r#r$rr-r-rr.r s )d   >5r )r$r%r&r%)r]r^r_r`r&r>)Cr: __future__renumrr)rP collectionsrcollections.abcr dataclassesrrrtypingrr r r mcp.typesr pydantic.networksr rMfastmcp.exceptionsr fastmcp.resourcesrrfastmcp.server.dependenciesrfastmcp.server.serverrfastmcp.tools.toolrrfastmcp.utilitiesrfastmcp.utilities.loggingrfastmcp.utilities.openapirrrrrrfastmcp.serverrr7rg HttpMethodr/ RouteMapFn ComponentFnEnumr0r;r>r3rrlrmrrr r-r-r-r.sb                 E -(y8