Skip to content

Commit d8cb5e1

Browse files
tylerhutchersontylerhutcherson
committed
update dependencies and use black for code formatting
1 parent 3e0990f commit d8cb5e1

27 files changed

+519
-2213
lines changed

‎langgraph/checkpoint/redis/__init__.py‎

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -6,25 +6,25 @@
66
from typing import Any, List, Optional, Tuple, cast
77

88
from langchain_core.runnables import RunnableConfig
9-
from redisvl.index import SearchIndex
10-
from redisvl.query import FilterQuery
11-
from redisvl.query.filter import Num, Tag
12-
from redisvl.redis.connection import RedisConnectionFactory
13-
149
from langgraph.checkpoint.base import (
1510
ChannelVersions,
1611
Checkpoint,
1712
CheckpointMetadata,
1813
CheckpointTuple,
1914
get_checkpoint_id,
2015
)
16+
from langgraph.constants import TASKS
17+
from redis import Redis
18+
from redisvl.index import SearchIndex
19+
from redisvl.query import FilterQuery
20+
from redisvl.query.filter import Num, Tag
21+
from redisvl.redis.connection import RedisConnectionFactory
22+
2123
from langgraph.checkpoint.redis.aio import AsyncRedisSaver
2224
from langgraph.checkpoint.redis.ashallow import AsyncShallowRedisSaver
2325
from langgraph.checkpoint.redis.base import BaseRedisSaver
2426
from langgraph.checkpoint.redis.shallow import ShallowRedisSaver
2527
from langgraph.checkpoint.redis.version import __lib_name__, __version__
26-
from langgraph.constants import TASKS
27-
from redis import Redis
2828

2929

