apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-29 15:59:42 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixbug: terminal

Browse source
This commit is contained in:
莘权 马 2024-04-08 23:54:55 +08:00
parent b5d79a378a
commit d8aea069fa
3 changed files with 16 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
metagpt/logs.py
View file

@ -26,7 +26,7 @@ class ToolLogItem(BaseModel):
TOOL_LOG_END_MARKER = ToolLogItem(
type="str", name="end_marker", value="#END#"
type="str", name="end_marker", value="\x18\x19\x1B\x18"
) # A special log item to suggest the end of a stream log

11
metagpt/tools/libs/terminal.py
View file

@ -39,7 +39,7 @@ class Terminal:
# Send the command
self.process.stdin.write(cmd + self.command_terminator)
self.process.stdin.write(
f'echo "{TOOL_LOG_END_MARKER.value}"' + self.command_terminator
f'echo "{TOOL_LOG_END_MARKER.value}"' + self.command_terminator # write EOF
) # Unique marker to signal command end
self.process.stdin.flush()
log_tool_output(
@ -49,7 +49,14 @@ class Terminal:
# Read the output until the unique marker is found
while True:
line = self.process.stdout.readline()
if line.strip() == TOOL_LOG_END_MARKER.value:
ix = line.rfind(TOOL_LOG_END_MARKER.value)
if ix >= 0:
line = line[0:ix]
if line:
log_tool_output(
output=ToolLogItem(name="output", value=line), tool_name="Terminal"
) # log stdout in real-time
cmd_output.append(line)
log_tool_output(TOOL_LOG_END_MARKER)
break
log_tool_output(