style: format Redis Cluster services

Author: Wenvii

tests/memory/test_redis_cluster_memory_service.py

@@ -12,14 +12,13 @@
 
 
 class TestRedisClusterMemoryService:
+
     @patch("trpc_agent_sdk.memory._redis_cluster_memory_service.RedisClusterStorage")
     def test_uses_cluster_storage(self, storage_cls):
         storage = MagicMock()
         storage_cls.return_value = storage
 
-        service = RedisClusterMemoryService(
-            db_url="redis://seed:6379/0", is_async=True, max_connections=20)
+        service = RedisClusterMemoryService(db_url="redis://seed:6379/0", is_async=True, max_connections=20)
 
         assert service._redis_storage is storage
-        storage_cls.assert_called_once_with(
-            is_async=True, redis_url="redis://seed:6379/0", max_connections=20)
+        storage_cls.assert_called_once_with(is_async=True, redis_url="redis://seed:6379/0", max_connections=20)

tests/sessions/test_redis_cluster_session_service.py

@@ -12,14 +12,13 @@
 
 
 class TestRedisClusterSessionService:
+
     @patch("trpc_agent_sdk.sessions._redis_cluster_session_service.RedisClusterStorage")
     def test_uses_cluster_storage(self, storage_cls):
         storage = MagicMock()
         storage_cls.return_value = storage
 
-        service = RedisClusterSessionService(
-            db_url="redis://seed:6379/0", is_async=True, max_connections=20)
+        service = RedisClusterSessionService(db_url="redis://seed:6379/0", is_async=True, max_connections=20)
 
         assert service._redis_storage is storage
-        storage_cls.assert_called_once_with(
-            is_async=True, redis_url="redis://seed:6379/0", max_connections=20)
+        storage_cls.assert_called_once_with(is_async=True, redis_url="redis://seed:6379/0", max_connections=20)

tests/storage/test_redis_cluster_storage.py

@@ -45,6 +45,7 @@ async def close(self):
 
 
 class TestRedisClusterStorage:
+
     async def test_query_scans_all_cluster_keys_and_deduplicates(self):
         storage = RedisClusterStorage(redis_url="redis://seed:6379/0", is_async=True)
         client = _FakeClusterClient()
@@ -80,8 +81,7 @@ async def test_async_client_is_created_from_seed_url(self):
             client_cls.from_url.return_value = client
             await storage.create_redis_engine()
 
-        client_cls.from_url.assert_called_once_with(
-            "redis://seed:6379/0", decode_responses=True, max_connections=20)
+        client_cls.from_url.assert_called_once_with("redis://seed:6379/0", decode_responses=True, max_connections=20)
         assert storage._redis_client is client
 
     async def test_close_releases_cluster_client(self):

trpc_agent_sdk/storage/_redis_cluster.py

@@ -13,7 +13,6 @@
 
 from __future__ import annotations
 
-import asyncio
 import inspect
 import json
 from typing import Any
@@ -27,10 +26,8 @@
 
 from ._redis import RedisCommand
 from ._redis import RedisCondition
-from ._redis import RedisExpire
 from ._redis import RedisStorage
 
-
 RedisClusterClient = Union[AsyncRedisCluster, SyncRedisCluster]
 
 
@@ -97,8 +94,7 @@ async def delete(self, conn: RedisClusterClient, key: str, conditions: Optional[
         if inspect.isawaitable(ret):
             await ret
 
-    async def query(self, conn: RedisClusterClient, key: str,
-                    conditions: RedisCondition) -> list[tuple[str, Any]]:
+    async def query(self, conn: RedisClusterClient, key: str, conditions: RedisCondition) -> list[tuple[str, Any]]:
         """Query keys across all cluster primaries using cursor-based scanning."""
         keys = await self._scan_keys(conn, key)
         if conditions.limit > 0: