o i" @sdZddlmZmZmZddlmZmZmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZddlm Zddlm Zdd lm Zdd lm Zdd lm Zdd lmZmZdd lmZddlmZmZmZm Z m!Z!m"Z"ddl#m$Z$m%Z%ee&Z'edeeZ(ede eZ)ede eZ*edeeZ+ede eZ,ede eZ-edeeZ.ede eZ/de0e1efde2efddZ3Gdddee(e*e)e+e,e-e.e/fZ4ddgZ5dS) zHOpenAPI parsing logic for converting OpenAPI specs to HTTPRoute objects.)AnyGenericTypeVar)OpenAPI Operation ParameterPathItem Reference RequestBodyResponseSchema)r)r)r)r)r )r )r )r ) BaseModelValidationError) get_logger) HTTPRoute JsonSchema ParameterInfoParameterLocationRequestBodyInfo ResponseInfo)_combine_schemas_and_map_params_replace_ref_with_defsTOpenAPITSchema TReference TParameter TRequestBody TResponse TOperation TPathItem openapi_dictreturnc Cs|dd}zC|dr+t|}td|jt|tt t t t t t| }|WSt|}td|jt|ttttttt| }|WStyq}ztd||}td|td||d }~ww) z Parses an OpenAPI schema dictionary into a list of HTTPRoute objects using the openapi-pydantic library. Supports both OpenAPI 3.0.x and 3.1.x versions. openapiz3.0z0Successfully parsed OpenAPI 3.0 schema version: z0Successfully parsed OpenAPI 3.1 schema version: z"OpenAPI schema validation failed: zValidation errors: zInvalid OpenAPI schema: N)get startswith OpenAPI_30model_validateloggerdebugr# OpenAPIParser Reference_30 Schema_30 Parameter_30RequestBody_30 Response_30 Operation_30 PathItem_30parserr r rr r rrrerrorerrors ValueError)r!openapi_version openapi_30parser openapi_31e error_detailsr=s/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/experimental/utilities/openapi/parser.pyparse_openapi_to_http_routes7sT       r?c@seZdZdZdedeedeedeedee dee dee d ee d e fd d Zd e defddZdedefddZdedefddZ  d5deedBdeedBdeefddZdededBfddZde defd d!Zd"ee efdee efdBfd#d$Zd"ee efdBdee effd%d&Z d6d'ed(ee efd)e!e dBde!e fd*d+Z"d,eed-edBd(ee efdee effd.d/Z#d0ee efd(ee efdee effd1d2Z$dee%fd3d4Z&dS)7r+z[Unified parser for OpenAPI schemas with generic type parameters to handle both 3.0 and 3.1.r# reference_cls schema_cls parameter_clsrequest_body_cls response_cls operation_cls path_item_clsr7c Cs:||_||_||_||_||_||_||_||_| |_dS)z?Initialize the parser with the OpenAPI schema and type classes.N) r#r@rArBrCrDrErFr7) selfr#r@rArBrCrDrErFr7r=r=r>__init__{s  zOpenAPIParser.__init__param_inr"cCs"|dvr|Std|ddS)z@Convert string parameter location to our ParameterLocation type.)pathqueryheadercookiezUnknown parameter location: z, defaulting to 'query'rK)r)warning)rGrIr=r=r>_convert_to_parameter_locationsz,OpenAPIParser._convert_to_parameter_locationitemc Cst||jr|j}t|ts|Sz|dstd||dd}|j}|D]s}| rrcsT           zOpenAPIParser._resolve_ref schema_objc Csz0||}t||jr|jdddd}nt|tr|}n tdt|di}t|}|WSt yU}zdt |vr?tj d|dd iWYd }~Sd }~wt yr}ztj d|dd iWYd }~Sd }~ww) z1Resolves a schema and returns it as a dictionary.jsonT)modeby_alias exclude_nonez%Expected Schema after resolving, got z. Returning empty dict.-External or non-local reference not supportedz"Failed to extract schema as dict: Fexc_infoN) rcrUrA model_dumprbr)rNtyperr6rWr4 Exception)rGrlresolved_schemaresultr;r=r=r>_extract_schema_as_dicts2    z%OpenAPIParser._extract_schema_as_dictNoperation_paramspath_item_paramsc Csg}i}|pg|p g}|D]}z||}t||js)tdt|dWq|j}ddlm} t|| r:|j n|} | | } |j } |j | f} | |vrOWqd|| <i}| rw| | }|| }t||jsvt|drv|jdurv|j|d<n>t|dr|jrtt|jd}|rt|d r|jr|j}| |}||}t||jst|dr|jdur|j|d<t|d d}t|d d}t|j | |j||j||d }||Wqty}zt|d t|dd}tjd|d|ddWYd}~qd}~ww|S)z_extract_parameterss                  z!OpenAPIParser._extract_parametersrequest_body_or_refc Cs|sdSz||}t||jstdt|dWdSt|j|jd}t |dr|j r|j D]\\}}|rt |dr|j rz | |j }||j|<Wq3tyr}zdt|vr]td|d |WYd}~q3d}~wty}ztd|d |WYd}~q3d}~wwq3|WSty}z!dt|vrt|d d }tjd |d |d dWYd}~dSd}~wty}zt|d d }tjd |d |d dWYd}~dSd}~ww)z-Extract and resolve request body information.Nz*Expected RequestBody after resolving, got z. Returning None.)rrr~rrq)Failed to extract schema for media type 'rTrVrz Failed to extract request body 'Frr)rcrUrCr)rNrurrrrar~itemsrrycontent_schemar6rWr4rvr_) rGr request_bodyrequest_body_infomedia_type_strmedia_type_obj schema_dictr;ref_namer=r=r>_extract_request_bodySsz      z#OpenAPIParser._extract_request_body status_codec CsHzt|}d|kodkWSWSttfy#|dvYSw)z>Check if a status code represents a successful response (2xx).i,)r}2xx)r\r6rglower)rGrcode_intr=r=r>_is_success_status_codes z%OpenAPIParser._is_success_status_codeoperation_responsescCs^|sdSgd}|D]}||vr|||fSq |D]\}}||r,||fSqdS)zTGet the primary success response for an MCP tool. We only need one success response.N)200201202204207)rr)rGrpriority_codescoder resp_or_refr=r=r>_get_primary_success_responses  z+OpenAPIParser._get_primary_success_responsec Csti}|s|S||}|std|S|\}}td|z||}t||js=td|dt|d|WSt|j d}t |dr|j r|j D]\}} | rt | dr| j rz1d } | j } t| |jr~| j} t| tr~| d r~| d d } || } | r| | d <| |j|<WqPty}zdt|vrtd|d|d|WYd }~qPd }~wty}ztd|d|d|WYd }~qPd }~wwqP||t|<W|Sty}z$dt|vrt|dd}tjd|d|d|ddWYd }~|Sd }~wty9}zt|dd}tjd|d|d|ddWYd }~|Sd }~ww)zcExtract and resolve response information. Only includes the primary success response for MCP tools.z;No success responses found, tool will have no output schemaz Using primary success response: z2Expected Response after resolving for status code z, got z. Returning empty responses.)rr~rN#/components/schemas/rQx-fastmcp-top-level-schemarqrz' in response : rVrz+Failed to extract response for status code z from reference 'rTFrr)rr)r*rcrUrDrNrurrrar~rrr@rVrWr&rYryrr6r4rvr_)rGrextracted_responsesprimary_responserrresponse resp_inforrtop_level_schema_namerrhrr;rr=r=r>_extract_responsess           z OpenAPIParser._extract_responsesr all_schemas collectedcs*durtfdd|S)al Extract all schema names referenced by a schema (including transitive dependencies). Args: schema: The schema to analyze all_schemas: All available schema definitions collected: Set of already collected schema names (for recursion) Returns: Set of schema names that are referenced Ncst|trGd|vr:t|dtr:|d}|dr!|dd}ndSdur:|vr:|vr:|||D]}|q>dSt|trU|D]}|qNdSdS)z%Recursively find all $ref references.z$ref)z#/$defs/rrQrN)rUrbrWr&rYaddrr[)objrV schema_namerrPrr find_refsr=r>r's&        z=OpenAPIParser._extract_schema_dependencies..find_refs)set)rGrrrr=rr>_extract_schema_dependenciess z*OpenAPIParser._extract_schema_dependencies parametersrcspt}|D]}|jr||j}||q|r/|jr/|jD] }||}||q!fdd|DS)aZ Extract only the schema definitions needed for input (parameters and request body). Args: parameters: Route parameters request_body: Route request body all_schemas: All available schema definitions Returns: Dictionary containing only the schemas needed for input ci|] }|vr||qSr=r=.0rrr=r> fzDOpenAPIParser._extract_input_schema_dependencies..)rschema_rupdaterr)rGrrrneeded_schemasparamdepsrr=rr>"_extract_input_schema_dependenciesFs     z0OpenAPIParser._extract_input_schema_dependencies responsescs|rsiSt}|D]6}|jsq |jD]*}||}|||d}t|trB|vrB|||j||dqq fdd|DS)a Extract only the schema definitions needed for outputs (responses). Args: responses: Route responses all_schemas: All available schema definitions Returns: Dictionary containing only the schemas needed for outputs r)rcrr=r=rrr=r>rrzEOpenAPIParser._extract_output_schema_dependencies..) rrrrrr%rUrWr)rGrrrrrrrr=rr>#_extract_output_schema_dependenciesjs,      z1OpenAPIParser._extract_output_schema_dependenciescCsg}t|jdr |jjstdgSi}t|jdrn|jjrn|jj}t|drn|jrn|jD]=\}}zt||j rH| |}| |||<n| |||<Wq0t ym}ztd|d|WYd}~q0d}~ww|r|D]\}}t|t rt|||<qt|jjD]E\}} t| |jstd|d t| d qt| d r| jnd} gd } | D]} t| | d} | rt| |jr| }z|t| d d| }|t| d d}|t| dd}i}t| dr| jrdd| jD}||||}|||}t||t| ddt| ddt| ddt| dgp#g|||||||jd }zt|dd\}}||_||_Wn*t yh}ztd|d|d|did|_i|_WYd}~nd}~ww| |Wqt!y}z'dt"|vrt| dd}tj#d |d|d!|d"|d#d$WYd}~qd}~wt y}zt| dd}tj#d |d|d!|d"|d#d$WYd}~qd}~wwqqt$d%t%|d&|S)'z*Parse the OpenAPI schema into HTTP routes.pathsz$OpenAPI schema has no paths defined.rRschemasz%Failed to extract schema definition 'rTNz%Skipping invalid path item for path 'z ' (type: )r)r%putpostdeleteoptionsheadpatchtrace requestBodyrr`cSs i|] \}}|dr||qS)zx-)r&)rkvr=r=r>rs z'OpenAPIParser.parse.. operationIdsummaryrtags) rJmethod operation_idrrrrrrrequest_schemasresponse_schemas extensionsr7F) convert_refsz)Failed to pre-calculate schema for route  robject)ru propertiesrqrzFailed to process operation z (ID: z): TrrzFinished parsing. Extracted z HTTP routes.)&rar#rr)rNrRrrrUr@rcryrvrbrrFrurr_rEupperrrrr`rrrr7rflat_param_schema parameter_maprr6rWr4r*len)rGroutesschema_definitionsrRrrrwr;path_str path_item_objpath_level_params http_methods method_lower operation method_upperrrrr input_schemasoutput_schemasroute flat_schema param_map schema_errorop_errorop_idr=r=r>r3s                   ]zOpenAPIParser.parse)NN)N)'__name__ __module__ __qualname____doc__rrurrrrrrr rWrHrrOrrcrryr[rrrrboolrrbtuplerrrrrrrrr3r=r=r=r>r+ms   7"   cB    b   3   $    )r+N)6r typingrrropenapi_pydanticrrrrr r r r openapi_pydantic.v3.v3_0r'r1r.r2r,r/r0r-pydanticr rfastmcp.utilities.loggingrmodelsrrrrrrrrrr r)rrrrrrrr rbrWr[r?r+__all__r=r=r=r>s\(                  6K