wip on key sep

Author: tylerhutcherson

langgraph/checkpoint/redis/__init__.py

@@ -303,7 +303,7 @@ def get_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:
 
         # Fetch pending_sends from parent checkpoint
         pending_sends = []
-        if doc["parent_checkpoint_id"]:
+        if "parent_checkpoint_id" in doc:
             pending_sends = self._load_pending_sends(
                 thread_id=thread_id,
                 checkpoint_ns=checkpoint_ns,

langgraph/checkpoint/redis/ashallow.py

@@ -38,7 +38,8 @@
     {
         "index": {
             "name": "checkpoints",
-            "prefix": CHECKPOINT_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": CHECKPOINT_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [
@@ -51,7 +52,8 @@
     {
         "index": {
             "name": "checkpoints_blobs",
-            "prefix": CHECKPOINT_BLOB_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": CHECKPOINT_BLOB_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [
@@ -64,7 +66,8 @@
     {
         "index": {
             "name": "checkpoint_writes",
-            "prefix": CHECKPOINT_WRITE_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": CHECKPOINT_WRITE_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [

langgraph/checkpoint/redis/base.py

@@ -31,7 +31,8 @@
     {
         "index": {
             "name": "checkpoints",
-            "prefix": CHECKPOINT_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": CHECKPOINT_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [
@@ -46,7 +47,8 @@
     {
         "index": {
             "name": "checkpoints_blobs",
-            "prefix": CHECKPOINT_BLOB_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": CHECKPOINT_BLOB_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [
@@ -60,7 +62,8 @@
     {
         "index": {
             "name": "checkpoint_writes",
-            "prefix": CHECKPOINT_WRITE_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": CHECKPOINT_WRITE_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [

langgraph/checkpoint/redis/shallow.py

@@ -31,7 +31,8 @@
     {
         "index": {
             "name": "checkpoints",
-            "prefix": CHECKPOINT_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": CHECKPOINT_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [
@@ -44,7 +45,8 @@
     {
         "index": {
             "name": "checkpoints_blobs",
-            "prefix": CHECKPOINT_BLOB_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": CHECKPOINT_BLOB_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [
@@ -57,7 +59,8 @@
     {
         "index": {
             "name": "checkpoint_writes",
-            "prefix": CHECKPOINT_WRITE_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": CHECKPOINT_WRITE_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [

langgraph/store/redis/__init__.py

@@ -25,7 +25,7 @@
 from redisvl.query import FilterQuery, VectorQuery
 from redisvl.redis.connection import RedisConnectionFactory
 from redisvl.utils.token_escaper import TokenEscaper
-from ulid import ULID
+from redisvl.utils.utils import create_ulid
 
 from langgraph.store.redis.aio import AsyncRedisStore
 from langgraph.store.redis.base import (
@@ -223,7 +223,7 @@ def _batch_put_ops(
         # Generate IDs for PUT operations
         for _, op in put_ops:
             if op.value is not None:
-                generated_doc_id = str(ULID())
+                generated_doc_id = create_ulid()
                 namespace = _namespace_to_text(op.namespace)
                 doc_ids[(namespace, op.key)] = generated_doc_id
 

langgraph/store/redis/base.py

@@ -44,7 +44,8 @@
     {
         "index": {
             "name": "store",
-            "prefix": STORE_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": STORE_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [
@@ -57,7 +58,8 @@
     {
         "index": {
             "name": "store_vectors",
-            "prefix": STORE_VECTOR_PREFIX + REDIS_KEY_SEPARATOR,
+            "prefix": STORE_VECTOR_PREFIX,
+            "key_separator": REDIS_KEY_SEPARATOR,
             "storage_type": "json",
         },
         "fields": [