Add reload_lms method and debug forward()

This commit is contained in:
2026-01-24 00:41:45 -08:00
parent fa1a5847cf
commit 2cd7286320

View File

@@ -223,7 +223,7 @@ class RLMCodingProgram(PrecompiledProgram):
self.agent.set_lm(self.lm) self.agent.set_lm(self.lm)
self.set_lm(self.lm) self.set_lm(self.lm)
print(f"{BLUE}CONFIG WAS SET: {self.config}{RESET}") print(f"{BLUE}CONFIG WAS SET: {self.config}{RESET}")
print(f"{BLUE}LM WAS SET: {self.agent.get_lm().model}{RESET}") print(f"{BLUE}LM WAS SET: {self.agent.get_lm()}{RESET}")
print(f"{BLUE}SUB LM WAS SET: {self.agent.sub_lm.model}{RESET}") print(f"{BLUE}SUB LM WAS SET: {self.agent.sub_lm.model}{RESET}")
def forward(self, task: str) -> str: def forward(self, task: str) -> str: