Spaces:
Sleeping
Sleeping
Commit ·
8fa387f
1
Parent(s): 6a29bd4
feat(environment): add conversation history builder from last 5 attack turns
Browse files- 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
|