o 2j@sJUdZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z e e Zd Zd ZGd d d ed dZGdddeZgidZeed<daedBed<deefddZdefddZdeeefdefddZdefddZdefddZd ed!edBdedBfd"d#Zd ed$eddfd%d&Z dedBfd'd(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@s"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__rr^/var/www/html/whisper/venv/lib/python3.10/site-packages/huggingface_hub/utils/_detect_agent.pyr 4s r F)totalc@s.eZdZUdZeeed<eeefed<dS)RegistryzJThe agent harness registry, as served by `{ENDPOINT}/api/agent-harnesses`.standardEnvVars harnessesN) r r rrlistrrrr rrrrr:s  r)rr_EMPTY_REGISTRY _registryreturnc Cst}|dp g}|dpi}|D]+\}}|pid}|r*t|r*|S|D]}tj|d|kr?|Sq,qdd|D}|D]}tj|d}rf||vrc|SdSqLdS) 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 cSsh|]}|qSr)lower).0krrr ^szdetect_agent..unknownN) _get_registrygetitems_env_vars_matchosenvironstripkeysr) registry standard_varsr harness_idinfoenv_varsvarlowercased_harnessesvaluerrr detect_agentIs(  r3cCs tduS)zDReturn `True` if the process is being invoked by an AI coding agent.N)r3rrrris_agenths r4r/cCsD|D]\}}tj|}|sq|dkrdS||krdSqdS)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$)r/r0patternr2rrrr&ms r&cCs@tdurztaWtStytjdddtaYtSwtS)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 Exceptionloggerdebugrrrrrr#s r#cCsNtj}t|td}r |St}durt|||St|dd}r%|StS)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)pathcachedfetchedstalerrrr9s r9rCr=cCsz;tj|s WdS|durttj||krWdSt|dd}t|WdWS1s4wYWdStyLt j dddYdSw)zBReturn the cached registry, or `None` if missing/stale/unreadable.Nutf-8encodingz/Could not read cached agent harnesses registry.Tr7) r'rCexiststimegetmtimeopenjsonloadr:r;r<)rCr=frrrr?s  ( r?r+cCs|z,t|jjdddt|ddd}t||WdWdS1s%wYWdSty=tjdddYdSw)NT)parentsexist_okwrGrHz)Could not cache agent harnesses registry.r7) rparentmkdirrMrNdumpr:r;r<)rCr+rPrrrrBs& rBcCsdtjrdSzddlm}|jtjdtd}||WSt y1t j dddYdSw) 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.Tr7) rHF_HUB_OFFLINE_httprWr$ENDPOINT_REGISTRY_FETCH_TIMEOUTraise_for_statusrNr:r;r<)rWresponserrrrAs    rA)"rrNr'rKpathlibrtypingrrrrr get_loggerr r;r@r\r rrrrrr3boolr4rr&r#r9intr?rBrArrrrs.