Skip to content

Commit 5c2c8ab

Browse files
rycerzesswapnil
authored andcommitted
feat: serialization utilities and route config
1 parent a9038dc commit 5c2c8ab

File tree

6 files changed

+403
-286
lines changed

6 files changed

+403
-286
lines changed

src/core/env_server/__init__.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,13 @@
99
from .base_transforms import CompositeTransform, NullTransform
1010
from .http_server import HTTPEnvServer, create_app, create_fastapi_app
1111
from .interfaces import Environment, Message, ModelTokenizer, Transform
12-
from .types import Action, Observation, State
12+
from .route_config import GetEndpointConfig
13+
from .serialization import (
14+
deserialize_action,
15+
deserialize_action_with_preprocessing,
16+
serialize_observation,
17+
)
18+
from .types import Action, Observation, State, SchemaResponse
1319
from .web_interface import create_web_interface_app, WebInterfaceManager
1420

1521
__all__ = [
@@ -22,6 +28,7 @@
2228
"Action",
2329
"Observation",
2430
"State",
31+
"SchemaResponse",
2532
# Base transforms
2633
"CompositeTransform",
2734
"NullTransform",
@@ -32,4 +39,10 @@
3239
# Web Interface
3340
"create_web_interface_app",
3441
"WebInterfaceManager",
42+
# Serialization utilities
43+
"deserialize_action",
44+
"deserialize_action_with_preprocessing",
45+
"serialize_observation",
46+
# Route configuration
47+
"GetEndpointConfig",
3548
]

0 commit comments

Comments
 (0)