File size: 464 Bytes
9acd478
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from langgraph.graph import StateGraph, START, END , MessagesState
from langgraph.checkpoint.memory import MemorySaver
from .utils.nodes import run_refiner
from .utils.state import State
memory = MemorySaver()


def human_refined_idea():
    graph_builder= StateGraph(State)
    graph_builder.add_node(run_refiner)
    graph_builder.add_edge(START, "run_refiner")
    graph_builder.add_edge("run_refiner",END)
    return graph_builder.compile(checkpointer=memory)