From cc9eae5d181cf687c900d284f23b4fb9e60af1f0 Mon Sep 17 00:00:00 2001
From: Debanjum Singh Solanky <debanjum@gmail.com>
Date: Sun, 26 Nov 2023 03:46:01 -0800
Subject: [PATCH] Update default chat model to Mistral in GPT4AllProcessor
 config

---
 src/khoj/utils/config.py | 2 +-
 src/khoj/utils/state.py  | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/khoj/utils/config.py b/src/khoj/utils/config.py
index aba6366a..abda12b6 100644
--- a/src/khoj/utils/config.py
+++ b/src/khoj/utils/config.py
@@ -80,7 +80,7 @@ class GPT4AllProcessorConfig:
 class GPT4AllProcessorModel:
     def __init__(
         self,
-        chat_model: str = "llama-2-7b-chat.ggmlv3.q4_0.bin",
+        chat_model: str = "mistral-7b-instruct-v0.1.Q4_0.gguf",
     ):
         self.chat_model = chat_model
         self.loaded_model = None
diff --git a/src/khoj/utils/state.py b/src/khoj/utils/state.py
index ce4d5804..b54cf4b3 100644
--- a/src/khoj/utils/state.py
+++ b/src/khoj/utils/state.py
@@ -7,6 +7,7 @@ from collections import defaultdict
 # External Packages
 from pathlib import Path
 from khoj.processor.embeddings import CrossEncoderModel, EmbeddingsModel
+from whisper import Whisper
 
 # Internal Packages
 from khoj.utils import config as utils_config
@@ -21,7 +22,7 @@ embeddings_model: EmbeddingsModel = None
 cross_encoder_model: CrossEncoderModel = None
 content_index = ContentIndex()
 gpt4all_processor_config: GPT4AllProcessorModel = None
-whisper_model = None
+whisper_model: Whisper = None
 config_file: Path = None
 verbose: int = 0
 host: str = None