kKjs`ddlmZddlZddlmZddlmZddlmZm Z ddZ ddZ ddZ dd Z dd Zdd Z d dd Zdd ZddZehdZehdZehdZehdZehdZehdZ d ddZ d d!dZy)") annotationsN) defaultdict)Any) JsonRefError replace_refsc syttdfd jD]\}}||d\tt d fd t  fdDS)uJCheck whether any definitions in ``$defs`` form a reference cycle. A cycle means a definition directly or transitively references itself (e.g. Node → children → Node, or A → B → A). ``jsonref.replace_refs`` silently produces Python-level object cycles for these, which Pydantic's serializer rejects. FcVt|trw|jd}t|tr7|j dr&|j |j dd|jD] }|| yt|tr|D] }|| yy)N$ref#/$defs//) isinstancedictgetstr startswithaddsplitvalueslist)objsourcerefvitem _collect_refsedgess n/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/utilities/json_schema.pyrz(_defs_have_cycles.._collect_refss c4 ''&/C#s#z(Bf !!#))C."45ZZ\a(" T "dF+#)rc|<j|dD]$}|vr|k(ry|k(s|s$y|<y)NTF)r) nodeneighborDONEIN_STACK UNVISITED _has_cycledefsrstates rr)z%_defs_have_cycles.._has_cycle*s_d  $+Ht#X(*X)+ 80D ,d rc3BK|]}|k(xr|ywNr#).0namer(r)r+s r z$_defs_have_cycles..6s(NuT{i'enumconstdefaultexamples)rrrequire_discriminator_propertyr5_strip_discriminatorr)rskip _DATA_KEYSr=rrs rrGrGMs#t#%L7c>+KW^ 05C=   #1(T Q*_*>q*A A#  #t7:;st$T*s;; J >> 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"}}} $defsF)proxies lazy_load) r<r9rresolve_root_refr_merge_ref_siblingsrrr5rGrRecursionError)rL dereferencedr*mergedr=rs rdereference_refsr_sF  'F GR01''!($FEUK zz'2&$V\4@&$'''  l "-9-?-?-AR-ATQQ'\AqD-ALR,L9 S . )(  '' (s*AB4 B.B.B4.B44CCc "| 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 Nr )r rWr r r F)strict) r4rrr5rrrr[updaterminlenzip)originalr]r*visitedrr=rsiblingsdef_namer^rRrSvaluemin_lenods rr[r[s*%(D!jt&D X 6"C)1)9X)9AQFW=W1)9HX#s#z(B99S>"-t#(?#6X dGxj s FF!F cHd|vrd|vrd|vr|d}t|trz|jdri|jdd}|d}||vrLt ||}|j |j Dcic] \}}|dvr||c}}||d<|S|Scc}}w)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 rWtyper r r >r rW)rrrrrrbr5)rLrrir*resolvedrSrjs rrZrZ&s0Gv-&2FVn c3 CNN:$>yy~b)H'?D4X/+1,,.*8JC&77U *8%)! Ms<B ctj|}|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`. propertiesNrK)rQdeepcopyrpopremove)rLparampropsremoveds r _prune_paramry[s]]6 "F JJ|R (Eiit$G !F<  :r**z!!%(j! JJz " Mr>roallOfr@r5rArKrrr >titlereadOnly writeOnly deprecated description>rBrCrDexamplerE>rr definitions dependenciesdependentSchemaspatternPropertiesrW> ifnotelsethenr5contains contentSchema propertyNamesadditionalItemsunevaluatedItemsadditionalPropertiesunevaluatedProperties>rzr@rA prefixItemsc sss|St tt|jd} d d  fd  |ddrz|rx|j D]\}} ||dd 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} rWNTc|dkDryttr rnjd}t|trM|j dr<|j dd}|r |j |nj||rzrRdvrNtdtr;tfdtDstd Drjd r$jd d urjd jD]\}}|r|d k(r|s|||d zd  |tvr5t|tr%|jD]} | ||d zd]|tvrf|t vr't|t"r|D]} | ||d zd|t$vr|||d zd|||d zd ytt"rD]} | ||d z|yy)uCTraverse schema tree, collecting $ref info and applying cleanups. The `in_schema` flag tracks whether the current node is reached via a known JSON-Schema-valued position (root, `properties` value, `items`, `allOf` element, etc.). When False — e.g. we descended through a user extension key like `x-ui` whose payload is opaque to us — we still collect `$ref` references (they may point at `$defs` the user cares about) but we skip all cleanups so we don't mutate user data that happens to look metadata-shaped. 2Nr r r r r{c3&K|]}|v ywr-r#)r.r=r$s rr0zD_single_pass_optimize..traverse_and_clean..0s@/?!AI/?sc3,K|] }|tvywr-)_METADATA_KEYS)r.r=s rr0zD_single_pass_optimize..traverse_and_clean..1sADqqN2DsrFrWr )depth in_schemaT)rrrrrrappendrr7_SCHEMA_KEYWORDSallrtr5_SUBSCHEMA_MAP_KEYSr_LITERAL_KEYWORDS_SUBSCHEMA_LIST_KEYSr_SUBSCHEMA_VALUE_KEYS)r$current_def_nameskip_defs_sectionrrrreferenced_defrSrj sub_schemardef_dependenciesprune_additional_properties prune_defs prune_titles root_refstraverse_and_cleans` rrz1_single_pass_optimize..traverse_and_cleans#" 2:  dD !hhv&c3'CNN:,F%(YYs^B%7N'(8??@PQ" n5!4"4=#6@/?@@ADAAHHW%0!78EAHH34#jjl U$ !&/uqyE--*UD2I&+lln *&,"'!)&* '5 ++..:eT3J %* ,"'!)&* !&//&(#ai"&   #+519w+~d #"*%!)y$r)rr)rrc|vryj|g}|r.| t}||vry||hz}|D]}||vs||syy)z.is_def_usedsv9$ 033HbA #"uHx' #xj0(8O&h6;'< $ (8 r)NFrT) r$objectrz str | Nonerr3rr6rr3r1r2r-)rirrset[str] | Noner1r3)r4rrrr5keysrt) rLrrrr*ri def_schemarrrrs ``` @@@@r_single_pass_optimizersd ,*E %I4? 5 ::g D(,"' CC$C C C  C  CCLvFd$(JJL Hj zHPT U%1  4TYY[)Hx("*  JJw % Mrc|r t|}t|}|xsgD]}t||}t|||d}|S)a Compress and optimize a JSON schema for MCP compatibility. Args: schema: The schema to compress prune_params: List of parameter names to remove from properties prune_additional_properties: Whether to remove additionalProperties: false. Defaults to False to maintain MCP client compatibility, as some clients (e.g., Claude) require additionalProperties: false for strict validation. prune_titles: Whether to remove title fields from the schema dereference: Whether to dereference $ref by inlining definitions. Defaults to False; dereferencing is typically handled by middleware at serve-time instead. )rvT)rrr)r_rZryr)rL prune_paramsrr dereferencervs rcompress_schemars\*!&)f %F##fE2$ #!$? F Mr)r*dict[str, Any]r1r3)rrr1r)rLrrMrr1r)rLrr1rr-) rfrr]rr*rrgrr1r)rLrrvrr1r)FFT) rLrrr3rr3rr3r1r)NFFF) rLrrzlist[str] | Nonerr3rr3rr3r1r) __future__rrQ collectionsrtypingrjsonrefrrr9r<rGrNrFr_r[rZry frozensetrrrrrrrrr#rrrss" #.,O^(:>M(h $ @@@ @ @  @F2j0   QR  " $!!KL (- m mm"&m m  md&*(- ( ("("&( (  (  (r