feat: server/env.py - OpenEnv 5-step structure
Browse files- server/env.py +10 -6
server/env.py
CHANGED
|
@@ -22,8 +22,8 @@ Key enforcement:
|
|
| 22 |
|
| 23 |
from __future__ import annotations
|
| 24 |
|
| 25 |
-
from dataclasses import dataclass, field
|
| 26 |
from typing import Any, Dict, Optional
|
|
|
|
| 27 |
|
| 28 |
from server.task_generator import TaskGenerator
|
| 29 |
from server.session_manager import SessionManager
|
|
@@ -32,6 +32,14 @@ from server.rewards.rubric import ContinuityRubric
|
|
| 32 |
from server.rewards.auxiliary import AuxiliaryRewarder
|
| 33 |
from server.handoff_validator import HandoffValidator
|
| 34 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
# ---------------------------------------------------------------------------
|
| 36 |
# OpenEnv base class — openenv-core package
|
| 37 |
# ---------------------------------------------------------------------------
|
|
@@ -40,14 +48,10 @@ try:
|
|
| 40 |
from openenv.core.env_server.types import State
|
| 41 |
_HAS_OPENENV = True
|
| 42 |
except ImportError:
|
| 43 |
-
# Fallback stub when openenv-core is not installed (local dev / CI)
|
| 44 |
class State: # type: ignore[no-redef]
|
| 45 |
def __init__(self, **kwargs):
|
| 46 |
-
for k, v in kwargs.items():
|
| 47 |
-
setattr(self, k, v)
|
| 48 |
-
|
| 49 |
class _EnvBase: # type: ignore[no-redef]
|
| 50 |
-
"""Stub base: used when openenv-core is not installed."""
|
| 51 |
pass
|
| 52 |
_HAS_OPENENV = False
|
| 53 |
|
|
|
|
| 22 |
|
| 23 |
from __future__ import annotations
|
| 24 |
|
|
|
|
| 25 |
from typing import Any, Dict, Optional
|
| 26 |
+
from uuid import uuid4
|
| 27 |
|
| 28 |
from server.task_generator import TaskGenerator
|
| 29 |
from server.session_manager import SessionManager
|
|
|
|
| 32 |
from server.rewards.auxiliary import AuxiliaryRewarder
|
| 33 |
from server.handoff_validator import HandoffValidator
|
| 34 |
|
| 35 |
+
try:
|
| 36 |
+
from models import ContinuityAction, ContinuityObservation
|
| 37 |
+
except ImportError:
|
| 38 |
+
try:
|
| 39 |
+
from ..models import ContinuityAction, ContinuityObservation
|
| 40 |
+
except ImportError:
|
| 41 |
+
from models import ContinuityAction, ContinuityObservation
|
| 42 |
+
|
| 43 |
# ---------------------------------------------------------------------------
|
| 44 |
# OpenEnv base class — openenv-core package
|
| 45 |
# ---------------------------------------------------------------------------
|
|
|
|
| 48 |
from openenv.core.env_server.types import State
|
| 49 |
_HAS_OPENENV = True
|
| 50 |
except ImportError:
|
|
|
|
| 51 |
class State: # type: ignore[no-redef]
|
| 52 |
def __init__(self, **kwargs):
|
| 53 |
+
for k, v in kwargs.items(): setattr(self, k, v)
|
|
|
|
|
|
|
| 54 |
class _EnvBase: # type: ignore[no-redef]
|
|
|
|
| 55 |
pass
|
| 56 |
_HAS_OPENENV = False
|
| 57 |
|