mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-25 21:36:52 +08:00
f292f3a84d
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: sp.wack <83104063+amanape@users.noreply.github.com> Co-authored-by: Engel Nyst <enyst@users.noreply.github.com>
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
"""Runtime Images router for OpenHands Server."""
|
|
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Query
|
|
|
|
from openhands.app_server.config import depends_sandbox_spec_service
|
|
from openhands.app_server.sandbox.sandbox_spec_models import (
|
|
SandboxSpecInfo,
|
|
SandboxSpecInfoPage,
|
|
)
|
|
from openhands.app_server.sandbox.sandbox_spec_service import (
|
|
SandboxSpecService,
|
|
)
|
|
|
|
router = APIRouter(prefix='/sandbox-specs', tags=['Sandbox'])
|
|
sandbox_spec_service_dependency = depends_sandbox_spec_service()
|
|
|
|
|
|
# Read methods
|
|
|
|
|
|
@router.get('/search')
|
|
async def search_sandbox_specs(
|
|
page_id: Annotated[
|
|
str | None,
|
|
Query(title='Optional next_page_id from the previously returned page'),
|
|
] = None,
|
|
limit: Annotated[
|
|
int,
|
|
Query(title='The max number of results in the page', gt=0, lte=100),
|
|
] = 100,
|
|
sandbox_spec_service: SandboxSpecService = sandbox_spec_service_dependency,
|
|
) -> SandboxSpecInfoPage:
|
|
"""Search / List sandbox specs."""
|
|
assert limit > 0
|
|
assert limit <= 100
|
|
return await sandbox_spec_service.search_sandbox_specs(page_id=page_id, limit=limit)
|
|
|
|
|
|
@router.get('')
|
|
async def batch_get_sandbox_specs(
|
|
id: Annotated[list[str], Query()],
|
|
sandbox_spec_service: SandboxSpecService = sandbox_spec_service_dependency,
|
|
) -> list[SandboxSpecInfo | None]:
|
|
"""Get a batch of sandbox specs given their ids, returning null for any missing."""
|
|
assert len(id) <= 100
|
|
sandbox_specs = await sandbox_spec_service.batch_get_sandbox_specs(id)
|
|
return sandbox_specs
|