vrr/agent-zero
2
0
Fork 0
mirror of https://github.com/agent0ai/agent-zero.git synced 2026-04-28 11:40:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

RFC & SSH exchange for development

Browse source
This commit is contained in:
frdel 2024-12-03 14:22:18 +01:00
parent 19f50d6d95
commit 46689d6477
22 changed files with 418 additions and 157 deletions
Download patch file Download diff file Expand all files Collapse all files

5
python/tools/code_execution_tool.py
View file

@ -3,7 +3,7 @@ from dataclasses import dataclass
import shlex
import time
from python.helpers.tool import Tool, Response
from python.helpers import files
from python.helpers import files, rfc_exchange
from python.helpers.print_style import PrintStyle
from python.helpers.shell_local import LocalInteractiveSession
from python.helpers.shell_ssh import SSHInteractiveSession
@ -93,12 +93,13 @@ class CodeExecution(Tool):
# initialize local or remote interactive shell insterface
if self.agent.config.code_exec_ssh_enabled:
pswd = self.agent.config.code_exec_ssh_pass if self.agent.config.code_exec_ssh_pass else await rfc_exchange.get_root_password()
shell = SSHInteractiveSession(
self.agent.context.log,
self.agent.config.code_exec_ssh_addr,
self.agent.config.code_exec_ssh_port,
self.agent.config.code_exec_ssh_user,
self.agent.config.code_exec_ssh_pass,
pswd,
)
else:
shell = LocalInteractiveSession()