mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
from dataclasses import dataclass
|
|
from typing import ClassVar
|
|
|
|
from openhands.core.schema import ActionType
|
|
from openhands.events.action.action import (
|
|
Action,
|
|
ActionConfirmationStatus,
|
|
ActionSecurityRisk,
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class CmdRunAction(Action):
|
|
command: str
|
|
thought: str = ''
|
|
keep_prompt: bool = True
|
|
# if True, the command prompt will be kept in the command output observation
|
|
# Example of command output:
|
|
# root@sandbox:~# ls
|
|
# file1.txt
|
|
# file2.txt
|
|
# root@sandbox:~# <-- this is the command prompt
|
|
|
|
action: str = ActionType.RUN
|
|
runnable: ClassVar[bool] = True
|
|
is_confirmed: ActionConfirmationStatus = ActionConfirmationStatus.CONFIRMED
|
|
security_risk: ActionSecurityRisk | None = None
|
|
|
|
@property
|
|
def message(self) -> str:
|
|
return f'Running command: {self.command}'
|
|
|
|
def __str__(self) -> str:
|
|
ret = f'**CmdRunAction (source={self.source})**\n'
|
|
if self.thought:
|
|
ret += f'THOUGHT: {self.thought}\n'
|
|
ret += f'COMMAND:\n{self.command}'
|
|
return ret
|
|
|
|
|
|
@dataclass
|
|
class IPythonRunCellAction(Action):
|
|
code: str
|
|
thought: str = ''
|
|
action: str = ActionType.RUN_IPYTHON
|
|
runnable: ClassVar[bool] = True
|
|
is_confirmed: ActionConfirmationStatus = ActionConfirmationStatus.CONFIRMED
|
|
security_risk: ActionSecurityRisk | None = None
|
|
kernel_init_code: str = '' # code to run in the kernel (if the kernel is restarted)
|
|
|
|
def __str__(self) -> str:
|
|
ret = '**IPythonRunCellAction**\n'
|
|
if self.thought:
|
|
ret += f'THOUGHT: {self.thought}\n'
|
|
ret += f'CODE:\n{self.code}'
|
|
return ret
|
|
|
|
@property
|
|
def message(self) -> str:
|
|
return f'Running Python code interactively: {self.code}'
|