@xjdZddlZddgfddgfddgfd d gfd d d gfdgdfddgfddgfdgdfddgfddgfddgfddgfddgfd d!gfd"d#gfd$d%gfd&d'gfd(d)gfd*d+gfd,d-gfgZd.gZd/Zd0Zd1Zd2ZdS)3aDetects which AI coding agent (if any) is driving the SDK. Detection is based on the environment variables that agent harnesses set in the processes they spawn. The registry mirrors Hugging Face's public agent-harnesses list so that traffic is attributed under the same agent identifiers across tools: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/src/agent-harnesses.ts N antigravityANTIGRAVITY_AGENTz augment-cli AUGMENT_AGENTcline CLINE_ACTIVEcoworkCLAUDE_CODE_IS_COWORKz claude-code CLAUDECODE CLAUDE_CODEcodex) CODEX_SANDBOXCODEX_CICODEX_THREAD_IDcrushCRUSHz gemini-cli GEMINI_CLIzgithub-copilot) COPILOT_MODELCOPILOT_ALLOW_ALLCOPILOT_GITHUB_TOKENgooseGOOSE_TERMINALz hermes-agentHERMES_SESSION_IDz kilo-codeKILOCODE_FEATUREkiroAGENT_CONTEXT_OUTopenclawOPENCLAW_SHELLopencodeOPENCODE_CLIENTpiPI_CODING_AGENTreplitREPL_IDtraeTRAE_AI_SHELL_IDzedZED_TERMz cursor-cli CURSOR_AGENTcursorCURSOR_TRACE_IDAI_AGENTcxg}tD]\}}|||ttzS)zReturns every environment variable the registry inspects, including the standard AI_AGENT signal. Useful for tests that need to isolate detection from the ambient environment. ) HARNESSESextendlistSTANDARD_ENV_VARS) variables_env_varss =C:\Projects\Historical\venv\Lib\site-packages\runpod/agent.pyknown_env_varsr50sG I ## 8"""" t-.. ..c|}g}|D]X}t|dkrnB|r-|s|dvr||Yd|S)zKeeps only User-Agent-safe characters ([A-Za-z0-9._-]), capped at 64 characters, so an arbitrary env value cannot produce a malformed header. @z._-)striplenisasciiisalnumappendjoin)valuesafechars r4 _sanitizerC;s KKMME D t99?? E <<>> t||~~  KK    774==r6ctD]7\}}|D]/}tj|dr|ccS08tD]*}t tj|d}|r|cS+dS)zReturns the identifier of the AI coding agent driving the SDK, or an empty string if none is detected. Specific harness markers take priority over the generic AI_AGENT signal. r9)r-osgetenvr:r0rC)agent_idr3envr@s r4detectrIIs (  (  Cyb!!''))   !")C,,--  LLL  2r6c2t}|rd|dSdS)zReturns the " (via )" User-Agent fragment for the detected agent, or an empty string when none is detected. Centralizing the fragment here keeps the tag format identical across every client's User-Agent. z (via )r9)rI)rGs r4suffixrL[s+ xxH$##### 2r6)__doc__rEr-r0r5rCrIrLr6r4rOsk ()*_%& ~ '()\=12 >>>? wiL>"UUUV  !)*+%&' ! "#"#$#$%   {  !" ZLN#$ !"#+  8 L///   $r6