Add reload_lms method and debug forward()
This commit is contained in:
@@ -223,13 +223,13 @@ 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:
|
||||||
if not task:
|
if not task:
|
||||||
return dspy.Prediction(answer="No Task Given.")
|
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() - 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() - self.lm.model: {self.lm.model}{RESET}")
|
||||||
print(f"{YELLOW}DEBUG forward() - agent.sub_lm.model: {self.agent.sub_lm.model}{RESET}")
|
print(f"{YELLOW}DEBUG forward() - agent.sub_lm.model: {self.agent.sub_lm.model}{RESET}")
|
||||||
|
|||||||
Reference in New Issue
Block a user