Backwards compat exports

langchain-ai · Nov 25, 2024 · a13e657 · a13e657
1 parent 625b530
commit a13e657
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/libs/checkpoint-postgres/langgraph/checkpoint/postgres/__init__.py b/libs/checkpoint-postgres/langgraph/checkpoint/postgres/__init__.py
@@ -21,6 +21,8 @@
 from langgraph.checkpoint.postgres.base import BasePostgresSaver
 from langgraph.checkpoint.serde.base import SerializerProtocol
 
+Conn = _internal.Conn  # For backward compatibility
+
 
 class PostgresSaver(BasePostgresSaver):
     lock: threading.Lock

diff --git a/libs/checkpoint-postgres/langgraph/checkpoint/postgres/aio.py b/libs/checkpoint-postgres/langgraph/checkpoint/postgres/aio.py
@@ -21,6 +21,8 @@
 from langgraph.checkpoint.postgres.base import BasePostgresSaver
 from langgraph.checkpoint.serde.base import SerializerProtocol
 
+Conn = _ainternal.Conn  # For backward compatibility
+
 
 class AsyncPostgresSaver(BasePostgresSaver):
     lock: asyncio.Lock
@@ -457,3 +459,6 @@ def put_writes(
         return asyncio.run_coroutine_threadsafe(
             self.aput_writes(config, writes, task_id), self.loop
         ).result()
+
+
+__all__ = ["AsyncPostgresSaver", "Conn"]