File size: 845 Bytes
301a0ac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from python.helpers.extension import Extension
from agent import LoopData
from python.extensions.message_loop_prompts_after._50_recall_memories import DATA_NAME_TASK as DATA_NAME_TASK_MEMORIES
from python.extensions.message_loop_prompts_after._51_recall_solutions import DATA_NAME_TASK as DATA_NAME_TASK_SOLUTIONS
class RecallWait(Extension):
async def execute(self, loop_data: LoopData = LoopData(), **kwargs):
task = self.agent.get_data(DATA_NAME_TASK_MEMORIES)
if task and not task.done():
# self.agent.context.log.set_progress("Recalling memories...")
await task
task = self.agent.get_data(DATA_NAME_TASK_SOLUTIONS)
if task and not task.done():
# self.agent.context.log.set_progress("Recalling solutions...")
await task
|