@=jdZddlZddlZddlZddlmZdaejZdZ ddddd e d e d e d e d e f dZ ded e fdZde dzde dzded efdZddd e de dzde dzdede de d ee e dzffdZdS)uRetry utilities — jittered backoff for decorrelated retries. Replaces fixed exponential backoff with jittered delays to prevent thundering-herd retry spikes when multiple sessions hit the same rate-limited provider concurrently. N)Any)g>@gN@gV@^@g@rg? base_delay max_delay jitter_ratioattemptrrrreturncnt5tdz at}dddn #1swxYwYtd|dz }|dks|dkr|}nt|d|zz|}t j|dzz dz}t j|}|d||z} || zS)aCCompute a jittered exponential backoff delay. Args: attempt: 1-based retry attempt number. base_delay: Base delay in seconds for attempt 1. max_delay: Maximum delay cap in seconds. jitter_ratio: Fraction of computed delay to use as random jitter range. 0.5 means jitter is uniform in [0, 0.5 * delay]. Returns: Delay in seconds: min(base * 2^(attempt-1), max_delay) + jitter. The jitter decorrelates concurrent retries so multiple sessions hitting the same provider don't all retry at the same instant. Nr?lyn<l) _jitter_lock_jitter_countermaxmintimetime_nsrandomRandomuniform) r rrrtickexponentdelayseedrngjitters 7/home/rurouni/.hermes/hermes-agent/agent/retry_utils.pyjittered_backoffrs. 11gk""H2~~qJ!x-0)<< LNNdZ/ 0J >D -  C [[L50 1 1F 6>s &**errorc|t|ddt|ddt|ddg}dd|DS)zCBest-effort flattened provider error text for retry classification.messageNbodyresponse c38K|]}|t|VdS)N)str).0parts r z_error_text..Ms-DD$43CCII3C3C3C3CDD)getattrjoinlower)r partss r _error_textr0Esj y$''vt$$z4((  E 88DD%DDD D D J J L LLr+base_urlmodelc|pd}|pd}t|dd}t|}|dkod|vo d|vod|vpd|vS) aLReturn True for Z.AI Coding Plan transient overload 429s. The coding-plan endpoint reports overload as HTTP 429 with body code 1305 and message "The service may be temporarily overloaded...". Treat only that narrow shape specially so ordinary quota/billing 429s still fail fast through the existing classifier. status_codeNizapi.z.ai/api/coding/paas/v4zglm-5.21305ztemporarily overloaded)r.r,r0)r1r2r base model_namestatustexts ris_zai_coding_overload_errorr;Ps N ! ! # #D+2$$&&J UM4 0 0F u  D#  A )T 1 A  # At^ ?74? r+)short_attempts default_waitr=ct|||s|dfS||kr|dfSt||z dz ttdz }t|}t d||ddfS)u7Provider-aware rate-limit backoff. For most providers this returns ``default_wait`` unchanged. For Z.AI Coding Plan GLM-5.2 overloads, keep the first ``short_attempts`` retries on the normal short exponential schedule, then switch to progressively longer waits (30s → 60s → 90s → 120s, capped) plus light jitter. ``attempt`` is 1-based, matching the retry loop's logged attempt number. Returns ``(wait_seconds, reason_label)`` where ``reason_label`` is suitable for status/log decoration when a provider-specific policy fired. )r1r2r Nzai_coding_overload_shortr g?rzai_coding_overload_long)r;rlen!_ZAI_CODING_OVERLOAD_LONG_BACKOFFr)r r1r2r r>r=idxrs radaptive_rate_limit_backoffrEds( (e T T T"T!!.  888 g&*C0Q,R,RUV,V W WC237J A* Y\ ] ] ]_y yyr+)__doc__r threadingrtypingrrLockrrCintfloatrr'r0boolr;tuplerEr+rrOs   y~ %>!  &&& && &  &  &&&&RMsMsMMMMcDjtTW\`6zzz zDjz : z  z  zz 5#* zzzzzzr+