eildZddlmZddlZddlmZmZmZehdZ 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|vr3t|dtr|dd|Sd |vr|St|S) 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. c0g|]}t|dSTr_repair_schema).0items 7/home/div/.hermes/hermes-agent/agent/moonshot_schema.py z"_repair_schema..4s%FFFtt444FFFc8i|]\}}|t|dSrr)rsub_keysub_vals r z"_repair_schema..?s;$GW4@@@r"c0g|]}t|dSrr)rvs r r!z"_repair_schema..Ds%NNN1^A>>>NNNr"Trr typeNz$ref) isinstancelistdictr_SCHEMA_MAP_KEYS_SCHEMA_LIST_KEYS_SCHEMA_NODE_KEYSrpop_fill_missing_type)rrrepairedkeyvalues r rr)s$GGFFFFF dD ! !  "Hjjll"" U " " "z%'>'> "(- HSMM% % %*UD*A*A %NNNNNHSMM % % %%&& & .u E E E % "HSMM (z(7*;TBB VT""" h ' ''r"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))Nrrequiredrobjectrrarrayenumrbooleanintegernumberstring)r*r+rintfloat)rinferredsamples r r1r1_s ~~$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. r9)r)rTrr)r)r*r,rcopydeepcopyget)rDr2s r !sanitize_moonshot_tool_parametersrIzs 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.FfunctionrDT)r*r,appendrHrI)rJ sanitized any_changetoolfnparamsr2new_fns r sanitize_moonshot_toolsrUs  &(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/zkimi-kimiTmoonshotz/kimi)striplowerrsplit startswith)rVbaretails r is_moonshot_modelrds u ;;==   D ;;sA  r "D w46>>tTW__0G0G_t 5r")T)rrrrrr)rr5rr5)rDrrr5)rJrKrrK)rVrWrr)__doc__ __future__rrFtypingrrr frozensetr-r.r/rr1rIrUrdr"r rjs .#""""" """""""""" 9XXXYYIHHHIIIcccdd3(3(3(3(3(l&&&&6,....6r"