AsyncPostgresSaver with interrupts in Python 3.10 seems to be broken

[RecallSpace]

I am trying to build a graph that takes human input and uses the asynchronous PostgreSQL checkpointer. However, I couldn't make it work in my usual Python 3.10 environment. I keep getting the following error:

RuntimeError: Called get_config outside of a runnable context

After some research, I found a few posts discussing the differences between psycopg2 and psycopg, and others mentioning compatibility issues. Consequently, I tried using Python 3.12, and then it worked. Here is the code:

from typing import TypedDict
from langgraph.constants import START
from langgraph.graph import StateGraph
from langgraph.types import interrupt
from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
from psycopg_pool import AsyncConnectionPool

class State(TypedDict):
   """The graph state."""
   some_text: str

def human_node(state: State):
   value = interrupt(
      # Any JSON serializable value to surface to the human.
      # For example, a question or a piece of text or a set of keys in the state
      {
         "text_to_revise": state["some_text"]
      }
   )
   return {
      # Update the state with the human's input
      "some_text": value
   }


# Build the graph
graph_builder = StateGraph(State)
# Add the human-node to the graph
graph_builder.add_node("human_node", human_node)
graph_builder.add_edge(START, "human_node")

DB_URI = "postgresql://user:xxxxx@localhost:5432/demo-db?sslmode=disable"

connection_kwargs = {
    "autocommit": True,
    "prepare_threshold": 0,
}

async def main():
   async with AsyncConnectionPool(
      # Example configuration
      conninfo=DB_URI,
      max_size=20,
      kwargs=connection_kwargs,
   ) as pool:
      # A checkpointer is required for `interrupt` to work.
      checkpointer = AsyncPostgresSaver(pool)
      graph = graph_builder.compile(
         checkpointer=checkpointer
      )

      # Pass a thread ID to the graph to run it.
      thread_config = {"configurable": {"thread_id": '42'}}

      r =  await graph.ainvoke(
         {"some_text": "Original text"}, 
         config=thread_config)
      
      return r
   
if __name__ == '__main__':
    import asyncio
    res = asyncio.run(main())
    print(res)

I tried the same with synchronous client, that one worked both on 3.10 and 3.12:

from langgraph.checkpoint.postgres import PostgresSaver
from psycopg_pool import ConnectionPool

def main():
    with ConnectionPool(
        # Example configuration
        conninfo=DB_URI,
        max_size=20,
        kwargs=connection_kwargs,
    ) as pool:
        # A checkpointer is required for `interrupt` to work.
        checkpointer = PostgresSaver(pool)
        graph = graph_builder.compile(
            checkpointer=checkpointer
        )

        # Pass a thread ID to the graph to run it.
        thread_config = {"configurable": {"thread_id": '42'}}

        r = graph.invoke(
            {"some_text": "Original text"}, 
            config=thread_config)

        return r
   
if __name__ == '__main__':
    res =  main()
    print(res)
        

I am using ubuntu WSL with:

psycopg                             3.2.4
psycopg-binary                      3.2.4
psycopg-pool                        3.2.4
langchain-core                      0.3.31
langgraph                           0.2.67
langgraph-checkpoint                2.0.10
langgraph-checkpoint-postgres       2.0.13
langgraph-sdk                       0.1.51
langsmith                           0.3.1

[hinthornw]

Yes - most async work in langgraph requires >=3.11 - tasks don't propagate context correctly prior to that, unfortunately