uPj&*UdZddlmZddlZddlmZmZddlmZm Z ddl m Z m Z m Z mZmZeGddZeGd d ZeGd d Zej&d Zded<d,dZd-dZej&dej0ej&dej0ej&dej0gZded<ej&dej0ej&dej0ej&dej0gZded<d.dZd/dZd0dZd0dZd1dZd2dZ d3d Z!hd!Z"ded"<d#Z#ded$<d4d%Z$ d5d&Z% d6d'Z&d7d(Z'd8d)Z(d9d*Z)d:d+Z*y);a Incremental signal detection for the adaptive router. Each session maintains a SessionState. On every turn, we call apply_turn(state, turn) which mutates the state in place and returns a SignalDelta listing which signals fired on THIS turn. The router then queues the delta to be flushed to DB. Design constraint: O(1) work per turn. No re-scanning the full session history. We keep small bounded windows: last_user_content, last_assistant_content, and a bounded list of recent tool call signatures. ) annotationsN) dataclassfield)AnyFinal)LOOP_REPEAT_THRESHOLDMIN_TURNS_FOR_CLEAN_CREDITMISALIGNMENT_JACCARD_THRESHOLDSTAGNATION_JACCARD_NEAR_DUPTOOL_CALL_HISTORY_MAXc|eZdZUdZdZded<dZded<dZded<dZded<dZ ded<dZ ded <dZ ded <d d Z y ) SignalDeltazMWhich signals fired on a single turn. Counts are 0 or 1 (one delta per turn).rint misalignment stagnation disengagement satisfactionfailureloop exhaustionc t|j|j|j|j|j |j |jgSN)anyrrrrrrr)selfs t/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/router_strategy/adaptive_router/signals.py any_firedzSignalDelta.any_fired*sK!!""!!     N)returnbool) __name__ __module__ __qualname____doc__r__annotations__rrrrrrrrrrrsOWL#JM3L#GSD#MJ  rrc(eZdZUdZded<ded<ded<ded<dZded <dZded <dZded <dZded <dZ ded <dZ ded<dZ ded<dZ ded<dZ ded<eeZded<eeZded<dZded<dZded<dZded<dZded <y)! SessionStateuIn-memory rolling state for one session. Mirrors the LiteLLM_AdaptiveRouterSession DB row (Wave 0 schema). The flusher later persists this. We keep this as a plain dataclass — no DB coupling. str session_id router_name model_nameclassified_typerrmisalignment_countstagnation_countdisengagement_countsatisfaction_count failure_count loop_countexhaustion_countN str | Nonelast_user_contentlast_assistant_contentdefault_factory list[str]tool_call_historyzdict[str, str]pending_tool_calls turn_countlast_processed_turnFrclean_credit_awarded int | Noneterminal_status)r r!r"r#r$r-r.r/r0r1r2r3r5r6rlistr:dictr;r<r>r?rAr%rrr'r'8s OOc  M3Jc$(z()-J-#(#>y>).t)DDJ!!!&$&"&OZ&rr'cteZdZUdZdZded<dZded<eeZ ded<eeZ ded <dZ d ed <y) TurnzCOne turn of input. Caller assembles this from the request/response.Nr4 user_contentassistant_contentr7list[dict[str, Any]] tool_calls tool_resultsr@response_status) r r!r"r#rFr$rGrrBrIrJrKr%rrrErEXsDM#L*#$(z(',T'BJ$B).t)DL&D"&OZ&rrEz [A-Za-z0-9]+r _TOKEN_REc|s tStj|Dchc]}|jc}Scc}wr)setrLfindalllower)textts r_tokensrShs1 u (006 7!AGGI 77 7s>cF||z}|syt||zt|z S)N)len)abunions r_jaccardrZns(q5E  q1u:E ""rz>\b(forget it|never mind|give up|talk to (?:a )?human|cancel)\bz/\b(this (?:isn'?t|is not) working|stop|abort)\bz#\bi'?ll do it (?:myself|manually)\b_DISENGAGEMENT_PATTERNSz?\b(that worked|that did it|works now|fixed it|solved it|nice)\bz4\b(thanks|thank you|thx|appreciated|appreciate it)\bz%\b(perfect|great|excellent|exactly)\b_SATISFACTION_PATTERNScp|r|sytt|t|}d|cxkxr tkScS)uFires when consecutive user messages share *some* topic (jaccard > 0) but are sufficiently different (jaccard < threshold) — i.e. user is rephrasing, not changing topic, not repeating.FrU)rZrSr ) prev_user curr_userjs r_detect_misalignmentras9 I *GI,>?A  33 33 33rcZ|r|sytt|t|}|tk\S)z>Fires when consecutive assistant messages are near-duplicates.F)rZrSr ) prev_asst curr_asstr`s r_detect_stagnationres, I *GI,>?A + ++rc:sytfdtDS)NFc3@K|]}|jywrsearch.0pr_s r z(_detect_disengagement..sDqqxx "D)rr[r_s`r_detect_disengagementrps  D,CD DDrc:sytfdtDS)NFc3@K|]}|jywrrhrjs rrmz'_detect_satisfaction..sCqqxx "Crn)rr\ros`r_detect_satisfactionrss  C,BC CCrc8|D]}|jdsyy)uAny tool result explicitly flagged as an error. We do NOT treat empty content as failure — many tools legitimately return empty output (zero-result searches, silent bash commands, void writes) and penalizing the model for those would corrupt the bandit posterior. is_errorTF)get)rJrs r_detect_failurerxs' 55  rcj|jdxs"|jdijdd}|jd"|jdijddttr1djfdt j D|ddS) zAStable signature for loop detection: name + sorted JSON-ish args.namefunction arguments,c32K|]}|d|yw)=Nr%)rkk call_argss rrmz_signature..s Uqs!IaL>2Us())rv isinstancerCjoinsortedkeys)callrzrs @r _signaturers((6"Ndhhz2&>&B&B62&ND%IHHZ,00bA )T"HHUF9>>CS= LOOP_REPEAT_THRESHOLD-1 times in recent history (so this call would be the Nth).FT)rcountr)history new_callsrsig recent_counts r _detect_looprsG }}S) 014 4  r>_EXHAUSTION_STATUSES)zcontext lengthzcontext windowz token limitz rate limitztoo many requeststimeout_EXHAUSTION_KEYWORDSc| |tvry|D]E}t|jddjt fdt DsEyy)NTcontentr|c3&K|]}|v ywrr%)rkkwrs rrmz%_detect_exhaustion..s<rW}.@++@c34K|]}|jywr)rrs rrmz&merge_signal_deltas...sB%%--Brc34K|]}|jywr)rrs rrmz&merge_signal_deltas..rrc34K|]}|jywr)rrs rrmz&merge_signal_deltas..s6eEMM6rc34K|]}|jywr)rrs rrmz&merge_signal_deltas..s00rc34K|]}|jywr)rrs rrmz&merge_signal_deltas..rr)rrrrrrr)rsum)deltass rmerge_signal_deltasrsq @@@)rturnrs radvance_session_staters "&"3"3 '+'='=$9 &&z$'789 5 " "#&;;"'"9"9;P:P:Q"R ' $ 4 4  % 0 0Erct|j|j|j|j xr|j dzt k\}t|j|j|j|j|j|j}t||}t|||jrd|_t!|||S)z Detect signals on this turn, mutate state, return the delta. O(1) per turn (no full-history rescan). Only inspects last_*, recent tool history (which is bounded at TOOL_CALL_HISTORY_MAX), and the new turn payload. r)rT)rr5rFrJr?r<r rr6rGr:rIrKrrrr)rrfeedback_deltaresponse_deltars r apply_turnr+s1    % : ::qu?O?ORS?SWq?q N 4 $$      N'Eue$ %)"%& Lr)rQr4rset[str])rWrrXrrfloat)r^r4r_r4rr)rcr4rdr4rr)r_r4rr)rJrHrr)rzdict[str, Any]rr()rr9rrHrr)rr@rJrHrr) rr4rr4rJrHrrrr)rr4rr4r:r9rIrHrJrHrKr@rr)rrrr)rr'rrrNone)rr'rrErr)rr'rrErr)+r# __future__rre dataclassesrrtypingrr.litellm.router_strategy.adaptive_router.configrr r r r rr'rEcompilerLr$rSrZ IGNORECASEr[r\rarerprsrxrrrrrrrrrrrr%rrrs  # (     2 '' '> '' '2::o. 5.8 #BJJPRTR_R_`BJJA2==QBJJ5r}}E"BJJJ BJJF VBJJ7G !4,E D " 8e7e % $ '     *)!%  '   ( /1*r