Don't import postgres storage if not specified in config (#318)

This commit is contained in:
Sarah Wooders
2023-11-05 09:52:18 -08:00
committed by GitHub
parent ca0ad1ecc1
commit d2afc1e86f

View File

@@ -53,13 +53,15 @@ class StorageConnector:
@staticmethod
def list_loaded_data():
from memgpt.connectors.db import PostgresStorageConnector
from memgpt.connectors.local import LocalStorageConnector
storage_type = MemGPTConfig.load().archival_storage_type
if storage_type == "local":
from memgpt.connectors.db import PostgresStorageConnector
return LocalStorageConnector.list_loaded_data()
elif storage_type == "postgres":
from memgpt.connectors.local import LocalStorageConnector
return PostgresStorageConnector.list_loaded_data()
else:
raise NotImplementedError(f"Storage type {storage_type} not implemented")