sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
Show stats 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):