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

Only create new chat on get if a specific chat id, slug isn't requested

Browse source
This commit is contained in:
Debanjum Singh Solanky 2024-03-15 11:58:39 +05:30
parent 8cdfaf41ec
commit cac26dafe3
2 changed files with 9 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/khoj/database/adapters/__init__.py
View file

@ -439,16 +439,13 @@ class ConversationAdapters:
user: KhojUser, client_application: ClientApplication = None, conversation_id: int = None, slug: str = None
):
if conversation_id:
conversation = Conversation.objects.filter(user=user, client=client_application, id=conversation_id)
return await Conversation.objects.filter(user=user, client=client_application, id=conversation_id).afirst()
elif slug:
conversation = Conversation.objects.filter(user=user, client=client_application, slug=slug)
return await Conversation.objects.filter(user=user, client=client_application, slug=slug).afirst()
else:
conversation = Conversation.objects.filter(user=user, client=client_application).order_by("-updated_at")
if await conversation.aexists():
return await conversation.afirst()
return await Conversation.objects.acreate(user=user, client=client_application, slug=slug)
return await (
Conversation.objects.filter(user=user, client=client_application).order_by("-updated_at").afirst()
) or Conversation.objects.acreate(user=user, client=client_application, slug=slug)
@staticmethod
async def adelete_conversation_by_user(

7
src/khoj/routers/api_chat.py
View file

@ -250,9 +250,10 @@ async def chat(
formatted_help = help_message.format(model=model_type, version=state.khoj_version, device=get_device())
return StreamingResponse(iter([formatted_help]), media_type="text/event-stream", status_code=200)
meta_log = (
await ConversationAdapters.aget_conversation_by_user(user, request.user.client_app, conversation_id, slug)
).conversation_log
conversation = await ConversationAdapters.aget_conversation_by_user(
user, request.user.client_app, conversation_id, slug
)
meta_log = conversation.conversation_log if conversation else {}
if conversation_commands == [ConversationCommand.Default]:
conversation_commands = await aget_relevant_information_sources(q, meta_log)