From b2952236c433e8ab9a121be2dc23f449a384d3f6 Mon Sep 17 00:00:00 2001 From: Debanjum Date: Thu, 10 Jul 2025 23:01:53 -0700 Subject: [PATCH] Log conversation id to help troubleshoot errors faster --- src/khoj/database/adapters/__init__.py | 3 ++- src/khoj/processor/conversation/utils.py | 5 +++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/khoj/database/adapters/__init__.py b/src/khoj/database/adapters/__init__.py index 5f0ec103..fe53a73c 100644 --- a/src/khoj/database/adapters/__init__.py +++ b/src/khoj/database/adapters/__init__.py @@ -1488,9 +1488,10 @@ class ConversationAdapters: conversation.updated_at = django_timezone.now() await conversation.asave() else: - await Conversation.objects.acreate( + conversation = await Conversation.objects.acreate( user=user, conversation_log=cleaned_conversation_log, client=client_application, slug=slug ) + return conversation @staticmethod def get_conversation_processor_options(): diff --git a/src/khoj/processor/conversation/utils.py b/src/khoj/processor/conversation/utils.py index 09297f30..d154a965 100644 --- a/src/khoj/processor/conversation/utils.py +++ b/src/khoj/processor/conversation/utils.py @@ -491,8 +491,9 @@ async def save_to_conversation_log( updated_conversation = None logger.error(f"Error constructing chat history: {e}") + db_conversation = None if updated_conversation: - await ConversationAdapters.save_conversation( + db_conversation = await ConversationAdapters.save_conversation( user, updated_conversation, client_application=client_application, @@ -505,7 +506,7 @@ async def save_to_conversation_log( logger.info( f""" -Saved Conversation Turn +Saved Conversation Turn ({db_conversation.id if db_conversation else 'N/A'}): You ({user.username}): "{q}" Khoj: "{chat_response}"