#xj HUddlZddlZddlmZddlmZddlmZmZm Z edZ edZ edZ e hdZe e ed <ejd Zed Gd dZ ddddeeeefde deefdZdS)N)Mapping) dataclass)LiteralOptionalcasthighmediumlow) environmentpathz user-agent)amp antigravity augment-cli claude-codecodex copilot-clicoworkcursor cursor-cli gemini-cligoosegrokgrok-botkiroopencodepireplit>rrrrrrrrrrrrrrrrr KNOWN_AGENTSz$(?:^|[\\/])\.pi[\\/]agent(?:[\\/]|$)T)frozenc<eZdZUdZeed<eed<eed<eed<dS)DetectionResultzKEvidence that an AI coding agent is driving the current process or request.agent confidencesourcesignalN) __name__ __module__ __qualname____doc__ AgentName__annotations__AgentConfidenceDetectionSourcestrIC:\Projects\Historical\venv\Lib\site-packages\agent_detector/_detector.pyr"r"6sCUU   KKKKKr1r"r )minimum_confidenceenvironr3returnc  | tjn|}|dvrtdd|g}|dd}|t vr8|ttt|ddd|dd kr%|td ddd|d r%|td d dd d D]<}||r%|td dd|=|dr%|tdddd|dr%|tdd dd|dr%|tdddd|dr%|tdd dd|dr%|tdd dd|dr%|tdddd|dr%|tdddddD]<}||r%|tdd d|=|dr%|tdddd|dr%|tdd dd|d rO|d!r:|d"r%|td#ddd!|d$r%|td%d dd$|d r%|td&ddd |d'd(kr%|td&d dd'|d)d*kr%|td*d+dd)|d,d}t |r%|td-d d.d,|d/r%|td0d+dd/|d1r%|td2d+dd1tfd3|DdS)4aDetect the AI coding agent driving the current process, if any. Detection is best-effort. A ``None`` result means "unattributed", not "human". Values from the environment are inspected locally but are never included in the returned result. Nrz5minimum_confidence must be 'high', 'medium', or 'low'AI_AGENTr r AGENTrAMP_CURRENT_THREAD_IDr )CODEX_THREAD_IDCODEX_CI CODEX_SANDBOXr GEMINI_CLIr COPILOT_CLIrOPENCODErANTIGRAVITY_AGENTr AUGMENT_AGENTrCLAUDE_CODE_IS_COWORKrCLAUDE_CODE_CHILD_SESSIONr) CLAUDECODE CLAUDE_CODEGROK_SESSION_IDr GROK_AGENT CURSOR_AGENTSAND_BOX_BOOT_IDSAND_BOX_STORE_IDrCURSOR_TRACE_IDrrCURSOR_EXTENSION_HOST_ROLEz agent-exec TERM_PROGRAMrr PATHrr REPL_IDrGOOSE_PROVIDERrc3XK|]$}|jk |V%dSN)indexr$).0 candidateconfidence_orderminimum_confidence_indexs r2 zdetect_agent..sL  %%i&:;;?WWW WWWW  r1) osr4 ValueErrorrTgetrappendr"rr+_PI_AGENT_PATHsearchnext) r4r3values candidatesexplicit_agentr&r rWrXs @@r2 detect_agentrd@s#?RZZF!:::PQQQ4M/556HII(*JZZ B//N%% DN;;V]T^ _ _   zz'e##/%PPQQQ zz)**d/%=JabbcccCWW ::f   W   ogv}fUU V V Vzz,^/, |\\]]]zz-  b/-=R_``aaazz*Z/*fmZXXYYY zz%&&  M8]DW X X   zz/""d/-=Rabbccczz)**e/(FMKbccdddzz-..  M6=B] ^ ^   0__ ::f   _   omX}V\]] ^ ^ ^zz#$$]/&&-IZ[[\\\zz,Z/&(M<XXYYY  >""b JJ) * *b JJ* + +b /*fmM_``aaazz#$$a/(HmM^__``` zz.!!`/, ~^^___ zz.//<?? L(MC_ ` `   zz.!!V++/&%WWXXX ::fb ! !DT""K/$&&IIJJJ zz)V/(E=)TTUUU zz"##\/'5-IYZZ[[[      '     r1rS)rZrecollections.abcr dataclassesrtypingrrrr-r.r+ frozensetrr,compiler^r"r/rdr0r1r2rks ######!!!!!!**********12=>   (&/Y&& i ",CDD $,0A+0AAA gc3h' (A(Ao AAAAAAr1