klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-10 13:26:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use default agent when creating a new conversation without agent specified

Browse Source
This commit is contained in:
sabaimran
2024-03-28 11:36:27 +05:30
parent fdf78525b4
commit 8397187231
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -477,9 +477,10 @@ class ConversationAdapters:
.first() .first()
) )
else: else:
agent = AgentAdapters.get_default_agent()
conversation = ( conversation = (
Conversation.objects.filter(user=user, client=client_application).order_by("-updated_at").first() 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 return conversation
@@ -513,7 +514,8 @@ class ConversationAdapters:
if agent is None: if agent is None:
raise HTTPException(status_code=400, detail="No such agent currently exists.") 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=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 @staticmethod
async def aget_conversation_by_user( async def aget_conversation_by_user(