`gj2QdZddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZddlmZddd Zddd Zddd Z d dd Zd Z d ddZdddZy)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)declare_stateless_channel)get_fallback_chaincn|syt|tr|gn|}t|ttfs|g}g}|D]b}t|tr(|j d|j dD;|j t|jd|Dcgc]}|s| c}xsdScc}w)Nc3<K|]}|jywN)strip).0parts E/root/.hermes/venv/lib/python3.12/site-packages/hermes_cli/oneshot.py z&_normalize_toolsets...sGtdjjlGs,) isinstancestrlisttupleextendsplitappendr )toolsets raw_items normalizeditems r_normalize_toolsetsr#s (37 XI i$ /K J1 dC   Gtzz#G G   c$ioo/ 0 1 ( 0T4D 0 8D8 0s B2'B2ct|}|y ddlm}|Dcgc] }||s |}}|Dcgc] }||vs| }}|rM ddlm}||Dcgc] }||s |}}|r%|j ||Dcgc] }||vs| }}td|DrI|Dcgc] }|dvs| } }| r2tjjd d j| dyt} t} |r dd l m} dd lm} | }t#|j%d t&r|j%d ni}|j)D]e\}}t#|t&s| |j%dddr| j+t-|L| j+t-|g |Dcgc] }|| vs| }}|Dcgc] }|| vs| }}|Dcgc]}|| vs|| vs |}}||z}|r2tjjdd j|d|r2tjjdd j|d|sy|dfS#t$r}dd|dfcYd}~Sd}~wwxYwcc}wcc}wcc}w#t$rg}YGwxYwcc}wcc}w#t$rt} t} YwxYwcc}wcc}wcc}w)N)NNr)validate_toolsetz*hermes -z: failed to validate --toolsets:  )discover_pluginsc3$K|]}|dv yw)>*allN)rnames rrz._validate_explicit_toolsets..Os 5D4<  5s>r$r%zNhermes -z: --toolsets all enables every toolset; ignoring additional entries: z, )read_raw_config)_parse_enabled_flag mcp_serversenabledT)defaultz0hermes -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. )rrr Exceptionhermes_cli.pluginsr"ranysysstderrwritejoinsethermes_cli.configr(hermes_cli.tools_configr)rgetdictitemsaddr)rrr excr'built_in unresolvedr" plugin_validignored mcp_names mcp_disabledr(r)cfgr* server_cfg mcp_validdisabledunknownvalids r_validate_explicit_toolsetsrH5s$X.JJ-",F/?/EFHF#-F4X1E$FJF  ;  -7RT;KD;QDRLR  OOL )+5R4\9Q$RJR 5H 55$.KD$l2J4KK  JJ  004 '0B/C2G %I UL ! 9 C!#C4>sww}?UW[4\#''-0bdK$/$5$5$7 0 j!*d3&z~~i'FPTUMM#d), $$SY/  0#-B$ 0ABIB!+Dt|/CDHD *ad).CT`H`taGa y E KDIIV]L^K__abc  byy"#2 ' R $;A JA#bIIIJGFS L  SL0 !I5L !CDasI6 JJ JJJ# J$J(J# J5 J5' J:1J:B=J? K#K# K(#K(- K-7K-<K-6 J?J J JJ## J21J2?K K c\|sy ddl}id|jdd|jdd|jdd|jdd|jdd|jdd |jd d |jd d |jd d |jd d |jd d|jdd|jdd|jddt|jdxs|dud|jd}|||d<t|j }|j j dd|j|j|ddzdy#t$rYywxYw)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 service_tierfailureT)parentsexist_ok)indentr!utf-8encoding) jsonr7boolr expanduserparentmkdir write_textdumpsr-)pathresultrZrbreportouts r_write_usage_filerms    &**-A"B 6::m4  6::m4  FJJ~6  VZZ8  ,?!@   !&**-A"B   += >  FJJ~6  K0  VZZ(   :.  &**\2  K0  d6::h/0GG44G * FJJ~6+ .   'F9 4j##% 5 tzz&z3d:WM    sFF F+*F+c $tjtjtjddj }|r6|xsdj s"|s t jjdyt|\}}|r t jj|yt|du}dtjd<dtjd<tt j} t j} ttjd d } d} i} d} t!| 5t#| 5 t%||||| \} } dddddd | j)|lt-|t.t0frt3|| t5| |t3|| t7| | jd|d| j9yt3|| | rC| j| | j;ds| jd| j9| j=ds| j=dr| xsdj sy| xsdj s"| jd| j9yy#t&$r }|}Yd}~_d}~wwxYw#1swYixYw#1swYnxYw#t*$rYlwxYw# | j)w#t*$rYwwxYwxYw)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. The caller owns process termination. HERMES_INFERENCE_MODELzhermes -z: --provider requires --model (or HERMES_INFERENCE_MODEL). Pass both explicitly, or neither to use your configured defaults. r]N1HERMES_YOLO_MODEHERMES_ACCEPT_HOOKSwr_r`)rTrUruse_config_toolsets)rZzhermes -z: agent failed: r!rXpartialzGhermes -z: no final response was produced; treating the run as failed. r)loggingdisableCRITICALosgetenvr r0r1r2rHrenvironrstdoutopendevnullrr _run_agent BaseExceptioncloser-rKeyboardInterrupt SystemExitrmreprrflushendswithr7)promptrTrUr usage_fileenv_model_earlyexplicit_toolsetstoolsets_errorru real_stdout real_stderrrresponserjrZr;s r run_oneshotrs8 OOG$$%ii 8"=CCEO%+2,,./  R (CH(M%~ (-h74?&)BJJ!"(+BJJ$%**K**K2::sW5G"HF$(G W % w'?  #-%.(; $ &  &  MMO g 1:> ? j&$w- HM*fc'lC5gYbABj&)(#  &   d # 8 9 5B?U?U?W N ! ! #de U!      *     MMO   s0 K-; KK J*K$K--K* K3J;5K;KKK KKK- K*)K*-L/L?L L  L L  Lcv ddlm}|S#t$r }tjd|Yd}~yd}~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_staterr-rxdebug)rr;s r_create_session_db_for_oneshotr)s5*{  OQTUs  838c ddlm}ddlm}ddlm}ddlm}ddlm } |} | jdxsi} t| tr| } n(| jdxs| jdxsd } tjd d j} |xsd jxs| xs| }|xsd jxsd }d }|F|s| rA|xsd jxs| }|r& dd lm}|j%|j&j|jj)}|@|j,}|j.}|j0r|j0j3d }nd }t| t4r:t| jdxsd jj)}|xs6tjdd jj)xsd}|||}|r|\}}|||xsd |}t7|}||rt9|| d}t;}d } t=| }| |jd|jd|jd|jd||dd||jd|xsd t> }d|_ d |_!d |_"|jG|}|jdxsd |f|Q tI|dd }t|tJr|jM|n|jM |jS| |jSSS#t*$rd }Y9wxYw#t*$rtOjPddYZwxYw#t*$rtOjPddYowxYw#t*$rtOjPddYSwxYw#| tI|dd }t|tJr|jM|n|jMn&#t*$rtOjPddYnwxYw |jSn&#t*$rtOjPddYnwxYw|8 |jSw#t*$rtOjPddYwwxYwwxYw)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)AIAgentrTr,rproN) model_switch/rUHERMES_INFERENCE_PROVIDERauto) requested target_modelexplicit_base_urlcliapi_keybase_urlapi_modeTcredential_pool) rrrUrrTenabled_toolsets quiet_modeplatform session_dbrfallback_modelclarify_callbackfinal_response_session_messagesz%oneshot memory/context cleanup failed)exc_infozoneshot agent cleanup failedz$oneshot session store cleanup failed)*r5rhermes_cli.modelsrhermes_cli.runtime_providerrr6r run_agentrr7rrr{r|r hermes_clir_ensure_direct_aliasesDIRECT_ALIASESlowerr-rTrUrrstripr8rsortedrr _oneshot_clarify_callbacksuppress_status_outputstream_delta_callbacktool_gen_callbackrun_conversationgetattrrshutdown_memory_providerrxrr)rrTrUrrurrrrrrB model_cfg cfg_model env_modeleffective_modeleffective_providerexplicit_base_url_from_aliasexplicit_model_msdirect cfg_providercurrent_providerdetectedruntime toolsets_listragent_fbrjsession_messagess rrr9s.;D;! -C &BI)S! MM),L g0FL"  2B7==?I{))+EyEIO#.b//19T26 !u  +2,,.;)  :**,++//0D0D0F0L0L0NO!"(,,%+__"??39??3I3I#3N0! i.#&y}}Z'@'FB#G#M#M#O#U#U#WL yy!? /1J E@U!%KK *[[,[[,[[,!*!#KK(9:;$7/ 8(,$&*#"&''/ +,2F;   V#*52Et#L .5223CD224 M  ! U  " "G  t V EPTU V M N$O , M;:M;> N! N!$ OO O-,O-0S 4?P43S 4 QS QS Q,+S , R S RS R&%S & S S S  S c|rd|dS y)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&)questionchoicess rrrs'Ji> @  ,r )robjectreturnzlist[str] | None)rrrz#tuple[list[str] | None, str | None])ri Optional[str]rjr8rZrrNone)NNNN) rrrTrrUrrrrrrint)NNNT) rrrTrrUrrrrurcrztuple[str, dict])rrrr)__doc__ __future__rrxr{r0 contextlibrrpathlibrtypingrgateway.session_contextrhermes_cli.fallback_configr rrHrmrrrrr&rrrs*# 7=99$GT( Z " $ | | | |  |  |  | ~ $ " $ [U [U [U[U [U  [U  [U| r