uPj0UdZddlmZmZddlmZddlmZddlm Z m Z m Z ddl m Z mZdZe ed<e d Ze d Ze ed <ed d GddZehdZe eeed<ddddd Ze eeeeedfdffed<dededzfdZdededeedffdZdededeedffdZeiZe eeefed <dedeeeffd!Zd"eeefdeedffd#Zd$eeefdzdedzfd%Z ed&ed'DZ!e eeed(<dedzd)eede"fd*Z#d"eeefdzdedzfd+Z$ded)eededzfd,Z%y)-asNaming contract for strategy-router (auto-router) pseudo-models. A deployment whose ``litellm_params.model`` starts with ``auto_router/`` does not name a provider model; the string is the discriminator that selects which pre-routing strategy owns the deployment. This module is the single source of truth for classifying that string (``Router._is_*_router_deployment`` delegates here) and for checking that a client-supplied write leaves the deployment coherent, so management endpoints can reject corruption with a 400 instead of the router silently dropping the deployment at load time under ``ignore_invalid_deployments``. )MappingSequence) dataclass)MappingProxyType)FinalLiteral TypeAlias)COMPLEXITY_ROUTER_CONFIG_KEYSLLM_CLASSIFIER_TYPESz auto_router/AUTO_ROUTER_MODEL_PREFIX)semantic complexityadaptivequality)tierdefault classifier embeddingStrategyRouterDependencyRoleT)frozenslotsc&eZdZUdZeed<eed<y)StrategyRouterDependencyzCA model name a strategy router must be able to reach to do its job. model_nameroleN)__name__ __module__ __qualname____doc__str__annotations__rr/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/router_utils/auto_router_model_naming.pyrrsMO &&r#r> auto_router_configquality_router_configadaptive_router_configauto_router_config_pathcomplexity_router_configauto_router_default_modelauto_router_embedding_modelauto_router_max_input_charsquality_router_default_modelcomplexity_router_default_modelSTRATEGY_ROUTER_PARAM_FIELDS))r%r()r*)r+))r)r.))r'))r&r-._REQUIRED_FIELD_GROUPSmodelreturnNc|jtsy|ttd}|jdry|jdry|jdryy) aClassify a ``litellm_params.model`` string the way the Router does. Returns None for regular provider models. Mirrors Router registration exactly: reserved names are matched by prefix, everything else under ``auto_router/`` is a semantic router. Ncomplexity_routerradaptive_routerrquality_routerrr ) startswithr len)r1 remainders r$classify_strategy_router_modelr:Asb   4 5S!9:<=I/0-.,- r#valuercDt|tr|r t||fSdS)zLOne dependency from a scalar field, or none when it is absent or not a name.r") isinstancer rr;rs r$_namedr?Ts%7A%7MRW $UD 1 3_]__r#ct|tr t|St|trt fd|DSy)zIDependencies from a field holding either a single name or a pool of them.c3DK|]}t|D]}|ywN)r?).0entrydeprs r$ z_pool..^s$KUveT7JKSKSKs r")r=r r?rtupler>s `r$_poolrHYs8%eT""%"KeKKK r# _NO_CONFIGc2t|tr|StSrB)r=rrI)r;s r$_mappingrKesug.5>J>r#litellm_paramsc ftt|jdd}|y|dk(r7t|jddt|jddzS|d k(r3t t |jd jd d S|d k(rt |jd}t tjt |jd d t|jdxs|jddzSt |jd}t |jd}t tjt dt |jdjDt|jddz|jdtvrt|jddndz|jdr%t|jddzSdzS)aThe model names a strategy-router deployment must reach, in no particular order. A field is a dependency only under the condition the runtime itself reads it: the classifier model needs `classifier_type: llm`, and the complexity embedding model needs `semantic_keyword_matching`. Listing one the router never calls reds a working deployment. The two default-model spellings are not symmetric. A quality router falls back to its config's `default_model`, so both are read. A complexity router ignores that field and derives its default from the tiers instead (`fallback_tier`, then MEDIUM, then SIMPLE), overwriting the config value at init, so only the `litellm_params` spelling is a dependency here; the derived one is already covered as a tier. Returns empty for a regular deployment, and for any name this module cannot reach from the deployment dict alone: a semantic router's routes live in an `auto_router_config` JSON string or an `auto_router_config_path` file, so only its default and embedding models are enumerable here. Every field is read defensively, since a caller may hold a config the router itself would refuse, and a health check must not raise on one. r1r"r r*rr+rrr'available_modelsrrr&r- default_modelr)classifier_llm_configc3BK|]}t|dD]}|yw)rN)rH)rCrrEs r$rFz/strategy_router_dependencies..s'l$X]^bdjXklQT#l#lstiersr.classifier_typersemantic_keyword_matchingembedding_model) r:r getr?rHrKrGdictfromkeysvaluesr )rLkindrrrs r$strategy_router_dependenciesr\is*1^5G5GQS5T1UVD | zn(()DEyQTZ   < ={U    zXn001IJKOOPbcekll y!."4"45L"MN MMgkk"45v>"&&'EFf'++VeJf   !!3!34N!OPJ 0G!HIJ  l(:>>'+B"C"J"J"Ll l^''(IJIV W>>"348LLz~~g. =  >>"=>z~~&78+F      r#r)cddlm}ddlm}|y |j |}y#|$rb}|j d}dj d|jddDxsd }d |d |jd d dcYd}~Sd}~wwxYw)a3Reject a complexity config the router would refuse to build a deployment from. Parsed with the router's own ``ComplexityRouterConfig`` rather than a copy of its rules, so the boundary rejects exactly what the load would. Plugins are resolved from dotted paths only on the config.yaml path, so a written config reaches this function in the same shape the load hands to the same model. Judged on the config alone: a patch may write one without naming a model, and the stored model is encrypted at rest, so it cannot be classified here. r)ValidationError)ComplexityRouterConfigN.c32K|]}t|ywrB)r )rCparts r$rFz:validate_complexity_router_config_write..s"N3t9"Nslocr"r)z'complexity_router_config is invalid at z: msgz invalid valuezW. The router would drop this deployment at load time, so the write is rejected instead.)pydanticr^0litellm.router_strategy.complexity_router.configr_model_validateerrorsjoinrW)r)r^r__excfirstlocations r$'validate_complexity_router_config_writerns)W' " 1 12J K   zz|A(("N5"9M"NNlRl5hZr%))ESbBcAded d  s#B AB?B B c#.K|] }|D]}|ywrBr")rCgroupfields r$rFrFs'==@EE= =sr_COMPLEXITY_ROUTER_FIELDSpresent_fieldscNt|xsddk(xst|tzS)aWhether this deployment configures a complexity router, so is judged on its key set. Scoped rather than applied to every deployment because the setting names are only unambiguous in this context: ``embedding_model``, for one, is a legitimate flat param on an s3_vectors vector store. ``present_fields`` carries the same merged view ``validate_strategy_router_model_write`` is judged on, so a router named only by its default model is in scope, and a field added to the table above is covered here for free. rNr)r:boolrr)r1rss r$"carries_complexity_router_settingsrvs0 *%+2 6, F $22Kr#c|yttt|tz}|syddj |dS)aReject a complexity-router setting written beside ``complexity_router_config``. The router reads its settings only from ``litellm_params.complexity_router_config``, so a key one level too high configures nothing. It does not stay inert: the alias-marker forwarding carries every unrecognized ``litellm_params`` key onto the outbound request, where the provider rejects it as an unknown body field, and the deployment then fails every call with an error naming an internal config key. Caller scopes; this judges. Nz@litellm_params sets complexity_router_config settings directly: , z. The router reads these only from complexity_router_config, so there they configure nothing and are forwarded to the provider as unknown request params, which rejects the call. Move them under complexity_router_config.)rGsorted frozensetr ri)rL misplaceds r$+validate_complexity_router_config_placementr|sOVIn$=@]$]^_I  J499U^K_J`a4 4r#c t|}|;ttz}|r&d|dtddj |dtd Sy|t td}|j trd|dtd |d S|s d|d td Stfd t|D}|rd|d|ddj |dSy)aCheck that writing ``model`` leaves a deployment the router can load. ``present_fields`` is the set of strategy-router param fields that are non-None on the deployment after the write (stored fields merged with the incoming ones). Returns a human-readable violation, or None when coherent. A config's contents are ``validate_complexity_router_config_write``'s to judge, since a write may carry one without naming a model at all. Nzlitellm_params.model='z' does not start with 'z3' but the deployment carries auto-router settings (rxz.), so the router could not load it. Keep the 'zN' prefix; to change the name clients call, edit the public model_name instead.z' repeats the 'z0' prefix, so the router could not load it. Use 'zG'; to change the name clients call, edit the public model_name instead.z(' is missing the router name after the 'z ' prefix.c3jK|]*}tfd|Drdj|,yw)c3&K|]}|v ywrBr")rCfrss r$rFzAvalidate_strategy_router_model_write... sOshiPQUcPcOssz or N)anyri)rCrprss r$rFz7validate_strategy_router_model_write.. s-$COsmrOsLs Es33z' selects the z router, which requires z; z in litellm_params.) r:ryr/r rir8r7rGr0)r1rsr[ offendingr9missings ` r$$validate_strategy_router_model_writers,17D |!.3O"OP (/FG_F`a<<@IIi%?@66  S!9:<=I45$UG?;S:TU''0k2" " $UG+STlSmmv w (>t(DG$UG>$?Wyy!""5 7  r#)&rcollections.abcrr dataclassesrtypesrtypingrrr rfr r r r!StrategyRouterKindrrrzr/r r0rGr:objectr?rHrIrKr\rnrrrurvr|rr"r#r$rsl .!",, #1%0LM*12^*_i_ $d#''$'7@  7eIcN3   U.K [g&8%c3hQT@T:U&UVW #2Dt2K&`&` <`G_adGdAe` ;F^`cFc@d+;2*> E'#v+& '>?F?wsF{3?:CK(: #S ():zgcSYkFZ]aFafilpfp84==-l;=4530 cDj )TW. ]a V @TW[@[`cfj`j,''Ys^'X[^bXb'r#