Aswini-Kumar commited on
Commit
8aedea9
·
verified ·
1 Parent(s): 7bff66e

feat: server/env.py - OpenEnv 5-step structure

Browse files
Files changed (1) hide show
  1. 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