From 0415c02b4084c1215f62cc3b2530eb918d4bf253 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@boringbits.io>
Date: Fri, 14 Feb 2025 18:34:06 -0800
Subject: [PATCH] Fix sqlite tests

---
 libs/checkpoint-sqlite/tests/test_aiosqlite.py | 6 +++---
 libs/checkpoint-sqlite/tests/test_sqlite.py    | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/libs/checkpoint-sqlite/tests/test_aiosqlite.py b/libs/checkpoint-sqlite/tests/test_aiosqlite.py
index dbb6e327d..4681a0b89 100644
--- a/libs/checkpoint-sqlite/tests/test_aiosqlite.py
+++ b/libs/checkpoint-sqlite/tests/test_aiosqlite.py
@@ -72,7 +72,6 @@ class TestAsyncSqliteSaver:
             assert checkpoint.metadata == {
                 **self.metadata_2,
                 "thread_id": "thread-2",
-                "checkpoint_ns": "",
                 "run_id": "my_run_id",
             }
 
@@ -94,14 +93,15 @@ class TestAsyncSqliteSaver:
             search_results_1 = [c async for c in saver.alist(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 = [c async for c in saver.alist(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,
             }
 
diff --git a/libs/checkpoint-sqlite/tests/test_sqlite.py b/libs/checkpoint-sqlite/tests/test_sqlite.py
index 39371c764..88b7027b4 100644
--- a/libs/checkpoint-sqlite/tests/test_sqlite.py
+++ b/libs/checkpoint-sqlite/tests/test_sqlite.py
@@ -73,7 +73,6 @@ class TestSqliteSaver:
             assert checkpoint.metadata == {
                 **self.metadata_2,
                 "thread_id": "thread-2",
-                "checkpoint_ns": "",
                 "run_id": "my_run_id",
             }
 
@@ -97,14 +96,15 @@ class TestSqliteSaver:
             search_results_1 = list(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(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,
             }