diff --git a/src/khoj/utils/rawconfig.py b/src/khoj/utils/rawconfig.py index 0d17e9e4..15ecb85d 100644 --- a/src/khoj/utils/rawconfig.py +++ b/src/khoj/utils/rawconfig.py @@ -53,17 +53,8 @@ class NotionContentConfig(ConfigBase): token: str -class ImageContentConfig(ConfigBase): - input_directories: Optional[List[Path]] = None - input_filter: Optional[List[str]] = None - embeddings_file: Path - use_xmp_metadata: bool - batch_size: int - - class ContentConfig(ConfigBase): org: Optional[TextContentConfig] = None - image: Optional[ImageContentConfig] = None markdown: Optional[TextContentConfig] = None pdf: Optional[TextContentConfig] = None plaintext: Optional[TextContentConfig] = None diff --git a/tests/conftest.py b/tests/conftest.py index 1c84c7d1..d513a574 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -30,12 +30,7 @@ from khoj.utils import fs_syncer, state from khoj.utils.config import SearchModels from khoj.utils.constants import web_directory from khoj.utils.helpers import resolve_absolute_path -from khoj.utils.rawconfig import ( - ContentConfig, - ImageContentConfig, - ImageSearchConfig, - SearchConfig, -) +from khoj.utils.rawconfig import ContentConfig, ImageSearchConfig, SearchConfig from tests.helpers import ( ChatModelOptionsFactory, OfflineChatProcessorConversationConfigFactory, @@ -223,13 +218,6 @@ def content_config(tmp_path_factory, search_models: SearchModels, default_user: # Generate Image Embeddings from Test Images content_config = ContentConfig() - content_config.image = ImageContentConfig( - input_filter=None, - input_directories=["tests/data/images"], - embeddings_file=content_dir.joinpath("image_embeddings.pt"), - batch_size=1, - use_xmp_metadata=False, - ) LocalOrgConfig.objects.create( input_files=None, diff --git a/tests/test_rawconfig.py b/tests/test_rawconfig.py deleted file mode 100644 index 6598f25b..00000000 --- a/tests/test_rawconfig.py +++ /dev/null @@ -1,15 +0,0 @@ -import pytest - -from khoj.utils.rawconfig import ImageContentConfig, TextContentConfig - - -# Test -# ---------------------------------------------------------------------------------------------------- -def test_input_filter_or_directories_required_in_image_content_config(): - # Act - with pytest.raises(ValueError): - ImageContentConfig( - input_directories=None, - input_filter=None, - embeddings_file="note_embeddings.pt", - )