import asyncio

from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
from psycopg_pool import AsyncConnectionPool


async def test():
    pool = AsyncConnectionPool("", open=False)
    saver = AsyncPostgresSaver(pool)
    print(f"Attributes with 'pool': {[a for a in dir(saver) if 'pool' in a.lower()]}")
    await pool.close()


if __name__ == "__main__":
    asyncio.run(test())
