^jpUdZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z e je Zd Zd ZGd d ed ZGddeZgidZeed<daedzed<deefdZdefdZdeeefdefdZdefdZdefdZdededzdedzfdZdededdfdZ dedzfd Z!y)!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<y) 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__n/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/huggingface_hub/utils/_detect_agent.pyr r 4sl #s(^rr F)totalc6eZdZUdZeeed<eeefed<y)RegistryzJThe agent harness registry, as served by `{ENDPOINT}/api/agent-harnesses`.standardEnvVars harnessesN) rrrrlistrrrr rrrrr:sT#YC$%%rr)rr_EMPTY_REGISTRY _registryreturncbt}|jdxsg}|jdxsi}|jD]i\}}|xsijd}|rt|r|cS|D]9}tj j|dj |k(s5|ccSk|jDchc]}|j}}|D]J}tj j|dj jx} sB| |vr| cSyycc}w)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 unknownN) _get_registrygetitems_env_vars_matchosenvironstripkeyslower) registry standard_varsr harness_idinfoenv_varsvarklowercased_harnessesvalues r detect_agentr6Is HLL!239rM [)/RI%OO-" DJB##I. 1   "Czz~~c2&,,.*<!! " "09~~/?@!AGGI@@JJNN3+11399; ;5 ;,,   AsD,ctduS)zDReturn `True` if the process is being invoked by an AI coding agent.N)r6rrris_agentr8hs > %%rr1c|jD]5\}}tjj|}|s(|dk(ry||k(s5yy)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%)r1r2patternr5s rr'r'msP!( W s#  c> G  rct tatStS#t$r%tj ddt aYtSwxYw)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. z+Could not resolve agent harnesses registry.Texc_info)r_load_registry Exceptionloggerdebugrrrrr$r$sR (&(I 9 ( LLFQUL V'I  (s &A  A ctj}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 rr?r?sg  ) )D't5JKKvK #$$1tW-&dD99u9 rrJrDct tjj|sy|8tjtjj |z |k\ryt |d5}t j|cdddS#1swYyxYw#t$rtjddYywxYw)zBReturn the cached registry, or `None` if missing/stale/unreadable.Nutf-8encodingz/Could not read cached agent harnesses registry.Tr=) r(rJexiststimegetmtimeopenjsonloadr@rArB)rJrDfs rrFrFs ww~~d#  DIIK"''2B2B42H$HW#T $ ) Q99Q<    FQU Vs:B9B B)B> BB BB B76B7r-c t|jjddt|dd5}t j ||dddy#1swYyxYw#t $rtjddYywxYw)NT)parentsexist_okwrOrPz)Could not cache agent harnesses registry.r=) rparentmkdirrUrVdumpr@rArB)rJr-rXs rrIrIswQ T t< $g . #! IIh " # # # Q @4 PQs.4A"A A"AA"A"" BBctjry ddlm}|j tj dt }|j|jS#t$rtjddYywxYw) 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_httprar%ENDPOINT_REGISTRY_FETCH_TIMEOUTraise_for_statusrVr@rArB)raresponses rrHrHs &=$$!!""6 7+%  !!#}}  MX\ ]sAA&& B B )"rrVr(rSpathlibrtypingrrr"rr get_loggerrrArGrfr rrrrrr6boolr8rr'r$r?intrFrIrHrrrrns& &   H %")5 &y&13DD" 8d?!hsm>&$& d38n$ x  .  cDj X_ QQQTQDr