o i5[ @sFdZddlmZddlmZddlmZmZmZee Z dedBdedBfd d Z dd e e efd e dBde e effd dZde e efde e effddZ ddededee e efe e e e e ffffddZdede e effddZ  d de e efde e efdBde dBde e efdBfddZgdZdS)!z5Schema manipulation utilities for OpenAPI operations.)Any) get_logger) HTTPRoute JsonSchema ResponseInfoschemaNreturncCs|rt|ts |S|}gd}|D] }||vr||qd|vr9dd|dD|d<|ds9|dd|vrNt|d|d<|dsN|dd|vrjt|dtrct|d|d<|S|ddurj |S)zW Clean up a schema dictionary for display by removing internal/complex fields. ) allOfanyOfoneOfnotnullable discriminatorreadOnly writeOnly deprecatedxml externalDocs propertiescSi|] \}}|t|qS)clean_schema_for_display).0kvrrt/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/experimental/utilities/openapi/schemas.py 0sz,clean_schema_for_display..itemsadditionalPropertiesT) isinstancedictcopypoprr)rcleanedfields_to_removefieldrrrr s6      rinfo descriptionc Cs4|}|d}r2t|tr1|dr$|dd}d||d<n9|ds1td|dn+|d }rPd|vrDt||d <nd d |D|d <n |d }r]t||d <d D]}t ||gD] \}} t| |||<qiq_|d} rt| t st| |d<|d|r|ds||d<|S)a Replace openapi $ref with jsonschema $defs recursively. Examples: - {"type": "object", "properties": {"$ref": "#/components/schemas/..."}} - {"type": "object", "additionalProperties": {"$ref": "#/components/schemas/..."}, "properties": {...}} - {"$ref": "#/components/schemas/..."} - {"items": {"$ref": "#/components/schemas/..."}} - {"anyOf": [{"$ref": "#/components/schemas/..."}]} - {"allOf": [{"$ref": "#/components/schemas/..."}]} - {"oneOf": [{"$ref": "#/components/schemas/..."}]} Args: info: dict[str, Any] description: str | None Returns: dict[str, Any] $refz#/components/schemas//#/$defs/z#/z/External or non-local reference not supported: z. FastMCP only supports local schema references starting with '#/'. Please include all schema definitions within the OpenAPI document.rcSrr)_replace_ref_with_defs)r prop_name prop_schemarrrrpsz*_replace_ref_with_defs..r)r r r rr() r"getr str startswithsplit ValueErrorr-r enumeratebool) r'r(rref_path schema_namer item_schemasectioniitemrrrrr-Js>       r-cCsd|vs d|vs d|vr|St|dtrd|dvr|Sd|vr|d}t|tr|dvr|dkr[d|vr[|}gd }i}|D] }||vrP||||<qC|ddig|d<|S|d krd |vr|}gd }i}|D] }||vr|||||<qo|ddig|d<|Si}|}gd }|D] }||vr||||<q|ddig|d<|Sgd }i}|}|D] }||vr||||<q|ddig|d<|S|S) z Make an optional parameter schema nullable to allow None values. For optional parameters, we need to allow null values in addition to the specified type to handle cases where None is passed for optional parameters. r r r typenull)arrayobjectr?r)defaultr(titleexampler@r)r r0listr1r"r#)r original_type array_schematop_level_fieldsnullable_schemar& object_schemaoriginal_schemarrr!_make_optional_parameter_nullables^ rKTroute convert_refsc si}g}i}ttttd}i}|jD] }||j|jq|jr|jjrtt|jj}|rsz3_combine_schemas_and_map_params..N__)location openapi_name(z parameter) r)bodyrBrz [^a-zA-Z0-9_]_ body_datar@)r=rrRcst|tr0d|vr#t|dtr#|d}|dr#|dd|D]}|q'dSt|tr>|D]}|q7dSdS)z%Recursively find all $ref references.r)r,r*r+N)r r!r1r2rSr3valuesrD)valuerefrr<)find_refs_in_value used_refsrrrcs      z;_combine_schemas_and_map_params..find_refs_in_valueFcsi|] \}}|vr||qSrr)rname def_schema)rdrrrs z3_combine_schemas_and_map_params..$defs) set parametersrXrSre request_bodycontent_schemanextiterr-r(r0r rDr!updateextendr#r`keysrRappendschema_r" capitalizerlowerresubisdigitrequest_schemaslen) rLrMrrR parameter_mapparam_names_by_location body_propsparam content_type body_schema merged_propsmerged_required sub_schemaall_non_body_paramslocation_paramsbody_param_namescolliding_params suffixed_name param_schema original_desc location_descr.r/ param_nameruresult schema_defsall_defsrer collected_all initial_countr)rcrUrdr_combine_schemas_and_map_paramss                                       rcCst|\}}|S)a@ Combines parameter and request body schemas into a single schema. Handles parameter name collisions by adding location suffixes. This is a backward compatibility wrapper around _combine_schemas_and_map_params. Args: route: HTTPRoute object Returns: Combined schema dictionary )r)rLrr^rrr_combine_schemass r responsesschema_definitionsopenapi_versioncCs|sdSgd}d}|D] }||vr||}nq |dur/|D] \}}|dr.|}nq!|dus6|js8dSgd}d}|D]} | |jvrN|j| }nq@|durj|jrjtt|j} |j| }td| |rqt|tssdSt |} d| vr|r| d} | dr| dd } | |vrt || } |r|d rd d l m }|| |} | d dkrdd| idgdd}|} |r|}|D]\}}t|trt |||<q|r|d rd d l m }t|D] }||||||<q|| d<| S)a Extract output schema from OpenAPI responses for use as MCP tool output schema. This function finds the first successful response (200, 201, 202, 204) with a JSON-compatible content type and extracts its schema. If the schema is not an object type, it wraps it to comply with MCP requirements. Args: responses: Dictionary of ResponseInfo objects keyed by status code schema_definitions: Optional schema definitions to include in the output schema openapi_version: OpenAPI version string, used to optimize nullable field handling Returns: dict: MCP-compliant output schema with potential wrapping, or None if no suitable schema found N)2002012022042)zapplication/jsonzapplication/vnd.api+jsonzapplication/hal+jsonzapplication/ld+jsonz text/jsonz/Using non-JSON content type for output schema: r)r,r*r+z3.0r)%convert_openapi_schema_to_json_schemar=r@rT)r=rrRzx-fastmcp-wrap-resultrg)rr2rkrlrmloggerdebugr r!r-r3json_schema_converterrr0r"rDrp)rrr success_codes response_info status_code resp_infojson_compatible_typesrr~first_content_type output_schemar7r8rwrapped_schemaprocessed_defsredef_namerrr$extract_output_schema_from_responsess            r)rrrKrr)N)T)NN)__doc__typingrfastmcp.utilities.loggingrmodelsrrr__name__rrr!r1r-rKr6tuplerrr__all__rrrrsJ  ?   "9P" l  ~