This commit is contained in:
2026-01-22 02:30:55 -08:00
parent 22b7e87aaf
commit fdb81ee671

View File

@@ -323,11 +323,13 @@ class RLMCodingProgram(PrecompiledProgram):
max_iterations=self.config.max_iters, max_iterations=self.config.max_iters,
verbose=self.config.verbose, verbose=self.config.verbose,
) )
agent.set_lm(self.lm) agent.set_lm(self.lm)
print(f"Using model: {self.lm.model}")
print(f"Using sub-model: {self.sub_lm.model}")
self.agent = agent self.agent = agent
print(f"Using model: {self.agent.get_lm().model}")
print(f"Using sub-model: {self.agent.get_lm().model}")
def forward(self, task: str) -> str: def forward(self, task: str) -> str:
assert task, "Task cannot be empty" assert task, "Task cannot be empty"
return self.agent(task=task) return self.agent(task=task)