yj+dZddlmZddlZddlmZmZmZehdZ ehdZ ehdZ dddZ ddZ ddZddZddZddZdS) 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 propertyNamesadditionalPropertiesTnoder is_schemaboolreturnc t|tr d|DSt|ts|Si}|D]\}}|tvr7t|tr"d|D||<E|t vr%t|trd|D||<s|t vr0t|trt|d||<|||<|||<|s|Sd|vrt|dtr|ddd |dD}|r|t|t|dkrVt|d krz"_repair_schema..7s%FFFtt444FFFc8i|]\}}|t|dSrr)rsub_keysub_vals r z"_repair_schema..Bs;$GW4@@@r"c0g|]}t|dSrrrvs r r!z"_repair_schema..Gs%NNN1^A>>>NNNr"Trr typeNcng|]2}t|tr|ddk0|3S)r*null) isinstancedictget)rbs r r!z"_repair_schema..\sLHHH!!!T**H/0uuV}}/F/F/F/F/Fr"c&i|]\}}|dk ||S)r )rkr)s r r&z"_repair_schema..ds#KKK$!Qa7llAlllr"rnullablez$refenum>numberstringbooleanintegerc"g|] }||dk | S)Nr3r(s r r!z"_repair_schema..s,555Q-AGG,3GGr"object)r-listr.r_SCHEMA_MAP_KEYS_SCHEMA_LIST_KEYS_SCHEMA_NODE_KEYSrpoplenupdate_fill_missing_typer/_ensure_required_array) rrrepairedkeyvaluenon_nullmerge node_typecleaneds r rr,s$GGFFFFF dD ! !  "Hjjll"" U " " "z%'>'> "(- HSMM% % %*UD*A*A %NNNNNHSMM % % %%&& & .u E E E % "HSMM  (z(7*;TBB VT"""HHx0HHH  H HW,=(>(>>> 8}}!!KK(..*:*:KKK Xa[))) $,!O LLT""" X%h//j&)94@@LL(( B B B55(6"2555G %#*   V$$$||Fx'')(33 Or"Dict[str, Any]c|d|d}t|tr'ttrfd|D|d<ng|d<|S)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``. rrequiredcg|]}|v| Sr3r3)rrpropss r r!z*_ensure_required_array..s===a!u*****r")r/r-r>r.)rreqrSs @r rFrFsy HH\ " "E ((:  C#t eT " " >====3===D Z Kr"c|d}t|tr"td|Dd}i|d|iSd|vr|dvr|Sd|vsd|vsd|vrd}nd |vsd |vrd }nd |vr|t|d tra|d rY|d d }t|trd}n5t|t rd}nt|t rd}nd}nd}i|d|iS)z9Infer a reasonable ``type`` if this schema node has none.r*c3LK|]}t|t|dv|V dS)>r<r,N)r-str)rts r z%_fill_missing_type..s; R R1Z3%7%7 RA\Nr<rrPrr=rrarrayr6rr9r:r7)r/r-r>nextrintfloat)rrLconcreteinferredsamples r rErEsV  I)T""* R R R R R   *$)))) ~~):55 tzT115Kt5S5S DMT11 4JtF|T::tF|fa fd # #  HH  $ $  HH  & & HHHH %d %FH % %%r" parametersc$t|tsdigdSttj|d}t|tsdigdS|ddkrd|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=)r*rrPTrr*r)r-r.rcopydeepcopyr/rF)rarGs r !sanitize_moonshot_tool_parametersres j$ ' 'D CCCdmJ774HHHH h % %D CCC||Fx''#8##!#8$$$ Or"toolsList[Dict[str, Any]]c|s|Sg}d}|D]}t|ts||-|d}t|ts||m|d}t |}||ur&d}i|d|i}|i|d|i|||r|n|S)zGApply ``sanitize_moonshot_tool_parameters`` to every tool's parameters.FfunctionraT)r-r.appendr/re)rf sanitized any_changetoolfnparamsrGnew_fns r sanitize_moonshot_toolsrqs  &(IJ##$%%    T " " "  XXj ! !"d##    T " " "  %%4V<< 6 ! !J33L(33F   99j&99 : : : :   T " " " "" -99-r"model str | NonecD|sdS|}|ddd}|ds|dkrdS|dks|d rdSd |vsd |vs|drdSdS) 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/r1zkimi-kimiTk3)zk3.zk3-moonshotz/kimi)striplowerrsplit startswith)rrbaretails r is_moonshot_modelrs u ;;==   D ;;sA  r "D w46>>t t||t~66|tTW__0G0G_t 5r")T)rrrrrr)rrNrrN)rarrrN)rfrgrrg)rrrsrr)__doc__ __future__rrctypingrrr frozensetr?r@rArrFrErerqrr3r"r rs#4#""""" """""""""" 9XXXYYIHHHIIIcccdd`````F&&&&&D.....6r"