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}"