subhdotsol commited on
Commit
8fa387f
·
1 Parent(s): 6a29bd4

feat(environment): add conversation history builder from last 5 attack turns

Browse files
Files changed (1) hide show
  1. server/environment.py +6 -0
server/environment.py CHANGED
@@ -62,4 +62,10 @@ class RedTeamEnvironment:
62
  raise ValueError("Episode already complete. Call /reset to start a new one.")
63
 
64
  self.turn += 1
 
 
 
 
 
 
65
  return None # Placeholder
 
62
  raise ValueError("Episode already complete. Call /reset to start a new one.")
63
 
64
  self.turn += 1
65
+
66
+ # Last 5 turns as conversation context for the defender
67
+ conversation_history = [
68
+ {"role": "user", "content": h["framing"]}
69
+ for h in self.attack_history[-5:]
70
+ ]
71
  return None # Placeholder