klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 21:29:13 +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

38
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 (
(Q(slug__iexact=agent_slug.lower())) await Agent.objects.filter(
& ( (Q(slug__iexact=agent_slug.lower()))
Q(privacy_level=Agent.PrivacyLevel.PUBLIC) & (
| Q(privacy_level=Agent.PrivacyLevel.PROTECTED) Q(privacy_level=Agent.PrivacyLevel.PUBLIC)
| Q(creator=user) | Q(privacy_level=Agent.PrivacyLevel.PROTECTED)
| 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 (
(Q(slug__iexact=agent_slug.lower())) & (Q(privacy_level=Agent.PrivacyLevel.PUBLIC) | Q(creator=user)) await Agent.objects.filter(
).afirst() (Q(slug__iexact=agent_slug.lower())) & (Q(privacy_level=Agent.PrivacyLevel.PUBLIC) | Q(creator=user))
)
.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 (
(Q(name__iexact=agent_name.lower())) & (Q(privacy_level=Agent.PrivacyLevel.PUBLIC) | Q(creator=user)) await Agent.objects.filter(
).afirst() (Q(name__iexact=agent_name.lower())) & (Q(privacy_level=Agent.PrivacyLevel.PUBLIC) | Q(creator=user))
)
.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):