3030
class RedisSaver(BaseRedisSaver[Redis, SearchIndex]):
@@ -152,9 +152,9 @@ def list(
152152

153153
# Ensure metadata matches CheckpointMetadata type
154154
sanitized_metadata = {
155-
k.replace("\u0000", ""): v.replace("\u0000", "")
156-
if isinstance(v, str)
157-
else v
155+
k.replace("\u0000", ""): (
156+
v.replace("\u0000", "") if isinstance(v, str) else v
157+
)
158158
for k, v in metadata_dict.items()
159159
}
160160
metadata = cast(CheckpointMetadata, sanitized_metadata)
@@ -318,9 +318,9 @@ def get_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:
318318

319319
# Ensure metadata matches CheckpointMetadata type
320320
sanitized_metadata = {
321-
k.replace("\u0000", ""): v.replace("\u0000", "")
322-
if isinstance(v, str)
323-
else v
321+
k.replace("\u0000", ""): (
322+
v.replace("\u0000", "") if isinstance(v, str) else v
323+
)
324324
for k, v in metadata_dict.items()
325325
}
326326
metadata = cast(CheckpointMetadata, sanitized_metadata)

‎langgraph/checkpoint/redis/aio.py‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -11,11 +11,6 @@
1111
from typing import Any, List, Optional, Sequence, Tuple, Type, cast
1212

1313
from langchain_core.runnables import RunnableConfig
14-
from redisvl.index import AsyncSearchIndex
15-
from redisvl.query import FilterQuery
16-
from redisvl.query.filter import Num, Tag
17-
from redisvl.redis.connection import RedisConnectionFactory
18-
1914
from langgraph.checkpoint.base import (
2015
WRITES_IDX_MAP,
2116
ChannelVersions,
@@ -25,10 +20,15 @@
2520
PendingWrite,
2621
get_checkpoint_id,
2722
)
28-
from langgraph.checkpoint.redis.base import BaseRedisSaver
2923
from langgraph.constants import TASKS
3024
from redis.asyncio import Redis as AsyncRedis
3125
from redis.asyncio.client import Pipeline
26+
from redisvl.index import AsyncSearchIndex
27+
from redisvl.query import FilterQuery
28+
from redisvl.query.filter import Num, Tag
29+
from redisvl.redis.connection import RedisConnectionFactory
30+
31+
from langgraph.checkpoint.redis.base import BaseRedisSaver
3232

3333

3434
async def _write_obj_tx(
@@ -189,9 +189,9 @@ async def aget_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:
189189

190190
# Ensure metadata matches CheckpointMetadata type
191191
sanitized_metadata = {
192-
k.replace("\u0000", ""): v.replace("\u0000", "")
193-
if isinstance(v, str)
194-
else v
192+
k.replace("\u0000", ""): (
193+
v.replace("\u0000", "") if isinstance(v, str) else v
194+
)
195195
for k, v in metadata_dict.items()
196196
}
197197
metadata = cast(CheckpointMetadata, sanitized_metadata)
@@ -308,9 +308,9 @@ async def alist(
308308

309309
# Ensure metadata matches CheckpointMetadata type
310310
sanitized_metadata = {
311-
k.replace("\u0000", ""): v.replace("\u0000", "")
312-
if isinstance(v, str)
313-
else v
311+
k.replace("\u0000", ""): (
312+
v.replace("\u0000", "") if isinstance(v, str) else v
313+
)
314314
for k, v in metadata_dict.items()
315315
}
316316
metadata = cast(CheckpointMetadata, sanitized_metadata)

‎langgraph/checkpoint/redis/ashallow.py‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,6 @@
1010
from typing import Any, AsyncIterator, Dict, List, Optional, Sequence, Tuple, Type, cast
1111

1212
from langchain_core.runnables import RunnableConfig
13-
from redisvl.index import AsyncSearchIndex
14-
from redisvl.query import FilterQuery
15-
from redisvl.query.filter import Num, Tag
16-
from redisvl.redis.connection import RedisConnectionFactory
17-
1813
from langgraph.checkpoint.base import (
1914
WRITES_IDX_MAP,
2015
ChannelVersions,
@@ -23,16 +18,21 @@
2318
CheckpointTuple,
2419
PendingWrite,
2520
)
21+
from langgraph.constants import TASKS
22+
from redis.asyncio import Redis as AsyncRedis
23+
from redis.asyncio.client import Pipeline
24+
from redisvl.index import AsyncSearchIndex
25+
from redisvl.query import FilterQuery
26+
from redisvl.query.filter import Num, Tag
27+
from redisvl.redis.connection import RedisConnectionFactory
28+
2629
from langgraph.checkpoint.redis.base import (
2730
CHECKPOINT_BLOB_PREFIX,
2831
CHECKPOINT_PREFIX,
2932
CHECKPOINT_WRITE_PREFIX,
3033
REDIS_KEY_SEPARATOR,
3134
BaseRedisSaver,
3235
)
33-
from langgraph.constants import TASKS
34-
from redis.asyncio import Redis as AsyncRedis
35-
from redis.asyncio.client import Pipeline
3636

3737
SCHEMAS = [
3838
{

‎langgraph/checkpoint/redis/base.py‎

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
from typing import Any, Generic, List, Optional, cast
88

99
from langchain_core.runnables import RunnableConfig
10-
1110
from langgraph.checkpoint.base import (
1211
WRITES_IDX_MAP,
1312
BaseCheckpointSaver,
@@ -200,12 +199,14 @@ def _dump_blobs(
200199
"checkpoint_ns": checkpoint_ns,
201200
"channel": k,
202201
"version": cast(str, ver),
203-
"type": self._get_type_and_blob(values[k])[0]
204-
if k in values
205-
else "empty",
206-
"blob": self._get_type_and_blob(values[k])[1]
207-
if k in values
208-
else None,
202+
"type": (
203+
self._get_type_and_blob(values[k])[0]
204+
if k in values
205+
else "empty"
206+
),
207+
"blob": (
208+
self._get_type_and_blob(values[k])[1] if k in values else None
209+
),
209210
},
210211
)
211212
for k, ver in versions.items()

‎langgraph/checkpoint/redis/shallow.py‎

Lines changed: 21 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -5,27 +5,27 @@
55
from typing import Any, Dict, Iterator, List, Optional, Sequence, Tuple, cast
66

77
from langchain_core.runnables import RunnableConfig
8-
from redisvl.index import SearchIndex
9-
from redisvl.query import FilterQuery
10-
from redisvl.query.filter import Num, Tag
11-
from redisvl.redis.connection import RedisConnectionFactory
12-
138
from langgraph.checkpoint.base import (
149
WRITES_IDX_MAP,
1510
ChannelVersions,
1611
Checkpoint,
1712
CheckpointMetadata,
1813
CheckpointTuple,
1914
)
15+
from langgraph.constants import TASKS
16+
from redis import Redis
17+
from redisvl.index import SearchIndex
18+
from redisvl.query import FilterQuery
19+
from redisvl.query.filter import Num, Tag
20+
from redisvl.redis.connection import RedisConnectionFactory
21+
2022
from langgraph.checkpoint.redis.base import (
2123
CHECKPOINT_BLOB_PREFIX,
2224
CHECKPOINT_PREFIX,
2325
CHECKPOINT_WRITE_PREFIX,
2426
REDIS_KEY_SEPARATOR,
2527
BaseRedisSaver,
2628
)
27-
from langgraph.constants import TASKS
28-
from redis import Redis
2929

3030
SCHEMAS = [
3131
{
@@ -250,9 +250,9 @@ def list(
250250

251251
# Ensure metadata matches CheckpointMetadata type
252252
sanitized_metadata = {
253-
k.replace("\u0000", ""): v.replace("\u0000", "")
254-
if isinstance(v, str)
255-
else v
253+
k.replace("\u0000", ""): (
254+
v.replace("\u0000", "") if isinstance(v, str) else v
255+
)
256256
for k, v in metadata_dict.items()
257257
}
258258
metadata = cast(CheckpointMetadata, sanitized_metadata)
@@ -342,9 +342,9 @@ def get_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:
342342

343343
# Ensure metadata matches CheckpointMetadata type
344344
sanitized_metadata = {
345-
k.replace("\u0000", ""): v.replace("\u0000", "")
346-
if isinstance(v, str)
347-
else v
345+
k.replace("\u0000", ""): (
346+
v.replace("\u0000", "") if isinstance(v, str) else v
347+
)
348348
for k, v in metadata_dict.items()
349349
}
350350
metadata = cast(CheckpointMetadata, sanitized_metadata)
@@ -478,12 +478,14 @@ def _dump_blobs(
478478
"thread_id": thread_id,
479479
"checkpoint_ns": checkpoint_ns,
480480
"channel": k,
481-
"type": self._get_type_and_blob(values[k])[0]
482-
if k in values
483-
else "empty",
484-
"blob": self._get_type_and_blob(values[k])[1]
485-
if k in values
486-
else None,
481+
"type": (
482+
self._get_type_and_blob(values[k])[0]
483+
if k in values
484+
else "empty"
485+
),
486+
"blob": (
487+
self._get_type_and_blob(values[k])[1] if k in values else None
488+
),
487489
},
488490
)
489491
for k, ver in versions.items()

‎langgraph/checkpoint/redis/types.py‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
11
from typing import Any, Optional, TypeVar, Union
22

3-
from redisvl.index import AsyncSearchIndex, SearchIndex
4-
53
from redis import Redis
64
from redis.asyncio import Redis as AsyncRedis
5+
from redisvl.index import AsyncSearchIndex, SearchIndex
76

87
RedisClientType = TypeVar("RedisClientType", bound=Union[Redis, AsyncRedis])
98
IndexType = TypeVar("IndexType", bound=Union[SearchIndex, AsyncSearchIndex])

‎langgraph/docs/.gitignore‎

Lines changed: 0 additions & 4 deletions
This file was deleted.

‎langgraph/docs/Dockerfile.jupyter‎

Lines changed: 0 additions & 35 deletions
This file was deleted.

‎langgraph/docs/README.md‎

Lines changed: 0 additions & 29 deletions
This file was deleted.

0 commit comments

Comments
 (0)