# 对照完整文件阅读这一段；此处保留当前步骤使用的函数或类型。
def approve(state: RunState) -> dict:
    # 恢复时会从这个函数开头重走；这里先展示动作，不执行写入。
    decision = interrupt({"action": "save_plan", "operation_id": state["operation_id"],
                          "question": state["question"], "minutes": state["minutes"]})
    if type(decision) is not bool:
        raise ValueError("确认结果需要是布尔值")
    return {"approved": decision, "status": "approved" if decision else "rejected"}

"""用进入记录核对 interrupt 恢复时从函数开头重新执行。"""
from typing import TypedDict
from langgraph.checkpoint.memory import InMemorySaver
from langgraph.graph import END, START, StateGraph
from langgraph.types import Command, interrupt


class State(TypedDict, total=False):
    minutes: int
    approved: bool


entered = []  # 这里只记录函数执行顺序，正式计划不在这里写入。


def approve(state: State):
    entered.append("进入 approve")
    decision = interrupt({"minutes": state["minutes"]})
    entered.append(f"拿到决定：{decision}")
    return {"approved": decision}


builder = StateGraph(State)
builder.add_node("approve", approve)
builder.add_edge(START, "approve")
builder.add_edge("approve", END)
graph = builder.compile(checkpointer=InMemorySaver())
config = {"configurable": {"thread_id": "reentry"}}
first = graph.invoke({"minutes": 35}, config)
assert first["__interrupt__"][0].value == {"minutes": 35}
assert entered == ["进入 approve"]
print("暂停：", entered)
# 同一进程中先看清重入；持久化重开见 checkpoint_walk，跨进程恢复用 CLI 核对。
second = graph.invoke(Command(resume=True), config)
assert second["approved"] is True
assert entered == ["进入 approve", "进入 approve", "拿到决定：True"]
print("恢复：", entered)

# 对照完整文件阅读这一段；此处保留当前步骤使用的函数或类型。
def pause_and_close(root: Path):
    config = run_config("alice", "walk")
    context = context_for(root, "alice")
    # 演示把随机编号换成固定文字，便于沿图追踪同一项业务。
    initial = {"question": QUESTION, "write_plan": True, "minutes": 35,
               "operation_id": "op-demo-1", "hits": [], "approved": False, "status": "new"}
    with SqliteSaver.from_conn_string(str(root / "checkpoints.sqlite3")) as saver:
        graph = build_graph(saver)
        updates = list(graph.stream(initial, config, context=context, stream_mode="updates"))
        snapshot = describe(graph, config)
        assert [next(iter(item)) for item in updates] == ["retrieve", "draft", "check", "__interrupt__"]
        assert snapshot["state"]["status"] == "answered"
        assert snapshot["next"] == ["approve"] and snapshot["pending"]
        assert "plan_id" not in snapshot["state"] and context.plans.list("alice") == []
    return updates, snapshot

# 对照完整文件阅读这一段；此处保留当前步骤使用的函数或类型。
def reopen_and_approve(root: Path):
    # 重新建立图、连接与上下文，再按同一个 thread_id 读取检查点。
    config = run_config("alice", "walk")
    context = context_for(root, "alice")
    with SqliteSaver.from_conn_string(str(root / "checkpoints.sqlite3")) as saver:
        graph = build_graph(saver)
        before = describe(graph, config)
        assert before["pending"][0]["operation_id"] == "op-demo-1"
        updates = list(graph.stream(Command(resume=True), config, context=context, stream_mode="updates"))
        after = describe(graph, config)
        assert [next(iter(item)) for item in updates] == ["approve", "save"]
        assert after["state"]["status"] == "saved" and after["next"] == []
        assert after["pending"] == [] and after["state"]["plan_id"] == 1
    assert len(context.plans.list("alice")) == 1
    return updates, after

# 对照完整文件阅读这一段；此处保留当前步骤使用的函数或类型。
def resume(graph, context: RunContext, thread: str, approved: bool) -> dict:
    config = run_config(context.owner, thread)
    if not describe(graph, config)["pending"]:
        raise ValueError("这个任务没有等待确认的动作")
    graph.invoke(Command(resume=approved), config=config, context=context)
    return describe(graph, config)