Fix type checking after removing types-redis

Author: abrookins

langgraph/checkpoint/redis/aio.py

@@ -131,8 +131,10 @@ async def __aexit__(
     ) -> None:
         """Async context manager exit."""
         if self._owns_its_client:
-            await self._redis.aclose()  # type: ignore[attr-defined]
-            await self._redis.connection_pool.disconnect()
+            await self._redis.aclose()
+            coro = self._redis.connection_pool.disconnect()
+            if coro:
+                await coro
 
             # Prevent RedisVL from attempting to close the client
             # on an event loop in a separate thread.

langgraph/checkpoint/redis/ashallow.py

@@ -134,8 +134,10 @@ async def __aexit__(
         tb: Optional[TracebackType],
     ) -> None:
         if self._owns_its_client:
-            await self._redis.aclose()  # type: ignore[attr-defined]
-            await self._redis.connection_pool.disconnect()
+            await self._redis.aclose()
+            coro = self._redis.connection_pool.disconnect()
+            if coro:
+                await coro
 
             # Prevent RedisVL from attempting to close the client
             # on an event loop in a separate thread.

langgraph/checkpoint/redis/base.py

@@ -154,7 +154,7 @@ def set_client_info(self) -> None:
 
         try:
             # Try to use client_setinfo command if available
-            self._redis.client_setinfo("LIB-NAME", __full_lib_name__)  # type: ignore
+            self._redis.client_setinfo("LIB-NAME", __full_lib_name__)
         except (ResponseError, AttributeError):
             # Fall back to a simple echo if client_setinfo is not available
             try:
@@ -174,7 +174,7 @@ async def aset_client_info(self) -> None:
 
         try:
             # Try to use client_setinfo command if available
-            await self._redis.client_setinfo("LIB-NAME", client_info)  # type: ignore
+            await self._redis.client_setinfo("LIB-NAME", client_info)
         except (ResponseError, AttributeError):
             # Fall back to a simple echo if client_setinfo is not available
             try:
@@ -468,17 +468,17 @@ def put_writes(
                     # UPSERT case - only update specific fields
                     if key_exists:
                         # Update only channel, type, and blob fields
-                        pipeline.set(key, "$.channel", write_obj["channel"])  # type: ignore[arg-type]
-                        pipeline.set(key, "$.type", write_obj["type"])  # type: ignore[arg-type]
-                        pipeline.set(key, "$.blob", write_obj["blob"])  # type: ignore[arg-type]
+                        pipeline.set(key, "$.channel", write_obj["channel"])
+                        pipeline.set(key, "$.type", write_obj["type"])
+                        pipeline.set(key, "$.blob", write_obj["blob"])
                     else:
                         # For new records, set the complete object
-                        pipeline.set(key, "$", write_obj)  # type: ignore[arg-type]
+                        pipeline.set(key, "$", write_obj)
                         created_keys.append(key)
                 else:
                     # INSERT case - only insert if doesn't exist
                     if not key_exists:
-                        pipeline.set(key, "$", write_obj)  # type: ignore[arg-type]
+                        pipeline.set(key, "$", write_obj)
                         created_keys.append(key)
 
             pipeline.execute()

langgraph/checkpoint/redis/shallow.py

@@ -555,10 +555,10 @@ def put_writes(
                         pipeline.set(key, "$.blob", write_obj["blob"])
                     else:
                         # For new records, set the complete object
-                        pipeline.set(key, "$", write_obj)  # type: ignore[arg-type]
+                        pipeline.set(key, "$", write_obj)
                 else:
                     # INSERT case
-                    pipeline.set(key, "$", write_obj)  # type: ignore[arg-type]
+                    pipeline.set(key, "$", write_obj)
 
             pipeline.execute()
 

langgraph/store/redis/aio.py

@@ -352,7 +352,7 @@ async def __aexit__(
 
         # Close Redis connections if we own them
         if self._owns_its_client:
-            await self._redis.aclose()  # type: ignore[attr-defined]
+            await self._redis.aclose()
             await self._redis.connection_pool.disconnect()
 
     async def abatch(self, ops: Iterable[Op]) -> list[Result]:

langgraph/store/redis/base.py

@@ -272,7 +272,7 @@ def set_client_info(self) -> None:
 
         try:
             # Try to use client_setinfo command if available
-            self._redis.client_setinfo("LIB-NAME", client_info)  # type: ignore
+            self._redis.client_setinfo("LIB-NAME", client_info)
         except (ResponseError, AttributeError):
             # Fall back to a simple echo if client_setinfo is not available
             try:
@@ -292,7 +292,7 @@ async def aset_client_info(self) -> None:
 
         try:
             # Try to use client_setinfo command if available
-            await self._redis.client_setinfo("LIB-NAME", client_info)  # type: ignore
+            await self._redis.client_setinfo("LIB-NAME", client_info)
         except (ResponseError, AttributeError):
             # Fall back to a simple echo if client_setinfo is not available
             try: