Avoid using "transactions" when they are not needed

Author: abrookins

langgraph/store/redis/aio.py

@@ -170,23 +170,12 @@ async def from_conn_string(
         index: Optional[IndexConfig] = None,
     ) -> AsyncIterator[AsyncRedisStore]:
         """Create store from Redis connection string."""
-        store = cls(redis_url=conn_string, index=index)
-        try:
+        async with cls(redis_url=conn_string, index=index) as store:
             store._task = store.loop.create_task(
                 store._run_background_tasks(store._aqueue, weakref.ref(store))
             )
             await store.setup()
             yield store
-        finally:
-            if hasattr(store, "_task"):
-                store._task.cancel()
-                try:
-                    await store._task
-                except asyncio.CancelledError:
-                    pass
-            if store._owns_client:
-                await store._redis.aclose()  # type: ignore[attr-defined]
-                await store._redis.connection_pool.disconnect()
 
     def create_indexes(self) -> None:
         """Create async indices."""
@@ -212,8 +201,9 @@ async def __aexit__(
             except asyncio.CancelledError:
                 pass
 
-        # if self._owns_client:
-        await self._redis.aclose()  # type: ignore[attr-defined]
+        if self._owns_client:
+            await self._redis.aclose()  # type: ignore[attr-defined]
+            await self._redis.connection_pool.disconnect()
 
     async def abatch(self, ops: Iterable[Op]) -> list[Result]:
         """Execute batch of operations asynchronously."""
@@ -301,7 +291,7 @@ async def _batch_get_ops(
             res = await self.store_index.search(search_query)
 
             # Use pipeline to get the actual JSON documents
-            pipeline = self._redis.pipeline()
+            pipeline = self._redis.pipeline(transaction=False)
             doc_ids = []
             for doc in res.docs:
                 # The id is already in the correct format (store:prefix:key)
@@ -481,7 +471,7 @@ async def _batch_search_ops(
                 )
 
                 # Get matching store docs in pipeline
-                pipeline = self._redis.pipeline()
+                pipeline = self._redis.pipeline(transaction=False)
                 result_map = {}  # Map store key to vector result with distances
 
                 for doc in vector_results: