From 2e290ea6908d9eb62447caf916975ab7e8939d8f Mon Sep 17 00:00:00 2001
From: Debanjum <debanjum@gmail.com>
Date: Wed, 7 May 2025 19:16:21 -0600
Subject: [PATCH] Pass conversation history to generate non-streaming chat
 model responses

Allows send_message_to_model_wrapper func to also use conversation
logs as context to generate response. This is an optional parameter
---
 src/khoj/routers/helpers.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/khoj/routers/helpers.py b/src/khoj/routers/helpers.py
index f0e2d974..1b2831bd 100644
--- a/src/khoj/routers/helpers.py
+++ b/src/khoj/routers/helpers.py
@@ -1152,6 +1152,7 @@ async def send_message_to_model_wrapper(
     query_images: List[str] = None,
     context: str = "",
     query_files: str = None,
+    conversation_log: dict = {},
     agent_chat_model: ChatModel = None,
     tracer: dict = {},
 ):
@@ -1189,6 +1190,7 @@ async def send_message_to_model_wrapper(
         user_message=query,
         context_message=context,
         system_message=system_message,
+        conversation_log=conversation_log,
         model_name=chat_model_name,
         loaded_model=loaded_model,
         tokenizer_name=tokenizer,
@@ -1254,6 +1256,7 @@ def send_message_to_model_wrapper_sync(
     user: KhojUser = None,
     query_images: List[str] = None,
     query_files: str = "",
+    conversation_log: dict = {},
     tracer: dict = {},
 ):
     chat_model: ChatModel = ConversationAdapters.get_default_chat_model(user)
@@ -1282,6 +1285,7 @@ def send_message_to_model_wrapper_sync(
     truncated_messages = generate_chatml_messages_with_context(
         user_message=message,
         system_message=system_message,
+        conversation_log=conversation_log,
         model_name=chat_model_name,
         loaded_model=loaded_model,
         max_prompt_size=max_tokens,