Zj UdZddlZddlmZdZeedfed<dZeeeedfefdfed<d heDchc]\}}| c}}zZ d eefd Z d e fd Z ycc}}w) 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_AGENTSdevinreturnctD]P}tjj|dj j }|sB|t vr|cSdcStD]\}}td|Ds|cSy)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"``. unknownc3ZK|]#}tjj|%yw)N)osenvironget).0vars b/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/huggingface_hub/utils/_detect_agent.py zdetect_agent..Gs7srzz~~c"7s)+N) rr0r1r2striplower _KNOWN_AGENTSr)any)r4nameenv_vars agent_names r5 detect_agentr>:s$@zz~~c2&,,.446 =04 ?i ?@ !-* 7h7 7  ctduS)zFReturn ``True`` if the process is being invoked by an AI coding agent.N)r>r?r5is_agentrBMs > %%r?) __doc__r0typingrrtuplestr__annotations__r)r9r>boolrB)_agents00r5rKs )>eCHo= 9 eE%S/3./45( <@xq%U@@ hsm&&$&-As A