state.py

This file is where definition of what information flows through the graph gets stored. LangGraph passes a shared state object, basically saying that this graph will store order_id and messages. This will be more directly relevant in our testing.

from typing_extensions import TypedDict
from langgraph.graph import StateGraph
import pandas as pd

class GraphState(TypedDict):
    order_id: str
    messages: list
    df: pd.DataFrame