o i"@sLddlmZddlmZddd Z ddddZ  ddddZdS)) annotations) defaultdictschemadictparamstrreturncCs`|di}||d}|dur|S||d<||dgvr.|d||ds.|d|S)zwReturn a new schema with *param* removed from `properties`, `required`, and (if no longer referenced) `$defs`. propertiesNrequired)getpopremove)rrpropsremovedrc/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/utilities/json_schema.py _prune_params   rFT prune_titlesboolprune_additional_properties prune_defscssss|Sttt|d}   ddfdd |ddra|ra|D] \}}||dq2ddfdd t|D] }|sX||qM|sa|dd|S)aK Optimize JSON schemas in a single traversal for better performance. This function combines three schema cleanup operations that would normally require separate tree traversals: 1. **Remove unused definitions** (prune_defs): Finds and removes `$defs` entries that aren't referenced anywhere in the schema, reducing schema size. 2. **Remove titles** (prune_titles): Strips `title` fields throughout the schema to reduce verbosity while preserving functional information. 3. **Remove restrictive additionalProperties** (prune_additional_properties): Removes `"additionalProperties": false` constraints to make schemas more flexible. **Performance Benefits:** - Single tree traversal instead of multiple passes (2-3x faster) - Immutable design prevents shared reference bugs - Early termination prevents runaway recursion on deeply nested schemas **Algorithm Overview:** 1. Traverse main schema, collecting $ref references and applying cleanups 2. Traverse $defs section to map inter-definition dependencies 3. Remove unused definitions based on reference analysis Args: schema: JSON schema dict to optimize (not modified) prune_titles: Remove title fields for cleaner output prune_additional_properties: Remove "additionalProperties": false constraints prune_defs: Remove unused $defs entries to reduce size Returns: A new optimized schema dict Example: >>> schema = { ... "type": "object", ... "title": "MySchema", ... "additionalProperties": False, ... "$defs": {"UnusedDef": {"type": "string"}} ... } >>> result = _single_pass_optimize(schema, prune_titles=True, prune_defs=True) >>> # Result: {"type": "object", "additionalProperties": False} $defsNFrnodeobjectcurrent_def_name str | Noneskip_defs_sectionrdepthintrNonec s@|dkrdSttrr2d}t|tr2|dr2|dd}|r-||n|rHdvrHtfdd d DrH drVd d urV d  D],\}}|re|d kreqZ|dvr}t|t r}|D] }|||ddqpqZ|||ddqZdStt rD] }|||ddqdSdS)zATraverse schema tree, collecting $ref info and applying cleanups.2N$refz#/$defs//titlec3s|]}|vVqdSNr).0krrr us  zD_single_pass_optimize..traverse_and_clean..)typer r!itemsallOfoneOfanyOfr additionalPropertiesFr)r,r-r.)r) isinstancerr r startswithsplitappendaddanyr r+list) rrrrrefreferenced_defkeyvalueitem)def_dependenciesrrr root_refstraverse_and_cleanr(rr?XsB         z1_single_pass_optimize..traverse_and_cleanT)r)rdef_namervisitingset[str] | Nonecsf|vrdS|g}|r1|durt}||vrdS||hB}|D]}||vr0||r0dSq"dS)zrrrFs    z*_single_pass_optimize..is_def_used)NFr) rrrrrrrrrrr%)r@rrArBrr)rCrr7r r+keysr )rrrrdefsr@ def_schemar)r=rFrrrr>r?r_single_pass_optimizes. 2  C  rJN prune_paramslist[str] | NonecCs:|pgD]}t||d}q|s|s|rt||||d}|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)rrr)rrJ)rrKrrrrrrrcompress_schemas  rM)rrrrrr)FFT) rrrrrrrrrr)NTTF) rrrKrLrrrrrrrr) __future__r collectionsrrrJrMrrrrs   0