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

Prefetch agents related database models

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2024-10-08 21:59:15 -07:00
parent 8de67c5d4d
commit d628f89ce9
1 changed files with 25 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -556,14 +556,18 @@ class AgentAdapters:
@staticmethod @staticmethod
async def aget_readonly_agent_by_slug(agent_slug: str, user: KhojUser): async def aget_readonly_agent_by_slug(agent_slug: str, user: KhojUser):
return await Agent.objects.filter( return (
await Agent.objects.filter(
(Q(slug__iexact=agent_slug.lower())) (Q(slug__iexact=agent_slug.lower()))
& ( & (
Q(privacy_level=Agent.PrivacyLevel.PUBLIC) Q(privacy_level=Agent.PrivacyLevel.PUBLIC)
| Q(privacy_level=Agent.PrivacyLevel.PROTECTED) | Q(privacy_level=Agent.PrivacyLevel.PROTECTED)
| Q(creator=user) | Q(creator=user)
) )
).afirst() )
.prefetch_related("creator", "chat_model", "fileobject_set")
.afirst()
)
@staticmethod @staticmethod
async def adelete_agent_by_slug(agent_slug: str, user: KhojUser): async def adelete_agent_by_slug(agent_slug: str, user: KhojUser):
@@ -579,15 +583,23 @@ class AgentAdapters:
@staticmethod @staticmethod
async def aget_agent_by_slug(agent_slug: str, user: KhojUser): async def aget_agent_by_slug(agent_slug: str, user: KhojUser):
return await Agent.objects.filter( return (
await Agent.objects.filter(
(Q(slug__iexact=agent_slug.lower())) & (Q(privacy_level=Agent.PrivacyLevel.PUBLIC) | Q(creator=user)) (Q(slug__iexact=agent_slug.lower())) & (Q(privacy_level=Agent.PrivacyLevel.PUBLIC) | Q(creator=user))
).afirst() )
.prefetch_related("creator", "chat_model", "fileobject_set")
.afirst()
)
@staticmethod @staticmethod
async def aget_agent_by_name(agent_name: str, user: KhojUser): async def aget_agent_by_name(agent_name: str, user: KhojUser):
return await Agent.objects.filter( return (
await Agent.objects.filter(
(Q(name__iexact=agent_name.lower())) & (Q(privacy_level=Agent.PrivacyLevel.PUBLIC) | Q(creator=user)) (Q(name__iexact=agent_name.lower())) & (Q(privacy_level=Agent.PrivacyLevel.PUBLIC) | Q(creator=user))
).afirst() )
.prefetch_related("creator", "chat_model", "fileobject_set")
.afirst()
)
@staticmethod @staticmethod
def get_agent_by_slug(slug: str, user: KhojUser = None): def get_agent_by_slug(slug: str, user: KhojUser = None):