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>
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
from dataclasses import dataclass
|
|
|
|
from openhands.app_server.sandbox.sandbox_spec_models import (
|
|
SandboxSpecInfo,
|
|
SandboxSpecInfoPage,
|
|
)
|
|
from openhands.app_server.sandbox.sandbox_spec_service import (
|
|
SandboxSpecService,
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class PresetSandboxSpecService(SandboxSpecService):
|
|
"""Service which uses a preset set of sandbox specs."""
|
|
|
|
specs: list[SandboxSpecInfo]
|
|
|
|
async def search_sandbox_specs(
|
|
self, page_id: str | None = None, limit: int = 100
|
|
) -> SandboxSpecInfoPage:
|
|
"""Search for sandbox specs with pagination support."""
|
|
# Apply pagination
|
|
start_idx = 0
|
|
if page_id:
|
|
try:
|
|
start_idx = int(page_id)
|
|
except ValueError:
|
|
start_idx = 0
|
|
|
|
end_idx = start_idx + limit
|
|
paginated_specs = self.specs[start_idx:end_idx]
|
|
|
|
# Determine next page ID
|
|
next_page_id = None
|
|
if end_idx < len(self.specs):
|
|
next_page_id = str(end_idx)
|
|
|
|
return SandboxSpecInfoPage(items=paginated_specs, next_page_id=next_page_id)
|
|
|
|
async def get_sandbox_spec(self, sandbox_spec_id: str) -> SandboxSpecInfo | None:
|
|
"""Get a single sandbox spec by ID, returning None if not found."""
|
|
for spec in self.specs:
|
|
if spec.id == sandbox_spec_id:
|
|
return spec
|
|
return None
|
|
|
|
async def get_default_sandbox_spec(self) -> SandboxSpecInfo:
|
|
return self.specs[0]
|