From a0a7ab7ec82e5975812f2367e3bd7ebdf762f63d Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Sun, 26 Nov 2023 02:34:23 -0800 Subject: [PATCH] Rename conversation.gpt4all package to conversation.offline --- .../processor/conversation/{gpt4all => offline}/__init__.py | 0 .../processor/conversation/{gpt4all => offline}/chat_model.py | 0 src/khoj/processor/conversation/{gpt4all => offline}/utils.py | 0 src/khoj/routers/api.py | 2 +- src/khoj/routers/helpers.py | 2 +- src/khoj/utils/config.py | 2 +- tests/test_gpt4all_chat_actors.py | 4 ++-- 7 files changed, 5 insertions(+), 5 deletions(-) rename src/khoj/processor/conversation/{gpt4all => offline}/__init__.py (100%) rename src/khoj/processor/conversation/{gpt4all => offline}/chat_model.py (100%) rename src/khoj/processor/conversation/{gpt4all => offline}/utils.py (100%) diff --git a/src/khoj/processor/conversation/gpt4all/__init__.py b/src/khoj/processor/conversation/offline/__init__.py similarity index 100% rename from src/khoj/processor/conversation/gpt4all/__init__.py rename to src/khoj/processor/conversation/offline/__init__.py diff --git a/src/khoj/processor/conversation/gpt4all/chat_model.py b/src/khoj/processor/conversation/offline/chat_model.py similarity index 100% rename from src/khoj/processor/conversation/gpt4all/chat_model.py rename to src/khoj/processor/conversation/offline/chat_model.py diff --git a/src/khoj/processor/conversation/gpt4all/utils.py b/src/khoj/processor/conversation/offline/utils.py similarity index 100% rename from src/khoj/processor/conversation/gpt4all/utils.py rename to src/khoj/processor/conversation/offline/utils.py diff --git a/src/khoj/routers/api.py b/src/khoj/routers/api.py index 95b257ce..67b959a7 100644 --- a/src/khoj/routers/api.py +++ b/src/khoj/routers/api.py @@ -30,7 +30,7 @@ from khoj.database.models import ( LocalPlaintextConfig, NotionConfig, ) -from khoj.processor.conversation.gpt4all.chat_model import extract_questions_offline +from khoj.processor.conversation.offline.chat_model import extract_questions_offline from khoj.processor.conversation.openai.gpt import extract_questions from khoj.processor.conversation.openai.whisper import transcribe_audio from khoj.processor.conversation.prompts import help_message, no_entries_found diff --git a/src/khoj/routers/helpers.py b/src/khoj/routers/helpers.py index c6fcb436..e1ab05b5 100644 --- a/src/khoj/routers/helpers.py +++ b/src/khoj/routers/helpers.py @@ -15,7 +15,7 @@ from fastapi import Depends, Header, HTTPException, Request from khoj.database.adapters import ConversationAdapters from khoj.database.models import KhojUser, Subscription from khoj.processor.conversation import prompts -from khoj.processor.conversation.gpt4all.chat_model import converse_offline, send_message_to_model_offline +from khoj.processor.conversation.offline.chat_model import converse_offline, send_message_to_model_offline from khoj.processor.conversation.openai.gpt import converse, send_message_to_model from khoj.processor.conversation.utils import ThreadedGenerator, message_to_log diff --git a/src/khoj/utils/config.py b/src/khoj/utils/config.py index 7795d695..aba6366a 100644 --- a/src/khoj/utils/config.py +++ b/src/khoj/utils/config.py @@ -11,7 +11,7 @@ from typing import TYPE_CHECKING, List, Optional, Union, Any import torch # Internal Packages -from khoj.processor.conversation.gpt4all.utils import download_model +from khoj.processor.conversation.offline.utils import download_model logger = logging.getLogger(__name__) diff --git a/tests/test_gpt4all_chat_actors.py b/tests/test_gpt4all_chat_actors.py index 782b54f2..7b59e1e3 100644 --- a/tests/test_gpt4all_chat_actors.py +++ b/tests/test_gpt4all_chat_actors.py @@ -19,8 +19,8 @@ except ModuleNotFoundError as e: print("There was an error importing GPT4All. Please run pip install gpt4all in order to install it.") # Internal Packages -from khoj.processor.conversation.gpt4all.chat_model import converse_offline, extract_questions_offline, filter_questions -from khoj.processor.conversation.gpt4all.utils import download_model +from khoj.processor.conversation.offline.chat_model import converse_offline, extract_questions_offline, filter_questions +from khoj.processor.conversation.offline.utils import download_model from khoj.processor.conversation.utils import message_to_log