ArabicNewsAnalyzer commited on
Commit
88cd064
·
verified ·
1 Parent(s): 3e43e8b

Update app/memory/redis_checkpointer.py

Browse files
Files changed (1) hide show
  1. app/memory/redis_checkpointer.py +3 -8
app/memory/redis_checkpointer.py CHANGED
@@ -1,5 +1,5 @@
1
  from __future__ import annotations
2
- from redis.exceptions import ConnectionError as RedisConnectionError, TimeoutError as RedisTimeoutError
3
  from langgraph.checkpoint.redis.aio import AsyncRedisSaver
4
 
5
  from app.config import get_settings
@@ -9,6 +9,7 @@ _checkpointer: AsyncRedisSaver | None = None
9
 
10
 
11
  async def get_checkpointer() -> AsyncRedisSaver:
 
12
  global _checkpointer_cm, _checkpointer
13
  if _checkpointer is None:
14
  settings = get_settings()
@@ -16,12 +17,6 @@ async def get_checkpointer() -> AsyncRedisSaver:
16
 
17
  _checkpointer_cm = AsyncRedisSaver.from_conn_string(
18
  settings.redis_url,
19
- connection_args={
20
- "health_check_interval": 30,
21
- "socket_keepalive": True,
22
- "retry_on_timeout": True,
23
- "retry_on_error": [ConnectionError, TimeoutError],
24
- },
25
  ttl={"default_ttl": ttl_minutes, "refresh_on_read": True},
26
  )
27
  _checkpointer = await _checkpointer_cm.__aenter__()
@@ -34,4 +29,4 @@ async def close_checkpointer() -> None:
34
  if _checkpointer_cm is not None:
35
  await _checkpointer_cm.__aexit__(None, None, None)
36
  _checkpointer = None
37
- _checkpointer_cm = None
 
1
  from __future__ import annotations
2
+
3
  from langgraph.checkpoint.redis.aio import AsyncRedisSaver
4
 
5
  from app.config import get_settings
 
9
 
10
 
11
  async def get_checkpointer() -> AsyncRedisSaver:
12
+
13
  global _checkpointer_cm, _checkpointer
14
  if _checkpointer is None:
15
  settings = get_settings()
 
17
 
18
  _checkpointer_cm = AsyncRedisSaver.from_conn_string(
19
  settings.redis_url,
 
 
 
 
 
 
20
  ttl={"default_ttl": ttl_minutes, "refresh_on_read": True},
21
  )
22
  _checkpointer = await _checkpointer_cm.__aenter__()
 
29
  if _checkpointer_cm is not None:
30
  await _checkpointer_cm.__aexit__(None, None, None)
31
  _checkpointer = None
32
+ _checkpointer_cm = None