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

Simplify get_conversation_by_user DB adapter code

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2024-07-22 20:34:30 +05:30
parent daec439d52
commit b224d7ffad
1 changed files with 8 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -680,34 +680,18 @@ class ConversationAdapters:
async def aget_conversation_by_user( async def aget_conversation_by_user(
user: KhojUser, client_application: ClientApplication = None, conversation_id: int = None, title: str = None user: KhojUser, client_application: ClientApplication = None, conversation_id: int = None, title: str = None
) -> Optional[Conversation]: ) -> Optional[Conversation]:
query = Conversation.objects.filter(user=user, client=client_application).prefetch_related("agent")
if conversation_id: if conversation_id:
return ( return await query.filter(id=conversation_id).afirst()
await Conversation.objects.filter(user=user, client=client_application, id=conversation_id)
.prefetch_related("agent")
.afirst()
)
elif title: elif title:
return ( return await query.filter(title=title).afirst()
await Conversation.objects.filter(user=user, client=client_application, title=title)
.prefetch_related("agent")
.afirst()
)
else:
conversation = (
Conversation.objects.filter(user=user, client=client_application)
.prefetch_related("agent")
.order_by("-updated_at")
)
if await conversation.aexists(): conversation = await query.order_by("-updated_at").afirst()
return await conversation.prefetch_related("agent").afirst()
return await ( return conversation or await Conversation.objects.prefetch_related("agent").acreate(
Conversation.objects.filter(user=user, client=client_application) user=user, client=client_application
.prefetch_related("agent") )
.order_by("-updated_at")
.afirst()
) or await Conversation.objects.prefetch_related("agent").acreate(user=user, client=client_application)
@staticmethod @staticmethod
async def adelete_conversation_by_user( async def adelete_conversation_by_user(