feat: extend RedisVL version support to 0.7.0

README.md

@@ -327,7 +327,7 @@ We welcome contributions! Here's how you can help:
 2. Install dependencies:
 
    ```bash
-   poetry install --all-extras
+   `poetry install --all-extras`
    ```
 
 ### Available Commands

pyproject.toml

@@ -20,7 +20,7 @@ packages = [{ include = "langgraph" }]
 [tool.poetry.dependencies]
 python = ">=3.9,<3.14"
 langgraph-checkpoint = "^2.0.24"
-redisvl = "^0.5.1"
+redisvl = ">=0.5.1,<0.8.0"
 redis = "^5.2.1"
 python-ulid = "^3.0.0"
 

tests/test_interruption.py

@@ -24,7 +24,7 @@ class InterruptionError(Exception):
     pass
 
 
-class MockRedis:
+class MockRedis(Redis):
     """Mock Redis class that can simulate interruptions during operations."""
 
     def __init__(self, real_redis: Redis, interrupt_on: str = None) -> None:
@@ -34,14 +34,36 @@ def __init__(self, real_redis: Redis, interrupt_on: str = None) -> None:
             real_redis: The real Redis client to delegate to
             interrupt_on: Operation name to interrupt on (e.g., 'json().set', 'Pipeline.execute')
         """
+        # Copy connection info from real_redis to satisfy Redis base class
+        super().__init__(
+            connection_pool=real_redis.connection_pool,
+            single_connection_client=real_redis.single_connection_client,
+        )
         self.real_redis = real_redis
         self.interrupt_on = interrupt_on
         self.operations_count = {}
         self.interrupt_after_count = {}
 
-    def __getattr__(self, name):
+    def __getattribute__(self, name):
         """Proxy attribute access to the real Redis client, but track operations."""
-        attr = getattr(self.real_redis, name)
+        # For special attributes we've set in __init__, use the parent implementation
+        if name in [
+            "real_redis",
+            "interrupt_on",
+            "operations_count",
+            "interrupt_after_count",
+        ]:
+            return super().__getattribute__(name)
+
+        # For Redis base class attributes
+        if name in ["connection_pool", "single_connection_client", "_parser", "_lock"]:
+            return super().__getattribute__(name)
+
+        try:
+            attr = getattr(self.real_redis, name)
+        except AttributeError:
+            # Fall back to parent class
+            return super().__getattribute__(name)
 
         # For methods we want to potentially interrupt
         if callable(attr) and name == self.interrupt_on:

tests/test_shallow_sync.py

@@ -269,7 +269,10 @@ def test_from_conn_string_errors(redis_url: str) -> None:
             saver.setup()
 
     # Test with empty URL
-    with pytest.raises(ValueError, match="REDIS_URL env var not set"):
+    # Handle both old and new RedisVL error message formats
+    with pytest.raises(
+        ValueError, match="REDIS_URL (env var|environment variable) not set"
+    ):
         with ShallowRedisSaver.from_conn_string("") as saver:
             saver.setup()
 

tests/test_sync.py

@@ -451,7 +451,10 @@ def test_from_conn_string_errors() -> None:
             saver.setup()
 
     # Test with empty URL
-    with pytest.raises(ValueError, match="REDIS_URL env var not set"):
+    # Handle both old and new RedisVL error message formats
+    with pytest.raises(
+        ValueError, match="REDIS_URL (env var|environment variable) not set"
+    ):
         with RedisSaver.from_conn_string("") as saver:
             saver.setup()