From 90efc2ea7aa5f315b634b2244b71368d7789302a Mon Sep 17 00:00:00 2001
From: sabaimran <narmiabas@gmail.com>
Date: Tue, 1 Aug 2023 09:24:03 -0700
Subject: [PATCH] Update comments and add explanations

---
 src/khoj/processor/conversation/gpt4all/chat_model.py | 2 --
 src/khoj/processor/conversation/gpt4all/utils.py      | 1 +
 tests/conftest.py                                     | 2 +-
 3 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/khoj/processor/conversation/gpt4all/chat_model.py b/src/khoj/processor/conversation/gpt4all/chat_model.py
index 199f6e44..e153e0eb 100644
--- a/src/khoj/processor/conversation/gpt4all/chat_model.py
+++ b/src/khoj/processor/conversation/gpt4all/chat_model.py
@@ -124,11 +124,9 @@ def converse_offline(
     """
     gpt4all_model = loaded_model or GPT4All(model)
     # Initialize Variables
-    current_date = datetime.now().strftime("%Y-%m-%d")
     compiled_references_message = "\n\n".join({f"{item}" for item in references})
 
     # Get Conversation Primer appropriate to Conversation Type
-    # TODO If compiled_references_message is too long, we need to truncate it.
     if compiled_references_message == "":
         conversation_primer = user_query
     else:
diff --git a/src/khoj/processor/conversation/gpt4all/utils.py b/src/khoj/processor/conversation/gpt4all/utils.py
index 86bfc0f5..b7f953e4 100644
--- a/src/khoj/processor/conversation/gpt4all/utils.py
+++ b/src/khoj/processor/conversation/gpt4all/utils.py
@@ -19,6 +19,7 @@ def download_model(model_name: str):
     if os.path.exists(filename):
         return GPT4All(model_name)
 
+    # Download the model to a tmp file. Once the download is completed, move the tmp file to the actual file
     tmp_filename = filename + ".tmp"
 
     try:
diff --git a/tests/conftest.py b/tests/conftest.py
index 6f0d4bf5..871154e7 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -174,7 +174,7 @@ def processor_config(tmp_path_factory):
 def processor_config_offline_chat(tmp_path_factory):
     processor_dir = tmp_path_factory.mktemp("processor")
 
-    # Setup conversation processor, if OpenAI API key is set
+    # Setup conversation processor
     processor_config = ProcessorConfig()
     processor_config.conversation = ConversationProcessorConfig(
         enable_offline_chat=True,