style: format code with black and isort

Author: bsbodden

langgraph/checkpoint/redis/aio.py

@@ -41,11 +41,11 @@
 from ulid import ULID
 
 from langgraph.checkpoint.redis.base import (
-    BaseRedisSaver,
     CHECKPOINT_BLOB_PREFIX,
     CHECKPOINT_PREFIX,
     CHECKPOINT_WRITE_PREFIX,
     REDIS_KEY_SEPARATOR,
+    BaseRedisSaver,
 )
 from langgraph.checkpoint.redis.key_registry import (
     AsyncCheckpointKeyRegistry as AsyncKeyRegistry,

langgraph/checkpoint/redis/ashallow.py

@@ -38,6 +38,7 @@
     to_storage_safe_str,
 )
 
+
 class AsyncShallowRedisSaver(BaseRedisSaver[AsyncRedis, AsyncSearchIndex]):
     """Async Redis implementation that only stores the most recent checkpoint."""
 
@@ -799,10 +800,8 @@ def _make_redis_checkpoint_writes_key_cached(
             f"writes:{thread_id}:{checkpoint_ns}:{checkpoint_id}:{task_id}:{idx}"
         )
         if cache_key not in self._key_cache:
-            self._key_cache[cache_key] = (
-                self._make_redis_checkpoint_writes_key(
-                    thread_id, checkpoint_ns, checkpoint_id, task_id, idx
-                )
+            self._key_cache[cache_key] = self._make_redis_checkpoint_writes_key(
+                thread_id, checkpoint_ns, checkpoint_id, task_id, idx
             )
         return self._key_cache[cache_key]
 

langgraph/checkpoint/redis/shallow.py

@@ -701,10 +701,8 @@ def _make_redis_checkpoint_writes_key_cached(
             if len(self._key_cache) >= self._key_cache_max_size:
                 # Remove least recently used (first item)
                 self._key_cache.popitem(last=False)
-            self._key_cache[cache_key] = (
-                self._make_redis_checkpoint_writes_key(
-                    thread_id, checkpoint_ns, checkpoint_id, task_id, idx
-                )
+            self._key_cache[cache_key] = self._make_redis_checkpoint_writes_key(
+                thread_id, checkpoint_ns, checkpoint_id, task_id, idx
             )
         return self._key_cache[cache_key]
 

tests/test_issue_125_custom_checkpoint_prefixes.py

@@ -180,7 +180,7 @@ def test_custom_checkpoint_prefix_isolation_sync(
 
         # Saver 2 with prefix "app2_checkpoint"
         with RedisSaver.from_conn_string(
-                redis_url, checkpoint_prefix="app2_checkpoint"
+            redis_url, checkpoint_prefix="app2_checkpoint"
         ) as saver2:
             saver2.setup()
 
@@ -300,7 +300,9 @@ def test_custom_checkpoint_prefix_delete_thread(
         saver.delete_thread(thread_id)
 
         # Verify all keys are deleted
-        checkpoint_keys_after = list(saver._redis.scan_iter(f"{custom_checkpoint_prefix}:*"))
+        checkpoint_keys_after = list(
+            saver._redis.scan_iter(f"{custom_checkpoint_prefix}:*")
+        )
         write_keys_after = list(saver._redis.scan_iter(f"{custom_write_prefix}:*"))
         assert len(checkpoint_keys_after) == 0
         assert len(write_keys_after) == 0
@@ -431,7 +433,7 @@ async def test_custom_checkpoint_prefix_isolation_async(
 
         # Saver 2 with prefix "async_app2_checkpoint"
         async with AsyncRedisSaver.from_conn_string(
-                redis_url, checkpoint_prefix="async_app2_checkpoint"
+            redis_url, checkpoint_prefix="async_app2_checkpoint"
         ) as saver2:
             await saver2.setup()
 
@@ -446,7 +448,6 @@ async def test_custom_checkpoint_prefix_isolation_async(
             }
             await saver1.aput(config, checkpoint1, simple_metadata, {})
 
-
             checkpoint2: Checkpoint = {
                 "v": 1,
                 "id": checkpoint_id,
@@ -489,19 +490,29 @@ async def test_custom_checkpoint_prefix_async_delete_thread(
         await saver.setup()
 
         await saver.aput(config, simple_checkpoint, simple_metadata, {})
-        await saver.aput_writes(config, writes=[("channel1", "value1")], task_id="task_1")
+        await saver.aput_writes(
+            config, writes=[("channel1", "value1")], task_id="task_1"
+        )
 
         # Verify keys exist
-        checkpoint_keys = [k async for k in saver._redis.scan_iter(f"{custom_checkpoint_prefix}:*")]
-        write_keys = [k async for k in saver._redis.scan_iter(f"{custom_write_prefix}:*")]
+        checkpoint_keys = [
+            k async for k in saver._redis.scan_iter(f"{custom_checkpoint_prefix}:*")
+        ]
+        write_keys = [
+            k async for k in saver._redis.scan_iter(f"{custom_write_prefix}:*")
+        ]
         assert len(checkpoint_keys) > 0
         assert len(write_keys) > 0
 
         # Delete thread
         await saver.adelete_thread(thread_id)
 
         # Verify all keys are deleted
-        checkpoint_keys_after = [k async for k in saver._redis.scan_iter(f"{custom_checkpoint_prefix}:*")]
-        write_keys_after = [k async for k in saver._redis.scan_iter(f"{custom_write_prefix}:*")]
+        checkpoint_keys_after = [
+            k async for k in saver._redis.scan_iter(f"{custom_checkpoint_prefix}:*")
+        ]
+        write_keys_after = [
+            k async for k in saver._redis.scan_iter(f"{custom_write_prefix}:*")
+        ]
         assert len(checkpoint_keys_after) == 0
         assert len(write_keys_after) == 0

tests/test_shallow_sync.py

@@ -9,13 +9,13 @@
     create_checkpoint,
     empty_checkpoint,
 )
+from redis import Redis
+from redis.exceptions import ConnectionError as RedisConnectionError
+
 from langgraph.checkpoint.redis.base import (
     CHECKPOINT_BLOB_PREFIX,
     CHECKPOINT_WRITE_PREFIX,
 )
-from redis import Redis
-from redis.exceptions import ConnectionError as RedisConnectionError
-
 from langgraph.checkpoint.redis.shallow import ShallowRedisSaver
 
 
@@ -377,7 +377,9 @@ def test_key_generation_inconsistency(redis_url: str) -> None:
         )
 
         # The base key uses storage-safe transformations
-        expected_base_key = f"{CHECKPOINT_BLOB_PREFIX}:test_thread:__empty__:test_channel:1"
+        expected_base_key = (
+            f"{CHECKPOINT_BLOB_PREFIX}:test_thread:__empty__:test_channel:1"
+        )
         assert base_blob_key == expected_base_key
 
         # The shallow pattern now uses storage-safe transformations (fixed!)
@@ -402,9 +404,7 @@ def test_key_generation_inconsistency(redis_url: str) -> None:
         )
 
         # The base key uses storage-safe transformations
-        expected_base_key = (
-            f"{CHECKPOINT_WRITE_PREFIX}:test_thread:__empty__:test_checkpoint:test_task:0"
-        )
+        expected_base_key = f"{CHECKPOINT_WRITE_PREFIX}:test_thread:__empty__:test_checkpoint:test_task:0"
         assert base_writes_key == expected_base_key
 
         # The shallow pattern now uses storage-safe transformations (fixed!)