Ki" dZddlmZmZmZddlmZmZmZm 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/de0e1efde2efdZ3Gddee(e*e)e+e,e-e.e/fZ4ddgZ5y) 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 |jdd} |jdrvtj|}tj d|j t|ttttttt| }|jSt!j|}tj d|j t|t"t$t&t(t*t,t.| }|jS#t0$rT}tj3d||j5}tj3d|t7d||d }~wwxYw) 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 rr rerrorerrors ValueError)r"openapi_version openapi_30parser openapi_31e error_detailss l/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/utilities/openapi/parser.pyparse_openapi_to_http_routesr@7sJ#&&y"5O)L  % %e ,#22<@J LLB:CUCUBVW # F<<> !!// =J LLB:CUCUBVW # F<<> ! L 9!=>  *=/:;3M?CD!K Ls BDA5D E.AE))E.ceZdZdZdedeedeedeedee dee dee d ee d e fd Zd e d efdZded efdZded efdZ d(deedzdeedzd eefdZded edzfdZde d efdZdee efd ee efdzfdZdee efdzd ee effdZ d)dedee efd e!e dzd e!e fd!Z"d"eed#edzdee efd ee effd$Z#d%ee efdee efd ee effd&Z$d ee%fd'Z&y)*r-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_clsr9c ||_||_||_||_||_||_||_||_| |_y)z?Initialize the parser with the OpenAPI schema and type classes.N) r%rBrCrDrErFrGrHr9) selfr%rBrCrDrErFrGrHr9s r?__init__zOpenAPIParser.__init__{sJ *$* 0(**.param_inr#cB|dvr|Stjd|dy)z@Convert string parameter location to our ParameterLocation type.)pathqueryheadercookiezUnknown parameter location: z, defaulting to 'query'rP)r+warning)rJrMs r?_convert_to_parameter_locationz,OpenAPIParser._convert_to_parameter_locations* < <O5hZ?VWXrLitemct||jr|j}t|ts|S |j dst d||j djd}|j}|D]}|jrt|tr|t|}nt|tr||jjvrt||d}n|j r||j vr|j |}no|dk(rt#|dr |j$}nQt#||rt||d}n7d}n4t|t&r|j)|}nt d|d|d|t d |d |dt||jr|j+|S|S|S#t,t.t0t2t f$r}t d |d ||d}~wwxYw) z.Resolves a reference to its target definition.z#/z/External or non-local reference not supported: /N componentszCannot traverse part 'z' in reference ''zReference part 'z' not found in path 'zFailed to resolve reference '': ) isinstancerBrefstrr(r8stripsplitr%isdigitlistintr __class__ model_fieldsgetattr model_extrahasattrrXdictr' _resolve_refAttributeErrorKeyError IndexError TypeError)rJrUref_strpartstargetpartr=s r?rizOpenAPIParser._resolve_refs dD.. /hhGgs+ , Y))$/$I'S d+11#6!D||~*VT*B!'D !2#FI66#3#3#@#@@%,VT4%@F#//DF#a&H LL=aSAEL RI  LL=aSAEL RI s0A?B C, )B:4C,: C,C'!C,'C,Noperation_paramspath_item_paramsc <g}i}|xsg|xsgz}|D]8} |j|}t||js#tj dt |dO|j }ddlm} t|| r |jn|} |j| } |j} |j| f} | |vrd|| <i}| r`|j| }|j| }t||jst|dr|j |j |d<nt|dr|j"rt%t'|j"j)d}|rt|d rw|j*rk|j*}|j|}|j|}t||js't|dr|j |j |d<t-|d d}t-|d d}t/|j| |j0||j2|| }|j5|;|S#t6$r@}t-|d t-|dd}tj9d|d|dYd}~d}~wwxYw)z>'7'I'I ,0,H,H,V)150A0A,0O- *+@$BTBT U '(=y I 5 = = I;P;X;X-i8")Y= 7D9+"+&//, ) 5 5# !'' 3YS j  $ &',y*Q  3J#a&H2E9EH LL28*CsCe   2E9EH LL28*CsCe    ssA EAE+ E8 E*C40E4 E=)D+&E+ E7EEEE G%(8F%% G%1*G  G% status_codec t|}d|cxkxrdkScS#ttf$r|jdvcYSwxYw)z>Check if a status code represents a successful response (2xx).i,)r2xx)rbr8rmlower)rJrcode_ints r?_is_success_status_codez%OpenAPIParser._is_success_status_codesP =;'H((S( (( (I& =$$&*<< < =s!AAoperation_responsesc|sygd}|D]}||vs|||fcS|jD]\}}|j|s||fcSy)zTGet the primary success response for an MCP tool. We only need one success response.N)200201202204207)rr)rJrpriority_codescoder resp_or_refs r?_get_primary_success_responsez+OpenAPIParser._get_primary_success_responsesy#=# 9D**1$788 9 )<(A(A(C 2 $K++K8#[11 2 rLc Ti}|s|S|j|}|stjd|S|\}}tjd| |j|}t ||j s'tj d|dt|d|St|j}t|dr|jr|jjD]\}} | s t| ds| js# d } | j} t | |jrA| j} t | t r%| j#d r| j%d d } |j'| } | r| | d <| |j(|<||t!|<|S#t*$r6}dt!|vrtj-d|d|d|Yd }~d }~wt.$r)}tj-d|d|d|Yd }~,d }~wwxYw#t*$rF}dt!|vrt1|dd}tj-d|d|d|dYd }~|Sd }~wt.$r8}t1|dd}tj-d|d|d|dYd }~|Sd }~wwxYw)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.)rrrN#/components/schemas/rWx-fastmcp-top-level-schemarxrz' in response : r\rz+Failed to extract response for status code z from reference 'rZFry)rr+r,rir[rFrSr|rrrgrrrrBr\r]r(r_rrr8r6r}re)rJrextracted_responsesprimary_responserrresponse resp_inforrtop_level_schema_namerrnrr=rs r?_extract_responsesz OpenAPIParser._extract_responsess"8:"& & ==>QR LLV W&  [ 7 }EFH ((5Hh(9(9:H V>**FH+*%1E1EFIx+0@0@6>6F6F6L6L6N&2NN&#N4GH*<< 481+9+K+KL),8J8JK*6*:*:#-gs#;@R@R$;A"=DMM# K 0 1&#"G * NRU !S !&"LL"KNK[\//:m2aS!B )"LL"KNK[\//:m2aS!B >#a&H{E9=H LL=k]K##+*Cs4   #" {E9=H LL=k]K##+*Cs4    #" suA HAH. H; H B F'H' H0,G!H! H-H HHH J'";I## J'/-J""J'r all_schemas collectedcB tfd|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 ctt|trd|vret|dtrR|d}|jdr|j dd}ny$|vr |vrj |||j D] }| yt|tr|D] }| yy)z%Recursively find all $ref references.z$ref)z#/$defs/rrWrN)r[rhr]r(r_addrra)objr\ schema_namerrUrr find_refss r?rz=OpenAPIParser._extract_schema_dependencies..find_refs's#t$S=ZF S%Af+C~~&KL&)iinR&8 "-'y8';6! k2!+k":;!ZZ\%Ee$%C&$DdO$'rL)set)rJrrrrs ``@r?_extract_schema_dependenciesz*OpenAPIParser._extract_schema_dependenciess'"  I $8 &rL parametersrctt}|D]<}|js|j|j|}|j|>|rN|jrB|jj D]%}|j||}|j|'|Dcic] }||vs|||c}Scc}w)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 )rschema_rupdaterr) rJrrrneeded_schemasparamdepsrrs r?"_extract_input_schema_dependenciesz0OpenAPIParser._extract_input_schema_dependenciesFs"  ,E}}88 T%%d+ , L77"."="="D"D"F ,88U%%d+ , 1? (,$+BUD+d# #   s B5*B5 responsesc|r|siSt}|jD]}|js|jjD]t}|j||}|j ||j d}t |tsH||vsM|j||j||||v|Dcic] }||vs|||c}Scc}w)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)r) rrrrrr'r[r]r) rJrrrrrrrrs r?#_extract_output_schema_dependenciesz1OpenAPIParser._extract_output_schema_dependenciesjs I#&5!((* H**"*"9"9"@"@"B 88U%%d+,001MN k3/K;4N"&&{355#K0#"06   $1? (,$+BUD+d# #   s  CCcF g}t|jdr|jjstj dgSi}t|jdr|jj r|jj }t|dr|j rs|j jD]V\}} t||jr&|j|}|j|||<n|j|||<X|r7|jD]$\}}t|tst|||<&|jjjD]\}} t| |js&tj d|d t!| d Ct| d r | j"nd} gd } | D]} t%| | d} | st| |j&s+| j)} |j+t%| d d| }|j-t%| d d}|j/t%| dd}i}t| drJ| j0r>| j0jDcic]\}}|j3dr||}}}|j5|||}|j7||}t9||t%| ddt%| ddt%| ddt%| dgxsg|||||||j: } t=|d\}}||_||_ |jC|tjKd$tM|d%|S#t$r&}tj d|d|Yd}~d}~wwxYwcc}}w#t$r9}tj d|d|d|did|_i|_ Yd}~d}~wwxYw#tD$rI}dtG|vrt%| dd}tjId|d|d |d!|d"#Yd}~d}~wt$r;}t%| dd}tjId|d|d |d!|d"#Yd}~d}~wwxYw)&z*Parse the OpenAPI schema into HTTP routes.pathsz$OpenAPI schema has no paths defined.rXschemasz%Failed to extract schema definition 'rZNz%Skipping invalid path item for path 'z ' (type: )r)r'putpostdeleteoptionsheadpatchtrace requestBodyrrfzx- operationIdsummaryrtags) rOmethod operation_idrrrrrrrequest_schemasresponse_schemas extensionsr9F) convert_refsz)Failed to pre-calculate schema for route  robject)r| propertiesrxrzFailed to process operation z (ID: z): TryzFinished parsing. Extracted z HTTP routes.)'rgr%rr+rSrXrrr[rBrirr}rhrrHr|rrerGupperrrrrfr(rrrr9rflat_param_schema parameter_maprr8r]r6r,len)rJroutesschema_definitionsrXrrr~r=path_str path_item_objpath_level_params http_methods method_lower operation method_upperrrrrkv input_schemasoutput_schemasroute flat_schema param_map schema_errorop_errorop_ids r?r5zOpenAPIParser.parses"$t||W-T\\5G5G NNA BI  4<< .4<<3J3J00Jz9-*2D2D$.$6$6$<$<$>LD& %fd.@.@A.2.?.?.GO7;7S7S /8.t48<7S7S &8.t4"  2 8 8 : N ffd+/Ef/M&t, N (,||'9'9'?'?'As  #HmmT-?-?@;H:YtTaObNccde=,7((  L!-[  #M<F It7I7I!J#/#5#5#7LT%)%=%=#I|TBDU& -1,F,F#I}dC-)%)$;$;#I{DA% &( "9m<AVAV-6,A,A,G,G,I*$(Aq#$<<#5!"1*J*)-(O(O&-.) *.)Q)Q%.* !*!)#/)0M4)P$+Iy$$G(/ =$(O!(FB!?!E2'1):&/,9-;'1,0,@,@!"55T %-262K7BE32;E/ e,U[ 1s j  3CK= NO G%CD6QCPv*T ) 5"NN"KL>YZ[cZddfgsft u )1.07E335E// 5& Jc$O" ' =) L :<.(SYZ_Y``cdlcmn%)%% ' =) L :<.(SYZ_Y``cdlcmn%)%s{ANB P O ;A:P 6OP  N>N99N>P  P /P?P P  P  R >Q R %0RR )NN)N)'__name__ __module__ __qualname____doc__rr|rrrrrr r!r]rKrrTrrirrrarrrrboolrrhtuplerrrrrrrrr5rLr?r-r-msif//J'/M / J' / |, /9o/J'/I//.s?P555n#*D.2-1a s)d*a s)d*a  m  a F@@SWAW@D=3=4=#'S> sCx4 .\##'S>D#8\# c< \#D&* 11#s(^1s8d? 1 S 1f" '" &," #s(^ " c3h " H' \)*' #s(^'  c3h ' R[tI[rLr-r@N)6rtypingrrropenapi_pydanticrrrr r r r r openapi_pydantic.v3.v3_0r)r3r0r4r.r1r2r/pydanticrrfastmcp.utilities.loggingrmodelsrrrrrrrrrrr+rrrrrrr r!rhr]rar@r-__all__rrLr?r"s.N((   ;>><>B<8/0 H  :w 3 )VY / \9l ; \9l ; ~{NC K; 7 \9l ; K; 7 3LtCH~3L$y/3LlA    A L" rL