Solution ready for review: update hitl_agent.py
This commit is contained in:
+2
-1
@@ -3,6 +3,7 @@ from __future__ import annotations
|
|||||||
from typing import TypedDict
|
from typing import TypedDict
|
||||||
|
|
||||||
from langgraph.graph import END, START, StateGraph
|
from langgraph.graph import END, START, StateGraph
|
||||||
|
from langgraph.checkpoint.memory import InMemorySaver
|
||||||
from langgraph.types import Command, interrupt
|
from langgraph.types import Command, interrupt
|
||||||
|
|
||||||
|
|
||||||
@@ -50,7 +51,7 @@ def build_graph():
|
|||||||
graph.add_edge("intro", "ask_human")
|
graph.add_edge("intro", "ask_human")
|
||||||
graph.add_edge("ask_human", "resolve")
|
graph.add_edge("ask_human", "resolve")
|
||||||
graph.add_edge("resolve", END)
|
graph.add_edge("resolve", END)
|
||||||
return graph.compile()
|
return graph.compile(checkpointer=InMemorySaver())
|
||||||
|
|
||||||
|
|
||||||
def main() -> None:
|
def main() -> None:
|
||||||
|
|||||||
Reference in New Issue
Block a user