chore: support alembic (#1867)

Co-authored-by: Shubham Naik <shub@memgpt.ai> Co-authored-by: Sarah Wooders <sarahwooders@gmail.com>
2024-10-11 15:51:14 -07:00
parent 02f4098553
commit 4fbc8c9fbb
15 changed files with 334 additions and 112 deletions
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -385,7 +385,7 @@ def test_sources(client: Union[LocalClient, RESTClient], agent: AgentState):
    # list archival memory
    archival_memories = client.get_archival_memory(agent_id=agent.id)
    # print(archival_memories)
-    assert len(archival_memories) == created_passages
+    assert len(archival_memories) == created_passages, f"Mismatched length {len(archival_memories)} vs. {created_passages}"

    # check number of passages
    sources = client.list_sources()