Fix: inherit from openenv Environment base class to resolve reset_async/concurrency errors

server/pm_ops_environment.py

@@ -1,9 +1,9 @@
 import copy
 import random
-import uuid
 from typing import Any, Dict, Optional
 
-from models import PMOpsAction, PMOpsObservation
+from openenv.core.env_server.interfaces import Environment
+from models import PMOpsAction, PMOpsObservation, PMOpsState
 from server.apps.ticketing import TicketingApp
 from server.apps.codebase import CodebaseApp
 from server.apps.chat import ChatApp
@@ -34,10 +34,11 @@ def _oracle_check(scenario: Dict[str, Any]) -> bool:
     return bool(channel) and (bool(team) or scenario["type"] == "release_notes")
 
 
-class PMOpsEnvironment:
+class PMOpsEnvironment(Environment[PMOpsAction, PMOpsObservation, PMOpsState]):
     SUPPORTS_CONCURRENT_SESSIONS = True
 
     def __init__(self):
+        super().__init__()
         self._ticketing: Optional[TicketingApp] = None
         self._codebase: Optional[CodebaseApp] = None
         self._chat: Optional[ChatApp] = None
@@ -46,8 +47,9 @@ class PMOpsEnvironment:
         self._step_count: int = 0
         self._done: bool = False
 
-    def reset(self) -> PMOpsObservation:
-        seed = random.randint(0, 2 ** 31)
+    def reset(self, seed: Optional[int] = None, episode_id: Optional[str] = None, **kwargs) -> PMOpsObservation:
+        if seed is None:
+            seed = random.randint(0, 2 ** 31)
         rng = random.Random(seed)
         difficulty = rng.choice(_DIFFICULTY_POOL)
         task_type = rng.choice(_TASK_TYPES)
@@ -84,7 +86,7 @@ class PMOpsEnvironment:
             done=False,
         )
 
-    def step(self, action: PMOpsAction) -> PMOpsObservation:
+    def step(self, action: PMOpsAction, timeout_s: Optional[float] = None, **kwargs) -> PMOpsObservation:
         if self._ticketing is None:
             raise RuntimeError("Call reset() before step()")
 
@@ -129,6 +131,18 @@ class PMOpsEnvironment:
             done=done,
         )
 
+    @property
+    def state(self) -> PMOpsState:
+        return PMOpsState(
+            org_config=self._org_config or {},
+            task_config=self._scenario or {},
+            ticketing=self._ticketing.snapshot() if self._ticketing else {},
+            chat=self._chat.snapshot() if self._chat else {},
+            codebase={},
+            step_count=self._step_count,
+            finished=self._done,
+        )
+
     def _dispatch(self, action: PMOpsAction) -> Dict[str, Any]:
         at = action.action_type
         args = action.args or {}
@@ -156,12 +170,12 @@ class PMOpsEnvironment:
         if at.startswith("ticketing."):
             op = at.split(".", 1)[1]
             handlers = {
-                "create_ticket":  self._ticketing.create_ticket,
-                "update_ticket":  self._ticketing.update_ticket,
-                "get_ticket":     self._ticketing.get_ticket,
-                "list_tickets":   self._ticketing.list_tickets,
-                "assign_ticket":  self._ticketing.assign_ticket,
-                "comment_ticket": self._ticketing.comment_ticket,
+                "create_ticket":     self._ticketing.create_ticket,
+                "update_ticket":     self._ticketing.update_ticket,
+                "get_ticket":        self._ticketing.get_ticket,
+                "list_tickets":      self._ticketing.list_tickets,
+                "assign_ticket":     self._ticketing.assign_ticket,
+                "comment_ticket":    self._ticketing.comment_ticket,
                 "transition_ticket": self._ticketing.transition_ticket,
             }
             if op not in handlers:
@@ -182,10 +196,10 @@ class PMOpsEnvironment:
         if at.startswith("chat."):
             op = at.split(".", 1)[1]
             handlers = {
-                "post_message": self._chat.post_message,
-                "read_channel": self._chat.read_channel,
+                "post_message":  self._chat.post_message,
+                "read_channel":  self._chat.read_channel,
                 "list_channels": self._chat.list_channels,
-                "search":       self._chat.search,
+                "search":        self._chat.search,
             }
             if op not in handlers:
                 return {"ok": False, "error": f"Unknown chat action: {op}"}