from pydantic import BaseModel, Field
from langgraph.graph import StateGraph

class Foo(BaseModel):
    foo: str = Field(alias='bar')

# Define a new graph
graph_builder = StateGraph(
    Foo
)


def node(foo: Foo):
    print(foo)
    return None

# Add the node to the graph
graph_builder.add_node("node", node)
# Set the entrypoint as `call_model`
graph_builder.add_edge("__start__", "node")

# Compile the workflow into an executable graph
graph = graph_builder.compile()


graph.invoke({'bar': 'hello'})