jV ddlZddlZddlZddlmZmZmZmZmZddl m Z ddl m Z ddl mZmZmZej"eZGddeZdefd Zdefd ZGd d e Zy) N)AnyListOptionalTypeUnion)CallbackManagerForToolRun)BaseTool) BaseModelFieldmodel_validatorcteZdZUdZeddZeeeefe d< e de de d e fd Zy ) ShellInputz!Commands for the Bash Shell tool..z ::j)(D)"*F:  P  N)__name__ __module__ __qualname____doc__r rrstrr__annotations__r classmethoddictrrrrrrs[+&+ R'HeCcN#)(#   $ rrrcR ddlm}|dS#t$r tdwxYw)zGet default bash process.r BashProcesszBashProcess has been moved to langchain experimental.To use this tool, install langchain-experimental with `pip install langchain-experimental`.T)return_err_output)$langchain_experimental.llm_bash.bashr* ImportErrorr)s r_get_default_bash_processr.'s: D  ..   9   s&c:tj}|dk(ry|S)z Get platform.DarwinMacOS)platformsystem)r3s r _get_platformr44s __ F  MrceZdZUdZeeZeed< dZ e ed< de dZ e ed< e Zeeed < d Zeed < dd ee ee fdeede fdZy ) ShellToolzTool to run shell commands.)default_factoryprocessterminalnamezRun shell commands on this z machine.r args_schemaFask_human_inputNr run_managerrchtd| |jrOtdj}|dk(r|jj |St jdy|jj |S#t$r"}t jd|Yd}~yd}~wwxYw)z%Run commands and return final output.zExecuting command: z'Proceed with command execution? (y/n): yz.Invalid input. User aborted command execution.Nz Error during command execution: ) printr<inputlowerr8runloggerinfo Exceptionerror)selfrr= user_inputes r_runzShellTool._runQs %hZ01 ##"#LMSSU $<<++H55KK PQ||''11  LL;A3? @ s$ABB+B B1B,,B1)N)rr r!r"r r.r8rr$r:r#r4rrr;rr r<boolrrrrrKr'rrr6r6<s%)BCGSC'D#4]_4EYOKO#-Ki-%!OT!<@T#Y'78 rr6)loggingr2rtypingrrrrrlangchain_core.callbacksrlangchain_core.toolsr pydanticr r r getLoggerrrDrr.r#r4r6r'rrrSsf33*66   8 $0 /3 /s++r