Spaces:
Runtime error
Runtime error
| import logging | |
| from llama_index.storage.docstore import BaseDocumentStore, SimpleDocumentStore | |
| from llama_index.storage.index_store import SimpleIndexStore | |
| from llama_index.storage.index_store.types import BaseIndexStore | |
| from app.paths import local_data_path | |
| logger = logging.getLogger(__name__) | |
| class NodeStoreComponent: | |
| index_store: BaseIndexStore | |
| doc_store: BaseDocumentStore | |
| def __init__(self) -> None: | |
| try: | |
| self.index_store = SimpleIndexStore.from_persist_dir( | |
| persist_dir=str(local_data_path) | |
| ) | |
| except FileNotFoundError: | |
| logger.debug("Local index store not found, creating a new one") | |
| self.index_store = SimpleIndexStore() | |
| try: | |
| self.doc_store = SimpleDocumentStore.from_persist_dir( | |
| persist_dir=str(local_data_path) | |
| ) | |
| except FileNotFoundError: | |
| logger.debug("Local document store not found, creating a new one") | |
| self.doc_store = SimpleDocumentStore() | |