feat: add NotSupportedOperationError and update E2BEnvironment to raise it for unsupported command status retrieval

This commit is contained in:
Yeuoly 2025-12-31 18:09:14 +08:00
parent 41565e91ed
commit 62e422f75a
2 changed files with 11 additions and 2 deletions

View File

@ -8,3 +8,9 @@ class VirtualEnvironmentLaunchFailedError(Exception):
"""Exception raised when launching the virtual environment fails."""
pass
class NotSupportedOperationError(Exception):
"""Exception raised when an operation is not supported."""
pass

View File

@ -10,7 +10,7 @@ from uuid import uuid4
from e2b_code_interpreter import Sandbox
from core.virtual_environment.__base.entities import Arch, CommandStatus, ConnectionHandle, FileState, Metadata
from core.virtual_environment.__base.exec import ArchNotSupportedError
from core.virtual_environment.__base.exec import ArchNotSupportedError, NotSupportedOperationError
from core.virtual_environment.__base.virtual_environment import VirtualEnvironment
from core.virtual_environment.channel.queue_transport import QueueTransportReadCloser
from core.virtual_environment.channel.transport import (
@ -180,7 +180,10 @@ class E2BEnvironment(VirtualEnvironment):
)
def get_command_status(self, connection_handle: ConnectionHandle, pid: str) -> CommandStatus:
return super().get_command_status(connection_handle, pid)
"""
Nop, E2B does not support getting command status yet.
"""
raise NotSupportedOperationError("E2B does not support getting command status yet.")
def _cmd_thread(
self,