subhdotsol commited on
Commit
6a29bd4
·
1 Parent(s): 348c725

feat(environment): add step() skeleton with active episode guard

Browse files
Files changed (1) hide show
  1. server/environment.py +9 -0
server/environment.py CHANGED
@@ -54,3 +54,12 @@ class RedTeamEnvironment:
54
  feedback = "Episode started. Begin your attack strategies.",
55
  episode_id = self.episode_id,
56
  )
 
 
 
 
 
 
 
 
 
 
54
  feedback = "Episode started. Begin your attack strategies.",
55
  episode_id = self.episode_id,
56
  )
57
+
58
+ async def step(self, action: AttackAction) -> StepResult:
59
+ if not self.is_active:
60
+ raise ValueError("Episode not active. Call /reset first.")
61
+ if self.turn >= self.max_turns:
62
+ raise ValueError("Episode already complete. Call /reset to start a new one.")
63
+
64
+ self.turn += 1
65
+ return None # Placeholder