openenv/echo_env

1# SPDX-License-Identifier: BSD-3-Clause2 3"""Core environment interfaces and types."""4 5from .base_transforms import CompositeTransform, NullTransform6from .exceptions import (7    ConcurrencyConfigurationError,8    EnvironmentFactoryError,9    OpenEnvError,10    SessionCapacityError,11    SessionCreationError,12    SessionNotFoundError,13)14from .http_server import create_app, create_fastapi_app, HTTPEnvServer15from .interfaces import Environment, Message, ModelTokenizer, TaskProvider, Transform16 17try:18    from .mcp_environment import MCPEnvironment19except ModuleNotFoundError:20    MCPEnvironment = None  # type: ignore[assignment]21 22from .mcp_types import (23    CallToolAction,24    CallToolObservation,25    JsonRpcError,26    # JSON-RPC types27    JsonRpcErrorCode,28    JsonRpcRequest,29    JsonRpcResponse,30    ListToolsAction,31    ListToolsObservation,32    McpMethod,33    RESERVED_TOOL_NAMES,34    Tool,35    ToolError,36    ToolErrorType,37    WSMCPMessage,38    WSMCPResponse,39)40from .route_config import GetEndpointConfig41from .serialization import (42    deserialize_action,43    deserialize_action_with_preprocessing,44    serialize_observation,45)46from .types import (47    Action,48    BaseMessage,49    ConcurrencyConfig,50    GetTaskRangeRequest,51    GetTaskRequest,52    HealthResponse,53    HealthStatus,54    ListTasksRequest,55    NumTasksRequest,56    Observation,57    SchemaResponse,58    ServerCapacityStatus,59    ServerMode,60    SessionInfo,61    State,62    WSCloseMessage,63    WSErrorCode,64    WSErrorResponse,65    WSIncomingMessage,66    WSObservationResponse,67    WSResetMessage,68    WSStateMessage,69    WSStateResponse,70    WSStepMessage,71)72 73try:74    from .web_interface import create_web_interface_app, WebInterfaceManager75except ModuleNotFoundError:76    create_web_interface_app = None  # type: ignore[assignment]77    WebInterfaceManager = None  # type: ignore[assignment]78 79__all__ = [80    # Core interfaces81    "Environment",82    "Transform",83    "TaskProvider",84    "Message",85    "ModelTokenizer",86    # Types87    "Action",88    "Observation",89    "State",90    "SchemaResponse",91    "HealthResponse",92    "ListTasksRequest",93    "NumTasksRequest",94    "GetTaskRequest",95    "GetTaskRangeRequest",96    # Enums97    "HealthStatus",98    "ServerMode",99    "WSErrorCode",100    # WebSocket message types101    "BaseMessage",102    "WSIncomingMessage",103    "WSResetMessage",104    "WSStepMessage",105    "WSStateMessage",106    "WSCloseMessage",107    "WSObservationResponse",108    "WSStateResponse",109    "WSErrorResponse",110    # Concurrency types111    "ConcurrencyConfig",112    "ServerCapacityStatus",113    "SessionInfo",114    # Exceptions115    "OpenEnvError",116    "ConcurrencyConfigurationError",117    "SessionCapacityError",118    "SessionNotFoundError",119    "SessionCreationError",120    "EnvironmentFactoryError",121    # Base transforms122    "CompositeTransform",123    "NullTransform",124    # HTTP Server125    "HTTPEnvServer",126    "create_app",127    "create_fastapi_app",128    # Web Interface129    "create_web_interface_app",130    "WebInterfaceManager",131    # Serialization utilities132    "deserialize_action",133    "deserialize_action_with_preprocessing",134    "serialize_observation",135    # Route configuration136    "GetEndpointConfig",137    # MCP types138    "Tool",139    "ToolError",140    "ToolErrorType",141    "ListToolsAction",142    "CallToolAction",143    "ListToolsObservation",144    "CallToolObservation",145    "WSMCPMessage",146    "WSMCPResponse",147    "RESERVED_TOOL_NAMES",148    "MCPEnvironment",149    # JSON-RPC types150    "JsonRpcErrorCode",151    "JsonRpcError",152    "JsonRpcRequest",153    "JsonRpcResponse",154    "McpMethod",155]156