diff --git a/private_gpt/components/node_store/node_store_component.py b/private_gpt/components/node_store/node_store_component.py index 1bbcc4ff..5cd0afbd 100644 --- a/private_gpt/components/node_store/node_store_component.py +++ b/private_gpt/components/node_store/node_store_component.py @@ -38,10 +38,10 @@ class NodeStoreComponent: case "postgres": try: - from llama_index.storage.docstore.postgres import ( # type: ignore + from llama_index.storage.docstore.postgres import ( # type: ignore PostgresDocumentStore, ) - from llama_index.storage.index_store.postgres import ( # type: ignore + from llama_index.storage.index_store.postgres import ( # type: ignore PostgresIndexStore, ) except ImportError: