o j @sUdZddlZddlmZdZeedfed<dZeeeedfefdfed<d hd d eDBZ d eefd dZ d e fddZ 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_AGENTSdevincCsh|]\}}|qSr*).0_agentr*r*V/home/kuhnn/.local/lib/python3.10/site-packages/huggingface_hub/utils/_detect_agent.py 7sr/returncCshtD]}tj|d}|r|tvr|SdSqtD]\}}tdd|Dr1|Sq 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"``. unknowncss|] }tj|VqdS)N)osenvironget)r+varr*r*r. Gszdetect_agent..N) rr3r4r5striplower _KNOWN_AGENTSr(any)r6nameenv_vars agent_namer*r*r. detect_agent:s r?cCs tduS)zFReturn ``True`` if the process is being invoked by an AI coding agent.N)r?r*r*r*r.is_agentMs r@) __doc__r3typingrrtuplestr__annotations__r(r:r?boolr@r*r*r*r.s  $