Multiple States, 1 Conditional Edge, but 2 Results...

[ahmed33033]

Hi there!

I was learning how to work with graphs in LangGraph, and I stumbled across a perplexing issue while working with conditional edges.

Graph Summary

I have 3 nodes. The first node, node_1, is only connected to node_2, while node_2 is connected to both node_3 and END via a conditional edge.

States Summary

I have 3 different states. Both PersonState and PurchasesState has unique attributes (seen below). On the other hand, OverallState combines the aforementioned states.

The Issue

node_2 issues an update to the state, which is not reflected correctly in the routing function (that controls the conditional edge).

Code

from operator import add
from typing import TypedDict, Annotated
from langgraph.graph import StateGraph, START, END

class PersonState(TypedDict):
    name: str
    money: Annotated[int, add]
    
class PurchasesState(TypedDict):
    purchases: Annotated[list[str], add]
    total_cost: Annotated[int, add]

class OverallState(PersonState, PurchasesState):
    pass


# node_1 operation. 
def pay_salary(state: PersonState) -> OverallState:
    print(f"Node 1: {state}")
    return {"money": 1000}

# node_2 operation. 
def subtract_expenses(state: PurchasesState) -> OverallState:
    print(f"Node 2: {state}")
    return {'money': -100, "purchases": ["new iphone"], 'total_cost': 100}

# node_3 operation. 
def add_assistance(state: OverallState) -> OverallState:
    print(f"Node 3: {state}")
    return {"money": 100}

# Routing Function. 
def check_if_poor(state: OverallState):
    print(f"Routing function: {state}")
    # if poor, add asssitance
    if (state['money'] < 100): return "node_3"
    else: return END


graph = StateGraph(OverallState, input = PersonState)
graph.add_edge(START, 'node_1')
graph.add_node('node_1', pay_salary)
graph.add_edge('node_1', 'node_2')

graph.add_node('node_2', subtract_expenses)
graph.add_conditional_edges('node_2', check_if_poor)

graph.add_node('node_3', add_assistance)
graph.add_edge("node_3", END)

workflow = graph.compile()
output_dict = workflow.invoke({"name": "Ahmed"})
print(f"Final output: {output_dict}")

Code Output

Node 1: {'name': 'Ahmed', 'money': 0}
Node 2: {'purchases': [], 'total_cost': 0}
Routing function: {'money': -100, 'purchases': ['new iphone'], 'total_cost': 100}
Node 3: {'name': 'Ahmed', 'money': 900, 'purchases': ['new iphone'], 'total_cost': 100}
Final output: {'name': 'Ahmed', 'money': 1000, 'purchases': ['new iphone'], 'total_cost': 100}

The output line that I can't understand

Routing function: {'money': -100, 'purchases': ['new iphone'], 'total_cost': 100}

How does money have a value of -100? In the method pay_salary, the money attribute is set to 1000. Then, in the subtract_expenses method, I subtract 100 from it. So, the money attribute should have a value of 900, not -100.

The line that is causing the weird output (after debugging)

def subtract_expenses(state: PurchasesState) -> OverallState:

How is it causing the issue?

When I change the inputted state from PurchasesState to OverallState, it works as expected, i.e. the output printed in the routing function check_if_poor is:

Routing function: {'money': 900, 'purchases': ['new iphone'], 'total_cost': 100, 'name': 'Ahmed'}

What's my question?

Why does changing the type of the inputted state in the function subtract_expenses lead me to receive different values for the money attribute in the routing function check_if_poor?

Thanks in advance!

P.S. Please don't mind how inefficient the graph setup is for the problem I'm trying to solve. I had simplified my original code a good bit of my code to make it easier to pinpoint what the issue was.