Rmj1dZddlmZddlmZmZmZmZddlm Z m Z m Z m Z ddl mZmZmZddlmZddlmZddlmZdd lmZGd d e ZGd d edZGdde ZGdde ZdJdZdKdZedGddZedGdd Z edGd!d"Z!edGd#d$Z"edGd%d&Z#edGd'd(Z$edGd)d*Z%edGd+d,Z&edGd-d.Z'd/Z(edGd0d1Z)dLd4Z*d5d5d5d6dMdDZdNdHZgdIZ+d5S)OzAdaptive plugin configuration helpers. Adaptive is configured as a single flat top-level plugin component. Hosted plugins remain separate top-level components managed through ``nemo_relay.plugin``. ) annotations) dataclassfieldfields is_dataclass)LiteralProtocol TypedDictcast)Json JsonObjectUnsupportedBehavior)AdaptiveRuntime)build_cache_telemetry_event)set_latency_sensitivity)validate_adaptive_configc.eZdZUded<ded<ded<dS)_ConfigDiagnosticRequiredzLiteral['warning', 'error']levelstrcodemessageN)__name__ __module__ __qualname____annotations__[/home/thesage/.hermes/hermes-agent/venv/lib/python3.11/site-packages/nemo_relay/adaptive.pyrrs.&&&& IIILLLLLrrc(eZdZUdZded<ded<dS)ConfigDiagnosticz#One adaptive validation diagnostic.r componentrNrrr__doc__rrrrr!r!s(--NNNJJJJJrr!F)totalceZdZUdZded<dS) ConfigReportz-Validation report for adaptive configuration.zlist[ConfigDiagnostic] diagnosticsNr#rrrr'r'#s$77''''''rr'ceZdZddZdS)_SupportsToDictreturnr cdSNrselfs rto_dictz_SupportsToDict.to_dict*srNr+r )rrrr0rrrr*r*)s((((((rr*valueobjectr+r ctdr'ttSt r1t t sfdtDSt tr dDSt trd DSttS)Nr0chi|].}t|jx|jt/Sr-)getattrname _normalize).0 field_info field_valuer2s r z_normalize..1sF   &ujo>>> K OZ 44KKKrc,g|]}t|Sr)r8)r9items r z_normalize..7s 333T 4  333rc^i|]*\}}|tt|t|+Sr-)r rr8)r9keyvals rr<z_normalize..9s2aaaHCQTQ`S# 3Q`Q`Q`r) hasattrr r*r0r isinstancetyperlistdictitemsr )r2r;s`@rr8r8-sui  6OU++33555E :eT#:#:      $Umm    %433U3333%baa aaaa e  rr cFttt|Sr-)r r r8)r2s r_normalize_objectrJ=s  Ju-- . ..rT)slotscFeZdZUdZdZded<dZded<dZded<d d Zd S) ConfigPolicyaPolicy for unsupported adaptive configuration. Args: unknown_component: How to handle unknown component kinds. unknown_field: How to handle unknown adaptive config fields. unsupported_value: How to handle known fields with unsupported values. warnrunknown_component unknown_fielderrorunsupported_valuer+r c,|j|j|jdS)z9Serialize this policy to the canonical JSON object shape.rOrPrRrTr.s rr0zConfigPolicy.to_dictOs%"&!7!/!%!7   rNr1) rrrr$rOrrPrRr0rrrrMrMAsk.43333)/M////-44444      rrMczeZdZUdZded<eeZded<edd Z eddd Z ddZ dS) BackendSpeczAdaptive state backend selection. Args: kind: Backend kind string such as ``"in_memory"`` or ``"redis"``. config: Backend-specific JSON object. rkinddefault_factoryr configr+ 'BackendSpec'c"tdS)z*Return an in-memory adaptive backend spec. in_memory)rWrVrrrr]zBackendSpec.in_memoryds ,,,,r nemo_relay:url key_prefixc*td||dS)z%Return a Redis adaptive backend spec.redisr`rarWrZr^rds rrczBackendSpec.redisis:0V0VWWWWrc:|jt|jdS)z?Serialize this backend spec to the canonical JSON object shape.re)rWrJrZr.s rr0zBackendSpec.to_dictns ->t{-K-KLLLrN)r+r[)r_)r`rrarr+r[r1) rrrr$rrrGrZ staticmethodr]rcr0rrrrVrVXsIIIt444F4444---\-XXXX\XMMMMMMrrVc&eZdZUdZded<ddZdS) StateConfigzAdaptive state configuration. Args: backend: State backend selection for adaptive features that persist or learn over time. rVbackendr+r c.dt|jiS)z?Serialize this state config to the canonical JSON object shape.rj)rJrjr.s rr0zStateConfig.to_dict~s,T\::;;rNr1)rrrr$rr0rrrririssB<<<<<>>F>>>>       rradaptivec4eZdZUdZded<dZded<d d Zd S) ComponentSpeczTop-level adaptive component wrapper. Args: config: ``AdaptiveConfig`` or an equivalent JSON object. enabled: Whether the adaptive component should be activated. Behavior: The component kind is always ``"adaptive"``. AdaptiveConfig | JsonObjectrZTrwenabledr+r cFt|jt|jdS)z7Serialize this component to the canonical plugin shape.)rWrrZ)ADAPTIVE_PLUGIN_KINDrrJrZr.s rr0zComponentSpec.to_dict:s()|' 44   rNr1)rrrr$rrr0rrrrr+sS('''G      rrrZrc`tttt|S)zDValidate an adaptive config document without constructing a runtime.)r r'_validate_adaptive_configrJ)rZs rvalidate_configrCs$  78I&8Q8QRR S SSrN)usage request_facts timestamprr request_idrJsonObject | Nonerrtemplate_version toolset_hash model_family tenant_scoperrnc `ttdzt||||||||||  S)zHBuild one canonical cache telemetry event from usage plus request facts.N rrrrrrrrrr)r r _build_cache_telemetry_eventrs rrrHsN T$!'-%%%   rrruNonec$t|dS)aSet a request-local latency-sensitivity hint. Args: level: Positive integer sensitivity value for the current execution context. Returns: `None`. Behavior: This is an execution-time hint for the current request/scope context, not persistent adaptive configuration. The native adaptive layer stores this as a positive integer. N)_set_latency_sensitivity)rs rrrgsU#####r)rrrrsrrVr!rMr'rrirmr}rr)r2r3r+r )r2r3r+r )rZrr+r')rrrrrrrrrrrrrrrrrrrrnr+r)rrur+r),r$ __future__r dataclassesrrrrtypingrr r r nemo_relayr r rnemo_relay._nativerrrrrrrrr!r'r*r8rJrMrVrirmrsr}rrrrrr__all__rrrrs #""""">>>>>>>>>>>>555555555555<<<<<<<<<<AAAAAAZZZZZZRRRRRRTTTTTT  0(((((9((( )))))h)))     ////         , MMMMMMMM4  < < < < < < < <         *         6 QQQQQQQQ$         2         6 & & & & & & & & R"         .TTTT $'+ !>$$$$$   r