Get rid of enable flag for the offline chat processor config

- Default, assume that offline chat is enabled if there is an offline chat model option configured
This commit is contained in:
sabaimran
2024-04-23 23:08:29 +05:30
parent ac474fce38
commit 60658a8037
13 changed files with 33 additions and 67 deletions

View File

@@ -33,7 +33,6 @@ from khoj.utils.helpers import resolve_absolute_path
from khoj.utils.rawconfig import ContentConfig, ImageSearchConfig, SearchConfig
from tests.helpers import (
ChatModelOptionsFactory,
OfflineChatProcessorConversationConfigFactory,
OpenAIProcessorConversationConfigFactory,
ProcessLockFactory,
SubscriptionFactory,
@@ -377,7 +376,12 @@ def client_offline_chat(search_config: SearchConfig, default_user2: KhojUser):
configure_content(all_files, user=default_user2)
# Initialize Processor from Config
OfflineChatProcessorConversationConfigFactory(enabled=True)
ChatModelOptionsFactory(
chat_model="NousResearch/Hermes-2-Pro-Mistral-7B-GGUF",
tokenizer=None,
max_prompt_size=None,
model_type="offline",
)
UserConversationProcessorConfigFactory(user=default_user2)
state.anonymous_mode = True

View File

@@ -9,7 +9,6 @@ from khoj.database.models import (
Conversation,
KhojApiUser,
KhojUser,
OfflineChatProcessorConversationConfig,
OpenAIProcessorConversationConfig,
ProcessLock,
SearchModelConfig,
@@ -55,13 +54,6 @@ class UserConversationProcessorConfigFactory(factory.django.DjangoModelFactory):
setting = factory.SubFactory(ChatModelOptionsFactory)
class OfflineChatProcessorConversationConfigFactory(factory.django.DjangoModelFactory):
class Meta:
model = OfflineChatProcessorConversationConfig
enabled = True
class OpenAIProcessorConversationConfigFactory(factory.django.DjangoModelFactory):
class Meta:
model = OpenAIProcessorConversationConfig

View File

@@ -24,7 +24,7 @@ from khoj.utils.constants import default_offline_chat_model
@pytest.fixture(scope="session")
def loaded_model():
return download_model(default_offline_chat_model)
return download_model(default_offline_chat_model, max_tokens=5000)
freezegun.configure(extend_ignore_list=["transformers"])