Praudyog

from langgraph.graph import StateGraph, MessagesState, START, END

def mock_llm(state:MessagesState):
    return("messages":[{"role":"ai", "content":"Hello World"}])
    
graph = StateGraph(MessagesState)

graph.add_node("mock_llm", mock_llm)

grapg.add_edge(START, "mock_llm")

graph.add_edge("mock_llm", END)

workflow = graph.compile()

workflow.invoke({"messages":[{"role":"user", "content":"Hi"}]})