Spaces:
Sleeping
Sleeping
Commit ·
d289b48
1
Parent(s): 061383a
feat(environment): add set_llm_pipeline() injection method for Person 3
Browse files- server/environment.py +4 -0
server/environment.py
CHANGED
|
@@ -27,3 +27,7 @@ class RedTeamEnvironment:
|
|
| 27 |
def set_reward_computer(self, compute_fn: Callable) -> None:
|
| 28 |
self.reward_computer = compute_fn
|
| 29 |
logger.info("Reward computer wired up (Person 2)")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
def set_reward_computer(self, compute_fn: Callable) -> None:
|
| 28 |
self.reward_computer = compute_fn
|
| 29 |
logger.info("Reward computer wired up (Person 2)")
|
| 30 |
+
|
| 31 |
+
def set_llm_pipeline(self, pipeline_fn: Callable) -> None:
|
| 32 |
+
self.llm_pipeline = pipeline_fn
|
| 33 |
+
logger.info("LLM pipeline wired up (Person 3)")
|