Looping issue using LangGraph with multiple agents

[pinktownscavenger]

I have this base code that I'm using to create a graph with three nodes; human (for human input), template_selection, and information_gathering. The problem is that there are multiple outputs, which is confusing. I appreciate any help you can provide.

Code:

def human_node(state: State, config) -> Command:
    user_input = interrupt(
        {
            'input': 'Enter'
        }
    )['input']
    ...
    return Command(update={"messages": updated_messages}, goto=state["next_node"])

def template_selection_node(state: State, config) -> Command[Literal["human","information_gathering"]]:
    ...
    if assistant_response == 'template_selection':
        return Command(update={"messages": new_messages, "next_node": assistant_response}, goto="human")
    else:
        return Command(update={"messages": new_messages, "next_node": assistant_response}, goto="information_gathering")

def information_gathering_node(state:State) -> Command[Literal["human"]]:
    ...
    return Command(update={"next_node": "information_gathering"},goto='human')

while True:
    for chunk in graph.stream(initial_state, config):
        for node_id, value in chunk.items():
            if node_id == "__interrupt__":
                user_input = input("Enter: ")
                current_state = graph.invoke(
                    Command(resume={"input": user_input}),
                    config
                )

Output:

Assistant Response: template_selection
Routing to human...
Enter: Hi
Assistant Response: template_selection
Routing to human...
Assistant Response: template_selection
Routing to human...
Enter: meow
Assistant Response: information_gathering
Routing to information gathering...
Entered Information Gathering with information_gathering.
Assistant Response: template_selection
Routing to human...
Enter: