diff --git a/nanocode.py b/nanocode.py index 9f79ed6..1227e8c 100644 --- a/nanocode.py +++ b/nanocode.py @@ -223,13 +223,13 @@ class RLMCodingProgram(PrecompiledProgram): self.agent.set_lm(self.lm) self.set_lm(self.lm) 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}") def forward(self, task: str) -> str: if not task: return dspy.Prediction(answer="No Task Given.") - + print(f"{YELLOW}DEBUG forward() - agent.get_lm(): {self.agent.get_lm()}{RESET}") print(f"{YELLOW}DEBUG forward() - self.lm.model: {self.lm.model}{RESET}") print(f"{YELLOW}DEBUG forward() - agent.sub_lm.model: {self.agent.sub_lm.model}{RESET}")