src/genai/human_refined_ideation/agent.py

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)