klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-10 05:39:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update comments and add explanations

Browse Source
This commit is contained in:
sabaimran
2023-08-01 09:24:03 -07:00
parent f7e03f6d63
commit 90efc2ea7a
3 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/khoj/processor/conversation/gpt4all/chat_model.py
View File

@@ -124,11 +124,9 @@ def converse_offline(
""" """
gpt4all_model = loaded_model or GPT4All(model) gpt4all_model = loaded_model or GPT4All(model)
# Initialize Variables # Initialize Variables
current_date = datetime.now().strftime("%Y-%m-%d")
compiled_references_message = "\n\n".join({f"{item}" for item in references}) compiled_references_message = "\n\n".join({f"{item}" for item in references})
# Get Conversation Primer appropriate to Conversation Type # Get Conversation Primer appropriate to Conversation Type
# TODO If compiled_references_message is too long, we need to truncate it.
if compiled_references_message == "": if compiled_references_message == "":
conversation_primer = user_query conversation_primer = user_query
else: else:

1
src/khoj/processor/conversation/gpt4all/utils.py
View File

@@ -19,6 +19,7 @@ def download_model(model_name: str):
if os.path.exists(filename): if os.path.exists(filename):
return GPT4All(model_name) 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" tmp_filename = filename + ".tmp"
try: try:

2
tests/conftest.py
View File

@@ -174,7 +174,7 @@ def processor_config(tmp_path_factory):
def processor_config_offline_chat(tmp_path_factory): def processor_config_offline_chat(tmp_path_factory):
processor_dir = tmp_path_factory.mktemp("processor") processor_dir = tmp_path_factory.mktemp("processor")
# Setup conversation processor, if OpenAI API key is set # Setup conversation processor
processor_config = ProcessorConfig() processor_config = ProcessorConfig()
processor_config.conversation = ConversationProcessorConfig( processor_config.conversation = ConversationProcessorConfig(
enable_offline_chat=True, enable_offline_chat=True,