NQjKFdZddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZd%d&d Zd%d'dZd%d(dZ d)d*dZdZ d+d,d"Zd%d-d$ZdS).uOneshot (-z) mode: send a prompt, get the final content block, exit. Bypasses cli.py entirely. No banner, no spinner, no session_id line, no stderr chatter. Just the agent's final text to stdout. Toolsets = explicit --toolsets when provided, otherwise whatever the user has configured for "cli" in `hermes tools`. Rules / memory / AGENTS.md / preloaded skills = same as a normal chat turn. Approvals = auto-bypassed (HERMES_YOLO_MODE=1 is set for the call). Working directory = the user's CWD (AGENTS.md etc. resolve from there as usual). Model / provider selection mirrors `hermes chat`: - Both optional. If omitted, use the user's configured default. - If both given, pair them exactly as given. - If only --model given, auto-detect the provider that serves it. - If only --provider given, error out (ambiguous — caller must pick a model). Env var fallbacks (used when the corresponding arg is not passed): - HERMES_INFERENCE_MODEL ) annotationsN)redirect_stderrredirect_stdout)Path)Optional)get_fallback_chaintoolsetsobjectreturnlist[str] | Nonec|sdSt|tr|gn|}t|ttfs|g}g}|D]~}t|tr3|d|dDJ|t|d|DpdS)Nc3>K|]}|VdSN)strip).0parts 6/home/alina/.hermes/hermes-agent/hermes_cli/oneshot.py z&_normalize_toolsets..-s*GGtdjjllGGGGGG,cg|]}||Sr)ritems r z'_normalize_toolsets..1s 0 0 0T4 0D 0 0 0r) isinstancestrlisttupleextendsplitappendr)r raw_items normalizedrs r_normalize_toolsetsr$"s t(377E XI i$ / / K J11 dC  1   GGtzz#GGG G G G G   c$iioo// 0 0 0 0 0 0Z 0 0 0 8D8r#tuple[list[str] | None, str | None]ct|}|dS ddlmn#t$r}dd|dfcYd}~Sd}~wwxYwfd|Dfd|D}|rW ddlm}|fd |Dn#t$rgYnwxYwr#fd |D}td DrFd |D}|r6tj d d |ddStt|r ddl m}ddlm}|}t#|dt&r|dni} | D]\} } t#| t&s|| dddr#t-| _t-| n,#t$rttYnwxYwfd|D} fd|D} fd|D}| z}|r6tj dd |d| r6tj dd | d|sdS|dfS)N)NNr)validate_toolsetz*hermes -z: failed to validate --toolsets:  c*g|]}| |Srrrnamer's rrz/_validate_explicit_toolsets..>s*FFF/?/?/E/EFFFFrcg|]}|v| Srr)rr+built_ins rrz/_validate_explicit_toolsets..?s#FFF4X1E1E$1E1E1Er)discover_pluginsc*g|]}| |Srrr*s rrz/_validate_explicit_toolsets..Fs*RRRT;K;KD;Q;QRDRRRrcg|]}|v| Srr)rr+ plugin_valids rrz/_validate_explicit_toolsets..Ls#RRR4\9Q9Q$9Q9Q9Qrc3K|]}|dvV dS)>*allNrrr+s rrz._validate_explicit_toolsets..Ns' 5 5D4<  5 5 5 5 5 5rcg|]}|dv| S)>r3r4rr5s rrz/_validate_explicit_toolsets..Os"KKKD$l2J2J42J2J2JrzNhermes -z: --toolsets all enables every toolset; ignoring additional entries: z, )read_raw_config)_parse_enabled_flag mcp_serversenabledT)defaultcg|]}|v| Srr)rr+ mcp_namess rrz/_validate_explicit_toolsets..ks#BBB$ 0A0A0A0A0Arcg|]}|v| Srr)rr+ mcp_disableds rrz/_validate_explicit_toolsets..ls#DDDt|/C/C/C/C/Crc$g|] }|v|v | Srr)rr+r?r=s rrz/_validate_explicit_toolsets..ms.aaad).C.CT`H`H`tH`H`H`rz0hermes -z: ignoring unknown --toolsets entries: zThermes -z: ignoring disabled MCP servers (set enabled: true in config.yaml to use): )Nz:hermes -z: --toolsets did not contain any valid toolsets. )r$r r' Exceptionhermes_cli.pluginsr.ranysysstderrwritejoinsethermes_cli.configr7hermes_cli.tools_configr8rgetdictitemsaddr)r r#exc unresolvedr.ignoredr7r8cfgr9r+ server_cfg mcp_validdisabledunknownvalidr-r?r=r1r's @@@@@r_validate_explicit_toolsetsrX4s $X..JzJ------- JJJI#IIIIIIIIIIJGFFFFFFHFFFF:FFFJ S  ; ; ; ; ; ;     RRRRZRRRLL   LLL   S OOL ) ) )RRRR:RRRJ 5 5H 5 5 555KKJKKK   J  G04 '0B0BGGG   z%%I UUL! ! 9 9 9 9 9 9 C C C C C C!/##C4>sww}?U?UW[4\4\d#''-000bdK$/$5$5$7$7 0 0 j!*d33&&z~~i'F'FPTUUU0MM#d)),,,, $$SYY////  0 ! ! !I55LLL !CBBB*BBBIDDDDDDDHaaaaa aaaG y Ed bDIIV]L^L^bbbccc   'yy"" ' ' '   SRR $;s9! =8==A>> B  B 4C+H &I I path Optional[str]resultrLfailureNonec|sdS ddl}|d|d|d|d|d|d|d |d |d |d |d |d|d|dt|dp|dud}|||d<t|}|jdd|||ddzddS#t$rYdSwxYw)uBest-effort JSON usage report for pipelines (``-z --usage-file``). Written even on failure so callers can always account for spend. Never raises — a broken usage write must not mask the run's own outcome. Nrestimated_cost_usd cost_status cost_source input_tokens output_tokenscache_read_tokenscache_write_tokensreasoning_tokens total_tokens api_callsmodelprovider session_id completedfailed)r_r`rarbrcrdrerfrgrhrirjrkrlrmr\T)parentsexist_ok)indentr(utf-8encoding) jsonrKboolr expanduserparentmkdir write_textdumpsrA)rYr[r\rureportouts r_write_usage_filer~~s   #)**-A"B"B!::m44!::m44"JJ~66#ZZ88!',?!@!@"(**-A"B"B & += > >"JJ~66K00ZZ(( :.. **\22K006::h//00GG44G  "   'F9 4jj##%% 555 tzz&z33d:WMMMMM      sF7F?? G  G promptrrirj usage_fileintc tjtjtjdd}|r9|pds#|s!t jddSt|\}}|r!t j|dSt|du}dtj d<dtj d<t j } t j} ttjd d } d} i} d} t| 5t!| 5 t#||||| \} } n#t$$r }|}Yd}~nd}~wwxYwdddn #1swxYwYdddn #1swxYwY | n:#t($rYn.wxYw# | w#t($rYwwxYwxYw|t+|t,t.fr!t1|| t3| |t1|| t5| | d|d| dSt1|| | rS| | | ds| d| | ds| dr| pdsdS| pds+| d| dSdS)uExecute a single prompt and print only the final content block. Args: prompt: The user message to send. model: Optional model override. Falls back to HERMES_INFERENCE_MODEL env var, then config.yaml's model.default / model.model. provider: Optional provider override. Falls back to config.yaml's model.provider, then "auto". toolsets: Optional comma-separated string or iterable of toolsets. usage_file: Optional path; when set, a JSON usage report (estimated cost, token counts, model, api_calls) is written there after the run — even when the run fails — so pipelines can account for spend per invocation. Returns the exit code. Caller should sys.exit() with the return. HERMES_INFERENCE_MODELzhermes -z: --provider requires --model (or HERMES_INFERENCE_MODEL). Pass both explicitly, or neither to use your configured defaults. rpN1HERMES_YOLO_MODEHERMES_ACCEPT_HOOKSwrrrs)rirjr use_config_toolsets)r\zhermes -z: agent failed: r(rmpartialzGhermes -z: no final response was produced; treating the run as failed. r)loggingdisableCRITICALosgetenvrrDrErFrXr$environstdoutopendevnullrr _run_agent BaseExceptioncloserArKeyboardInterrupt SystemExitr~reprrflushendswithrK)rrirjr renv_model_earlyexplicit_toolsetstoolsets_errorr real_stdout real_stderrrresponser[r\rOs r run_oneshotrs=8 OG$%%%i 8"==CCEEO%+2,,../  R   q(CH(M(M%~ (((q-h774?&)BJ!"(+BJ$%*K*K2:sW555G"HF$(G W % %  w'?'?   #-%.(; $$$ &&!                                  &  MMOOOO    D   MMOOOO    D  g 1:> ? ?  j&$w-- H H H HM*fc'llCCCCAgAAABBBqj&)))(###  && $   d # # # 8 9 5 5B?U?U?W?Wq N ! ! # #deeeq 1s G/F?E>EE> E/#E*%E>*E//E>2 F>F FF F GFGFG!F66 GGG-GG- G*'G-)G**G-c~ ddlm}|S#t$r }tjd|Yd}~dSd}~wwxYw)a!Best-effort SessionDB for ``hermes -z`` / oneshot mode. Oneshot bypasses ``HermesCLI._init_agent()``, so it must wire the SQLite session store itself. Without this, the ``session_search``/recall tool is advertised but every call returns "Session database not available.". r) SessionDBz7SQLite session store not available for oneshot mode: %sN) hermes_staterrArdebug)rrOs r_create_session_db_for_oneshotrse******y{{  OQTUUUttttts <7<Trrvtuple[str, dict]cddlm}ddlm}ddlm}ddlm}ddlm } |} | dpi} t| tr| } n,| dp| dpd } tjd d } |pd p| p| }|pd pd }d }|m|s| rh|pd p| }|rM dd lm}||j |}n#t*$rd }YnwxYw|0|j}|j}|jr|jd }nd }t| t4rHt| dpd }|p:tjdd pd}|||}|r|\}}|||pd |}t7|}||rt9|| d}t;}t=| }| | d| d| d| d||dd|| d|pd t> }d|_ d |_!d |_"|#|}| dpd |fS)zBuild an AIAgent exactly like a normal CLI chat turn would, then run a single conversation. Returns ``(final_response, run_result)``.r) load_config)detect_provider_for_model)resolve_runtime_provider)_get_platform_tools)AIAgentrir;rrN) model_switch/rjHERMES_INFERENCE_PROVIDERauto) requested target_modelexplicit_base_urlcliapi_keybase_urlapi_modeTcredential_pool) rrrjrrienabled_toolsets quiet_modeplatform session_dbrfallback_modelclarify_callbackfinal_response)$rIrhermes_cli.modelsrhermes_cli.runtime_providerrrJr run_agentrrKrrrrr hermes_clir_ensure_direct_aliasesDIRECT_ALIASESlowerrArirjrrstriprLr$sortedrr_oneshot_clarify_callbacksuppress_status_outputstream_delta_callbacktool_gen_callbackrun_conversation)rrirjr rrrrrrrR model_cfg cfg_model env_modeleffective_modeleffective_providerexplicit_base_url_from_aliasexplicit_model_msdirect cfg_providercurrent_providerdetectedruntime toolsets_listr_fbagentr[s rrr)s.-----;;;;;;DDDDDD;;;;;;!!!!!! +--C  &BI)S!!M MM),,L g0F0FL"  2B77==??I{))++EyEIO#.b//119T26 !u! ! +2,,..;)  C ::::::**,,,+//0D0D0F0F0L0L0N0NOO    !"(,%+_"?O39?3I3I#3N3N0! i..X#&y}}Z'@'@'FB#G#G#M#M#O#O#U#U#W#WL y!>?? /11J S ! !C G I&&Z((Z((Z((& $566{d3/   E8$(E "&E"E  # #F + +F JJ' ( ( .B 77sAE E+*E+questionc|rd|dS dS)uyClarify is disabled in oneshot mode — tell the agent to pick a default and proceed instead of stalling or erroring.z<[oneshot mode: no user available. Pick the best option from z' using your own judgment and continue.]z\[oneshot mode: no user available. Make the most reasonable assumption you can and continue.]r)rchoicess rrrs;  @ @ @ @  ,rr)r r r r )r r r r%)rYrZr[rLr\rZr r])NNNN) rrrirZrjrZr r rrZr r)NNNT) rrrirZrjrZr r rrvr r)rrr r)__doc__ __future__rrrrD contextlibrrpathlibrtypingrhermes_cli.fallback_configrr$rXr~rrrrrrrrsj*#""""" 7777777799999999999$GGGGGT" " " " " N " $ s s s s s l   $ " $ {8{8{8{8{8|       r