`gj+dZddlmZddlZddlmZmZmZehdZ ehdZ ehdZ dddZ dd Z dd Zdd Zdd Zdd Zy)uHelpers for translating OpenAI-style tool schemas to Moonshot's schema subset. Moonshot (Kimi) accepts a stricter subset of JSON Schema than standard OpenAI tool calling. Requests that violate it fail with HTTP 400: tools.function.parameters is not a valid moonshot flavored json schema, details: <...> Known rejection modes documented at https://forum.moonshot.ai/t/tool-calling-specification-violation-on-moonshot-api/102 and MoonshotAI/kimi-cli#1595: 1. Every property schema must carry a ``type``. Standard JSON Schema allows type to be omitted (the value is then unconstrained); Moonshot refuses. 2. When ``anyOf`` is used, ``type`` must be on the ``anyOf`` children, not the parent. Presence of both causes "type should be defined in anyOf items instead of the parent schema". 3. Every object schema must carry a ``required`` array, even an empty one. Standard JSON Schema allows omitting it; Moonshot 400s with "required must be an array". The ``#/definitions/...`` → ``#/$defs/...`` rewrite for draft-07 refs is handled separately in ``tools/mcp_tool._normalize_mcp_input_schema`` so it applies at MCP registration time for all providers. ) annotationsN)AnyDictList>$defs properties definitionspatternProperties>allOfanyOfoneOf prefixItems>notitemscontains propertyNamesadditionalPropertiesc Ht|tr|Dcgc]}t|dc}St|ts|Si}|j D]\}}|t vrCt|tr3|j Dcic]\}}|t|dc}}||<Q|t vr/t|tr|Dcgc]}t|dc}||<|tvr't|trt|d||<|||<|||<|s|Sd|vrt|dtr|jdd|dD cgc](} t| tr| jddk7r| *} } | rot| t|dkrUt| dk(r@|j D cic]\} }| dk7s | |} } }| j| d| }n | |d<|S|S|jd dd |vr t|}d |vrZt|d trG|jd} | d vr2|d Dcgc] }||d k7r| }}|r||d <n|jd |jddk(r t|}|Scc}wcc}}wcc}wcc} wcc}} wcc}w)a5Recursively apply Moonshot repairs to a schema node. ``is_schema=True`` means this dict is a JSON Schema node and gets the missing-type + anyOf-parent repairs applied. ``is_schema=False`` means it's a container map (e.g. the value of ``properties``) and we only recurse into its values. T is_schemar typeNnullrnullablez$refenum>numberstringbooleanintegerobject) isinstancelist_repair_schemadictr_SCHEMA_MAP_KEYS_SCHEMA_LIST_KEYS_SCHEMA_NODE_KEYSpopgetlenupdate_fill_missing_type_ensure_required_array)noderitemrepairedkeyvaluesub_keysub_valvbnon_nullkmerge node_typecleaneds H/root/.hermes/venv/lib/python3.12/site-packages/agent/moonshot_schema.pyr$r$,s$BFFtt4FF dD !  "Hjjl" U " "z%'> ). $GW4@@HSM% %*UD*AHMN1^A>NHSM % %%& .u E % "HSM'"*  (z(7*;TB VT"'0H!!!T*quuV}/FHH H HW,=(>> 8}!*2..*:K$!Qa7lAKK Xa[) $,!O LLT" X%h/j&)94@LL( B B"*6"25Q-AG5G5#*  V$||Fx')(3 OkG O*HL:5s)JJ =J-J. J<J8Jc|jd}|jd}t|tr*t|tr|Dcgc] }||vs| c}|d<|Sg|d<|Scc}w)uGuarantee an object schema carries a ``required`` array (Moonshot rule). Standard JSON Schema lets you omit ``required`` when nothing is required; Moonshot 400s on that ("required must be an array"). Ensure the key is a list. When ``properties`` is known, prune ``required`` entries that don't name a real property — defensive against dangling names, which Moonshot also rejects. Mutates and returns ``node``. rrequired)r*r"r#r%)r/propsreqrs r=r.r.sl HH\ "E ((: C#t eT "+.=a!u*=D  KZ K >s  A$A$c|jd}t|trtd|Dd}i|d|iSd|vr|dvr|Sd|vsd|vsd|vrd}nmd |vsd |vrd }nbd |vr\t|d trI|d rD|d d }t|trd}n+t|t rd}nt|t rd}nd}nd}i|d|iS)z9Infer a reasonable ``type`` if this schema node has none.rc3LK|]}t|ts|dvs|yw)>r rN)r"str).0ts r= z%_fill_missing_type..s R1Z3%7A\Nr rr?rr!rrarrayrrrrr)r*r"r#nextboolintfloat)r/r;concreteinferredsamples r=r-r-s I)T" R R  *$))) ~):5 tzT15Kt5S DMT1 4JtF|T:tF|fa fd # H  $ H  &HH %d %FH %%ct|tsdigdSttj|d}t|tsdigdS|j ddk7rd|d<d|vri|d<t ||S)zNormalize tool parameters to a Moonshot-compatible object schema. Returns a deep-copied schema with the two flavored-JSON-Schema repairs applied. Input is not mutated. r!)rrr?Trrr)r"r%r$copydeepcopyr*r.) parametersr1s r=!sanitize_moonshot_tool_parametersrVs j$ ' CCdmmJ74HH h % CC||Fx'#8#!#8$ OrQc||s|Sg}d}|D]}t|ts|j|%|jd}t|ts|j|X|jd}t |}||ur"d}i|d|i}|ji|d|i|j||r|S|S)zGApply ``sanitize_moonshot_tool_parameters`` to every tool's parameters.FfunctionrUT)r"r%appendr*rV)tools sanitized any_changetoolfnparamsr1new_fns r=sanitize_moonshot_toolsras  &(IJ#$%   T "  XXj !"d#   T "  %4V< 6 !J33L(3F   99j&9 :   T "#"#9--rQc|sy|jj}|jddd}|jds|dk(ry|dk(s|jd ryd |vsd |vs|jdryy) aTrue for any Kimi / Moonshot model slug, regardless of aggregator prefix. Matches bare names (``kimi-k2.6``, ``moonshotai/Kimi-K2.6``) and aggregator- prefixed slugs (``nous/moonshotai/kimi-k2.6``, ``openrouter/moonshotai/...``). Detection by model name covers Nous / OpenRouter / other aggregators that route to Moonshot's inference, where the base URL is the aggregator's, not ``api.moonshot.ai``. F/rzkimi-kimiTk3)zk3.zk3-moonshotz/kimi)striplowerrsplit startswith)modelbaretails r=is_moonshot_modelros~  ;;=   D ;;sA r "D w46> t|t~6TW_0G rQ)T)r/rrrKreturnr)r/Dict[str, Any]rprq)rUrrprq)rZList[Dict[str, Any]]rprr)rlz str | NonerprK)__doc__ __future__rrStypingrrr frozensetr&r'r(r$r.r-rVrarorQr=rxsc4# "" XYHIcd`F&&D..6rQ