i@+dZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z ejeZdZdZd)d Zd*dZddddd+dZd,dZd-dZd.dZdZdddd/d Zd0d"Zd1d%Zd2d(ZdS)3a)Cross-session rate limit guard for Nous Portal. Writes rate limit state to a shared file so all sessions (CLI, gateway, cron, auxiliary) can check whether Nous Portal is currently rate-limited before making requests. Prevents retry amplification when RPH is tapped. Each 429 from Nous triggers up to 9 API calls per conversation turn (3 SDK retries x 3 Hermes retries), and every one of those calls counts against RPH. By recording the rate limit state on first 429 and checking it before subsequent attempts, we eliminate the amplification effect. ) annotationsN)AnyMappingOptional)atomic_replace rate_limitsz nous.jsonreturnstrc ddlm}|}nM#t$r@tjtjdd}YnwxYwtj|ttS)z2Return the path to the Nous rate limit state file.r)get_hermes_home~z.hermes) hermes_constantsr ImportErrorospathjoin expanduser _STATE_SUBDIR_STATE_FILENAME)r bases :/home/piyush/.hermes/hermes-agent/agent/nous_rate_guard.py _state_pathrs@444444   @@@w||BG..s33Y??@ 7<<m_ = ==sAAAheadersOptional[Mapping[str, str]]Optional[float]c|sdSd|D}dD]K}||}|2 t|}|dkr|cS4#ttf$rYGwxYwLdS)ufExtract the best available reset-time estimate from response headers. Priority: 1. x-ratelimit-reset-requests-1h (hourly RPH window — most useful) 2. x-ratelimit-reset-requests (per-minute RPM window) 3. retry-after (generic HTTP header) Returns seconds-from-now, or None if no usable header found. Nc>i|]\}}||Slower.0kvs r z(_parse_reset_seconds..4&8881qwwyy!888)zx-ratelimit-reset-requests-1hzx-ratelimit-reset-requestsz retry-afterr)itemsgetfloat TypeError ValueError)rloweredkeyrawvals r_parse_reset_secondsr1's t88 888G   kk# ? Cjj77JJJz*      4sAA-,A-gr@)r error_contextdefault_cooldownr2Optional[dict[str, Any]]r3r*Nonectj}d}t|}|||z}|[t|trF|d}t|t t fr||krt |}|||z}t} tj |}tj |d||||z d} tj |d\} } tj| d5} tj| | dddn #1swxYwYt#| |n5#t$$r( tj| n#t($rYnwxYwwxYwt*d ||z |dS#t$$r&} t*d | Yd} ~ dSd} ~ wwxYw) aRecord that Nous Portal is rate-limited. Parses the reset time from response headers or error context. Falls back to ``default_cooldown`` (5 minutes) if no reset info is available. Writes to a shared file that all sessions can read. Args: headers: HTTP response headers from the 429 error. error_context: Structured error context from _extract_api_error_context(). default_cooldown: Fallback cooldown in seconds when no header data. Nreset_atT)exist_ok)r7 recorded_at reset_secondsz.tmp)dirsuffixwz3Nous rate limit recorded: resets in %.0fs (at %.0f)z)Failed to write Nous rate limit state: %s)timer1 isinstancedictr)intr*rrrdirnamemakedirstempfilemkstempfdopenjsondumpr ExceptionunlinkOSErrorloggerinfodebug)rr2r3nowr7header_seconds ctx_resetr state_dirstatefdtmp_pathfexcs rrecord_nous_rate_limitrXGsf" )++CH*'22N!'J}d;;!%%j11 i#u . . (9s??Y''H)) ==DGGOOD))  I----!%^   'IfEEE H 2s## $q %### $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ 8T * * * *     (####         A sNH      GGG @#FFFFFFFFFGsyAF 7E D." E .D22E 5D26E F E<E*)E<* E74E<6E77E<<"F G*G  Gct} t|5}tj|}dddn #1swxYwY|dd}|t jz }|dkr|S t j|n#t$rYnwxYwdS#ttj ttf$rYdSwxYw)zCheck if Nous Portal is currently rate-limited. Returns: Seconds remaining until reset, or None if not rate-limited. Nr7r) ropenrGloadr)r>rrJrKFileNotFoundErrorJSONDecodeErrorKeyErrorr+)rrVrSr7 remainings rnous_rate_limit_remainingr`s ==D $ZZ !1IaLLE ! ! ! ! ! ! ! ! ! ! ! ! ! ! !99Z++ty{{* q==   IdOOOO    D t t3Xy IttsWB'A B'AB'A6B'BB' B"B'!B""B''"C  C c tjtdS#t$rYdSt$r&}t d|Yd}~dSd}~wwxYw)zCClear the rate limit state (e.g., after a successful Nous request).z)Failed to clear Nous rate limit state: %sN)rrJrr\rKrLrN)rWs rclear_nous_rate_limitrbsG +--          GGG @#FFFFFFFFFGs $ A  A AA secondsctdt|}|dkr|dS|dkr"t|d\}}|r|d|dn|dSt|d\}}|dz}|r|d|dn|dS) z6Format seconds remaining into human-readable duration.r<sizm mzh h)maxrAdivmod)rcrfrgsecrh remainders rformat_remainingrms As7||A2vvwww4xx23!$1!~~s~~~~Q'''1!T??LAyRA )a<<1<<<= 60s) is proof of (a). 2. The last-known-good rate-limit state captured by ``_capture_rate_limits()`` on the previous successful response. If any bucket there was already near-exhausted with a substantial reset window, the current 429 is almost certainly (a) continuing from that condition. If neither signal fires, we treat the 429 as (b): fail the single request, let the retry loop or model-switch proceed, and do NOT write the cross-session breaker file. Returns True when the evidence points at (a). TNF)_parse_buckets_from_headers_has_exhausted_bucket_has_exhausted_bucket_in_object)rrnrSs ris_genuine_nous_rate_limitrusHT ( 0 0EU##t #(GHX(Y(Y#t 5r'0dict[str, tuple[Optional[int], Optional[float]]]c8|siSd|D}td|DsiSdd}dd }i}d D]O}||d |}||d |}||||f||<P|S)zExtract (remaining, reset_seconds) per bucket from x-ratelimit-* headers. Returns empty dict when no rate-limit headers are present. c>i|]\}}||Srrr!s rr%z/_parse_buckets_from_headers..r&r'c3@K|]}|dVdS)z x-ratelimit-N) startswith)r"r#s r z._parse_buckets_from_headers..s.==q||N++======r'r/ Optional[str]r Optional[int]ct|dS tt|S#ttf$rYdSwxYwN)rAr*r+r,r/s r _maybe_intz/_parse_buckets_from_headers.._maybe_intsJ ;4 uSzz?? ":&   44 s "77rcZ|dS t|S#ttf$rYdSwxYwr)r*r+r,rs r _maybe_floatz1_parse_buckets_from_headers.._maybe_float sD ;4 :: :&   44 s **)requestsz requests-1htokensz tokens-1hzx-ratelimit-remaining-zx-ratelimit-reset-N)r/r|r r})r/r|r r)r(anyr))rr-rrresulttagr_resets rrrrrs  88 888G ==W=== = = @BFA--Jw{{+IC+I+IJJKK  W[[)Cc)C)CDDEE  E$5$e,F3K Mr'buckets3Mapping[str, tuple[Optional[int], Optional[float]]]cn|D]\}}||dkr||tkrdS dS)zMReturn True when any bucket has remaining == 0 AND a meaningful reset window.NrTF)values_MIN_RESET_FOR_BREAKER_SECONDS)rr_rs rrsrssX$NN,, 5   A  =  2 2 244 3 5r'rSrcdD]~}t||d}|t|ddpd}t|ddpd}t|dd}|t|ddpd}|dkri|dkrp|tkrd Sd S) a1Check a RateLimitState-like object for an exhausted bucket. Accepts the dataclass from ``agent.rate_limit_tracker`` (buckets exposed as attributes ``requests_min``, ``requests_hour``, ``tokens_min``, ``tokens_hour``) and falls back gracefully for any object missing those attributes. ) requests_min requests_hour tokens_min tokens_hourNlimitrr_remaining_seconds_nowr:gTF)getattrr)rSattrbucketrr_rs rrtrt,sOd++ > ++0qFK338q  7>> =FOS99@SE A::  q==  2 2 244 3 5r')r r )rrr r)rrr2r4r3r*r r5)r r)r r5)rcr*r r )rrrnror rp)rrr rv)rrr rp)rSrr rp)__doc__ __future__rrGloggingrrDr>typingrrrutilsr getLogger__name__rLrrrr1rXr`rbrmrrurrrsrtrr'rrs  #"""""   ))))))))))  8 $ $ >>>>D,0.2# AGAGAGAGAGAGH0GGGG * * * * "& ,0&*444444n$$$$N    r'