Spaces:
Sleeping
Sleeping
Commit ·
6a29bd4
1
Parent(s): 348c725
feat(environment): add step() skeleton with active episode guard
Browse files- 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
|