a dg@sfdZddlmZddlZddlZddlZddlmZmZm Z ddl m Z erTddl Z GdddZ dS)z*Wrapper around subprocess to run commands.) annotationsN) TYPE_CHECKINGListUnion)uuid4c@seZdZUdZdZded<dZded<dZded<d dddddd Ze d d d d Z e ddddddZ dddddZ dddddZ ddddddZdddddZdS)! BashProcessa<Wrapper for starting subprocesses. Uses the python built-in subprocesses.run() Persistent processes are **not** available on Windows systems, as pexpect makes use of Unix pseudoterminals (ptys). MacOS and Linux are okay. Example: .. code-block:: python from langchain_community.utilities.bash import BashProcess bash = BashProcess( strip_newlines = False, return_err_output = False, persistent = False ) bash.run('echo 'hello world'') Fboolstrip_newlinesreturn_err_output persistent)r r r cCs<||_||_d|_d|_|r8tt|_|||j|_dS)z3 Initializes with default settings N)r r promptprocessstrr_initialize_persistent_process)selfr r r rr/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_experimental/llm_bash/bash.py__init__/s  zBashProcess.__init__pexpect)returncCs@tdkrtdz ddl}Wnty:tdYn0|S)z Import pexpect only when needed.Windowsz;Persistent bash processes are not yet supported on Windows.rNzVpexpect required for persistent bash processes. To install, run `pip install pexpect`.)platformsystem ValueErrorr ImportError)rrrr_lazy_import_pexpect@s    z BashProcess._lazy_import_pexpectrz pexpect.spawn)rr rcCs<|}|jdgddd}|d||j|dd|S)z Initializes a persistent bash setting in a clean environment. NOTE: Unavailable on Windows Args: Prompt(str): the bash command to execute env)z-ibashz--norcz --noprofilezutf-8)encodingzPS1= timeout)rspawnsendlineZ expect_exact)rr rrrrrrQs  z*BashProcess._initialize_persistent_processzUnion[str, List[str]])commandsrcCs<t|tr|g}d|}|jdur.||S||SdS)z Run commands in either an existing persistent subprocess or on in a new subprocess environment. Args: commands(List[str]): a list of commands to execute in the session ;N) isinstancerjoinr_run_persistent_run)rr%rrrrungs   zBashProcess.run)commandrc Csz"tj|ddtjtjdj}WnJtjyl}z0|jrP|jWYd}~St|WYd}~Sd}~00|j r|| }|S)z Runs a command in a subprocess and returns the output. Args: command: The command to run T)shellcheckstdoutstderrN) subprocessr+PIPESTDOUTr/decodeCalledProcessErrorr rr strip)rr,outputerrorrrrr*zszBashProcess._run)r7r,rcCs(t|d}tj|d|dd}|S)z Uses regex to remove the command from the output Args: output: a process' output string command: the executed command z\s*\nr )count)reescapesubr6)rr7r,patternrrrprocess_outputszBashProcess.process_outputcCs|}|jdurtd|j||jj|jdd|jdz|jj|j|jgddWn|jy~d|YS0|jj|jkrd|jj S|jj }| ||}|j r| S|S)z Runs commands in a persistent environment and returns the output. Args: command: the command to execute NzProcess not initializedr r!r z&Timeout error while executing command zExited with error status: )rrrr$expectr ZEOFTIMEOUTafterZ exitstatusbeforer?r r6)rr,rr7rrrr)s"    zBashProcess._run_persistentN)FFF)__name__ __module__ __qualname____doc__r __annotations__r r r staticmethodrrr+r*r?r)rrrrrs     r)rG __future__rrr;r1typingrrruuidrrrrrrrs