main
py 14 lines 495 Bytes
Raw
1 from helpers.tool import Tool, Response
2 from plugins._memory.helpers.memory import Memory
3
4
5
6 class MemoryDelete(Tool):
7
8 async def execute(self, ids="", **kwargs):
9 db = await Memory.get(self.agent)
10 ids = [id.strip() for id in ids.split(",") if id.strip()]
11 dels = await db.delete_documents_by_ids(ids=ids, cascade=True)
12
13 result = self.agent.read_prompt("fw.memories_deleted.md", memory_count=len(dels))
14 return Response(message=result, break_loop=False)