subhdotsol commited on
Commit
d289b48
·
1 Parent(s): 061383a

feat(environment): add set_llm_pipeline() injection method for Person 3

Browse files
Files changed (1) hide show
  1. 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)")