~hrddlmZddlZddZ d d dZd dZ d d dZy) ) annotationsNc|jdi}|j|d}||S||d<||jdgvr*|dj||ds|jd|S)zwReturn a new schema with *param* removed from `properties`, `required`, and (if no longer referenced) `$defs`. propertiesNrequired)getpopremove)schemaparampropsremoveds Y/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/fastmcp/utilities/json_schema.py _prune_paramrsy JJ|R (Eiit$G !F<  :r**z!!%(j! JJz " Mctdfd |rL|jdi}t|D]}|vs|j||s|jdd|S)zaWalk the schema and optionally prune titles, unused definitions, and additionalProperties: false.ct|trrU|jd}t|tr4|j dr#j |j ddrd|vr|jdr%|jdddur|jd|jD] }| yt|tr|D] }| yy)Nz$refz#/$defs//titleadditionalPropertiesF) isinstancedictrstr startswithaddsplitrvalueslist)noderefvprune_additional_properties prune_defs prune_titles used_defswalks rr&z_walk_and_prune..walk&s dD !hhv&c3'CNN:,FMM#))C."454!,HH3T:eC/0[[] Q d # Q $rz$defsN)robjectreturnNone)setrrr)r r#r$r"defsdef_namer%r&s ``` @@r_walk_and_pruner-ss%I8 Lzz'2&T  #Hy(" # JJw % MrcP|jdddur|jd|S)z;Remove additionalProperties from the schema if it is False.rNF)rr)r s r_prune_additional_propertiesr/Ps( zz($/58 )* Mrctj|}|xsgD]}t||}|s|s|rt||||}|S)a Remove the given parameters from the schema. Args: schema: The schema to compress prune_params: List of parameter names to remove from properties prune_defs: Whether to remove unused definitions prune_additional_properties: Whether to remove additionalProperties: false prune_titles: Whether to remove title fields from the schema )r )r#r$r")copydeepcopyrr-)r prune_paramsr#r"r$r s rcompress_schemar4Ws[$]]6 "F#3fE23\%@ !%(C   Mr)r rr rr(r)FFF) r rr#boolr$r5r"r5r(r)r rr(r)NTTF) r rr3zlist[str] | Noner#r5r"r5r$r5r(r) __future__rr1rr-r/r4rrr8s" .(- 2 222"& 2  2j&*(, ! !"!!"& !  !  !r