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>
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
import asyncio
|
|
from abc import ABC, abstractmethod
|
|
|
|
from openhands.app_server.errors import SandboxError
|
|
from openhands.app_server.sandbox.sandbox_spec_models import (
|
|
SandboxSpecInfo,
|
|
SandboxSpecInfoPage,
|
|
)
|
|
from openhands.app_server.services.injector import Injector
|
|
from openhands.sdk.utils.models import DiscriminatedUnionMixin
|
|
|
|
# The version of the agent server to use for deployments.
|
|
# Typically this will be the same as the values from the pyproject.toml
|
|
AGENT_SERVER_VERSION = 'f8ca02c4a3b847bfc50b3c5e579ce126c511fefc'
|
|
|
|
|
|
class SandboxSpecService(ABC):
|
|
"""Service for managing Sandbox specs.
|
|
|
|
At present this is read only. The plan is that later this class will allow building
|
|
and deleting sandbox specs and limiting access by user and group. It would also be
|
|
nice to be able to set the desired number of warm sandboxes for a spec and scale
|
|
this up and down.
|
|
"""
|
|
|
|
@abstractmethod
|
|
async def search_sandbox_specs(
|
|
self, page_id: str | None = None, limit: int = 100
|
|
) -> SandboxSpecInfoPage:
|
|
"""Search for sandbox specs."""
|
|
|
|
@abstractmethod
|
|
async def get_sandbox_spec(self, sandbox_spec_id: str) -> SandboxSpecInfo | None:
|
|
"""Get a single sandbox spec, returning None if not found."""
|
|
|
|
async def get_default_sandbox_spec(self) -> SandboxSpecInfo:
|
|
"""Get the default sandbox spec."""
|
|
page = await self.search_sandbox_specs()
|
|
if not page.items:
|
|
raise SandboxError('No sandbox specs available!')
|
|
return page.items[0]
|
|
|
|
async def batch_get_sandbox_specs(
|
|
self, sandbox_spec_ids: list[str]
|
|
) -> list[SandboxSpecInfo | None]:
|
|
"""Get a batch of sandbox specs, returning None for any not found."""
|
|
results = await asyncio.gather(
|
|
*[
|
|
self.get_sandbox_spec(sandbox_spec_id)
|
|
for sandbox_spec_id in sandbox_spec_ids
|
|
]
|
|
)
|
|
return results
|
|
|
|
|
|
class SandboxSpecServiceInjector(
|
|
DiscriminatedUnionMixin, Injector[SandboxSpecService], ABC
|
|
):
|
|
pass
|