Overriding interrupts based on a conditional in a node/edge

[ishaan812]

Wanted to override interrupt_after in some cases based on a condition, should I just use a nodeInterrupt here instead of intterupt after. Although I prefer the interrupt_after/interrupt_before since it means all the navigation logic stays within the graph. So would prefer a way to override an interrupt_after based on condition within the code.

In the code below, I would want to based on some user input/ message type be able to override the intterupt after the start node as declared.

class WorkflowState(TypedDict):
    tenantId: str
    user_id: str
    messages: Annotated[list, add_messages]
    prev_context: str
    cardJSON: dict
    html: str
    finished: bool
    file: any


class CardWorkflow(WorkflowInterface):
    def __init__(self, tenantId: str = None, Checkpointer=None):
        self.graph = StateGraph(WorkflowState)
        self.tenantId = tenantId
        self._initialize_graph()
        if Checkpointer:
            self.workflow_instance = self.graph.compile(
                interrupt_after=["start", "parse_to_cardJSON"],
                checkpointer=Checkpointer,
            )
        else:
            self.workflow_instance = self.graph.compile(
                interrupt_after=["start", "parse_to_cardJSON"],
            )
        self._save_workflow_diagram(os.path.dirname(__file__))

    def _initialize_graph(self):
        """Setup the state graph for the workflow process."""

        self.graph.add_node('start', start_node)
        self.graph.add_node('wait_for_input', wait_for_chat)
        self.graph.add_node('generate_html', generate_html_ai)
        self.graph.add_node('parse_to_cardJSON', parse_to_cardJSON)
        self.graph.add_node('wait_for_chat', wait_for_chat)
        self.graph.add_node('end_chat', end_chat)

        # Define edges and start/finish points
        self.graph.add_edge("start", "wait_for_input")
        self.graph.add_edge('wait_for_input', 'generate_html')
        self.graph.add_edge('generate_html', 'parse_to_cardJSON')
        self.graph.add_edge('parse_to_cardJSON', 'wait_for_chat')
        self.graph
        self.graph.add_conditional_edges(
            "wait_for_chat",
            lambda state: "end_chat" if state["finished"] else "wait_for_input"
        )

        self.graph.set_entry_point('start')
        self.graph.set_finish_point('end_chat')