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

intervention fix, timed output fix?, dirty json {{ fix, prompt additions

Browse source
This commit is contained in:
frdel 2024-07-25 23:47:32 +02:00
parent cc19171464
commit 5c592ca71f
10 changed files with 41 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

10
python/tools/code_execution_tool.py
View file

@ -21,6 +21,9 @@ class State:
class CodeExecution(Tool):
def execute(self,**kwargs):
if self.agent.handle_intervention(): return Response(message="", break_loop=False) # wait for intervention and handle it, if paused
self.prepare_state()
# os.chdir(files.get_abs_path("./work_dir")) #change CWD to work_dir
@ -77,6 +80,9 @@ class CodeExecution(Tool):
return self.terminal_session(command)
def terminal_session(self, command):
if self.agent.handle_intervention(): return "" # wait for intervention and handle it, if paused
self.state.shell.send_command(command)
PrintStyle(background_color="white",font_color="#1B4F72",bold=True).print(f"{self.agent.agent_name} code execution output:")
@ -84,9 +90,11 @@ class CodeExecution(Tool):
def get_terminal_output(self):
idle=0
while True:
while True:
time.sleep(0.1) # Wait for some output to be generated
full_output, partial_output = self.state.shell.read_output()
if self.agent.handle_intervention(): return full_output # wait for intervention and handle it, if paused
if partial_output:
PrintStyle(font_color="#85C1E9").stream(partial_output)