klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-08 05:39:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle file retrieval when agent is None

Browse Source
This commit is contained in:
sabaimran
2024-11-04 16:55:22 -08:00
parent e3ca52b7cb
commit b6145df3be
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/khoj/database/adapters/__init__.py
View File

@@ -1477,6 +1477,8 @@ class EntryAdapters:
@staticmethod @staticmethod
async def aagent_has_entries(agent: Agent): async def aagent_has_entries(agent: Agent):
if agent is None:
return False
return await Entry.objects.filter(agent=agent).aexists() return await Entry.objects.filter(agent=agent).aexists()
@staticmethod @staticmethod
@@ -1495,6 +1497,8 @@ class EntryAdapters:
@staticmethod @staticmethod
async def aget_agent_entry_filepaths(agent: Agent): async def aget_agent_entry_filepaths(agent: Agent):
if agent is None:
return []
return await sync_to_async(set)( return await sync_to_async(set)(
Entry.objects.filter(agent=agent).distinct("file_path").values_list("file_path", flat=True) Entry.objects.filter(agent=agent).distinct("file_path").values_list("file_path", flat=True)
) )