From 839718723182ba89c0c517333698775a3c46e318 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Thu, 28 Mar 2024 11:36:27 +0530 Subject: [PATCH] Use default agent when creating a new conversation without agent specified --- src/khoj/database/adapters/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/khoj/database/adapters/__init__.py b/src/khoj/database/adapters/__init__.py index 25f781fe..d7730c0e 100644 --- a/src/khoj/database/adapters/__init__.py +++ b/src/khoj/database/adapters/__init__.py @@ -477,9 +477,10 @@ class ConversationAdapters: .first() ) else: + agent = AgentAdapters.get_default_agent() conversation = ( Conversation.objects.filter(user=user, client=client_application).order_by("-updated_at").first() - ) or Conversation.objects.create(user=user, client=client_application) + ) or Conversation.objects.create(user=user, client=client_application, agent=agent) return conversation @@ -513,7 +514,8 @@ class ConversationAdapters: if agent is None: raise HTTPException(status_code=400, detail="No such agent currently exists.") return await Conversation.objects.acreate(user=user, client=client_application, agent=agent) - return await Conversation.objects.acreate(user=user, client=client_application) + agent = await AgentAdapters.aget_default_agent() + return await Conversation.objects.acreate(user=user, client=client_application, agent=agent) @staticmethod async def aget_conversation_by_user(