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

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

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