Skip to content

Commit 688f7ef

Browse files
Refactor capture_observation function formatting
1 parent 36a591c commit 688f7ef

1 file changed

Lines changed: 12 additions & 12 deletions

File tree

env.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -129,16 +129,16 @@ def list_workspace_files() -> list:
129129
ensure_workspace()
130130
return os.listdir(WORKSPACE_DIR)
131131

132-
132+
133133
def capture_observation(cmd_result: dict) -> str:
134-
"""Format command result as an observation string for the agent loop."""
135-
parts = []
136-
if cmd_result["stdout"].strip():
137-
parts.append(f"STDOUT:\n{cmd_result['stdout'].strip()}")
138-
if cmd_result["stderr"].strip():
139-
parts.append(f"STDERR:\n{cmd_result['stderr'].strip()}")
140-
parts.append(f"Return code: {cmd_result['returncode']}")
141-
files = list_workspace_files()
142-
if files:
143-
parts.append(f"Workspace files: {', '.join(files)}")
144-
return "\n".join(parts)
134+
"""Format command result as an observation string for the agent loop."""
135+
parts = []
136+
if cmd_result["stdout"].strip():
137+
parts.append(f"STDOUT:\n{cmd_result['stdout'].strip()}")
138+
if cmd_result["stderr"].strip():
139+
parts.append(f"STDERR:\n{cmd_result['stderr'].strip()}")
140+
parts.append(f"Return code: {cmd_result['returncode']}")
141+
files = list_workspace_files()
142+
if files:
143+
parts.append(f"Workspace files: {', '.join(files)}")
144+
return "\n".join(parts)

0 commit comments

Comments
 (0)