diff --git a/libs/checkpoint-postgres/langgraph/store/postgres/__init__.py b/libs/checkpoint-postgres/langgraph/store/postgres/__init__.py index 47b7836d2..d0432b588 100644 --- a/libs/checkpoint-postgres/langgraph/store/postgres/__init__.py +++ b/libs/checkpoint-postgres/langgraph/store/postgres/__init__.py @@ -1,4 +1,4 @@ from langgraph.store.postgres.aio import AsyncPostgresStore -from langgraph.store.postgres.base import PostgresStore +from langgraph.store.postgres.base import PoolConfig, PostgresStore -__all__ = ["AsyncPostgresStore", "PostgresStore"] +__all__ = ["AsyncPostgresStore", "PoolConfig", "PostgresStore"]