mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-25 21:36:52 +08:00
09af93a02a
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Engel Nyst <engel.nyst@gmail.com>
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
from typing import AsyncGenerator
|
|
|
|
from fastapi import Request
|
|
from pydantic import Field
|
|
|
|
from openhands.app_server.sandbox.preset_sandbox_spec_service import (
|
|
PresetSandboxSpecService,
|
|
)
|
|
from openhands.app_server.sandbox.sandbox_spec_models import (
|
|
SandboxSpecInfo,
|
|
)
|
|
from openhands.app_server.sandbox.sandbox_spec_service import (
|
|
SandboxSpecService,
|
|
SandboxSpecServiceInjector,
|
|
get_agent_server_env,
|
|
get_agent_server_image,
|
|
)
|
|
from openhands.app_server.services.injector import InjectorState
|
|
|
|
|
|
def get_default_sandbox_specs():
|
|
return [
|
|
SandboxSpecInfo(
|
|
id=get_agent_server_image(),
|
|
command=['/usr/local/bin/openhands-agent-server', '--port', '60000'],
|
|
initial_env={
|
|
'OPENVSCODE_SERVER_ROOT': '/openhands/.openvscode-server',
|
|
'LOG_JSON': 'true',
|
|
'OH_ENABLE_VNC': '0',
|
|
'OH_CONVERSATIONS_PATH': '/workspace/conversations',
|
|
'OH_BASH_EVENTS_DIR': '/workspace/bash_events',
|
|
'OH_VSCODE_PORT': '60001',
|
|
**get_agent_server_env(),
|
|
},
|
|
working_dir='/workspace/project',
|
|
)
|
|
]
|
|
|
|
|
|
class RemoteSandboxSpecServiceInjector(SandboxSpecServiceInjector):
|
|
specs: list[SandboxSpecInfo] = Field(
|
|
default_factory=get_default_sandbox_specs,
|
|
description='Preset list of sandbox specs',
|
|
)
|
|
|
|
async def inject(
|
|
self, state: InjectorState, request: Request | None = None
|
|
) -> AsyncGenerator[SandboxSpecService, None]:
|
|
yield PresetSandboxSpecService(self.specs)
|