iT%dZddlmZddlZddlmZmZmZehdZ ehdZ ehdZ dddZ ddZ ddZddZddZdS)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". 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_schemaboolreturnct|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..4s%FFFtt444FFFc8i|]\}}|t|dSrr)rsub_keysub_vals r z"_repair_schema..?s;$GW4@@@r"c0g|]}t|dSrrrvs r r!z"_repair_schema..Ds%NNN1^A>>>NNNr"Trr typeNcng|]2}t|tr|ddk0|3S)r*null) isinstancedictget)rbs r r!z"_repair_schema..YsLHHH!!!T**H/0uuV}}/F/F/F/F/Fr"c&i|]\}}|dk ||S)r )rkr)s r r&z"_repair_schema..as#KKK$!Qa7llAlllr"rnullablez$refenum)stringintegernumberbooleanc"g|] }||dk | SNr3r(s r r!z"_repair_schema..~s,555Q-AGG,3GGr") r-listr.r_SCHEMA_MAP_KEYS_SCHEMA_LIST_KEYS_SCHEMA_NODE_KEYSrpoplenupdate_fill_missing_typer/) 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$$$ Or"Dict[str, Any]cjd|vr |ddvr|Sd|vsd|vsd|vrd}nd|vsd|vrd }nd |vr|t|d tra|d rY|d d }t|trd }n5t|trd }nt|trd}nd}nd}i|d|iS)z9Infer a reasonable ``type`` if this schema node has none.r*r<rrequiredrobjectrrarrayr6rr:r8r9r7)r-r>rintfloat)rinferredsamples r rErEs ~~$v,j88 tzT115Kt5S5S DMT11 4JtF|T::tF|fa fd # #  HH  $ $  HH  & & HHHH %d %FH % %%r" parametersct|tsdidSttj|d}t|tsdidS|ddkrd|d<d|vri|d<|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. rP)r*rTrr*r)r-r.rcopydeepcopyr/)rVrFs r !sanitize_moonshot_tool_parametersrZs j$ ' '4 333dmJ774HHHH h % %4 333||Fx''#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.FfunctionrVT)r-r.appendr/rZ)r[ sanitized any_changetoolfnparamsrFnew_fns r sanitize_moonshot_toolsrfs  &(IJ##$%%    T " " "  XXj ! !"d##    T " " "  %%4V<< 6 ! !J33L(33F   99j&99 : : : :   T " " " "" -99-r"model str | Nonec |sdS|}|ddd}|ds|dkrdSd|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-kimiTmoonshotz/kimi)striplowerrsplit startswith)rgbaretails r is_moonshot_modelrts u ;;==   D ;;sA  r "D w46>>tTW__0G0G_t 5r")T)rrrrrr)rrMrrM)rVrrrM)r[r\rr\)rgrhrr)__doc__ __future__rrXtypingrrr frozensetr?r@rArrErZrfrtr3r"r rys.#""""" """""""""" 9XXXYYIHHHIIIcccdd\\\\\~&&&&6,....6r"