This commit is contained in:
Nuno Campos
2025-02-14 18:31:03 -08:00
parent 8ff5e79e70
commit e8665f84e7
+6 -5
View File
@@ -73,7 +73,6 @@ class TestMemorySaver:
assert checkpoint.metadata == {
**self.metadata_2,
"thread_id": "thread-2",
"checkpoint_ns": "",
"run_id": "my_run_id",
}
@@ -96,14 +95,15 @@ class TestMemorySaver:
search_results_1 = list(self.memory_saver.list(None, filter=query_1))
assert len(search_results_1) == 1
assert search_results_1[0].metadata == {
**self.config_1["configurable"],
"thread_id": "thread-1",
"thread_ts": "1",
**self.metadata_1,
}
search_results_2 = list(self.memory_saver.list(None, filter=query_2))
assert len(search_results_2) == 1
assert search_results_2[0].metadata == {
**self.config_2["configurable"],
"thread_id": "thread-2",
**self.metadata_2,
}
@@ -146,7 +146,8 @@ class TestMemorySaver:
]
assert len(search_results_1) == 1
assert search_results_1[0].metadata == {
**self.config_1["configurable"],
"thread_id": "thread-1",
"thread_ts": "1",
**self.metadata_1,
}
@@ -155,7 +156,7 @@ class TestMemorySaver:
]
assert len(search_results_2) == 1
assert search_results_2[0].metadata == {
**self.config_2["configurable"],
"thread_id": "thread-2",
**self.metadata_2,
}