Update state of graph in an edge

[kom-senapati]

I am implementing https://langchain-ai.github.io/langgraph/tutorials/rag/langgraph_self_rag. But it went into a loop and gave unwanted outputs as I added messages state as well.

def decide_to_generate(state):
    """
    Determines whether to generate an answer, or re-generate a question.

    Args:
        state (dict): The current graph state

    Returns:
        str: Binary decision for next node to call
    """

    print("---ASSESS GRADED DOCUMENTS---")
    state["question"]
    filtered_documents = state["documents"]

    if not filtered_documents:
        print(f"max_retries_rewrite: {state['max_retries_rewrite']}")
        if state["max_retries_rewrite"] < 2:
            state.update({"max_retries_rewrite": state["max_retries_rewrite"] + 1})

            print(
                "---DECISION: ALL DOCUMENTS ARE NOT RELEVANT TO QUESTION, TRANSFORM QUERY---"
            )
            return "transform_query"
        else:
            state.update({"max_retries_rewrite": 0})

            print("---DECISION: ALL DOCUMENTS ARE NOT RELEVANT TO QUESTION, STOP---")
            return "no_context"
    else:
        print("---DECISION: GENERATE---")
        return "generate"

Here I can't update the state, it always shows me 0.