|
from openhands.core.logger import openhands_logger as logger
|
|
from openhands.runtime.impl.docker.docker_runtime import (
|
|
DockerRuntime,
|
|
)
|
|
from openhands.runtime.impl.e2b.sandbox import E2BBox
|
|
from openhands.runtime.impl.modal.modal_runtime import ModalRuntime
|
|
from openhands.runtime.impl.remote.remote_runtime import RemoteRuntime
|
|
from openhands.runtime.impl.runloop.runloop_runtime import RunloopRuntime
|
|
|
|
|
|
def get_runtime_cls(name: str):
|
|
|
|
if name == 'eventstream' or name == 'docker':
|
|
return DockerRuntime
|
|
elif name == 'e2b':
|
|
return E2BBox
|
|
elif name == 'remote':
|
|
return RemoteRuntime
|
|
elif name == 'modal':
|
|
logger.debug('Using ModalRuntime')
|
|
return ModalRuntime
|
|
elif name == 'runloop':
|
|
return RunloopRuntime
|
|
else:
|
|
raise ValueError(f'Runtime {name} not supported')
|
|
|
|
|
|
__all__ = [
|
|
'E2BBox',
|
|
'RemoteRuntime',
|
|
'ModalRuntime',
|
|
'RunloopRuntime',
|
|
'DockerRuntime',
|
|
'get_runtime_cls',
|
|
]
|
|
|
