gjUdZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z e j e Zd Zd ZGd d ed ZGddeZgidZeed<daedzed<deefdZdefdZdeeefdefdZdefdZdefdZdededzdedzfdZdededdfdZ dedzfd Z!dS)!uDetect whether the process is being invoked by an AI coding agent. Detection is based on environment variables that AI agents set in their shell sessions. `AI_AGENT` and `AGENT` are treated as a universal standard (any tool can set its harness id there); the remaining checks are tool-specific and ordered by priority (first match wins). The list of known harnesses is maintained on the Hub and exposed at `{ENDPOINT}/api/agent-harnesses`. We fetch it at most once a day and cache it locally so the list can be updated without requiring a new client release. Detection is entirely best-effort: there is no hardcoded list of harnesses. When the registry cannot be fetched (and no cached copy is available), detection simply reports "no agent". Any error while fetching/reading the registry is swallowed — detection must never make a process fail. More details: https://huggingface.co/docs/hub/agents-overview#register-your-agent-harness N)Path)Optional TypedDict) constants)loggingiQc.eZdZUdZeeefed<dS) HarnessInfozbA single harness entry. `envVars` maps an env var name to a match pattern (see `_env_vars_match`).envVarsN)__name__ __module__ __qualname____doc__dictstr__annotations__g/var/www/app.qaans.com/backend/venv/lib/python3.11/site-packages/huggingface_hub/utils/_detect_agent.pyr r 4s,ll #s(^rr F)totalcDeZdZUdZeeed<eeefed<dS)RegistryzJThe agent harness registry, as served by `{ENDPOINT}/api/agent-harnesses`.standardEnvVars harnessesN) rrrrlistrrrr rrrrr:s>TT#YC$%%%%%%rr)rr_EMPTY_REGISTRY _registryreturnctt}|dpg}|dpi}|D]r\}}|pid}|rt|r|cS|D]>}tj|d|kr|ccS?sd|D}|D]S}tj|dx}r ||vr|cSdSTdS)a[Return the id of the detected AI agent harness or `None`. Harnesses are checked in registry order; for each one we match its env var pattern(s) and, failing that, the standard `AI_AGENT` / `AGENT` vars against the harness id. The first match wins. When a standard var is set to an unrecognized value, `"unknown"` is returned. rrr c6h|]}|Sr)lower).0ks r zdetect_agent..^s @@@!AGGII@@@runknownN) _get_registrygetitems_env_vars_matchosenvironstripkeysr$) registry standard_varsr harness_idinfoenv_varsvarlowercased_harnessesvalues r detect_agentr9IsqHLL!2339rM [))/RI%OO--"" DJB##I..  11      " "Cz~~c2&&,,..*<<!!!!!!= " A@y~~/?/?@@@JNN3++113399;; ;5 ,,, 99  4rc"tduS)zDReturn `True` if the process is being invoked by an AI coding agent.N)r9rrris_agentr;hs >> %%rr5c|D]9\}}tj|}|s'|dkrdS||krdS:dS)zReturn `True` if any `(var, pattern)` from the harness matches the environment. Supported patterns: - `"*"`: the variable is set to any non-empty value - `""`: the variable equals this exact value *TF)r+r-r.r*)r5r6patternr8s rr,r,msk!(( W s##   c>>44 G  44  5rctC tan3#t$r&tddt aYnwxYwtS)zReturn the harness registry, loading (and caching in-process) on first call. Best-effort: any unexpected error degrades to an empty registry so detection never raises. Nz+Could not resolve agent harnesses registry.Texc_info)r_load_registry Exceptionloggerdebugrrrrr)r)s` (&((II ( ( ( LLFQUL V V V'III ( s-AActj}t|tx}r|St x}t |||St|dx}r|St S)zResolve the registry from the local cache or the Hub. No hardcoded list: if the Hub is unreachable and no cached copy exists, an empty registry is returned (i.e. no agent is detected). )max_ageN)rAGENT_HARNESSES_PATH_read_cached_registry_REGISTRY_TTL_SECONDS_fetch_registry_write_cached_registryr)pathcachedfetchedstales rrBrBs}  )D't5JKKKKv #$$$1tW---&dD9999u rrMrGc tj|sdS|9tjtj|z |krdSt |d5}t j|cdddS#1swxYwYdS#t$r t ddYdSwxYw)zBReturn the cached registry, or `None` if missing/stale/unreadable.Nutf-8encodingz/Could not read cached agent harnesses registry.Tr@) r-rMexiststimegetmtimeopenjsonloadrCrDrE)rMrGfs rrIrIs  w~~d## 4  DIKK"'2B2B42H2H$HW#T#T4 $ ) ) ) Q9Q<<                   FQU VVVtts@B9BB/B BBBBB&CCr1c4 t|jddt|dd5}t j||ddddS#1swxYwYdS#t $r tddYdSwxYw)NT)parentsexist_okwrRrSz)Could not cache agent harnesses registry.r@) rparentmkdirrXrYdumprCrDrE)rMr1r[s rrLrLsQ T t<<< $g . . . #! Ih " " " # # # # # # # # # # # # # # # # # # QQQ @4 PPPPPPQs4;A-A  A- A$$A-'A$(A--&BBc8tjrdS ddlm}|tjdt }||S#t$r t ddYdSwxYw) zMFetch the registry from the Hub. Returns `None` when offline or on any error.Nr) get_sessionz/api/agent-harnesses)timeoutz6Could not fetch agent harnesses registry from the Hub.Tr@) rHF_HUB_OFFLINE_httprdr*ENDPOINT_REGISTRY_FETCH_TIMEOUTraise_for_statusrYrCrDrE)rdresponses rrKrKst &&&&&&;==$$! 7 7 7+%   !!###}}  MX\ ]]]ttsAA//&BB)"rrYr-rVpathlibrtypingrrr"rr get_loggerrrDrJrir rrrrrr9boolr;rr,r)rBintrIrLrKrrrrqsR& &&&&&&&&  H % %")5 &&&&&y&&&13DDDDD" 8d?!!!hsm>&$&&&& d38n$ x     .  cDj X_    QQQTQQQQDr