Add tool logs
This commit is contained in:
@@ -298,9 +298,6 @@ class RLMCodingProgram(PrecompiledProgram):
|
|||||||
agent.set_lm(self.lm)
|
agent.set_lm(self.lm)
|
||||||
self.agent = agent
|
self.agent = agent
|
||||||
|
|
||||||
print(f"Using model: {self.agent.get_lm().model}")
|
|
||||||
print(f"Using sub-model: {self.agent.sub_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)
|
||||||
@@ -569,5 +566,5 @@ def main():
|
|||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
agent = RLMCodingProgram(RLMCodingConfig())
|
agent = RLMCodingProgram(RLMCodingConfig())
|
||||||
agent.push_to_hub(MODAIC_REPO_PATH, commit_message="debug", branch="prod")
|
agent.push_to_hub(MODAIC_REPO_PATH, commit_message="Add tool logs", branch="prod")
|
||||||
#main()
|
#main()
|
||||||
|
|||||||
Reference in New Issue
Block a user