From 277d2f5c964d3174438cedc86c57ff9c68bffb67 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Sun, 28 May 2023 19:28:57 +0530 Subject: [PATCH] Do not add "Notes:" suffix to chat messages when no notes retrieved This was causing spurious "Notes:" suffix being added to Khoj Chat in response --- src/khoj/processor/conversation/utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/khoj/processor/conversation/utils.py b/src/khoj/processor/conversation/utils.py index f5d86e00..3e3d38e7 100644 --- a/src/khoj/processor/conversation/utils.py +++ b/src/khoj/processor/conversation/utils.py @@ -72,7 +72,11 @@ def generate_chatml_messages_with_context( ): """Generate messages for ChatGPT with context from previous conversation""" # Extract Chat History for Context - chat_logs = [f'{chat["message"]}\n\nNotes:\n{chat.get("context","")}' for chat in conversation_log.get("chat", [])] + chat_logs = [] + for chat in conversation_log.get("chat", []): + chat_notes = f'\n\n Notes:\n{chat.get("context")}' if chat.get("context") else "\n" + chat_logs += [chat["message"] + chat_notes] + rest_backnforths = [] # Extract in reverse chronological order for user_msg, assistant_msg in zip(chat_logs[-2::-2], chat_logs[::-2]):