3、grach基础流程

grach基础流程

• 定义状态
• 定义节点
• 定义边

from typing import TypedDict, Annotated  
from langgraph.graph import StateGraph,START,END  
from operator import add  
  
#1.定义状态  
class OverAllState(TypedDict):  
    logs: Annotated[list[str],add]  
    cur_id: str  
  
  
  
def node_1(state: OverAllState) -> OverAllState:  
    pre_id = state["cur_id"]  
    return {  
        "logs": ["node_1运行完毕"],  
        "cur_id": pre_id + ", node_1"  
    }  
  
def node_2(state: OverAllState) -> OverAllState:  
    pre_id = state["cur_id"]  
    return {  
        "logs": ["node_2运行完毕"],  
        "cur_id": pre_id + ", node_2"  
    }  
  
builder = StateGraph( state_schema = OverAllState)  
builder.add_node(node_1)  
builder.add_node(node_2)  
builder.add_edge(START, "node_1")  
builder.add_edge("node_1", "node_2")  
builder.add_edge("node_2", END)  
  
graph = builder.compile()  
result = graph.invoke({"cur_id" : "start"})  
print(result)

执行结果

{'logs': ['node_1运行完毕', 'node_2运行完毕'], 'cur_id': 'start, node_1, node_2'}