mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-04-29 03:00:45 -04:00
a57a213c7c
Disable Python linting by default, and turn it on for SWE Bench. It is turned off by default since this behavior is weird and somewhat annoying to end users. It is turned on for SWE Bench because linting python files gives LLM a chance to fix the indentations.
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
import json
|
|
import os
|
|
from abc import ABC, abstractmethod
|
|
|
|
from opendevin.core.config import config
|
|
from opendevin.core.schema import CancellableStream
|
|
from opendevin.runtime.docker.process import Process
|
|
from opendevin.runtime.plugins.mixin import PluginMixin
|
|
|
|
|
|
class Sandbox(ABC, PluginMixin):
|
|
background_commands: dict[int, Process] = {}
|
|
_env: dict[str, str] = {}
|
|
|
|
def __init__(self, **kwargs):
|
|
for key in os.environ:
|
|
if key.startswith('SANDBOX_ENV_'):
|
|
sandbox_key = key.removeprefix('SANDBOX_ENV_')
|
|
self.add_to_env(sandbox_key, os.environ[key])
|
|
if config.enable_auto_lint:
|
|
self.add_to_env('ENABLE_AUTO_LINT', 'true')
|
|
|
|
def add_to_env(self, key: str, value: str):
|
|
self._env[key] = value
|
|
# Note: json.dumps gives us nice escaping for free
|
|
self.execute(f'export {key}={json.dumps(value)}')
|
|
|
|
@abstractmethod
|
|
def execute(
|
|
self, cmd: str, stream: bool = False, timeout: int | None = None
|
|
) -> tuple[int, str | CancellableStream]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def execute_in_background(self, cmd: str) -> Process:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def kill_background(self, id: int) -> Process:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def read_logs(self, id: int) -> str:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def close(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def copy_to(self, host_src: str, sandbox_dest: str, recursive: bool = False):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_working_directory(self):
|
|
pass
|