From ad716ca58d0e91cf49f1a41c288b7c016ba26607 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Tue, 8 Oct 2024 15:00:21 -0700 Subject: [PATCH] Delete associated entries with an agent when it is deleted --- src/khoj/database/adapters/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/khoj/database/adapters/__init__.py b/src/khoj/database/adapters/__init__.py index 9485005a..a04d092c 100644 --- a/src/khoj/database/adapters/__init__.py +++ b/src/khoj/database/adapters/__init__.py @@ -568,6 +568,10 @@ class AgentAdapters: @staticmethod async def adelete_agent_by_slug(agent_slug: str, user: KhojUser): agent = await AgentAdapters.aget_agent_by_slug(agent_slug, user) + + async for entry in Entry.objects.filter(agent=agent).aiterator(): + await entry.adelete() + if agent: await agent.adelete() return True