From d2afc1e86f664f994ca5a03ff325b89969bb72ad Mon Sep 17 00:00:00 2001 From: Sarah Wooders Date: Sun, 5 Nov 2023 09:52:18 -0800 Subject: [PATCH] Don't import postgres storage if not specified in config (#318) --- memgpt/connectors/storage.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/memgpt/connectors/storage.py b/memgpt/connectors/storage.py index fc0beea2..00e5e702 100644 --- a/memgpt/connectors/storage.py +++ b/memgpt/connectors/storage.py @@ -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")