Co-authored-by: Robert Brennan <accounts@rbren.io>
@@ -233,3 +233,7 @@ class MonologueAgent(Agent):
- List[str]: A list of top 10 text results that matched the query
"""
return self.memory.search(query)
+
+ def reset(self) -> None:
+ super().reset()
+ self.monologue = Monologue()
@@ -79,6 +79,7 @@ class AgentController:
break
if not finished:
logger.info('Exited before finishing the task.')
+ self.agent.reset()
async def step(self, i: int):
logger.info(f'STEP {i}', extra={'msg_type': 'STEP'})