KiOBddlmZddlmZddlmZddlmZmZd dZ dddZ d ddZ d d Z dd Z d dd Z d dd Zy)) annotations) defaultdict)Any) JsonRefError replace_refsc: t|dd}|jdi}t|||}t|tsJ|}d|vr)|j Dcic]\}}|dk7s ||}}}t ||Scc}}w#t$rt|cYSwxYw)aResolve all $ref references in a JSON schema by inlining definitions. This function resolves $ref references that point to $defs, replacing them with the actual definition content while preserving sibling keywords (like description, default, examples) that Pydantic places alongside $ref. This is necessary because some MCP clients (e.g., VS Code Copilot) don't properly handle $ref in tool input schemas. For self-referencing/circular schemas where full dereferencing is not possible, this function falls back to resolving only the root-level $ref while preserving $defs for nested references. Args: schema: JSON schema dict that may contain $ref references Returns: A new schema dict with $ref resolved where possible and $defs removed when no longer needed Example: >>> schema = { ... "$defs": {"Category": {"enum": ["a", "b"], "type": "string"}}, ... "properties": {"cat": {"$ref": "#/$defs/Category", "default": "a"}} ... } >>> resolved = dereference_refs(schema) >>> # Result: {"properties": {"cat": {"enum": ["a", "b"], "type": "string", "default": "a"}}} F)proxies lazy_load$defs) rget_merge_ref_siblings isinstancedictitems_strip_discriminator_mappingsrresolve_root_ref)schema dereferenceddefsmergedkvs i/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/utilities/json_schema.pydereference_refsr s:($FEUK zz'2&$V\4@&$'''  l "-9-?-?-ARTQQ'\AqDRLR &l3 S ( ''(s*AB A=(A=-B=BBBc@|dkDst|tsyd|vr(t|dtr|djdd|jD]I}t|trt ||dz#t|t s4|D]}t ||dzKy)alRemove discriminator.mapping entries whose values are $defs references. Pydantic emits discriminator.mapping with plain-string references like ``"#/$defs/Cat"`` that become dangling after $defs are removed by dereference_refs(). The oneOf/anyOf variants already carry their own const fields, so the mapping is redundant once refs are inlined. 2N discriminatormapping)rrpopvaluesrlist)rdepthvalueitems rrrCs rzFD1& Z0G%N##It4? eT " )% ; t $ ?-dEAI> ? ?Nc "| t}t|trt|trd|vr|d}|jDcic] \}}|dvs ||}}}t|trB|j dr1|j dd}||vr||vrt||||||hz}|rt|} | j|| S|Si} |jD]#\} } | |vrt|| | ||| | <| | | <%| St|trgt|trWtt|t|} t|d| |d| dDcgc]\}}t||||c}}|| dzS|Scc}}wcc}}w) aMerge sibling keywords from original $ref nodes into dereferenced schema. When jsonref resolves $ref, it replaces the entire node with the referenced definition, losing any sibling keywords like description, default, or examples. This function walks both trees in parallel and merges those siblings back. Args: original: The original schema with $ref and potential siblings dereferenced: The schema after jsonref processing defs: The $defs from the original schema, for looking up referenced definitions visited: Set of definition names already being processed (prevents cycles) Returns: The dereferenced schema with sibling keywords restored N$ref)r(r #/$defs//F)strict) setrrrstr startswithsplitr updater"minlenzip)originalrrvisitedrefrrsiblingsdef_namerresultkeyr$min_lenods rr r Ws*%(D!jt&D X 6"C)1)9XAQFW=W1XHX#s#z(B99S>"-t#(?#6X dGxj s FF!F cd|vr`d|vr\d|vrX|d}t|trC|jdr2|jdd}|d}||vrt ||}||d<|S|S)aResolve $ref at root level to meet MCP spec requirements. MCP specification requires outputSchema to have "type": "object" at the root level. When Pydantic generates schemas for self-referential models, it uses $ref at the root level pointing to $defs. This function resolves such references by inlining the referenced definition while preserving $defs for nested references. Args: schema: JSON schema dict that may have $ref at root level Returns: A new schema dict with root-level $ref resolved, or the original schema if no resolution is needed Example: >>> schema = { ... "$defs": {"Node": {"type": "object", "properties": {...}}}, ... "$ref": "#/$defs/Node" ... } >>> resolved = resolve_root_ref(schema) >>> # Result: {"type": "object", "properties": {...}, "$defs": {...}} r(r typer)r*r+)rr.r/r0r)rr7r9rresolveds rrrs}0Gv-&2FVn c3 CNN:$>yy~b)H'?D4X/$(! Mr&c|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)r r remove)rparampropsremoveds r _prune_paramrIsy JJ|R (Eiit$G !F<  :r**z!!%(j! JJz " Mr&c sss|St tt|jd} d d  fd  |dry|rw|j D]\}} ||d d  fd t|j D]}|r |j ||s|j 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} r Nc|dkDryttr rnjd}t|trM|j dr<|j dd}|r |j |n j| r)dvr%tfdd Drjd r$jd d urjd jD]E\}}|r|d k(r|d vr&t|tr|D]}|||dz8|||dzGyttrD]}|||dzyy)zATraverse schema tree, collecting $ref info and applying cleanups.rNr(r)r*r+titlec3&K|]}|v ywN).0rnodes r zD_single_pass_optimize..traverse_and_clean..0s I s)r@rCr(rallOfoneOfanyOfrDadditionalPropertiesFr )rSrTrUr)r#) rrr r.r/r0appendaddanyr rr")rQcurrent_def_nameskip_defs_sectionr#r7referenced_defr;r$r%def_dependenciesprune_additional_properties prune_defs prune_titles root_refstraverse_and_cleans` rrbz1_single_pass_optimize..traverse_and_cleansp 2:  dD !hhv&c3'CNN:,F%(YYs^B%7N'(8??@PQ" n5 4    HHW%,HH34=/0#jjl Q U$55*UD:Q %T*41AQRST'u.>eaiP Qd # L"4)9K L$r&T)r[)rZc|vryj|g}|r.| t}||vry||hz}|D]}||vs||syy)z.is_def_used^s{9$ 033HbA #"uHx' #xj0(8$O&h6;'< $ $ r&)NFr) rQobjectrZz str | Noner[boolr#intreturnNonerN)r9r.rdset[str] | Nonerkri)r-rr"r rkeysr ) rr`r^r_rr9 def_schemar]rgrarbs ``` @@@@r_single_pass_optimizerpsd ,*E %I4? 5 ::g D(,"' @L@L$@L @L @L  @L@LFv6d$(JJL F Hj zH E F  4TYY[) #Hx(" #  JJw % Mr&cz|xsgD]}t||}|s|s|rt||||}|r t|}|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 dereference: Whether to inline $ref references for client compatibility )rF)r`r^r_)rIrpr)r prune_paramsr_r^r` dereferencerFs rcompress_schemartsX(#3fE232j& %(C!  !&) Mr&)rdict[str, Any]rkru)r)rrr#rjrkrlrN) r5rrrrrur6rmrkr)rrurFr.rkru)FFT) rrur`rir^rir_rirkru)NTTFF)rrurrzlist[str] | Noner_rir^rir`rirsrirkru) __future__r collectionsrtypingrjsonrefrrrrr rrIrprtrOr&rrzs"#.7(t?0 $ @@@ @ @  @F$N.(- j jj"&j j  j^&*(, $ $"$$"& $  $  $$r&