Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Add SQLite integration for recall memory

Browse Source
This commit is contained in:
Sarah Wooders
2023-12-22 11:33:37 +04:00
parent e2b29d8995
commit 5002b8e1dd
5 changed files with 100 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_storage.py
View File

@@ -56,8 +56,8 @@ def generate_messages():
return messages
@pytest.mark.parametrize("storage_connector", ["postgres", "chroma", "lancedb"])
# @pytest.mark.parametrize("storage_connector", ["postgres"])
@pytest.mark.parametrize("storage_connector", ["postgres", "chroma", "sqllite", "lancedb"])
# @pytest.mark.parametrize("storage_connector", ["sqllite"])
@pytest.mark.parametrize("table_type", [TableType.RECALL_MEMORY, TableType.ARCHIVAL_MEMORY])
def test_storage(storage_connector, table_type):
@@ -86,9 +86,9 @@ def test_storage(storage_connector, table_type):
return
config.archival_storage_type = "chroma"
config.archival_storage_path = "./test_chroma"
if storage_connector == "local":
if storage_connector == "sqllite":
if table_type == TableType.ARCHIVAL_MEMORY:
print("Skipping test, local only supported for recall memory")
print("Skipping test, sqllite only supported for recall memory")
return
config.recall_storage_type = "local"