i UdZddlZddlmZdZeedfed<dZeeeedfefdfed<d hd eDzZ d eefd Z d e fd Z dS)aDetect 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 it); the remaining checks are tool-specific and ordered by prevalence. Inspired by ``@vercel/detect-agent`` (https://github.com/vercel/vercel/tree/main/packages/detect-agent). N)Optional)AI_AGENTAGENT._STANDARD_AGENT_VARS)))ANTIGRAVITY_AGENT antigravity)) AUGMENT_AGENTz augment-cli)) CLINE_ACTIVEcline))CLAUDE_CODE_IS_COWORKcowork)) CLAUDECODE CLAUDE_CODEz claude-code)) CODEX_SANDBOXCODEX_CICODEX_THREAD_IDcodex))CURSOR_TRACE_IDcursor)) CURSOR_AGENTz cursor-cli)) GEMINI_CLIgemini)) COPILOT_MODELCOPILOT_ALLOW_ALLCOPILOT_GITHUB_TOKENzgithub-copilot))GOOSE_TERMINALgoose))OPENCLAW_SHELLopenclaw))OPENCODE_CLIENTopencode))PI_CODING_AGENTpi))REPL_IDreplit)) ROO_ACTIVEzroo-code))TRAE_AI_SHELL_IDtrae _TOOL_AGENTSdevinch|]\}}|Sr,).0_agents d/usr/local/lib/hermes-agent/venv/lib/python3.11/site-packages/huggingface_hub/utils/_detect_agent.py r17s@@@xq%U@@@returnctD]W}tj|d}|r|t vr|ndcSXtD]"\}}td|Dr|cS#dS)a!Return the name of the detected AI agent or ``None``. Checks environment variables in priority order and returns on the first match. When ``AI_AGENT`` or ``AGENT`` is set, the value is checked against known agent names, unrecognized values are returned as ``"unknown"``. unknownc3TK|]#}tj|V$dS)N)osenvironget)r-vars r0 zdetect_agent..Gs077srz~~c""777777r2N) rr8r9r:striplower _KNOWN_AGENTSr)any)r;nameenv_vars agent_names r0 detect_agentrD:s$@@z~~c2&&,,..4466  @=0044i ? ? ? @!-* 77h777 7 7      4r2c"tduS)zFReturn ``True`` if the process is being invoked by an AI coding agent.N)rDr,r2r0is_agentrFMs >> %%r2) __doc__r8typingrrtuplestr__annotations__r)r?rDboolrFr,r2r0rMs )>eCHo=== 9 eE%S/3./45( @@<@@@@ hsm&&$&&&&&&r2