߲j<dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZmZmZmZddlmZd#dZd$d Zd%d Zd&d Zd'd Zd(d Zd)dZ d* d+dZd,dZd-dZd.dZ d/dZ!d0dZ"d1dZ#e$hdZ%dZ&dZ'e$hdZ(dZ)e ee%Z*e e!e%Z+e e"e&Z,d2dZ-d*d3dZ.d.d Z/d4d!Z0 d5d"Z1y)6aShared runner for user-configured shell ("command") TTS/STT providers. ``tts.providers.: {type: command, command: "piper -f {output_path} < {input_path}"}`` (and the ``stt.`` twin): ``{placeholders}`` are shell-quoted for their surrounding quote context, ``{{``/``}}`` stay literal. Owns the quote-aware rendering, the idle-timeout process runner and the generic ``
.providers.`` readers, re-imported by ``tts_tool``/``transcription_tools`` under their historical private names. TTS placeholders: ``{input_path}``/``{text_path}``, ``{output_path}``, ``{format}``, ``{voice}``, ``{model}``, ``{speed}``. Built-in provider names always win over a same-named ``providers`` entry. ) annotationsN)partial)Path)AnyDict FrozenSetOptional)is_truthy_valuecd}d}d}||krH||}|dk(r|dk(r.d}n+|dk(r|rd}n!|dk(rd}n|dk(rd}n|dvr|}n |dk(r|d z }|d z }||krH|S) zRReturn the shell quote char (``'``/``"``) active right before *position*, or None.NFr'"\T)r r )command_templatepositionquoteescapedichars ?/home/ubuntu/.hermes/hermes-agent/tools/tts_command_provider.pyshell_quote_contextrsEG A h," C<s{ c\ Z E T\ FA Q! h," Lc:|dk(r|jddS|dk(rB|jddjddjddjd d Stjd k(rtj|gSt j |S) zGQuote a placeholder value for its position in a shell command template.r z'\''r rz\\z\"$z\$`z\`nt)replaceosname subprocess list2cmdlineshlexr)value quote_contexts rquote_command_placeholderr&6s}}S'**}}T6*223>FFsERZZ[^`eff/1ww$: " "E7 +VEKKPUDVVrc$djdD}tjd|d|d}gd fd }|j|j ddj d d }D]\}}|j ||}|S) zMReplace ``{name}`` placeholders (quote-aware) while preserving ``{{``/``}}``.|c3FK|]}tj|ywN)reescape).0r s r z*render_command_template..As>RYYt_>s!z(?z)\}\}|\{(?Pz)\})c|jdxs|jd}dtd}t|t|j }j ||f|S)Ndoublesingle__HERMES_CMD_PLACEHOLDER___)grouplenr&rstartappend)matchr tokenquotedr placeholders replacementss r replace_matchz.render_command_template..replace_matchEsn{{8$= H(=+C ,=+>bA*<+=?RScejepeper?stUFO, rz{{{z}}})r8z re.Match[str]returnstr)joinr+compilesubr) rr;namespatternr=renderedr9r$r<s `` @rrender_command_templaterH?s HH>> >Ejj7w>TUZT[[`abG*,L{{=*:;CCD#NVVW[]`aH$2 u##E512 Orc: |j|j}|jdD]} t||t||y#|j$rY:wxYw#|j$rYyt $rt||YywxYw)zIApply ``terminate``/``kill`` to *proc* and all descendants (best effort).T) recursiveN)Processpidchildrengetattr NoSuchProcess Exception)psutilprocmethodparentchilds r_signal_process_treerVRs  )__t_4 E &v&(  !''       f s:0A-AA-A*'A-)A**A--B>BBc jytjdk(r^ tjddddt j gtjtjdtjy d dl fd nfd }|d  jd y#t$rjYywxYw#t$rdYUwxYw#tj$r |dYywxYw)zCBest-effort termination of a shell process and all of its children.Nrtaskkillz/Fz/Tz/PID)stdoutstderrtimeoutstdinrc&t|Sr*)rN)mrRs rz0terminate_command_process_tree..rs)q)+rct|Sr*)rV)r_rRrQs rr`z0terminate_command_process_tree..ss264Cr terminater\kill)pollrr r!runrArLDEVNULLrPrerQ ImportErrorwaitTimeoutExpired)rRsignalrQs` @rterminate_command_process_treermbs yy{ ww$  NNJdFCMJS]SeSe","4"4azGYGY [ 17+C  ; !   IIK    $ $vs6AB3C C#3CC C C #DDc|jd}t|ttfrA|Dcgc]5}t |j st |j 7c}SgScc}w)z^``env_passthrough`` allowlist: parent env vars copied back into the secret-scrubbed child env.env_passthrough)get isinstancelisttuplerAstrip)configrawxs rcommand_env_passthroughrx{sO **& 'C>HtUZm>\C :q3q6<<>CFLLN :dbdd :s A+ A+cd|jfd|jffDcgc]\}}|s |d|j}}}dj|xsdScc}}w)zU``stderr: ...; stdout: ...`` for a failed command provider, or ``no command output``.r[rZ: z; zno command output)r[rZrtrB)excstreamtextpartss rcommand_failure_detailrsc?G>TW_adakakVl=m v\VTquxr$**, ( vE v 99U  222 ws AAc ddlm}ddlm}|d}|xsgD])}tj j |}|%|||<+tjdk(rdttddind d i}tj|fd tjtjd d d ||tjd |} tjggd} ddh} dfd } d| jfd| j ffD cgc]\} }t#j$| | |fd !}} }|D]}|j't)j*|z}d}| r|t)j*z }|dkrd }nb j t-d|\} }|| j1| U| | j3|t)j*|z}| r|s2 | j5t7d|t)j*z |rnt;| |D]}|j=dj?s:jA\} }|r| | j3|j?s:dj=| ddj=| d}}|r.tj8||||tj8||| jBr#tjD| jB|||tjF|| jB||Scc}} w#tj.$rYwxYw#tj8$rd }YQwxYw)u=Run a command-provider shell command with process-tree idle cleanup. ``timeout`` is an IDLE timeout, reset whenever the command emits output — a slow-but-alive provider survives, a silently stalled one is killed. Child env is scrubbed of Hermes secrets while propagating delegated-child lineage markers.r)delegated_child_subprocess_env)hermes_subprocess_envF)inherit_credentialsr creationflagsCREATE_NEW_PROCESS_GROUPstart_new_sessionTutf-8r)shellrZr[r}encodingerrorsenvr])rZr[rZr[c:t|ddxsd}tt|dddd} ||jdn|dj|d}|snj||fC j|dfy#j|dfwxYw)Nrrbufferread1ir)r)rNreaddecodeput)r r|rrchunk output_queues r read_streamz)run_command_provider..read_streams6:t4?$7$G +.3m E*uATATU]foATAp  $/    dD\ *L  dD\ *s ABB)targetargsdaemong?rdgg?)outputr[)r rAr|rr@None)$agent.delegation_contextrtools.environments.localrrenvironrpr rNr!PopenPIPErhqueueQueuerZr[ threadingThreadr6time monotonicminEmptydiscardr7rjmaxrkrmrBempty get_nowait returncodeCalledProcessErrorCompletedProcess)commandr\rorrscrubbedkeyr$r4rRchunks open_streamsrr r|readersreaderdeadline timed_out remainingrrZr[rs @rrun_command_providerrsSH>$?H$"" s#  !HSM" WYV]V]aeVeowz3MqQ R&-   G ?4 PZP_P_!% OmnvOw","4"4 ?8= ?D>C[[]L.0B#?Fh'L +&.t{{$;h =T#UW4{$PTUWGW ~~')HI t~~// >I  &**3tY3G*HKD% =   & t E">>#g-    IIc#x$..2B'BCI D&t, %F KKK $ %$$&&113KD%t ##E*$$&WWVH-.x8H0IFF''PVW  % %gw 7 9 ++DOOWV\bcc  & &w PPOW{{   (( I s*$L+=L1!1M 1MM M$#M$cpt|tr|j|nd}t|tr|SiS)z;Return ``config[name]`` if it's a dict, else an empty dict.N)rqdictrp)rur sections r_get_provider_sectionrs/",VT":fjjG $/77R7rct|dj|}t|tr|S|j |vr t||SiS)u``
.providers.`` (canonical), else ``
.`` for non-built-in names only — refused for built-ins so a user's ``openai:`` block still means OpenAI, not a command. providers)rrprqrlower)rur builtinsrs r_named_provider_configrsK$FK8<r " ( ( *C #/ "63 9E/65@D@rc|jd|jd|} t|}|dkDr|St|S#ttf$rt|cYSwxYw)z]Timeout in seconds (``timeout`` > ``timeout_seconds``); invalid or non-positive -> *default*.r\timeout_secondsr)rpfloat TypeError ValueError)rudefaultrvr$s r_command_timeoutrsc **Y +1 z "W~s AAAc|jdxs|jdxs|}t|jjj d}||vr|S|S)zEValidated ``format``/``output_format`` from *config*, else *default*.format output_format.)rprArrtlstrip)ruformatsrrvfmts r_command_output_formatrsX **X  H&**_"= HC c(..  " ) )# .C.3-g-r> xaiedgepipergeminineuttsopenaiminimaxmistral deepinfra kittentts elevenlabsxmp3>aacamrm4aroggwavflacopusi)r)r)rc#Kt|djD]?\}}t|ts|j t vs.t |s:||fAyw)zDYield (name, config) pairs for every declared command-type provider.rN)ritemsrqrArBUILTIN_TTS_PROVIDERSr) tts_configr cfgs r_iter_command_providersrsS*:{CIIK c dC TZZ\9N%NSnorSs)Os1A A  A  A c|r@t|jjjj dnd}|t vr|St |t tS)zZValidated output format: the output path's suffix wins, then ``format``/``output_format``.rr)rsuffixrrtrCOMMAND_TTS_OUTPUT_FORMATSr!DEFAULT_COMMAND_TTS_OUTPUT_FORMAT)ru output_pathrs r_get_command_tts_output_formatr#sSEPT+  % % + + - 3 3 5 < True only when the user explicitly opted in to voice delivery.voice_compatibleF)r rp)rus r _is_command_tts_voice_compatibler+s 6::&8%@ AArc<|jdt|S)zKReturn an output path whose extension matches the provider's output_format.r) with_suffixr)pathrus r#_configured_command_tts_output_pathr0s!   a >v FGH IIrct|jdxsdj}|std|dt |j }|j jdd|jr|jt|}tj5}t |dz } | j|d t| t| t|t|t|t|jd dt|jd dt|jd |jd dd } t|| } t!| |t#| ddd|jr|j1j2dkrt)d|d|t|S#t$j&$r} t)d|d|dd| d} ~ wt$j*$r-} t)d|d| j,dt/| | d} ~ wwxYw#1swYxYw)zGenerate speech by running a user-configured shell command; returns the audio path it wrote. Raises ``ValueError`` for bad provider config, ``RuntimeError`` for timeouts / bad exits / no output.rrztts.providers.z.command is not configuredT)parentsexist_okz input.txtr)rvoicemodelspeed) input_path text_pathrrrrr)rozTTS provider 'z' timed out after gsNz' exited with code rzrz' produced no output at )rArprtrr expanduserrTmkdirexistsunlink_get_command_tts_timeouttempfileTemporaryDirectory write_textrrHrrxr!rk RuntimeErrorrrrstatst_size) r}r provider_namerurrrr\tmpdirr r;rr{s r_generate_command_ttsr5s" 6::i06B7==? >-8RSTT +  ) ) +F MMt4 }} &v.G  $ $ &&L;. TG4i.s9~VYZ`Va4VS[IGR01C 7TV@WPQXYZP[[\]^dg g,,  /B3>>BRRTUkloUpTqr  s73CI 5GI1HI(III  I)rrArintr@ Optional[str])r$rAr%rr@rA)rrAr;zDict[str, str]r@rA)rQrrRsubprocess.PopenrSrAr@r)rRrr@r)ruDict[str, Any]r@rr)r{zsubprocess.CalledProcessErrorr@rAr*)rrAr\rrozOptional[list]r@zsubprocess.CompletedProcess)rurr rAr@r)rurr rArFrozenSet[str]r@r)rurr@r)rrArurrrr@zOptional[Dict[str, Any]])rurrrr@r)rurrrrrAr@rA)rr)rurrrr@rA)rrrurr@r) r}rArrArrArurrrr@rA)2__doc__ __future__rrrr+r#r!rrr functoolsrpathlibrtypingrrrr utilsr rr&rHrVrmrxrrrrrrrr frozensetr#DEFAULT_COMMAND_TTS_TIMEOUT_SECONDSrr#DEFAULT_COMMAND_TTS_MAX_TEXT_LENGTH_get_named_provider_config _resolve_command_provider_configrrrrrrrrrr+s #  11!2W&  2e 3EILQ LQ LQ3ALQ LQ`8 W[AA)A5CAH`A2."#12'*#$)!&'ab&*#%%;F[\#*+BMb#c "#3=`aiB J ! !!03!=K!Yg!!r