feat: add ttl support to vector store

libs/redis/langchain_redis/vectorstores.py

@@ -159,6 +159,8 @@ class RedisVectorStore(VectorStore):
             URL of the Redis instance to connect to.
         redis_client: Optional[Redis]
             Pre-existing Redis connection.
+        ttl: Optional[int]
+            Time-to-live for the Redis keys.
 
     Instantiate:
         .. code-block:: python
@@ -265,10 +267,12 @@ def __init__(
         self,
         embeddings: Embeddings,
         config: Optional[RedisConfig] = None,
+        ttl: Optional[int] = None,
         **kwargs: Any,
     ):
         self.config = config or RedisConfig(**kwargs)
         self._embeddings = embeddings
+        self.ttl = ttl
 
         if self.config.embedding_dimensions is None:
             self.config.embedding_dimensions = len(
@@ -441,10 +445,12 @@ def add_texts(
 
         result = (
             self._index.load(
-                datas, keys=[f"{self.config.key_prefix}:{key}" for key in keys]
+                datas,
+                keys=[f"{self.config.key_prefix}:{key}" for key in keys],
+                ttl=self.ttl,
             )
             if keys
-            else self._index.load(datas)
+            else self._index.load(datas, ttl=self.ttl)
         )
 
         return list(result) if result is not None else []

libs/redis/tests/unit_tests/test_vectorstores.py

@@ -70,7 +70,10 @@ def create(self, overwrite: bool = False) -> None:
         pass
 
     def load(
-        self, documents: List[Dict[str, Any]], keys: Optional[List[str]] = None
+        self,
+        documents: List[Dict[str, Any]],
+        keys: Optional[List[str]] = None,
+        ttl: Optional[int] = None,
     ) -> List[str]:
         for i, doc in enumerate(documents):
             key = keys[i] if keys else f"key_{i}"