From e8665f84e7d81549c36899d54dadecc58e868e10 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@boringbits.io>
Date: Fri, 14 Feb 2025 18:31:03 -0800
Subject: [PATCH] Update

---
 libs/checkpoint/tests/test_memory.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/libs/checkpoint/tests/test_memory.py b/libs/checkpoint/tests/test_memory.py
index c69872389..d347ebbab 100644
--- a/libs/checkpoint/tests/test_memory.py
+++ b/libs/checkpoint/tests/test_memory.py
@@ -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,
         }