mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-04-29 03:00:45 -04:00
4db4a84e2e
* simply jupyter execution via heredoc * make sure /tmp always exists * add integration test for jupyter exec
88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
from opendevin.events.action import (
|
|
AgentRecallAction,
|
|
BrowseURLAction,
|
|
CmdKillAction,
|
|
CmdRunAction,
|
|
FileReadAction,
|
|
FileWriteAction,
|
|
IPythonRunCellAction,
|
|
)
|
|
from opendevin.events.observation import (
|
|
CmdOutputObservation,
|
|
ErrorObservation,
|
|
IPythonRunCellObservation,
|
|
NullObservation,
|
|
Observation,
|
|
)
|
|
from opendevin.runtime.runtime import Runtime
|
|
|
|
from .browse import browse
|
|
from .files import read_file, write_file
|
|
|
|
|
|
class ServerRuntime(Runtime):
|
|
async def run(self, action: CmdRunAction) -> Observation:
|
|
return self._run_command(action.command, background=action.background)
|
|
|
|
async def kill(self, action: CmdKillAction) -> Observation:
|
|
cmd = self.sandbox.kill_background(action.id)
|
|
return CmdOutputObservation(
|
|
content=f'Background command with id {action.id} has been killed.',
|
|
command_id=action.id,
|
|
command=cmd.command,
|
|
exit_code=0,
|
|
)
|
|
|
|
async def run_ipython(self, action: IPythonRunCellAction) -> Observation:
|
|
obs = self._run_command(
|
|
('cat > /tmp/opendevin_jupyter_temp.py <<EOL\n' f'{action.code}\n' 'EOL'),
|
|
background=False,
|
|
)
|
|
|
|
# run the code
|
|
obs = self._run_command(
|
|
('cat /tmp/opendevin_jupyter_temp.py | execute_cli'), background=False
|
|
)
|
|
return IPythonRunCellObservation(content=obs.content, code=action.code)
|
|
|
|
async def read(self, action: FileReadAction) -> Observation:
|
|
working_dir = self.sandbox.get_working_directory()
|
|
return await read_file(action.path, working_dir, action.start, action.end)
|
|
|
|
async def write(self, action: FileWriteAction) -> Observation:
|
|
working_dir = self.sandbox.get_working_directory()
|
|
return await write_file(
|
|
action.path, working_dir, action.content, action.start, action.end
|
|
)
|
|
|
|
async def browse(self, action: BrowseURLAction) -> Observation:
|
|
return await browse(action, self.browser)
|
|
|
|
async def recall(self, action: AgentRecallAction) -> Observation:
|
|
return NullObservation('')
|
|
|
|
def _run_command(self, command: str, background=False) -> Observation:
|
|
if background:
|
|
return self._run_background(command)
|
|
else:
|
|
return self._run_immediately(command)
|
|
|
|
def _run_immediately(self, command: str) -> Observation:
|
|
try:
|
|
exit_code, output = self.sandbox.execute(command)
|
|
return CmdOutputObservation(
|
|
command_id=-1, content=output, command=command, exit_code=exit_code
|
|
)
|
|
except UnicodeDecodeError:
|
|
return ErrorObservation('Command output could not be decoded as utf-8')
|
|
|
|
def _run_background(self, command: str) -> Observation:
|
|
bg_cmd = self.sandbox.execute_in_background(command)
|
|
content = f'Background command started. To stop it, send a `kill` action with id {bg_cmd.pid}'
|
|
return CmdOutputObservation(
|
|
content=content,
|
|
command_id=bg_cmd.pid,
|
|
command=command,
|
|
exit_code=0,
|
|
)
|