LangGraph实战案例之LangGraph工作流构建

import os
from langgraph.graph import StateGraph, MessagesState
from langgraph.graph.state import CompiledStateGraph, StateGraph  # 重复了
from langgraph.constants import START,END
key_keywords="key_keywords"
key_search="key_search"
key_answer="key_answer"


def node_key_keywords(state:MessagesState):
    pass

def node_key_search(state:MessagesState):
    pass
def node_key_answer(state:MessagesState):
    pass



def output_graph_image(graph,file_name):
    png_data = graph.get_graph().draw_png()
    output_file_dir = os.path.dirname(__file__)
    output_file_path = os.path.join(output_file_dir,file_name+'.png')
    with open(output_file_path,'wb') as f:
        f.write(png_data)


state_graph = StateGraph(MessagesState)
state_graph.add_node(key_keywords,node_key_keywords)
state_graph.add_node(key_search,node_key_search)
state_graph.add_node(key_answer,node_key_answer)

state_graph.add_edge(START,key_keywords)
state_graph.add_edge(key_keywords,key_search)
state_graph.add_edge(key_search,key_answer)
state_graph.add_edge(key_answer,END)

compiled_graph = state_graph.compile()

output_graph_image(compiled_graph,'output_graph_image')