From b6145df3be5498c8a0402d4977a250c15591ee32 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Mon, 4 Nov 2024 16:55:22 -0800 Subject: [PATCH] Handle file retrieval when agent is None --- 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 164b4023..fbb00690 100644 --- a/src/khoj/database/adapters/__init__.py +++ b/src/khoj/database/adapters/__init__.py @@ -1477,6 +1477,8 @@ class EntryAdapters: @staticmethod async def aagent_has_entries(agent: Agent): + if agent is None: + return False return await Entry.objects.filter(agent=agent).aexists() @staticmethod @@ -1495,6 +1497,8 @@ class EntryAdapters: @staticmethod async def aget_agent_entry_filepaths(agent: Agent): + if agent is None: + return [] return await sync_to_async(set)( Entry.objects.filter(agent=agent).distinct("file_path").values_list("file_path", flat=True) )