Merge pull request #7 from facebookexternal/env_code

fix imports

Author: pankit-eng

src/core/__init__.py

@@ -7,7 +7,7 @@
 """Core components for agentic environments."""
 
 # Re-export main components from submodules for convenience
-from .env import *
+from .env_server import *
 from .docker import *
 
 # Note: MCP module doesn't export anything yet

src/core/env_server/__init__.py

@@ -8,28 +8,16 @@
 
 from .base_transforms import CompositeTransform, NullTransform
 from .interfaces import Environment, Transform
-from .types import (
-    Action,
-    CodeAction,
-    CodeObservation,
-    CodeState,
-    ExecutionResult,
-    Observation,
-    State,
-)
+from .types import Action, Observation, State
 
 __all__ = [
     # Core interfaces
     "Environment",
     "Transform",
     # Types
     "Action",
-    "CodeAction",
     "Observation",
-    "CodeObservation",
     "State",
-    "CodeState",
-    "ExecutionResult",
     # Base transforms
     "CompositeTransform",
     "NullTransform",