eiz!pdZddlmZddlZddlZddlZddlmZmZddl m Z  ddd Z dddZ dddZ dS)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 = 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 - HERMES_INFERENCE_PROVIDER (already read by resolve_runtime_provider) ) annotationsN)redirect_stderrredirect_stdout)Optionalpromptstrmodel Optional[str]providerreturnintctjtjtjdd}|r9|pds#|s!t jddSdtj d<dtj d<t j }ttj d} t|5t|5t||| }d d d n #1swxYwYd d d n #1swxYwY |n:#t"$rYn.wxYw# |w#t"$rYwwxYwxYw|rS|||d s|d |d S) aExecute 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 HERMES_INFERENCE_PROVIDER env var, then config.yaml's model.provider, then "auto". 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. 1HERMES_YOLO_MODEHERMES_ACCEPT_HOOKSw)r r N r)loggingdisableCRITICALosgetenvstripsysstderrwriteenvironstdoutopendevnullrr _run_agentclose Exceptionendswithflush)rr r env_model_early real_stdoutr#responses 4/home/div/.hermes/hermes-agent/hermes_cli/oneshot.py run_oneshotr-s, OG$%%%i 8"==CCEEO%+2,,../  R   q&)BJ!"(+BJ$%*K2:s##G W % % J Jw'?'? J J!&IIIH J J J J J J J J J J J J J J J J J J J J J J J J J J J J J J  MMOOOO    D   MMOOOO    D (###  && $   d # # # 1sE D%D8 DD D D D E DE "D#E 'D<< E E  E3E#"E3# E0-E3/E00E3c ddlm}ddlm}ddlm}ddlm}ddlm }|}| dpi} t| tr| } n,| dp| dpd } tjd d } |pd p| p| } |pd pd } | |s| r|pd p| }|rd }t| t rHt| d pd }|p:tjd d pd}|||}|r|\} } || | pd }t%||d}|| d| d| d | d| |dd| dt& }d|_d |_d |_||pd S)zBuild an AIAgent exactly like a normal CLI chat turn would, then run a single conversation. Returns the final response string.r) load_config)detect_provider_for_model)resolve_runtime_provider)_get_platform_tools)AIAgentr defaultrrNr HERMES_INFERENCE_PROVIDERauto) requested target_modelcliapi_keybase_urlapi_modeTcredential_pool) r:r;r r<r enabled_toolsets quiet_modeplatformr=clarify_callback)hermes_cli.configr/hermes_cli.modelsr0hermes_cli.runtime_providerr1hermes_cli.tools_configr2 run_agentr3get isinstancerrrrdictlowersorted_oneshot_clarify_callbacksuppress_status_outputstream_delta_callbacktool_gen_callbackchat)rr r r/r0r1r2r3cfg model_cfg cfg_model env_modeleffective_modeleffective_providerexplicit_model cfg_providercurrent_providerdetectedruntime toolsets_listagents r,r$r$_s.-----;;;;;;DDDDDD;;;;;;!!!!!! +--C  &BI)S!!M MM),,L g0F0FL"  2B77==??I{))++EyEIO#.b//119T!u! ! +2,,..;)  ?L)T** T"9==#<#<#BCCIIKKQQSS 98"==CCEEKKMM  10AQRRH ?6>3"O&&$$,G..sE::;;M G I&&Z((Z((Z((& $5663+   E4$(E "&E"E ::f   ##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_choicess r,rLrLs;  @ @ @ @  ,r^)NN)rrr r r r r r )rrr r r r r r)N)r_rr r)__doc__ __future__rrrr contextlibrrtypingrr-r$rLrar^r,rgs*#""""" 77777777  "= = = = = D "]$]$]$]$]$@       r^