from pathlib import Path

from llama_index.core import (
    MockEmbedding,
    Settings,
    SimpleDirectoryReader,
    StorageContext,
    VectorStoreIndex,
    load_index_from_storage,
)


KNOWLEDGE_BASE_DIR = Path("knowledge_base")
STORAGE_DIR = Path("storage")

Settings.embed_model = MockEmbedding(embed_dim=8)